apiVersion: kustomize.config.k8s.io/v1beta1
kind: Kustomization
namespace: "{{ namespace }}"
resources:
  - rendered.yaml
  - weights-pvc.yaml
patches:
  - target:
      group: apps
      version: v1
      kind: Deployment
      name: "{{ name }}"
    patch: |
      apiVersion: apps/v1
      kind: Deployment
      metadata:
        name: "{{ name }}"
        labels:
          app.kubernetes.io/component: ide
          app.kubernetes.io/part-of: ocdp-workload
      spec:
        template:
          metadata:
            labels:
              app.kubernetes.io/component: ide
              app.kubernetes.io/part-of: ocdp-workload
          spec:
            imagePullSecrets:
              - name: regcred
            schedulerName: hami-scheduler
            securityContext:
              fsGroup: 1000
            volumes:
              - name: data
                persistentVolumeClaim:
                  claimName: "{{ name }}"
              - name: model-weights
                persistentVolumeClaim:
                  claimName: "{{ name }}-weights"
            containers:
              - name: code-server
                image: harbor.bwgdi.com/library/earth2studio-demo:v6
                imagePullPolicy: IfNotPresent
                securityContext:
                  runAsUser: 1000
                command:
                  - code-server
                args:
                  - --bind-addr
                  - 0.0.0.0:8080
                  - --auth
                  - none
                  - /workspace
                ports:
                  - name: http
                    containerPort: 8080
                    protocol: TCP
                env:
                  - name: HOME
                    value: /workspace
                  - name: SHELL
                    value: /bin/bash
                  - name: XDG_CONFIG_HOME
                    value: /workspace/.config
                readinessProbe:
                  httpGet:
                    path: /
                    port: http
                livenessProbe:
                  httpGet:
                    path: /
                    port: http
                resources:
                  requests:
                    cpu: "{{ cpuRequestMillicores }}m"
                    memory: "{{ memoryRequestMiB }}Mi"
                  limits:
                    cpu: "{{ cpuLimitMillicores }}m"
                    memory: "{{ memoryLimitMiB }}Mi"
                    nvidia.com/gpu: "{{ gpuCount }}"
                    nvidia.com/gpumem: "{{ gpuMemoryMiB }}"
                volumeMounts:
                  - name: data
                    mountPath: /workspace
                  - name: model-weights
                    mountPath: /models
  - target:
      version: v1
      kind: Secret
      name: "{{ name }}"
    patch: |
      apiVersion: v1
      kind: Secret
      metadata:
        name: "{{ name }}"
        labels:
          app.kubernetes.io/component: auth
          app.kubernetes.io/part-of: ocdp-workload
        annotations: {}
      type: Opaque
      data:
        password: dW51c2Vk
  - target:
      version: v1
      kind: Service
      name: "{{ name }}"
    patch: |
      apiVersion: v1
      kind: Service
      metadata:
        name: "{{ name }}"
        labels:
          app.kubernetes.io/component: ide
          app.kubernetes.io/part-of: ocdp-workload
      spec:
        type: NodePort
        ports:
          - name: http
            port: 80
            targetPort: http
            protocol: TCP