context101-cli 0.1.0

package/src/embedding-models.js

@@ -0,0 +1,184 @@
+// Keep in sync with web/lib/embedding-models.ts — ListFoundationModels
+// does not return vector dimensions, and SKU variants (…:0:512) are not
+// valid Knowledge Base embeddingModelArn values.
+export const KNOWN_EMBEDDING_DIMENSIONS = {
+  "amazon.titan-embed-text-v2:0": {
+    supportedDimensions: [256, 512, 1024],
+    defaultDimension: 1024,
+    label: "Titan Text Embeddings V2",
+    provider: "aws",
+  },
+  "amazon.titan-embed-text-v1": {
+    supportedDimensions: [1536],
+    defaultDimension: 1536,
+    label: "Titan Embeddings G1 - Text",
+    provider: "aws",
+  },
+  "amazon.titan-embed-image-v1": {
+    supportedDimensions: [1024],
+    defaultDimension: 1024,
+    label: "Titan Multimodal Embeddings G1",
+    provider: "aws",
+  },
+  "cohere.embed-english-v3": {
+    supportedDimensions: [1024],
+    defaultDimension: 1024,
+    label: "Embed English v3",
+    provider: "cohere",
+  },
+  "cohere.embed-multilingual-v3": {
+    supportedDimensions: [1024],
+    defaultDimension: 1024,
+    label: "Embed Multilingual v3",
+    provider: "cohere",
+  },
+  "cohere.embed-english-light-v3": {
+    supportedDimensions: [384],
+    defaultDimension: 384,
+    label: "Embed English Light v3",
+    provider: "cohere",
+  },
+  "cohere.embed-multilingual-light-v3": {
+    supportedDimensions: [384],
+    defaultDimension: 384,
+    label: "Embed Multilingual Light v3",
+    provider: "cohere",
+  },
+};
+
+export function isKnownEmbeddingModel(modelId) {
+  return Boolean(modelId && KNOWN_EMBEDDING_DIMENSIONS[modelId]);
+}
+
+export function embeddingModelMeta(modelId) {
+  return KNOWN_EMBEDDING_DIMENSIONS[modelId] ?? null;
+}
+
+export function fallbackEmbeddingModels() {
+  return Object.keys(KNOWN_EMBEDDING_DIMENSIONS).map((id) => catalogEntry(id));
+}
+
+/** Live Bedrock list plus every curated embedding id brains can pick later. */
+export function allEmbeddingModels(listed = []) {
+  const byId = new Map();
+  for (const id of Object.keys(KNOWN_EMBEDDING_DIMENSIONS)) {
+    byId.set(id, catalogEntry(id));
+  }
+  for (const model of listed) {
+    if (model?.id) byId.set(model.id, model);
+  }
+  return [...byId.values()].sort((a, b) =>
+    a.provider === b.provider
+      ? a.label.localeCompare(b.label)
+      : a.provider.localeCompare(b.provider)
+  );
+}
+
+export function catalogEntry(modelId, extras = {}) {
+  const meta = KNOWN_EMBEDDING_DIMENSIONS[modelId];
+  if (meta) {
+    return {
+      id: modelId,
+      provider: meta.provider,
+      label: meta.label,
+      dimensions: meta.defaultDimension,
+    };
+  }
+  return {
+    id: modelId,
+    provider: extras.provider || providerFromId(modelId),
+    label: extras.label || modelId,
+    dimensions: extras.dimensions ?? null,
+  };
+}
+
+export function formatEmbeddingChoice(model) {
+  return `${model.label} (${model.id}, ${model.dimensions})`;
+}
+
+function providerFromName(name) {
+  if (name === "Amazon") return "aws";
+  if (name === "Cohere") return "cohere";
+  return null;
+}
+
+function providerFromId(id) {
+  if (String(id).startsWith("amazon.")) return "aws";
+  if (String(id).startsWith("cohere.")) return "cohere";
+  return null;
+}
+
+/** SKU variants (…:0:512) are not valid Knowledge Base embeddingModelArn values. */
+export function isEmbeddingSkuVariant(modelId) {
+  return (String(modelId).match(/:/g) || []).length >= 2;
+}
+
+export function parseEmbeddingCatalog(payload) {
+  const summaries = payload?.modelSummaries;
+  if (!Array.isArray(summaries)) return [];
+  const seen = new Set();
+  const models = [];
+  for (const row of summaries) {
+    const id = row?.modelId;
+    if (!id || seen.has(id)) continue;
+    if (row.modelLifecycle?.status && row.modelLifecycle.status !== "ACTIVE") {
+      continue;
+    }
+    if (isEmbeddingSkuVariant(id)) continue;
+    const provider = providerFromName(row.providerName) || providerFromId(id);
+    if (provider !== "aws" && provider !== "cohere") continue;
+    seen.add(id);
+    models.push(catalogEntry(id, { provider, label: row.modelName }));
+  }
+  models.sort((a, b) =>
+    a.provider === b.provider
+      ? a.label.localeCompare(b.label)
+      : a.provider.localeCompare(b.provider)
+  );
+  return models;
+}
+
+export function listEmbeddingModels({ exec, env, region } = {}) {
+  if (!exec || !region) {
+    return { models: fallbackEmbeddingModels(), source: "fallback", warning: null };
+  }
+  const result = exec({
+    command: "aws",
+    args: [
+      "bedrock",
+      "list-foundation-models",
+      "--by-output-modality",
+      "EMBEDDING",
+      "--region",
+      region,
+      "--output",
+      "json",
+    ],
+    env,
+  });
+  if (!result.ok) {
+    return {
+      models: fallbackEmbeddingModels(),
+      source: "fallback",
+      warning: "Could not list Bedrock embedding models — showing defaults.",
+    };
+  }
+  try {
+    const listed = parseEmbeddingCatalog(JSON.parse(result.stdout || "{}"));
+    const models = allEmbeddingModels(listed);
+    if (listed.length === 0) {
+      return {
+        models,
+        source: "fallback",
+        warning: "No embedding models returned by Bedrock — showing defaults.",
+      };
+    }
+    return { models, source: "live", warning: null };
+  } catch {
+    return {
+      models: fallbackEmbeddingModels(),
+      source: "fallback",
+      warning: "Could not list Bedrock embedding models — showing defaults.",
+    };
+  }
+}

package/src/env-file.js

@@ -0,0 +1,140 @@
+import { chmod, mkdir, readFile, writeFile } from "node:fs/promises";
+import { dirname } from "node:path";
+import {
+  ALLOW_PUBLIC_SIGNUP,
+  APP_MODE,
+  BILLING_ENABLED,
+  DRIVER_NEON,
+  DRIVER_POSTGRES,
+} from "./defaults.js";
+import { ownPublicUrl } from "./hosted-url.js";
+
+export function inferDriver(url) {
+  if (!url) return DRIVER_POSTGRES;
+  return url.includes("neon.tech") ? DRIVER_NEON : DRIVER_POSTGRES;
+}
+
+export function inferPrepare(url) {
+  if (!url) return true;
+  return !/pooler\.supabase\.com/i.test(url);
+}
+
+export function quoteShell(value) {
+  return `"${String(value).replace(/["\\$`]/g, "\\$&")}"`;
+}
+
+export function renderDeployEnv(values) {
+  const lines = [
+    "# Written by `context101 init`. Gitignored. chmod 600.",
+    "# Deploy with `context101 deploy` — never raw `cdk deploy`.",
+    "",
+    `CTX_TOKEN=${quoteShell(values.CTX_TOKEN)}`,
+  ];
+
+  if (values.CTX_GH_TOKEN) {
+    lines.push(`CTX_GH_TOKEN=${quoteShell(values.CTX_GH_TOKEN)}`);
+  } else {
+    lines.push(
+      "# CTX_GH_TOKEN omitted — only needed if REPOSITORY is set (Amplify watches a repo)."
+    );
+  }
+
+  lines.push("");
+  if (values.AWS_PROFILE) {
+    lines.push(`AWS_PROFILE=${quoteShell(values.AWS_PROFILE)}`);
+  }
+  if (values.AWS_ACCESS_KEY_ID) {
+    lines.push(`AWS_ACCESS_KEY_ID=${quoteShell(values.AWS_ACCESS_KEY_ID)}`);
+  }
+  if (values.AWS_SECRET_ACCESS_KEY) {
+    lines.push(`AWS_SECRET_ACCESS_KEY=${quoteShell(values.AWS_SECRET_ACCESS_KEY)}`);
+  }
+  if (values.AWS_REGION) {
+    lines.push(`AWS_REGION=${quoteShell(values.AWS_REGION)}`);
+  }
+
+  lines.push("");
+  if (values.CREATE_RDS) {
+    lines.push("# DATABASE_URL omitted — CDK creates RDS Postgres.");
+    lines.push(`CREATE_RDS=${quoteShell("true")}`);
+    lines.push(
+      `DATABASE_DRIVER=${quoteShell(values.DATABASE_DRIVER ?? DRIVER_POSTGRES)}`
+    );
+    lines.push(`DATABASE_PREPARE=${quoteShell("true")}`);
+  } else {
+    lines.push(`DATABASE_URL=${quoteShell(values.DATABASE_URL)}`);
+    lines.push(`DATABASE_DRIVER=${quoteShell(values.DATABASE_DRIVER)}`);
+    lines.push(
+      `DATABASE_PREPARE=${quoteShell(values.DATABASE_PREPARE ? "true" : "false")}`
+    );
+  }
+
+  lines.push("");
+  lines.push(`BETTER_AUTH_SECRET=${quoteShell(values.BETTER_AUTH_SECRET)}`);
+  lines.push(
+    "# BETTER_AUTH_URL / APP_URL: omit to use Amplify's default domain"
+  );
+  lines.push(
+    "# (https://main.<app-id>.amplifyapp.com), or set a domain you own."
+  );
+  lines.push("# Never the hosted Context101 product.");
+  const betterAuthUrl = ownPublicUrl(values.BETTER_AUTH_URL);
+  const appUrl = ownPublicUrl(values.APP_URL);
+  if (betterAuthUrl) {
+    lines.push(`BETTER_AUTH_URL=${quoteShell(betterAuthUrl)}`);
+  }
+  if (appUrl) {
+    lines.push(`APP_URL=${quoteShell(appUrl)}`);
+  }
+  lines.push(`MCP_TOKEN_PEPPER=${quoteShell(values.MCP_TOKEN_PEPPER)}`);
+
+  lines.push("");
+  lines.push(`APP_MODE=${quoteShell(values.APP_MODE ?? APP_MODE)}`);
+  lines.push(
+    `ALLOW_PUBLIC_SIGNUP=${quoteShell(values.ALLOW_PUBLIC_SIGNUP ?? ALLOW_PUBLIC_SIGNUP)}`
+  );
+  lines.push(
+    `BILLING_ENABLED=${quoteShell(values.BILLING_ENABLED ?? BILLING_ENABLED)}`
+  );
+
+  if (values.REPOSITORY) {
+    lines.push("");
+    lines.push(`REPOSITORY=${quoteShell(values.REPOSITORY)}`);
+  }
+
+  if (values.EMBED_MODEL_ID) {
+    lines.push(`EMBED_MODEL_ID=${quoteShell(values.EMBED_MODEL_ID)}`);
+  }
+
+  lines.push("");
+  return `${lines.join("\n")}\n`;
+}
+
+export async function writeDeployEnv(filePath, values) {
+  await mkdir(dirname(filePath), { recursive: true });
+  await writeFile(filePath, renderDeployEnv(values), { encoding: "utf8", mode: 0o600 });
+  await chmod(filePath, 0o600);
+  return filePath;
+}
+
+export async function readExampleToken(examplePath) {
+  try {
+    const text = await readFile(examplePath, "utf8");
+    const match = text.match(/^CTX_TOKEN=(.*)$/m);
+    if (!match) return null;
+    return unquote(match[1]);
+  } catch {
+    return null;
+  }
+}
+
+function unquote(raw) {
+  const trimmed = raw.trim();
+  if (
+    (trimmed.startsWith('"') && trimmed.endsWith('"')) ||
+    (trimmed.startsWith("'") && trimmed.endsWith("'"))
+  ) {
+    return trimmed.slice(1, -1);
+  }
+  return trimmed;
+}

package/src/exec.js

@@ -0,0 +1,34 @@
+import { spawn, spawnSync } from "node:child_process";
+import { runCdk } from "./cdk-invoke.js";
+
+export function createExec(baseEnv = process.env) {
+  return function exec({ command, args = [], env, cwd, timeout = 15_000 }) {
+    const result = spawnSync(command, args, {
+      encoding: "utf8",
+      timeout,
+      env: { ...baseEnv, ...env },
+      cwd,
+    });
+    const stdout = result.stdout ?? "";
+    const stderr = result.stderr ?? "";
+    const code = result.status ?? (result.error ? 1 : 0);
+    return {
+      ok: code === 0,
+      code,
+      stdout: stdout.trim(),
+      stderr: stderr.trim(),
+      error: result.error ?? null,
+    };
+  };
+}
+
+export function commandExists(exec, name) {
+  const result = exec({ command: "sh", args: ["-c", `command -v ${name}`] });
+  return result.ok && Boolean(result.stdout);
+}
+
+export function runDeployWrapper(opts) {
+  return runCdk(opts);
+}
+
+export { spawn };

package/src/hosted-url.js

@@ -0,0 +1,35 @@
+const HOSTED_ZONE = "context101.dev";
+
+/**
+ * The hosted product lives on this zone. Self-host must use the operator's
+ * domain or Amplify's default `main.<app-id>.amplifyapp.com`.
+ */
+export function isHostedContext101Url(raw) {
+  if (raw == null) return false;
+  const value = String(raw).trim();
+  if (!value) return false;
+  const host = hostnameOf(value);
+  return host === HOSTED_ZONE || host.endsWith(`.${HOSTED_ZONE}`);
+}
+
+export function ownPublicUrl(raw) {
+  if (raw == null) return undefined;
+  const value = String(raw).trim();
+  if (!value || isHostedContext101Url(value)) return undefined;
+  return value;
+}
+
+function hostnameOf(value) {
+  try {
+    const withScheme = /^[a-z][a-z0-9+.-]*:\/\//i.test(value)
+      ? value
+      : `https://${value}`;
+    return new URL(withScheme).hostname.toLowerCase();
+  } catch {
+    return value
+      .replace(/^[a-z][a-z0-9+.-]*:\/\//i, "")
+      .split("/")[0]
+      .split(":")[0]
+      .toLowerCase();
+  }
+}