1import * as _ from 'lodash-es'
2import type { KubeResource } from '../../ctxF/klusterCtxF.mts'
3import { getRegcacheTargetHost, regcacheImgRef, regcacheName } from './regcacheConstsF.mts'
4import { isRegcacheServing } from './regcacheServingAI.mts'
5
6const podSpecPaths = [
7  ['spec'],
8  ['spec', 'template', 'spec'],
9  ['spec', 'jobTemplate', 'spec', 'template', 'spec'],
10]
11
12const rewritePodSpec = (podSpec: Record<string, any>, cacheHost: string) => {
13  for (const key of ['containers', 'initContainers']) {
14    for (const container of _.castArray(podSpec[key] || [])) {
15      if (container.image) container.image = regcacheImgRef({image: container.image, cacheHost})
16    }
17  }
18}
19
20export const rewriteResourceImgRefs = async (resource: KubeResource): Promise<KubeResource> => {
21  const cacheHost = getRegcacheTargetHost()
22  if (!cacheHost) return resource
23  if (resource.metadata?.name === regcacheName) return resource
24  if (!await isRegcacheServing(cacheHost)) return resource
25  for (const path of podSpecPaths) {
26    const podSpec = _.get(resource, path)
27    if (podSpec?.containers || podSpec?.initContainers) rewritePodSpec(podSpec, cacheHost)
28  }
29  return resource
30}
31