pikasTech-unidesk/scripts/src/artifact-registry/options.ts

// SPEC: PJ2026-01060307 控制面模块化 draft-2026-06-25-p0. options module for scripts/src/artifact-registry.ts.

// Moved mechanically from scripts/src/artifact-registry.ts:881-978 for #903.

import { createHash } from "node:crypto";
import { existsSync, readFileSync, rmSync, writeFileSync } from "node:fs";
import { join } from "node:path";
import { runCommand, type CommandResult } from "../command";
import { readConfig, type UniDeskConfig, repoRoot, rootPath } from "../config";
import { startJob } from "../jobs";
import {
  compareDeployJsonExecutorMirrors,
  deployJsonCommitImage,
  deployJsonDriftResult,
  deployJsonSourceOfTruth,
  hasDeployJsonExecutorContract,
  k3sManifestExecutorMirror,
  parseDeployJsonServiceContractBase64,
  readDeployJsonServiceContractFromFile,
  type DeployJsonExecutorMirror,
  type DeployJsonServiceContract,
} from "../deploy-json-contract";
import { d601K3sGuardShellLines } from "../d601-k3s-guard";
import { composeRuntimeEnvValue } from "../runtime-env";

import type { ArtifactDeployEnvironment, ArtifactRegistryOptions } from "./types";
import { defaultOptions } from "./types";

export function isHelpArg(value: string | undefined): boolean {
  return value === undefined || value === "help" || value === "--help" || value === "-h";
}

export function requireValue(args: string[], index: number, option: string): string {
  const value = args[index + 1];
  if (value === undefined || value.length === 0) throw new Error(`${option} requires a non-empty value`);
  return value;
}

export function positiveInt(value: string, option: string): number {
  const parsed = Number(value);
  if (!Number.isInteger(parsed) || parsed <= 0) throw new Error(`${option} must be a positive integer`);
  return parsed;
}

export function absolutePath(value: string, option: string): string {
  if (!value.startsWith("/")) throw new Error(`${option} must be an absolute path`);
  if (value.includes("\n") || value.includes("\0")) throw new Error(`${option} must not contain control characters`);
  return value.replace(/\/+$/u, "");
}

export function commitValue(value: string, option: string): string {
  const normalized = value.toLowerCase();
  if (!/^[0-9a-f]{40}$/u.test(normalized)) throw new Error(`${option} must be a full 40-character commit SHA`);
  return normalized;
}

export function environmentValue(value: string, option: string): ArtifactDeployEnvironment {
  if (value === "prod" || value === "dev") return value;
  throw new Error(`${option} must be one of: prod, dev`);
}

export function parseArtifactRegistryOptions(args: string[]): ArtifactRegistryOptions {
  const options = { ...defaultOptions };
  for (let index = 0; index < args.length; index += 1) {
    const arg = args[index];
    if (arg === "--dry-run") {
      options.dryRun = true;
    } else if (arg === "--run-now") {
      options.runNow = true;
    } else if (arg === "--env" || arg === "--environment") {
      const environment = requireValue(args, index, arg);
      if (environment !== "dev" && environment !== "prod") throw new Error(`${arg} must be dev or prod`);
      options.environment = environment;
      index += 1;
    } else if (arg === "--provider-id") {
      options.providerId = requireValue(args, index, arg);
      index += 1;
    } else if (arg === "--host") {
      options.host = requireValue(args, index, arg);
      index += 1;
    } else if (arg === "--port") {
      options.port = positiveInt(requireValue(args, index, arg), arg);
      index += 1;
    } else if (arg === "--image") {
      options.image = requireValue(args, index, arg);
      index += 1;
    } else if (arg === "--base-dir") {
      options.baseDir = absolutePath(requireValue(args, index, arg), arg);
      index += 1;
    } else if (arg === "--storage-dir") {
      options.storageDir = absolutePath(requireValue(args, index, arg), arg);
      index += 1;
    } else if (arg === "--timeout-ms") {
      options.timeoutMs = positiveInt(requireValue(args, index, arg), arg);
      index += 1;
    } else if (arg === "--commit") {
      options.commit = commitValue(requireValue(args, index, arg), arg);
      index += 1;
    } else if (arg === "--target-image") {
      options.targetImage = requireValue(args, index, arg);
      index += 1;
    } else if (arg === "--service" || arg === "--service-id") {
      options.serviceId = requireValue(args, index, arg);
      index += 1;
    } else if (arg === "--source-repo") {
      options.sourceRepo = requireValue(args, index, arg);
      options.sourceRepoExplicit = true;
      index += 1;
    } else if (arg === "--deploy-ref") {
      options.deployRef = requireValue(args, index, arg);
      index += 1;
    } else if (arg === "--deploy-json-service") {
      options.deployJsonService = parseDeployJsonServiceContractBase64(requireValue(args, index, arg));
      index += 1;
    } else if (arg === "--env" || arg === "--environment") {
      options.environment = environmentValue(requireValue(args, index, arg), arg);
      index += 1;
    } else {
      throw new Error(`unknown artifact-registry option: ${arg}`);
    }
  }
  if (options.host !== "127.0.0.1") throw new Error("--host is first-stage restricted to 127.0.0.1");
  if (options.port !== 5000) throw new Error("--port is first-stage restricted to 5000");
  return options;
}