onadit 0.0.1 → 1.0.0

package/dist/src/onboarding/search-by-meaning.js

@@ -0,0 +1,163 @@
+/**
+ * "Search by meaning": give this install an embeddings key Onadit pays for.
+ *
+ * Onadit.app's Privacy screen asks for it, with the user's consent, because it
+ * changes what leaves the Mac: from then on the text of imported and captured
+ * chats is sent to Voyage AI (MongoDB) to be embedded. The flow:
+ *   1. ask Onadit's relay Worker for a key, sending only a random install ID
+ *   2. prove the key works: a one-line embedding of the size Onadit stores
+ *   3. point config.embedding at the key's endpoint and model, then save the
+ *      key to ~/.rift.env (config first: a key without its endpoint would be
+ *      sent to the wrong API, an endpoint without a key is just keyword mode)
+ *   4. mark the archive for re-embedding and restart the engine, which reads
+ *      the key only at boot
+ * The key is never logged or returned to the caller.
+ */
+import crypto from "node:crypto";
+import fs from "node:fs";
+import path from "node:path";
+import { EMBEDDING_DIMENSIONS } from "../providers/types.js";
+import { defaultRiftEnvPath } from "../runtime/rift-env.js";
+import { writeEnvFile } from "./env-file.js";
+import { validateVoyageKey } from "./voyage-validate.js";
+export const DEFAULT_KEY_RELAY_URL = "https://rift-relay.clem-rog.workers.dev";
+const KEY_PATH = "/v1/voyage-key";
+const STATE_FILE = "search-key.json";
+function statePath(configPath) {
+    return path.join(path.dirname(configPath), STATE_FILE);
+}
+function readState(configPath) {
+    try {
+        const parsed = JSON.parse(fs.readFileSync(statePath(configPath), "utf8"));
+        return typeof parsed.install_id === "string" ? parsed : null;
+    }
+    catch {
+        return null;
+    }
+}
+function writeJsonAtomic(file, value) {
+    const tmp = `${file}.tmp.${process.pid}`;
+    fs.writeFileSync(tmp, JSON.stringify(value, null, 2) + "\n", { encoding: "utf8", mode: 0o600 });
+    fs.renameSync(tmp, file);
+}
+function writeState(configPath, state) {
+    writeJsonAtomic(statePath(configPath), state);
+}
+/** Point config.embedding at the provisioned endpoint, keeping a backup. */
+function persistEmbedding(configPath, embedding, now) {
+    const raw = fs.readFileSync(configPath, "utf8");
+    fs.writeFileSync(`${configPath}.bak.${now.toISOString().replace(/[:.]/g, "-")}`, raw, {
+        encoding: "utf8",
+        mode: 0o600,
+    });
+    const parsed = JSON.parse(raw);
+    parsed["embedding"] = { provider: "voyage", ...embedding };
+    writeJsonAtomic(configPath, parsed);
+}
+function isProvisionedKey(value) {
+    const v = value;
+    return (!!v &&
+        typeof v.api_key === "string" &&
+        typeof v.base_url === "string" &&
+        // The key travels to this URL: https only, except a local test server.
+        /^(https:\/\/|http:\/\/(127\.0\.0\.1|localhost)[:/])/.test(v.base_url) &&
+        typeof v.model === "string" &&
+        (v.output_dimension === undefined || typeof v.output_dimension === "number"));
+}
+const MESSAGES = {
+    capacity: "Onadit can't turn on search by meaning for more people right now. Try again tomorrow.",
+    unavailable: "Onadit's key service didn't answer. Try again in a few minutes.",
+    network: "Onadit couldn't reach its key service. Check your connection and try again.",
+    key_rejected: "The new key didn't work. Try again in a few minutes.",
+};
+function failed(reason) {
+    return { status: "failed", reason, message: MESSAGES[reason] };
+}
+export async function enableSearchByMeaning(deps) {
+    if (deps.keyPresent)
+        return { status: "already_on" };
+    const now = deps.now?.() ?? new Date();
+    const f = deps.fetchFn ?? fetch;
+    const state = readState(deps.configPath) ?? { install_id: crypto.randomUUID() };
+    writeState(deps.configPath, state);
+    let res;
+    try {
+        res = await f(`${(deps.relayUrl ?? DEFAULT_KEY_RELAY_URL).replace(/\/+$/, "")}${KEY_PATH}`, {
+            method: "POST",
+            headers: { "content-type": "application/json" },
+            body: JSON.stringify({ install_id: state.install_id }),
+            signal: AbortSignal.timeout(20_000),
+        });
+    }
+    catch {
+        return failed("network");
+    }
+    if (res.status === 429)
+        return failed("capacity");
+    if (!res.ok)
+        return failed("unavailable");
+    let key;
+    try {
+        key = await res.json();
+    }
+    catch {
+        return failed("unavailable");
+    }
+    if (!isProvisionedKey(key) ||
+        (key.output_dimension !== undefined && key.output_dimension !== EMBEDDING_DIMENSIONS)) {
+        return failed("unavailable");
+    }
+    const check = await (deps.validate ?? validateVoyageKey)({
+        apiKey: key.api_key,
+        model: key.model,
+        baseUrl: key.base_url,
+        ...(key.output_dimension ? { outputDimension: key.output_dimension } : {}),
+        expectDimensions: EMBEDDING_DIMENSIONS,
+        ...(deps.fetchFn ? { fetchFn: deps.fetchFn } : {}),
+    });
+    if (!check.ok)
+        return failed("key_rejected");
+    persistEmbedding(deps.configPath, {
+        model: key.model,
+        base_url: key.base_url,
+        ...(key.output_dimension ? { output_dimension: key.output_dimension } : {}),
+    }, now);
+    writeEnvFile({ filePath: deps.envPath ?? defaultRiftEnvPath(), key: "VOYAGE_API_KEY", value: key.api_key, now: () => now });
+    writeState(deps.configPath, {
+        install_id: state.install_id,
+        provisioned_at: now.toISOString(),
+        backfill_pending: true,
+    });
+    return { status: "on" };
+}
+/** Whether the archive still waits for its first real embeddings. */
+export function backfillPending(configPath) {
+    return readState(configPath)?.backfill_pending === true;
+}
+/** Call once the re-embed job is queued, so the next boot doesn't repeat it. */
+export function clearBackfillPending(configPath) {
+    const state = readState(configPath);
+    if (state?.backfill_pending)
+        writeState(configPath, { ...state, backfill_pending: false });
+}
+/**
+ * A key the user pasted was checked against Voyage's own API, so drop a
+ * Onadit-provided endpoint (Atlas) from config.json. Model and output size
+ * stay: Voyage serves the same models, so stored vectors still match.
+ * Returns whether anything changed.
+ */
+export function useVoyageOwnEndpoint(configPath, now = new Date()) {
+    const raw = fs.readFileSync(configPath, "utf8");
+    const parsed = JSON.parse(raw);
+    const embedding = parsed["embedding"];
+    if (!embedding || embedding["base_url"] === undefined)
+        return false;
+    fs.writeFileSync(`${configPath}.bak.${now.toISOString().replace(/[:.]/g, "-")}`, raw, {
+        encoding: "utf8",
+        mode: 0o600,
+    });
+    delete embedding["base_url"];
+    writeJsonAtomic(configPath, parsed);
+    return true;
+}
+//# sourceMappingURL=search-by-meaning.js.map

package/dist/src/onboarding/search-by-meaning.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"search-by-meaning.js","sourceRoot":"","sources":["../../../src/onboarding/search-by-meaning.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AACH,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAEzD,MAAM,CAAC,MAAM,qBAAqB,GAAG,yCAAyC,CAAC;AAC/E,MAAM,QAAQ,GAAG,gBAAgB,CAAC;AAClC,MAAM,UAAU,GAAG,iBAAiB,CAAC;AA+BrC,SAAS,SAAS,CAAC,UAAkB;IACnC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,UAAU,CAAC,CAAC;AACzD,CAAC;AAED,SAAS,SAAS,CAAC,UAAkB;IACnC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,CAA4B,CAAC;QACrG,OAAO,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAE,MAAyB,CAAC,CAAC,CAAC,IAAI,CAAC;IACnF,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,IAAY,EAAE,KAAc;IACnD,MAAM,GAAG,GAAG,GAAG,IAAI,QAAQ,OAAO,CAAC,GAAG,EAAE,CAAC;IACzC,EAAE,CAAC,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IAChG,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,UAAU,CAAC,UAAkB,EAAE,KAAqB;IAC3D,eAAe,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;AAChD,CAAC;AAED,4EAA4E;AAC5E,SAAS,gBAAgB,CACvB,UAAkB,EAClB,SAAyE,EACzE,GAAS;IAET,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAChD,EAAE,CAAC,aAAa,CAAC,GAAG,UAAU,QAAQ,GAAG,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE;QACpF,QAAQ,EAAE,MAAM;QAChB,IAAI,EAAE,KAAK;KACZ,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAA4B,CAAC;IAC1D,MAAM,CAAC,WAAW,CAAC,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,SAAS,EAAE,CAAC;IAC3D,eAAe,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;AACtC,CAAC;AAUD,SAAS,gBAAgB,CAAC,KAAc;IACtC,MAAM,CAAC,GAAG,KAAuC,CAAC;IAClD,OAAO,CACL,CAAC,CAAC,CAAC;QACH,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ;QAC7B,OAAO,CAAC,CAAC,QAAQ,KAAK,QAAQ;QAC9B,uEAAuE;QACvE,qDAAqD,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC;QACtE,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ;QAC3B,CAAC,CAAC,CAAC,gBAAgB,KAAK,SAAS,IAAI,OAAO,CAAC,CAAC,gBAAgB,KAAK,QAAQ,CAAC,CAC7E,CAAC;AACJ,CAAC;AAED,MAAM,QAAQ,GAAG;IACf,QAAQ,EAAE,uFAAuF;IACjG,WAAW,EAAE,iEAAiE;IAC9E,OAAO,EAAE,6EAA6E;IACtF,YAAY,EAAE,sDAAsD;CAC5D,CAAC;AAEX,SAAS,MAAM,CAAC,MAA6B;IAC3C,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;AACjE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,IAAsB;IAChE,IAAI,IAAI,CAAC,UAAU;QAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC;IACrD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,IAAI,IAAI,IAAI,EAAE,CAAC;IACvC,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,IAAI,KAAK,CAAC;IAEhC,MAAM,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC;IAChF,UAAU,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IAEnC,IAAI,GAAa,CAAC;IAClB,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,qBAAqB,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,GAAG,QAAQ,EAAE,EAAE;YAC1F,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC;YACtD,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;SACpC,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC;IAC3B,CAAC;IACD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG;QAAE,OAAO,MAAM,CAAC,UAAU,CAAC,CAAC;IAClD,IAAI,CAAC,GAAG,CAAC,EAAE;QAAE,OAAO,MAAM,CAAC,aAAa,CAAC,CAAC;IAE1C,IAAI,GAAY,CAAC;IACjB,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;IACzB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,MAAM,CAAC,aAAa,CAAC,CAAC;IAC/B,CAAC;IACD,IACE,CAAC,gBAAgB,CAAC,GAAG,CAAC;QACtB,CAAC,GAAG,CAAC,gBAAgB,KAAK,SAAS,IAAI,GAAG,CAAC,gBAAgB,KAAK,oBAAoB,CAAC,EACrF,CAAC;QACD,OAAO,MAAM,CAAC,aAAa,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,IAAI,iBAAiB,CAAC,CAAC;QACvD,MAAM,EAAE,GAAG,CAAC,OAAO;QACnB,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,OAAO,EAAE,GAAG,CAAC,QAAQ;QACrB,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC1E,gBAAgB,EAAE,oBAAoB;QACtC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACnD,CAAC,CAAC;IACH,IAAI,CAAC,KAAK,CAAC,EAAE;QAAE,OAAO,MAAM,CAAC,cAAc,CAAC,CAAC;IAE7C,gBAAgB,CACd,IAAI,CAAC,UAAU,EACf;QACE,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC5E,EACD,GAAG,CACJ,CAAC;IACF,YAAY,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,OAAO,IAAI,kBAAkB,EAAE,EAAE,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;IAC5H,UAAU,CAAC,IAAI,CAAC,UAAU,EAAE;QAC1B,UAAU,EAAE,KAAK,CAAC,UAAU;QAC5B,cAAc,EAAE,GAAG,CAAC,WAAW,EAAE;QACjC,gBAAgB,EAAE,IAAI;KACvB,CAAC,CAAC;IACH,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;AAC1B,CAAC;AAED,qEAAqE;AACrE,MAAM,UAAU,eAAe,CAAC,UAAkB;IAChD,OAAO,SAAS,CAAC,UAAU,CAAC,EAAE,gBAAgB,KAAK,IAAI,CAAC;AAC1D,CAAC;AAED,gFAAgF;AAChF,MAAM,UAAU,oBAAoB,CAAC,UAAkB;IACrD,MAAM,KAAK,GAAG,SAAS,CAAC,UAAU,CAAC,CAAC;IACpC,IAAI,KAAK,EAAE,gBAAgB;QAAE,UAAU,CAAC,UAAU,EAAE,EAAE,GAAG,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;AAC7F,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,oBAAoB,CAAC,UAAkB,EAAE,MAAY,IAAI,IAAI,EAAE;IAC7E,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAChD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAA4B,CAAC;IAC1D,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAAwC,CAAC;IAC7E,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,SAAS;QAAE,OAAO,KAAK,CAAC;IACpE,EAAE,CAAC,aAAa,CAAC,GAAG,UAAU,QAAQ,GAAG,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE;QACpF,QAAQ,EAAE,MAAM;QAChB,IAAI,EAAE,KAAK;KACZ,CAAC,CAAC;IACH,OAAO,SAAS,CAAC,UAAU,CAAC,CAAC;IAC7B,eAAe,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IACpC,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/src/onboarding/setup-page.d.ts

@@ -0,0 +1,7 @@
+export interface SetupPageOptions {
+    token: string;
+    version: string;
+}
+export declare function renderSetupPage(opts: SetupPageOptions): string;
+export declare function renderExpiredSetupPage(): string;
+//# sourceMappingURL=setup-page.d.ts.map

package/dist/src/onboarding/setup-page.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup-page.d.ts","sourceRoot":"","sources":["../../../src/onboarding/setup-page.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;CACjB;AAwCD,wBAAgB,eAAe,CAAC,IAAI,EAAE,gBAAgB,GAAG,MAAM,CA2xB9D;AAED,wBAAgB,sBAAsB,IAAI,MAAM,CAE/C"}