@bcmyguest/openmem 0.0.0-stage → 0.8.0

package/wizard/lib/validate-setup.mjs

@@ -0,0 +1,62 @@
+import fs from "node:fs";
+import path from "node:path";
+import { envValue, indexerStateDir } from "./paths.mjs";
+import { DEFAULT_PROVIDER_ID, listAllProviders, validateManifest } from "./providers/registry.mjs";
+
+/**
+ * Provider-neutral checks: manifest validity, wizard.conf, the stack .env host ids,
+ * index.yml and the shared indexer state dir. Provider checks come from the provider's
+ * `extendValidation` hook, which appends to ctx.validationErrors / ctx.validationChecks.
+ *
+ * @param {import("./run-wizard.mjs").WizardContext} ctx
+ * @returns {{ ok: boolean, errors: string[], checks: string[] }}
+ */
+export function validateMemorySetup(ctx) {
+  const errors = [];
+  const checks = [];
+
+  if (ctx.memoryProvider) {
+    for (const e of validateManifest(ctx.memoryProvider)) {
+      errors.push(`manifest ${ctx.memoryProvider.id}: ${e}`);
+    }
+  }
+
+  if (!fs.existsSync(ctx.wizardConf)) {
+    errors.push(`missing ${ctx.wizardConf}`);
+  } else {
+    const provider = envValue(ctx.wizardConf, "MEMORY_PROVIDER") || DEFAULT_PROVIDER_ID;
+    checks.push(`wizard.conf MEMORY_PROVIDER=${provider}`);
+    if (ctx.memoryProvider && provider !== ctx.memoryProvider.id) {
+      errors.push(
+        `wizard.conf MEMORY_PROVIDER=${provider} does not match active ${ctx.memoryProvider.id}`,
+      );
+    }
+  }
+
+  const stackEnv = path.join(ctx.repoRoot, ".env");
+  if (!fs.existsSync(stackEnv)) errors.push(`missing repo ${stackEnv}`);
+  else {
+    for (const key of ["HOST_UID", "HOST_GID"]) {
+      if (!envValue(stackEnv, key)) errors.push(`repo .env missing ${key}`);
+    }
+    checks.push("repo .env host ids");
+  }
+
+  if (!fs.existsSync(ctx.indexFile)) errors.push(`missing ${ctx.indexFile}`);
+  else checks.push("index.yml present");
+
+  const stateDir = indexerStateDir(ctx);
+  if (!fs.existsSync(stateDir)) errors.push(`missing indexer state dir ${stateDir}`);
+  else checks.push("indexer state mount point");
+
+  return { ok: errors.length === 0, errors, checks };
+}
+
+/** @returns {{ ok: boolean, errors: string[] }} */
+export function validateAllProviderManifests() {
+  const errors = [];
+  for (const m of listAllProviders()) {
+    for (const e of validateManifest(m)) errors.push(`${m.id}: ${e}`);
+  }
+  return { ok: errors.length === 0, errors };
+}

package/wizard/lib/wizard-action.mjs

@@ -0,0 +1,34 @@
+/** @typedef {"configure"|"up"|"down"} WizardAction */
+
+/** @returns {WizardAction} */
+export function actionDefault() {
+  const fromEnv = process.env.PM_WIZARD_ACTION || process.env.WIZARD_ACTION;
+  if (fromEnv === "up" || fromEnv === "down" || fromEnv === "configure") return fromEnv;
+  return "configure";
+}
+
+/**
+ * @param {string[]} argv
+ * @returns {WizardAction}
+ */
+export function parseActionFromArgv(argv) {
+  if (argv.includes("--down")) return "down";
+  if (argv.includes("--up")) return "up";
+  const fromEnv = process.env.PM_WIZARD_ACTION || process.env.WIZARD_ACTION;
+  if (fromEnv === "up" || fromEnv === "down") return fromEnv;
+  return "configure";
+}
+
+/** @param {WizardAction} action */
+export function actionLabel(action) {
+  switch (action) {
+    case "configure":
+      return "Configure and start (full setup)";
+    case "up":
+      return "Start stack (existing config)";
+    case "down":
+      return "Stop stack";
+    default:
+      return action;
+  }
+}

package/wizard/lib/wizard-conf.mjs

@@ -0,0 +1,71 @@
+import fs from "node:fs";
+import { ensureHsDir, envValue } from "./paths.mjs";
+import { syncPmConfig } from "./pm-config.mjs";
+import { DEFAULT_PROVIDER_ID } from "./providers/registry.mjs";
+
+/** Meta keys owned outside provider extras; rewrite keeps them unless `extra` overrides. */
+const META_KEYS = ["NOTEBOOKLM_HOME"];
+
+/**
+ * Write wizard.conf (MEMORY_PROVIDER plus provider-owned keys) and the mise-like
+ * config.toml. Does nothing in a dry run; callers print their own dry-run line.
+ * Preserves meta keys already on disk (for example NOTEBOOKLM_HOME) when not in `extra`.
+ *
+ * @param {import("./run-wizard.mjs").WizardContext} ctx
+ * @param {Record<string, string>} [extra] provider keys, written after MEMORY_PROVIDER in order
+ * @param {{ pmSections?: Record<string, Record<string, unknown>> }} [opts]
+ */
+export function writeWizardConf(ctx, extra = {}, opts = {}) {
+  if (ctx.dry) return;
+  ensureHsDir(ctx);
+  /** @type {Record<string, string>} */
+  const preserved = {};
+  if (fs.existsSync(ctx.wizardConf)) {
+    for (const key of META_KEYS) {
+      if (key in extra) continue;
+      const v = envValue(ctx.wizardConf, key);
+      if (v) preserved[key] = v;
+    }
+  }
+  const providerId = ctx.memoryProvider?.id || DEFAULT_PROVIDER_ID;
+  let body = `MEMORY_PROVIDER=${providerId}\n`;
+  for (const [key, value] of Object.entries({ ...preserved, ...extra })) {
+    if (!/^[A-Z][A-Z0-9_]*$/.test(key)) throw new Error(`invalid wizard.conf key: ${key}`);
+    body += `${key}=${String(value ?? "").replace(/[\r\n]+/g, " ")}\n`;
+  }
+  fs.writeFileSync(ctx.wizardConf, body, { mode: 0o644 });
+  ctx.wizardConfWritten = true;
+  syncPmConfig(ctx, opts.pmSections || {});
+}
+
+/**
+ * Merge keys into the existing wizard.conf without dropping MEMORY_PROVIDER or other lines.
+ * Does nothing in a dry run.
+ *
+ * @param {import("./run-wizard.mjs").WizardContext} ctx
+ * @param {Record<string, string>} updates
+ */
+export function upsertWizardConf(ctx, updates) {
+  if (ctx.dry) return;
+  ensureHsDir(ctx);
+  /** @type {Record<string, string>} */
+  const keys = {};
+  if (fs.existsSync(ctx.wizardConf)) {
+    for (const line of fs.readFileSync(ctx.wizardConf, "utf8").split("\n")) {
+      const m = /^([A-Z][A-Z0-9_]*)=(.*)$/.exec(line);
+      if (m) keys[m[1]] = m[2];
+    }
+  }
+  const providerId = ctx.memoryProvider?.id || keys.MEMORY_PROVIDER || DEFAULT_PROVIDER_ID;
+  for (const [key, value] of Object.entries(updates)) {
+    if (!/^[A-Z][A-Z0-9_]*$/.test(key)) throw new Error(`invalid wizard.conf key: ${key}`);
+    keys[key] = String(value ?? "").replace(/[\r\n]+/g, " ");
+  }
+  keys.MEMORY_PROVIDER = providerId;
+  let body = `MEMORY_PROVIDER=${providerId}\n`;
+  for (const [key, value] of Object.entries(keys)) {
+    if (key === "MEMORY_PROVIDER") continue;
+    body += `${key}=${value}\n`;
+  }
+  fs.writeFileSync(ctx.wizardConf, body, { mode: 0o644 });
+}