@highstate/k8s 0.9.4 → 0.9.5

package/src/namespace.ts

@@ -0,0 +1,241 @@
+import type { k8s } from "@highstate/library"
+import { core, type types } from "@pulumi/kubernetes"
+import {
+  ComponentResource,
+  output,
+  Output,
+  type ComponentResourceOptions,
+  type Input,
+  type Inputs,
+} from "@pulumi/pulumi"
+import { getProvider, mapMetadata, type CommonArgs } from "./shared"
+
+export type NamespaceArgs = Omit<CommonArgs, "namespace"> & {
+  /**
+   * Whether to apply "pod-security.kubernetes.io/enforce=privileged" label to the namespace.
+   */
+  privileged?: boolean
+}
+
+export type CreateOrPatchNamespaceArgs = NamespaceArgs & {
+  /**
+   * The resource to use to determine the name of the namespace.
+   *
+   * If not provided, the namespace will be created, otherwise it will be retrieved/patched.
+   */
+  resource: Input<k8s.Resource> | undefined
+}
+
+export abstract class Namespace extends ComponentResource {
+  protected constructor(
+    type: string,
+    name: string,
+    args: Inputs,
+    opts: ComponentResourceOptions | undefined,
+
+    /**
+     * The cluster where the namespace is created.
+     */
+    readonly cluster: Output<k8s.Cluster>,
+
+    /*
+     * The metadata of the underlying Kubernetes namespace.
+     */
+    readonly metadata: Output<types.output.meta.v1.ObjectMeta>,
+
+    /**
+     * The spec of the underlying Kubernetes namespace.
+     */
+    readonly spec: Output<types.output.core.v1.NamespaceSpec>,
+
+    /**
+     * The status of the underlying Kubernetes namespace.
+     */
+    readonly status: Output<types.output.core.v1.NamespaceStatus>,
+  ) {
+    super(type, name, args, opts)
+  }
+
+  /**
+   * Creates a new namespace.
+   */
+  static create(name: string, args: NamespaceArgs, opts?: ComponentResourceOptions): Namespace {
+    return new CreatedNamespace(name, args, opts)
+  }
+
+  /**
+   * Creates a new namespace or patches an existing one.
+   *
+   * Will throw an error if the namespace does not exist when `args.resource` is provided.
+   */
+  static createOrPatch(
+    name: string,
+    args: CreateOrPatchNamespaceArgs,
+    opts?: ComponentResourceOptions,
+  ): Namespace {
+    if (!args.resource) {
+      return new CreatedNamespace(name, args, opts)
+    }
+
+    return new NamespacePatch(
+      name,
+      {
+        ...args,
+        name: output(args).apply(args => {
+          if (args.resource!.clusterId !== args.cluster.id) {
+            throw new Error(
+              `Cluster mismatch when patching namespace "${name}": "${args.resource!.clusterId}" != "${args.cluster.id}"`,
+            )
+          }
+
+          return args.resource!.metadata.namespace
+        }),
+      },
+      opts,
+    )
+  }
+
+  /**
+   * Creates a new namespace or gets an existing one.
+   *
+   * Will throw an error if the namespace does not exist when `args.resource` is provided.
+   */
+  static createOrGet(
+    name: string,
+    args: CreateOrPatchNamespaceArgs,
+    opts?: ComponentResourceOptions,
+  ): Namespace {
+    if (!args.resource) {
+      return new CreatedNamespace(name, args, opts)
+    }
+
+    return new ExternalNamespace(
+      name,
+      output(args).apply(args => {
+        if (args.resource!.clusterId !== args.cluster.id) {
+          throw new Error(
+            `Cluster mismatch when receiving namespace "${name}": "${args.resource!.clusterId}" != "${args.cluster.id}"`,
+          )
+        }
+
+        return args.resource!.metadata.namespace
+      }),
+      args.cluster,
+      opts,
+    )
+  }
+
+  /**
+   * Patches an existing namespace.
+   *
+   * Will throw an error if the namespace does not exist.
+   */
+  static patch(name: string, args: NamespaceArgs, opts?: ComponentResourceOptions): Namespace {
+    return new NamespacePatch(name, args, opts)
+  }
+
+  /**
+   * Gets an existing namespace.
+   *
+   * Will throw an error if the namespace does not exist.
+   */
+  static get(
+    name: string,
+    id: Input<string>,
+    cluster: Input<k8s.Cluster>,
+    opts?: ComponentResourceOptions,
+  ): Namespace {
+    return new ExternalNamespace(name, id, cluster, opts)
+  }
+}
+
+class CreatedNamespace extends Namespace {
+  constructor(name: string, args: NamespaceArgs, opts?: ComponentResourceOptions) {
+    const namespace = output(args).apply(async args => {
+      return new core.v1.Namespace(
+        name,
+        {
+          metadata: mapMetadata(args, name),
+        },
+        {
+          ...opts,
+          parent: this,
+          provider: await getProvider(args.cluster),
+        },
+      )
+    })
+
+    super(
+      "highstate:k8s:Namespace",
+      name,
+      args,
+      opts,
+      output(args.cluster),
+      namespace.metadata,
+      namespace.spec,
+      namespace.status,
+    )
+  }
+}
+
+class NamespacePatch extends Namespace {
+  constructor(name: string, args: NamespaceArgs, opts?: ComponentResourceOptions) {
+    const namespace = output(args).apply(async args => {
+      return new core.v1.NamespacePatch(
+        name,
+        {
+          metadata: mapMetadata(args, name),
+        },
+        {
+          ...opts,
+          parent: this,
+          provider: await getProvider(args.cluster),
+        },
+      )
+    })
+
+    super(
+      "highstate:k8s:NamespacePatch",
+      name,
+      args,
+      opts,
+      output(args.cluster),
+      namespace.metadata,
+      namespace.spec,
+      namespace.status,
+    )
+  }
+}
+
+class ExternalNamespace extends Namespace {
+  constructor(
+    name: string,
+    id: Input<string>,
+    cluster: Input<k8s.Cluster>,
+    opts?: ComponentResourceOptions,
+  ) {
+    const namespace = output(id).apply(async realName => {
+      return core.v1.Namespace.get(
+        //
+        name,
+        realName,
+        {
+          ...opts,
+          parent: this,
+          provider: await getProvider(cluster),
+        },
+      )
+    })
+
+    super(
+      "highstate:k8s:ExternalNamespace",
+      name,
+      { id, cluster },
+      opts,
+      output(cluster),
+      namespace.metadata,
+      namespace.spec,
+      namespace.status,
+    )
+  }
+}