operator/src/storage-provider/storage-provider.ts

import { mkdir } from 'fs/promises';

import { V1PersistentVolume, type V1PersistentVolumeClaim } from '@kubernetes/client-node';

import { Watcher, WatcherService } from '../services/watchers/watchers.ts';
import type { Services } from '../utils/service.ts';
import { ResourceService, type Resource } from '../services/resources/resources.ts';

const PROVISIONER = 'reuse-local-path-provisioner';
const LABEL_SELECTOR = `provisioner=${PROVISIONER}`;

class StorageProvider {
  #watcher: Watcher<V1PersistentVolumeClaim>;
  #services: Services;

  constructor(services: Services) {
    this.#services = services;
    const watchService = this.#services.get(WatcherService);
    this.#watcher = watchService.create({
      path: '/api/v1/persistantvolumeclaims',
      list: (k8s) =>
        k8s.api.listPersistentVolumeClaimForAllNamespaces({
          labelSelector: LABEL_SELECTOR,
        }),
      verbs: ['add', 'update', 'delete'],
      selector: LABEL_SELECTOR,
    });
    this.#watcher.on('changed', this.#handleChange);
  }

  #handleChange = async (pvc: Resource<V1PersistentVolumeClaim>) => {
    const target = `/data/volumes/${pvc.namespace}/${pvc.name}`;
    await mkdir(target, { recursive: true });
    const resourceService = this.#services.get(ResourceService);
    const pv = resourceService.get<V1PersistentVolume>({
      apiVersion: 'v1',
      kind: 'PersistantVolume',
      name: pvc.name,
      namespace: pvc.namespace,
    });
    await pv.load();
    await pv.patch({
      metadata: {
        labels: {
          provisioner: PROVISIONER,
        },
      },
      spec: {
        hostPath: {
          path: target,
        },
        capacity: {
          storage: pvc.spec?.resources?.requests?.storage ?? '1GB',
        },
        persistentVolumeReclaimPolicy: 'Retain',
        accessModes: pvc.spec?.accessModes,
        claimRef: {
          uid: pvc.metadata?.uid,
          resourceVersion: pvc.metadata?.resourceVersion,
          apiVersion: pvc.apiVersion,
          name: pvc.name,
          namespace: pvc.namespace,
        },
      },
    });
  };

  public start = async () => {
    await this.#watcher.start();
  };
}

export { StorageProvider };