function getHomePageHtml(): string { return ` CZL Docker镜像服务(仅内部用)

快捷命令

提示，支持docker.io, ghcr,quay,k8sgcr,gcr, 非docker.io需加上域名前缀
`; } import { TokenProvider } from './token' import { Backend } from './backend' const PROXY_HEADER_ALLOW_LIST: string[] = ["accept", "user-agent", "accept-encoding"] const validActionNames = new Set(["manifests", "blobs", "tags", "referrers"]) const ORG_NAME_BACKEND:{ [key: string]: string; } = { "gcr": "https://gcr.io", "k8sgcr": "https://k8s.gcr.io", "quay": "https://quay.io", "ghcr": "https://ghcr.io" } const DEFAULT_BACKEND_HOST: string = "https://registry-1.docker.io" export async function handleRequest(request: Request): Promise { const url = new URL(request.url) if (url.pathname === '/') { return new Response(getHomePageHtml(), { headers: { 'content-type': 'text/html;charset=UTF-8' }, }) } return handleRegistryRequest(request) } function copyProxyHeaders(inputHeaders: Headers) : Headers { const headers = new Headers; for(const pair of inputHeaders.entries()) { if (PROXY_HEADER_ALLOW_LIST.includes(pair[0].toLowerCase())) { headers.append(pair[0], pair[1]) } } return headers; } function orgNameFromPath(pathname: string): string|null { const splitedPath: string[] = pathname.split("/", 3) if (splitedPath.length === 3 && splitedPath[0] === "" && splitedPath[1] === "v2") { return splitedPath[2].toLowerCase() } return null } function hostByOrgName(orgName: string|null): string { if (orgName !== null && orgName in ORG_NAME_BACKEND) { return ORG_NAME_BACKEND[orgName] } return DEFAULT_BACKEND_HOST } function rewritePath(orgName: string | null, pathname: string): string { let splitedPath = pathname.split("/"); // /v2/repo/manifests/xxx -> /v2/library/repo/manifests/xxx // /v2/repo/blobs/xxx -> /v2/library/repo/blobs/xxx if (orgName === null && splitedPath.length === 5 && validActionNames.has(splitedPath[3])) { splitedPath = [splitedPath[0], splitedPath[1], "library", splitedPath[2], splitedPath[3], splitedPath[4]] } if (orgName === null || !(orgName in ORG_NAME_BACKEND)) { return pathname } const cleanSplitedPath = splitedPath.filter(function(value: string, index: number) { return value !== orgName || index !== 2; }) return cleanSplitedPath.join("/") } async function handleRegistryRequest(request: Request): Promise { const reqURL = new URL(request.url) const orgName = orgNameFromPath(reqURL.pathname) const pathname = rewritePath(orgName, reqURL.pathname) const host = hostByOrgName(orgName) const tokenProvider = new TokenProvider() const backend = new Backend(host, tokenProvider) const headers = copyProxyHeaders(request.headers) return backend.proxy(pathname, {headers: request.headers}) }