@pd4castr/cli 1.11.0 → 1.13.0

package/dist/index.js

@@ -90,6 +90,7 @@ var modelInputSchema = z.union([
 ]);
 var modelOutputSchema = z.object({
   name: z.string(),
+  key: z.string().regex(/^[a-z0-9][a-z0-9_-]*$/).optional(),
   type: z.enum([
     "float",
     "integer",
@@ -103,6 +104,7 @@ var modelOutputSchema = z.object({
 });
 var sensitivitySchema = z.object({
   name: z.string(),
+  key: z.string().regex(/^[a-z0-9][a-z0-9_-]*$/).optional(),
   query: z.string()
 });
 var inputAggregationSchema = z.object({
@@ -132,10 +134,45 @@ var projectConfigSchema = z.object({
   outputs: z.array(modelOutputSchema),
   sensitivities: z.array(sensitivitySchema).optional().default([]),
   [CONFIG_WARNING_KEY]: z.string().optional().default(""),
+  $$configVersion: z.number().int().min(0).optional().default(0),
   $$id: z.string().nullable().optional().default(null),
   $$modelGroupID: z.string().nullable().optional().default(null),
   $$revision: z.number().int().min(0).nullable().optional().default(null),
   $$dockerImage: z.string().nullable().optional().default(null)
+}).superRefine((data, ctx) => {
+  if (data.$$id) {
+    return;
+  }
+  for (const [i, output] of data.outputs.entries()) {
+    if (!output.key) {
+      ctx.addIssue({
+        code: "invalid_format",
+        format: "regex",
+        pattern: "^[a-z0-9][a-z0-9_-]*$",
+        path: [
+          "outputs",
+          i,
+          "key"
+        ],
+        message: "key is required (lowercase alphanumeric with hyphens/underscores)"
+      });
+    }
+  }
+  for (const [i, sensitivity] of data.sensitivities.entries()) {
+    if (!sensitivity.key) {
+      ctx.addIssue({
+        code: "invalid_format",
+        format: "regex",
+        pattern: "^[a-z0-9][a-z0-9_-]*$",
+        path: [
+          "sensitivities",
+          i,
+          "key"
+        ],
+        message: "key is required (lowercase alphanumeric with hyphens/underscores)"
+      });
+    }
+  }
 });
 
 // src/utils/is-existing-path.ts
@@ -164,6 +201,7 @@ async function loadProjectContext(configPath) {
     const config = projectConfigSchema.parse(rawConfig);
     return {
       config,
+      configPath: resolvedConfigPath,
       projectRoot
     };
   } catch (error) {
@@ -298,9 +336,83 @@ function getEnv() {
 }
 __name(getEnv, "getEnv");
 
+// package.json
+var package_default = {
+  name: "@pd4castr/cli",
+  version: "1.13.0",
+  description: "CLI tool for creating, testing, and publishing pd4castr models",
+  license: "MIT",
+  main: "dist/index.js",
+  private: false,
+  type: "module",
+  bin: {
+    pd4castr: "dist/index.js"
+  },
+  files: [
+    "dist/**/*",
+    "LICENSE.md",
+    "docs/assets/logo.png"
+  ],
+  engines: {
+    node: ">=20.0.0"
+  },
+  scripts: {
+    build: "tsup",
+    dev: "tsup --watch",
+    pd4castr: "node dist/index.js",
+    format: "prettier --write .",
+    lint: "eslint .",
+    typecheck: "tsc --noEmit",
+    prepublishOnly: "pnpm run build"
+  },
+  dependencies: {
+    "@inquirer/prompts": "7.7.1",
+    auth0: "4.28.0",
+    chalk: "5.6.0",
+    commander: "14.0.0",
+    execa: "9.6.0",
+    express: "4.21.2",
+    immer: "10.1.1",
+    ky: "1.14.2",
+    ora: "8.2.0",
+    "promise-retry": "2.0.1",
+    slugify: "1.6.6",
+    tiged: "2.12.7",
+    "tiny-invariant": "1.3.3",
+    zod: "4.0.14"
+  },
+  devDependencies: {
+    "@faker-js/faker": "10.0.0",
+    "@mswjs/data": "0.16.2",
+    "@types/express": "4.17.21",
+    "@types/node": "24.1.0",
+    "@types/supertest": "6.0.3",
+    "hook-std": "3.0.0",
+    "jest-extended": "7.0.0",
+    memfs: "4.49.0",
+    msw: "2.12.9",
+    "strip-ansi": "7.1.0",
+    supertest: "7.2.2",
+    tsup: "8.5.0",
+    "type-fest": "4.41.0",
+    typescript: "5.8.3",
+    vitest: "4.0.18"
+  },
+  publishConfig: {
+    access: "public"
+  },
+  keywords: [
+    "cli",
+    "pd4castr"
+  ]
+};
+
 // src/api/api.ts
 var api = ky.create({
-  prefixUrl: getEnv().apiURL
+  prefixUrl: getEnv().apiURL,
+  headers: {
+    "X-SDK-Version": `cli/${package_default.version}`
+  }
 });
 
 // src/api/query-data-fetcher.ts
@@ -427,7 +539,6 @@ __name(registerFetchCommand, "registerFetchCommand");
 // src/commands/init/handle-action.ts
 import path6 from "path";
 import * as inquirer from "@inquirer/prompts";
-import { ExecaError as ExecaError2 } from "execa";
 import ora2 from "ora";
 import tiged from "tiged";
 
@@ -480,16 +591,26 @@ async function handleAction2() {
     await fetchTemplate(template, projectName);
     spinner.succeed("Template fetched successfully");
   } catch (error) {
-    spinner.fail("Error fetching template");
-    if (error instanceof Error && error.cause instanceof ExecaError2) {
-      console.error(error.cause.shortMessage);
-    } else if (error instanceof Error && error.cause) {
-      console.error(error.cause);
+    if (isDegitError(error)) {
+      spinner.fail(formatErrorMessage(DEGIT_CODE_MAP[error.code]));
+    } else {
+      spinner.fail(formatErrorMessage("UNKNOWN_ERROR"));
+      if (error instanceof Error) {
+        console.error(error.stack ?? error.message);
+      }
     }
     process.exit(1);
   }
 }
 __name(handleAction2, "handleAction");
+var DEGIT_CODE_MAP = {
+  COULD_NOT_DOWNLOAD: "TEMPLATE_DOWNLOAD_FAILED",
+  COULD_NOT_FETCH: "TEMPLATE_DOWNLOAD_FAILED",
+  MISSING_REF: "TEMPLATE_NOT_FOUND",
+  BAD_SRC: "TEMPLATE_NOT_FOUND",
+  BAD_REF: "TEMPLATE_NOT_FOUND",
+  DEST_NOT_EMPTY: "DEST_NOT_EMPTY"
+};
 async function fetchTemplate(template, projectName) {
   const templatePath = getTemplatePath(templates[template]);
   const fetcher = tiged(templatePath, {
@@ -500,6 +621,20 @@ async function fetchTemplate(template, projectName) {
   await fetcher.clone(destination);
 }
 __name(fetchTemplate, "fetchTemplate");
+function isDegitError(error) {
+  return error instanceof Error && "code" in error && typeof error.code === "string" && error.code in DEGIT_CODE_MAP;
+}
+__name(isDegitError, "isDegitError");
+var ERROR_MESSAGES = {
+  TEMPLATE_DOWNLOAD_FAILED: "Failed to download template. Check your internet connection and try again.",
+  TEMPLATE_NOT_FOUND: "Failed to download template. Please try again or report this issue if it persists.",
+  DEST_NOT_EMPTY: "Destination directory is not empty.",
+  UNKNOWN_ERROR: "An unexpected error occurred. Please try again or report this issue if it persists."
+};
+function formatErrorMessage(code) {
+  return `${ERROR_MESSAGES[code]} [CODE:${code}]`;
+}
+__name(formatErrorMessage, "formatErrorMessage");
 
 // src/commands/init/index.ts
 function registerInitCommand(program2) {
@@ -508,7 +643,7 @@ function registerInitCommand(program2) {
 __name(registerInitCommand, "registerInitCommand");
 
 // src/commands/login/handle-action.ts
-import { ExecaError as ExecaError3 } from "execa";
+import { ExecaError as ExecaError2 } from "execa";
 import ora3 from "ora";
 import { ZodError as ZodError3 } from "zod";
 
@@ -624,7 +759,7 @@ async function handleAction3() {
     } else if (error instanceof Error) {
       spinner.fail(error.message);
     }
-    if (error instanceof Error && error.cause instanceof ExecaError3) {
+    if (error instanceof Error && error.cause instanceof ExecaError2) {
       console.error(error.cause.shortMessage);
     } else if (error instanceof Error && error.cause) {
       console.error(error.cause);
@@ -641,7 +776,7 @@ function registerLoginCommand(program2) {
 __name(registerLoginCommand, "registerLoginCommand");
 
 // src/commands/logout/handle-action.ts
-import { ExecaError as ExecaError4 } from "execa";
+import { ExecaError as ExecaError3 } from "execa";
 import ora4 from "ora";
 import { ZodError as ZodError4 } from "zod";
 async function handleAction4() {
@@ -664,7 +799,7 @@ async function handleAction4() {
     } else if (error instanceof Error) {
       spinner.fail(error.message);
     }
-    if (error instanceof Error && error.cause instanceof ExecaError4) {
+    if (error instanceof Error && error.cause instanceof ExecaError3) {
       console.error(error.cause.shortMessage);
     } else if (error instanceof Error && error.cause) {
       console.error(error.cause);
@@ -681,7 +816,7 @@ function registerLogoutCommand(program2) {
 __name(registerLogoutCommand, "registerLogoutCommand");
 
 // src/commands/publish/handle-action.ts
-import { ExecaError as ExecaError5 } from "execa";
+import { ExecaError as ExecaError4 } from "execa";
 import express2 from "express";
 import { HTTPError as HTTPError3 } from "ky";
 import ora5 from "ora";
@@ -840,7 +975,7 @@ async function getModelConfigFromProjectConfig(ctx) {
   const sensitivities = await getSensitivitiesWithInlinedSQL(ctx);
   const inputAggregations = await getInputAggregationsWithInlinedSQL(ctx);
   const runDatetimeQuery = await getrunDatetimeQuerySQL(ctx);
-  const { $$id, $$modelGroupID, $$revision, $$dockerImage, ...config } = ctx.config;
+  const { $$configVersion, $$id, $$modelGroupID, $$revision, $$dockerImage, ...config } = ctx.config;
   return {
     ...config,
     id: $$id,
@@ -1008,7 +1143,7 @@ function getModelSummaryLines(ctx) {
     ] : [],
     ...ctx.config.sensitivities.length > 0 ? [
       `	${chalk2.bold("Sensitivities:")}`,
-      ...ctx.config.sensitivities.map((s) => `	\u2022 ${s.name}`)
+      ...ctx.config.sensitivities.map((s) => `	\u2022 ${s.name}${s.key ? ` (${s.key})` : ""}`)
     ] : [],
     ""
   ];
@@ -1405,8 +1540,125 @@ import chalk6 from "chalk";
 import * as inquirer3 from "@inquirer/prompts";
 import chalk4 from "chalk";
 
-// src/commands/publish/utils/validate-local-model-state.ts
+// src/utils/migrate-config.ts
+import fs11 from "fs/promises";
+import invariant4 from "tiny-invariant";
+
+// src/migrations/01-backfill-output-variable-keys.ts
 import invariant2 from "tiny-invariant";
+var backfillOutputVariableKeys = {
+  version: 1,
+  description: "Backfill output variable keys from published model",
+  migrate(config, ctx) {
+    const outputsMissingKeys = config.outputs.filter((output) => !output.key);
+    if (outputsMissingKeys.length === 0) {
+      return config;
+    }
+    invariant2(ctx.publishedModel, "Published model is required to backfill output variable keys");
+    const publishedKeysByName = /* @__PURE__ */ new Map();
+    for (const published of ctx.publishedModel.outputVariables) {
+      publishedKeysByName.set(published.name, published.key);
+    }
+    const unmatchedOutputs = [];
+    const updatedOutputs = config.outputs.map((output) => {
+      if (output.key) {
+        return output;
+      }
+      const publishedKey = publishedKeysByName.get(output.name);
+      if (publishedKey) {
+        return {
+          ...output,
+          key: publishedKey
+        };
+      }
+      unmatchedOutputs.push(output.name);
+      return output;
+    });
+    if (unmatchedOutputs.length > 0) {
+      throw new Error(`The following outputs need a \`key\` defined in .pd4castrrc.json:
+${unmatchedOutputs.map((name) => `  - ${name}`).join("\n")}
+
+Set a unique key (lowercase alphanumeric with hyphens/underscores) for each output.`);
+    }
+    return {
+      ...config,
+      outputs: updatedOutputs
+    };
+  }
+};
+
+// src/migrations/02-backfill-sensitivity-keys.ts
+import invariant3 from "tiny-invariant";
+var backfillSensitivityKeys = {
+  version: 2,
+  description: "Backfill sensitivity keys from published model",
+  migrate(config, ctx) {
+    const sensitivitiesMissingKeys = (config.sensitivities ?? []).filter((sensitivity) => !sensitivity.key);
+    if (sensitivitiesMissingKeys.length === 0) {
+      return config;
+    }
+    invariant3(ctx.publishedModel, "Published model is required to backfill sensitivity keys");
+    const publishedKeysByName = /* @__PURE__ */ new Map();
+    for (const published of ctx.publishedModel.sensitivities) {
+      publishedKeysByName.set(published.name, published.key);
+    }
+    const unmatchedSensitivities = [];
+    const updatedSensitivities = config.sensitivities.map((sensitivity) => {
+      if (sensitivity.key) {
+        return sensitivity;
+      }
+      const publishedKey = publishedKeysByName.get(sensitivity.name);
+      if (publishedKey) {
+        return {
+          ...sensitivity,
+          key: publishedKey
+        };
+      }
+      unmatchedSensitivities.push(sensitivity.name);
+      return sensitivity;
+    });
+    if (unmatchedSensitivities.length > 0) {
+      throw new Error(`The following sensitivities need a \`key\` defined in .pd4castrrc.json:
+${unmatchedSensitivities.map((name) => `  - ${name}`).join("\n")}
+
+Set a unique key (lowercase alphanumeric with hyphens/underscores) for each sensitivity.`);
+    }
+    return {
+      ...config,
+      sensitivities: updatedSensitivities
+    };
+  }
+};
+
+// src/migrations/index.ts
+var migrations = [
+  backfillOutputVariableKeys,
+  backfillSensitivityKeys
+];
+
+// src/utils/migrate-config.ts
+async function migrateConfig(ctx, migrationCtx) {
+  const pendingMigrations = migrations.filter((migration) => migration.version > ctx.config.$$configVersion).sort((a, b) => a.version - b.version);
+  if (pendingMigrations.length === 0) {
+    return;
+  }
+  let config = ctx.config;
+  for (const migration of pendingMigrations) {
+    config = migration.migrate(config, migrationCtx);
+  }
+  const lastMigration = pendingMigrations.at(-1);
+  invariant4(lastMigration, "expected at least one pending migration");
+  config = {
+    ...config,
+    $$configVersion: lastMigration.version
+  };
+  ctx.config = config;
+  await fs11.writeFile(ctx.configPath, JSON.stringify(config, null, 2) + "\n", "utf8");
+}
+__name(migrateConfig, "migrateConfig");
+
+// src/commands/publish/utils/validate-local-model-state.ts
+import invariant5 from "tiny-invariant";
 
 // src/api/get-model.ts
 async function getModel(modelId, authCtx) {
@@ -1422,7 +1674,7 @@ __name(getModel, "getModel");
 
 // src/commands/publish/utils/validate-local-model-state.ts
 async function validateLocalModelState(ctx, authCtx) {
-  invariant2(ctx.config.$$id, "model ID is required to fetch published model");
+  invariant5(ctx.config.$$id, "model ID is required to fetch published model");
   const currentModel = await getModel(ctx.config.$$id, authCtx);
   if (currentModel.revision !== ctx.config.$$revision) {
     throw new Error(`OUT OF SYNC: Local revision (${ctx.config.$$revision}) does not match the current published revision (${currentModel.revision})`);
@@ -1430,6 +1682,7 @@ async function validateLocalModelState(ctx, authCtx) {
   if (currentModel.modelGroupId !== ctx.config.$$modelGroupID) {
     throw new Error(`OUT OF SYNC: Local model group ID (${ctx.config.$$modelGroupID}) does not match the current published model group ID (${currentModel.modelGroupId})`);
   }
+  return currentModel;
 }
 __name(validateLocalModelState, "validateLocalModelState");
 
@@ -1449,7 +1702,10 @@ async function handleModelRevisionCreateFlow(options, app, spinner, ctx, authCtx
     throw new Error("Model revision update cancelled");
   }
   spinner.start("Validating local model state...");
-  await validateLocalModelState(ctx, authCtx);
+  const publishedModel = await validateLocalModelState(ctx, authCtx);
+  await migrateConfig(ctx, {
+    publishedModel
+  });
   spinner.succeed("Local model is synced with published model");
   const dockerImage = getDockerImage(ctx);
   if (!options.skipChecks) {
@@ -1537,7 +1793,10 @@ async function handleModelRevisionUpdateFlow(options, app, spinner, ctx, authCtx
     throw new Error("Model revision update cancelled");
   }
   spinner.start("Validating local model state...");
-  await validateLocalModelState(ctx, authCtx);
+  const publishedModel = await validateLocalModelState(ctx, authCtx);
+  await migrateConfig(ctx, {
+    publishedModel
+  });
   spinner.succeed("Local model is synced with published model");
   const dockerImage = getDockerImage(ctx);
   if (!options.skipChecks) {
@@ -1724,7 +1983,7 @@ async function handleAction5(options) {
       } else if (error instanceof Error) {
         spinner.fail(error.message);
       }
-      if (error instanceof Error && error.cause instanceof ExecaError5) {
+      if (error instanceof Error && error.cause instanceof ExecaError4) {
         console.error(error.cause.stderr);
       } else if (error instanceof Error && error.cause) {
         console.error(error.cause);
@@ -1748,7 +2007,7 @@ __name(registerPublishCommand, "registerPublishCommand");
 
 // src/commands/test/handle-action.ts
 import path16 from "path";
-import { ExecaError as ExecaError6 } from "execa";
+import { ExecaError as ExecaError5 } from "execa";
 import express3 from "express";
 import ora6 from "ora";
 import { ZodError as ZodError6 } from "zod";
@@ -1879,7 +2138,7 @@ ${clickHereLink} to view output (${fileLink})
       } else if (error instanceof Error) {
         spinner.fail(error.message);
       }
-      if (error instanceof Error && error.cause instanceof ExecaError6) {
+      if (error instanceof Error && error.cause instanceof ExecaError5) {
         console.error(error.cause.stderr);
       } else if (error instanceof Error && error.cause) {
         console.error(error.cause);
@@ -1903,79 +2162,6 @@ __name(registerTestCommand, "registerTestCommand");
 
 // src/program.ts
 import { Command } from "commander";
-
-// package.json
-var package_default = {
-  name: "@pd4castr/cli",
-  version: "1.11.0",
-  description: "CLI tool for creating, testing, and publishing pd4castr models",
-  license: "MIT",
-  main: "dist/index.js",
-  private: false,
-  type: "module",
-  bin: {
-    pd4castr: "dist/index.js"
-  },
-  files: [
-    "dist/**/*",
-    "LICENSE.md",
-    "docs/assets/logo.png"
-  ],
-  engines: {
-    node: ">=20.0.0"
-  },
-  scripts: {
-    build: "tsup",
-    dev: "tsup --watch",
-    pd4castr: "node dist/index.js",
-    format: "prettier --write .",
-    lint: "eslint .",
-    typecheck: "tsc --noEmit",
-    prepublishOnly: "yarn build"
-  },
-  dependencies: {
-    "@inquirer/prompts": "7.7.1",
-    auth0: "4.28.0",
-    chalk: "5.6.0",
-    commander: "14.0.0",
-    execa: "9.6.0",
-    express: "4.21.2",
-    immer: "10.1.1",
-    ky: "1.14.2",
-    ora: "8.2.0",
-    "promise-retry": "2.0.1",
-    slugify: "1.6.6",
-    tiged: "2.12.7",
-    "tiny-invariant": "1.3.3",
-    zod: "4.0.14"
-  },
-  devDependencies: {
-    "@faker-js/faker": "10.0.0",
-    "@mswjs/data": "0.16.2",
-    "@types/express": "4.17.21",
-    "@types/node": "24.1.0",
-    "@types/supertest": "6.0.3",
-    "hook-std": "3.0.0",
-    "jest-extended": "7.0.0",
-    memfs: "4.49.0",
-    msw: "2.12.9",
-    "strip-ansi": "7.1.0",
-    supertest: "7.2.2",
-    tsup: "8.5.0",
-    "type-fest": "4.41.0",
-    typescript: "5.8.3",
-    vitest: "4.0.18"
-  },
-  publishConfig: {
-    access: "public"
-  },
-  keywords: [
-    "cli",
-    "pd4castr"
-  ]
-};
-
-// src/program.ts
 var program = new Command();
 program.name("pd4castr").description("CLI tool for pd4castr").version(package_default.version);
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pd4castr/cli",
-  "version": "1.11.0",
+  "version": "1.13.0",
   "description": "CLI tool for creating, testing, and publishing pd4castr models",
   "license": "MIT",
   "main": "dist/index.js",
@@ -24,7 +24,7 @@
     "format": "prettier --write .",
     "lint": "eslint .",
     "typecheck": "tsc --noEmit",
-    "prepublishOnly": "yarn build"
+    "prepublishOnly": "pnpm run build"
   },
   "dependencies": {
     "@inquirer/prompts": "7.7.1",