qinglong/packages/ql3-cluster-admin/src/modelInvocationMigrationCli.ts

#!/usr/bin/env node

import { migratePostgresModelInvocationFeature } from '@qinglong/ai/model-invocation-migration';
import { runPostgresMigrationProcess } from '@qinglong/cluster-postgres/migration-process';

const USAGE = 'Usage: ql3-ai-feature-migrate';

function failureFact(error: unknown): Readonly<Record<string, unknown>> {
  const candidate = error as {
    readonly name?: unknown;
    readonly code?: unknown;
  };
  return Object.freeze({
    schemaVersion: 1,
    component: 'qinglong3-ai-feature-migration',
    event: 'migration_failed',
    name:
      typeof candidate?.name === 'string' && candidate.name.length <= 128
        ? candidate.name
        : 'Error',
    ...(typeof candidate?.code === 'string' && candidate.code.length <= 128
      ? { code: candidate.code }
      : {}),
  });
}

async function main(argv: readonly string[]): Promise<void> {
  if (argv.length === 1 && (argv[0] === '--help' || argv[0] === '-h')) {
    process.stdout.write(`${USAGE}\n`);
    return;
  }
  if (argv.length !== 0) {
    process.stderr.write(
      `${JSON.stringify({
        code: 'QL3_AI_FEATURE_MIGRATION_CLI_USAGE_INVALID',
        message: USAGE,
      })}\n`,
    );
    process.exitCode = 64;
    return;
  }
  try {
    await runPostgresMigrationProcess({
      environment: process.env,
      migrate: ({ pool }) => migratePostgresModelInvocationFeature(pool),
    });
    process.stdout.write(
      `${JSON.stringify({
        schemaVersion: 1,
        component: 'qinglong3-ai-feature-migration',
        event: 'migration_completed',
      })}\n`,
    );
  } catch (error) {
    process.stderr.write(`${JSON.stringify(failureFact(error))}\n`);
    process.exitCode = 1;
  }
}

void main(process.argv.slice(2));