@shareai-lab/kode 1.1.12 → 1.1.13

package/dist/context.js

@@ -0,0 +1,259 @@
+import {
+  getCurrentProjectConfig,
+  saveCurrentProjectConfig
+} from "./utils/config.js";
+import { logError } from "./utils/log.js";
+import { getCodeStyle } from "./utils/style.js";
+import { getCwd } from "./utils/state.js";
+import { memoize, omit } from "lodash-es";
+import { LSTool } from "./tools/lsTool/lsTool.js";
+import { getIsGit } from "./utils/git.js";
+import { ripGrep } from "./utils/ripgrep.js";
+import * as path from "path";
+import { execFileNoThrow } from "./utils/execFileNoThrow.js";
+import { join } from "path";
+import { readFile } from "fs/promises";
+import { existsSync } from "fs";
+import { getModelManager } from "./utils/model.js";
+import { lastX } from "./utils/generators.js";
+import { getGitEmail } from "./utils/user.js";
+import { PROJECT_FILE } from "./constants/product.js";
+async function getClaudeFiles() {
+  const abortController = new AbortController();
+  const timeout = setTimeout(() => abortController.abort(), 3e3);
+  try {
+    const [codeContextFiles, claudeFiles] = await Promise.all([
+      ripGrep(
+        ["--files", "--glob", join("**", "*", PROJECT_FILE)],
+        getCwd(),
+        abortController.signal
+      ).catch(() => []),
+      ripGrep(
+        ["--files", "--glob", join("**", "*", "CLAUDE.md")],
+        getCwd(),
+        abortController.signal
+      ).catch(() => [])
+    ]);
+    const allFiles = [...codeContextFiles, ...claudeFiles];
+    if (!allFiles.length) {
+      return null;
+    }
+    const fileTypes = [];
+    if (codeContextFiles.length > 0) fileTypes.push("AGENTS.md");
+    if (claudeFiles.length > 0) fileTypes.push("CLAUDE.md");
+    return `NOTE: Additional project documentation files (${fileTypes.join(", ")}) were found. When working in these directories, make sure to read and follow the instructions in the corresponding files:
+${allFiles.map((_) => path.join(getCwd(), _)).map((_) => `- ${_}`).join("\n")}`;
+  } catch (error) {
+    logError(error);
+    return null;
+  } finally {
+    clearTimeout(timeout);
+  }
+}
+function setContext(key, value) {
+  const projectConfig = getCurrentProjectConfig();
+  const context = omit(
+    { ...projectConfig.context, [key]: value },
+    "codeStyle",
+    "directoryStructure"
+  );
+  saveCurrentProjectConfig({ ...projectConfig, context });
+}
+function removeContext(key) {
+  const projectConfig = getCurrentProjectConfig();
+  const context = omit(
+    projectConfig.context,
+    key,
+    "codeStyle",
+    "directoryStructure"
+  );
+  saveCurrentProjectConfig({ ...projectConfig, context });
+}
+const getReadme = memoize(async () => {
+  try {
+    const readmePath = join(getCwd(), "README.md");
+    if (!existsSync(readmePath)) {
+      return null;
+    }
+    const content = await readFile(readmePath, "utf-8");
+    return content;
+  } catch (e) {
+    logError(e);
+    return null;
+  }
+});
+const getProjectDocs = memoize(async () => {
+  try {
+    const cwd = getCwd();
+    const codeContextPath = join(cwd, "AGENTS.md");
+    const claudePath = join(cwd, "CLAUDE.md");
+    const docs = [];
+    if (existsSync(codeContextPath)) {
+      try {
+        const content = await readFile(codeContextPath, "utf-8");
+        docs.push(`# AGENTS.md
+
+${content}`);
+      } catch (e) {
+        logError(e);
+      }
+    }
+    if (existsSync(claudePath)) {
+      try {
+        const content = await readFile(claudePath, "utf-8");
+        docs.push(`# CLAUDE.md
+
+${content}`);
+      } catch (e) {
+        logError(e);
+      }
+    }
+    return docs.length > 0 ? docs.join("\n\n---\n\n") : null;
+  } catch (e) {
+    logError(e);
+    return null;
+  }
+});
+const getGitStatus = memoize(async () => {
+  if (process.env.NODE_ENV === "test") {
+    return null;
+  }
+  if (!await getIsGit()) {
+    return null;
+  }
+  try {
+    const [branch, mainBranch, status, log, authorLog] = await Promise.all([
+      execFileNoThrow(
+        "git",
+        ["branch", "--show-current"],
+        void 0,
+        void 0,
+        false
+      ).then(({ stdout }) => stdout.trim()),
+      execFileNoThrow(
+        "git",
+        ["rev-parse", "--abbrev-ref", "origin/HEAD"],
+        void 0,
+        void 0,
+        false
+      ).then(({ stdout }) => stdout.replace("origin/", "").trim()),
+      execFileNoThrow(
+        "git",
+        ["status", "--short"],
+        void 0,
+        void 0,
+        false
+      ).then(({ stdout }) => stdout.trim()),
+      execFileNoThrow(
+        "git",
+        ["log", "--oneline", "-n", "5"],
+        void 0,
+        void 0,
+        false
+      ).then(({ stdout }) => stdout.trim()),
+      execFileNoThrow(
+        "git",
+        [
+          "log",
+          "--oneline",
+          "-n",
+          "5",
+          "--author",
+          await getGitEmail() || ""
+        ],
+        void 0,
+        void 0,
+        false
+      ).then(({ stdout }) => stdout.trim())
+    ]);
+    const statusLines = status.split("\n").length;
+    const truncatedStatus = statusLines > 200 ? status.split("\n").slice(0, 200).join("\n") + '\n... (truncated because there are more than 200 lines. If you need more information, run "git status" using BashTool)' : status;
+    return `This is the git status at the start of the conversation. Note that this status is a snapshot in time, and will not update during the conversation.
+Current branch: ${branch}
+
+Main branch (you will usually use this for PRs): ${mainBranch}
+
+Status:
+${truncatedStatus || "(clean)"}
+
+Recent commits:
+${log}
+
+Your recent commits:
+${authorLog || "(no recent commits)"}`;
+  } catch (error) {
+    logError(error);
+    return null;
+  }
+});
+const getContext = memoize(
+  async () => {
+    const codeStyle = getCodeStyle();
+    const projectConfig = getCurrentProjectConfig();
+    const dontCrawl = projectConfig.dontCrawlDirectory;
+    const [gitStatus, directoryStructure, claudeFiles, readme, projectDocs] = await Promise.all([
+      getGitStatus(),
+      dontCrawl ? Promise.resolve("") : getDirectoryStructure(),
+      dontCrawl ? Promise.resolve("") : getClaudeFiles(),
+      getReadme(),
+      getProjectDocs()
+    ]);
+    return {
+      ...projectConfig.context,
+      ...directoryStructure ? { directoryStructure } : {},
+      ...gitStatus ? { gitStatus } : {},
+      ...codeStyle ? { codeStyle } : {},
+      ...claudeFiles ? { claudeFiles } : {},
+      ...readme ? { readme } : {},
+      ...projectDocs ? { projectDocs } : {}
+    };
+  }
+);
+const getDirectoryStructure = memoize(
+  async function() {
+    let lines;
+    try {
+      const abortController = new AbortController();
+      setTimeout(() => {
+        abortController.abort();
+      }, 1e3);
+      const model = getModelManager().getModelName("main");
+      const resultsGen = LSTool.call(
+        {
+          path: "."
+        },
+        {
+          abortController,
+          options: {
+            commands: [],
+            tools: [],
+            forkNumber: 0,
+            messageLogName: "unused",
+            maxThinkingTokens: 0
+          },
+          messageId: void 0,
+          readFileTimestamps: {}
+        }
+      );
+      const result = await lastX(resultsGen);
+      lines = result.data;
+    } catch (error) {
+      logError(error);
+      return "";
+    }
+    return `Below is a snapshot of this project's file structure at the start of the conversation. This snapshot will NOT update during the conversation.
+
+${lines}`;
+  }
+);
+export {
+  getClaudeFiles,
+  getContext,
+  getDirectoryStructure,
+  getGitStatus,
+  getProjectDocs,
+  getReadme,
+  removeContext,
+  setContext
+};
+//# sourceMappingURL=context.js.map

package/dist/context.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../src/context.ts"],
+  "sourcesContent": ["import {\n  getCurrentProjectConfig,\n  saveCurrentProjectConfig,\n} from './utils/config.js'\nimport { logError } from './utils/log'\nimport { getCodeStyle } from './utils/style'\nimport { getCwd } from './utils/state'\nimport { memoize, omit } from 'lodash-es'\nimport { LSTool } from './tools/lsTool/lsTool'\nimport { getIsGit } from './utils/git'\nimport { ripGrep } from './utils/ripgrep'\nimport * as path from 'path'\nimport { execFileNoThrow } from './utils/execFileNoThrow'\nimport { join } from 'path'\nimport { readFile } from 'fs/promises'\nimport { existsSync } from 'fs'\nimport { getModelManager } from './utils/model'\nimport { lastX } from './utils/generators'\nimport { getGitEmail } from './utils/user'\nimport { PROJECT_FILE } from './constants/product'\n/**\n * Find all AGENTS.md and CLAUDE.md files in the current working directory\n */\nexport async function getClaudeFiles(): Promise<string | null> {\n  const abortController = new AbortController()\n  const timeout = setTimeout(() => abortController.abort(), 3000)\n  try {\n    // Search for both AGENTS.md and CLAUDE.md files\n    const [codeContextFiles, claudeFiles] = await Promise.all([\n      ripGrep(\n        ['--files', '--glob', join('**', '*', PROJECT_FILE)],\n        getCwd(),\n        abortController.signal,\n      ).catch(() => []),\n      ripGrep(\n        ['--files', '--glob', join('**', '*', 'CLAUDE.md')],\n        getCwd(),\n        abortController.signal,\n      ).catch(() => []),\n    ])\n\n    const allFiles = [...codeContextFiles, ...claudeFiles]\n    if (!allFiles.length) {\n      return null\n    }\n\n    // Add instructions for additional project files\n    const fileTypes = []\n    if (codeContextFiles.length > 0) fileTypes.push('AGENTS.md')\n    if (claudeFiles.length > 0) fileTypes.push('CLAUDE.md')\n\n    return `NOTE: Additional project documentation files (${fileTypes.join(', ')}) were found. When working in these directories, make sure to read and follow the instructions in the corresponding files:\\n${allFiles\n      .map(_ => path.join(getCwd(), _))\n      .map(_ => `- ${_}`)\n      .join('\\n')}`\n  } catch (error) {\n    logError(error)\n    return null\n  } finally {\n    clearTimeout(timeout)\n  }\n}\n\nexport function setContext(key: string, value: string): void {\n  const projectConfig = getCurrentProjectConfig()\n  const context = omit(\n    { ...projectConfig.context, [key]: value },\n    'codeStyle',\n    'directoryStructure',\n  )\n  saveCurrentProjectConfig({ ...projectConfig, context })\n}\n\nexport function removeContext(key: string): void {\n  const projectConfig = getCurrentProjectConfig()\n  const context = omit(\n    projectConfig.context,\n    key,\n    'codeStyle',\n    'directoryStructure',\n  )\n  saveCurrentProjectConfig({ ...projectConfig, context })\n}\n\nexport const getReadme = memoize(async (): Promise<string | null> => {\n  try {\n    const readmePath = join(getCwd(), 'README.md')\n    if (!existsSync(readmePath)) {\n      return null\n    }\n    const content = await readFile(readmePath, 'utf-8')\n    return content\n  } catch (e) {\n    logError(e)\n    return null\n  }\n})\n\n/**\n * Get project documentation content (AGENTS.md and CLAUDE.md)\n */\nexport const getProjectDocs = memoize(async (): Promise<string | null> => {\n  try {\n    const cwd = getCwd()\n    const codeContextPath = join(cwd, 'AGENTS.md')\n    const claudePath = join(cwd, 'CLAUDE.md')\n\n    const docs = []\n\n    // Try to read AGENTS.md\n    if (existsSync(codeContextPath)) {\n      try {\n        const content = await readFile(codeContextPath, 'utf-8')\n        docs.push(`# AGENTS.md\\n\\n${content}`)\n      } catch (e) {\n        logError(e)\n      }\n    }\n\n    // Try to read CLAUDE.md\n    if (existsSync(claudePath)) {\n      try {\n        const content = await readFile(claudePath, 'utf-8')\n        docs.push(`# CLAUDE.md\\n\\n${content}`)\n      } catch (e) {\n        logError(e)\n      }\n    }\n\n    return docs.length > 0 ? docs.join('\\n\\n---\\n\\n') : null\n  } catch (e) {\n    logError(e)\n    return null\n  }\n})\n\nexport const getGitStatus = memoize(async (): Promise<string | null> => {\n  if (process.env.NODE_ENV === 'test') {\n    // Avoid cycles in tests\n    return null\n  }\n  if (!(await getIsGit())) {\n    return null\n  }\n\n  try {\n    const [branch, mainBranch, status, log, authorLog] = await Promise.all([\n      execFileNoThrow(\n        'git',\n        ['branch', '--show-current'],\n        undefined,\n        undefined,\n        false,\n      ).then(({ stdout }) => stdout.trim()),\n      execFileNoThrow(\n        'git',\n        ['rev-parse', '--abbrev-ref', 'origin/HEAD'],\n        undefined,\n        undefined,\n        false,\n      ).then(({ stdout }) => stdout.replace('origin/', '').trim()),\n      execFileNoThrow(\n        'git',\n        ['status', '--short'],\n        undefined,\n        undefined,\n        false,\n      ).then(({ stdout }) => stdout.trim()),\n      execFileNoThrow(\n        'git',\n        ['log', '--oneline', '-n', '5'],\n        undefined,\n        undefined,\n        false,\n      ).then(({ stdout }) => stdout.trim()),\n      execFileNoThrow(\n        'git',\n        [\n          'log',\n          '--oneline',\n          '-n',\n          '5',\n          '--author',\n          (await getGitEmail()) || '',\n        ],\n        undefined,\n        undefined,\n        false,\n      ).then(({ stdout }) => stdout.trim()),\n    ])\n    // Check if status has more than 200 lines\n    const statusLines = status.split('\\n').length\n    const truncatedStatus =\n      statusLines > 200\n        ? status.split('\\n').slice(0, 200).join('\\n') +\n          '\\n... (truncated because there are more than 200 lines. If you need more information, run \"git status\" using BashTool)'\n        : status\n\n    return `This is the git status at the start of the conversation. Note that this status is a snapshot in time, and will not update during the conversation.\\nCurrent branch: ${branch}\\n\\nMain branch (you will usually use this for PRs): ${mainBranch}\\n\\nStatus:\\n${truncatedStatus || '(clean)'}\\n\\nRecent commits:\\n${log}\\n\\nYour recent commits:\\n${authorLog || '(no recent commits)'}`\n  } catch (error) {\n    logError(error)\n    return null\n  }\n})\n\n/**\n * This context is prepended to each conversation, and cached for the duration of the conversation.\n */\nexport const getContext = memoize(\n  async (): Promise<{\n    [k: string]: string\n  }> => {\n    const codeStyle = getCodeStyle()\n    const projectConfig = getCurrentProjectConfig()\n    const dontCrawl = projectConfig.dontCrawlDirectory\n    const [gitStatus, directoryStructure, claudeFiles, readme, projectDocs] =\n      await Promise.all([\n        getGitStatus(),\n        dontCrawl ? Promise.resolve('') : getDirectoryStructure(),\n        dontCrawl ? Promise.resolve('') : getClaudeFiles(),\n        getReadme(),\n        getProjectDocs(),\n      ])\n    return {\n      ...projectConfig.context,\n      ...(directoryStructure ? { directoryStructure } : {}),\n      ...(gitStatus ? { gitStatus } : {}),\n      ...(codeStyle ? { codeStyle } : {}),\n      ...(claudeFiles ? { claudeFiles } : {}),\n      ...(readme ? { readme } : {}),\n      ...(projectDocs ? { projectDocs } : {}),\n    }\n  },\n)\n\n/**\n * Approximate directory structure, to orient Claude. Claude will start with this, then use\n * tools like LS and View to get more information.\n */\nexport const getDirectoryStructure = memoize(\n  async function (): Promise<string> {\n    let lines: string\n    try {\n      const abortController = new AbortController()\n      setTimeout(() => {\n        abortController.abort()\n      }, 1_000)\n      // \uD83D\uDD27 Fix: Use ModelManager instead of legacy function\n      const model = getModelManager().getModelName('main')\n      const resultsGen = LSTool.call(\n        {\n          path: '.',\n        },\n        {\n          abortController,\n          options: {\n            commands: [],\n            tools: [],\n            forkNumber: 0,\n            messageLogName: 'unused',\n            maxThinkingTokens: 0,\n          },\n          messageId: undefined,\n          readFileTimestamps: {},\n        },\n      )\n      const result = await lastX(resultsGen)\n      lines = result.data\n    } catch (error) {\n      logError(error)\n      return ''\n    }\n\n    return `Below is a snapshot of this project's file structure at the start of the conversation. This snapshot will NOT update during the conversation.\n\n${lines}`\n  },\n)\n"],
+  "mappings": "AAAA;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP,SAAS,gBAAgB;AACzB,SAAS,oBAAoB;AAC7B,SAAS,cAAc;AACvB,SAAS,SAAS,YAAY;AAC9B,SAAS,cAAc;AACvB,SAAS,gBAAgB;AACzB,SAAS,eAAe;AACxB,YAAY,UAAU;AACtB,SAAS,uBAAuB;AAChC,SAAS,YAAY;AACrB,SAAS,gBAAgB;AACzB,SAAS,kBAAkB;AAC3B,SAAS,uBAAuB;AAChC,SAAS,aAAa;AACtB,SAAS,mBAAmB;AAC5B,SAAS,oBAAoB;AAI7B,eAAsB,iBAAyC;AAC7D,QAAM,kBAAkB,IAAI,gBAAgB;AAC5C,QAAM,UAAU,WAAW,MAAM,gBAAgB,MAAM,GAAG,GAAI;AAC9D,MAAI;AAEF,UAAM,CAAC,kBAAkB,WAAW,IAAI,MAAM,QAAQ,IAAI;AAAA,MACxD;AAAA,QACE,CAAC,WAAW,UAAU,KAAK,MAAM,KAAK,YAAY,CAAC;AAAA,QACnD,OAAO;AAAA,QACP,gBAAgB;AAAA,MAClB,EAAE,MAAM,MAAM,CAAC,CAAC;AAAA,MAChB;AAAA,QACE,CAAC,WAAW,UAAU,KAAK,MAAM,KAAK,WAAW,CAAC;AAAA,QAClD,OAAO;AAAA,QACP,gBAAgB;AAAA,MAClB,EAAE,MAAM,MAAM,CAAC,CAAC;AAAA,IAClB,CAAC;AAED,UAAM,WAAW,CAAC,GAAG,kBAAkB,GAAG,WAAW;AACrD,QAAI,CAAC,SAAS,QAAQ;AACpB,aAAO;AAAA,IACT;AAGA,UAAM,YAAY,CAAC;AACnB,QAAI,iBAAiB,SAAS,EAAG,WAAU,KAAK,WAAW;AAC3D,QAAI,YAAY,SAAS,EAAG,WAAU,KAAK,WAAW;AAEtD,WAAO,iDAAiD,UAAU,KAAK,IAAI,CAAC;AAAA,EAA+H,SACxM,IAAI,OAAK,KAAK,KAAK,OAAO,GAAG,CAAC,CAAC,EAC/B,IAAI,OAAK,KAAK,CAAC,EAAE,EACjB,KAAK,IAAI,CAAC;AAAA,EACf,SAAS,OAAO;AACd,aAAS,KAAK;AACd,WAAO;AAAA,EACT,UAAE;AACA,iBAAa,OAAO;AAAA,EACtB;AACF;AAEO,SAAS,WAAW,KAAa,OAAqB;AAC3D,QAAM,gBAAgB,wBAAwB;AAC9C,QAAM,UAAU;AAAA,IACd,EAAE,GAAG,cAAc,SAAS,CAAC,GAAG,GAAG,MAAM;AAAA,IACzC;AAAA,IACA;AAAA,EACF;AACA,2BAAyB,EAAE,GAAG,eAAe,QAAQ,CAAC;AACxD;AAEO,SAAS,cAAc,KAAmB;AAC/C,QAAM,gBAAgB,wBAAwB;AAC9C,QAAM,UAAU;AAAA,IACd,cAAc;AAAA,IACd;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACA,2BAAyB,EAAE,GAAG,eAAe,QAAQ,CAAC;AACxD;AAEO,MAAM,YAAY,QAAQ,YAAoC;AACnE,MAAI;AACF,UAAM,aAAa,KAAK,OAAO,GAAG,WAAW;AAC7C,QAAI,CAAC,WAAW,UAAU,GAAG;AAC3B,aAAO;AAAA,IACT;AACA,UAAM,UAAU,MAAM,SAAS,YAAY,OAAO;AAClD,WAAO;AAAA,EACT,SAAS,GAAG;AACV,aAAS,CAAC;AACV,WAAO;AAAA,EACT;AACF,CAAC;AAKM,MAAM,iBAAiB,QAAQ,YAAoC;AACxE,MAAI;AACF,UAAM,MAAM,OAAO;AACnB,UAAM,kBAAkB,KAAK,KAAK,WAAW;AAC7C,UAAM,aAAa,KAAK,KAAK,WAAW;AAExC,UAAM,OAAO,CAAC;AAGd,QAAI,WAAW,eAAe,GAAG;AAC/B,UAAI;AACF,cAAM,UAAU,MAAM,SAAS,iBAAiB,OAAO;AACvD,aAAK,KAAK;AAAA;AAAA,EAAkB,OAAO,EAAE;AAAA,MACvC,SAAS,GAAG;AACV,iBAAS,CAAC;AAAA,MACZ;AAAA,IACF;AAGA,QAAI,WAAW,UAAU,GAAG;AAC1B,UAAI;AACF,cAAM,UAAU,MAAM,SAAS,YAAY,OAAO;AAClD,aAAK,KAAK;AAAA;AAAA,EAAkB,OAAO,EAAE;AAAA,MACvC,SAAS,GAAG;AACV,iBAAS,CAAC;AAAA,MACZ;AAAA,IACF;AAEA,WAAO,KAAK,SAAS,IAAI,KAAK,KAAK,aAAa,IAAI;AAAA,EACtD,SAAS,GAAG;AACV,aAAS,CAAC;AACV,WAAO;AAAA,EACT;AACF,CAAC;AAEM,MAAM,eAAe,QAAQ,YAAoC;AACtE,MAAI,QAAQ,IAAI,aAAa,QAAQ;AAEnC,WAAO;AAAA,EACT;AACA,MAAI,CAAE,MAAM,SAAS,GAAI;AACvB,WAAO;AAAA,EACT;AAEA,MAAI;AACF,UAAM,CAAC,QAAQ,YAAY,QAAQ,KAAK,SAAS,IAAI,MAAM,QAAQ,IAAI;AAAA,MACrE;AAAA,QACE;AAAA,QACA,CAAC,UAAU,gBAAgB;AAAA,QAC3B;AAAA,QACA;AAAA,QACA;AAAA,MACF,EAAE,KAAK,CAAC,EAAE,OAAO,MAAM,OAAO,KAAK,CAAC;AAAA,MACpC;AAAA,QACE;AAAA,QACA,CAAC,aAAa,gBAAgB,aAAa;AAAA,QAC3C;AAAA,QACA;AAAA,QACA;AAAA,MACF,EAAE,KAAK,CAAC,EAAE,OAAO,MAAM,OAAO,QAAQ,WAAW,EAAE,EAAE,KAAK,CAAC;AAAA,MAC3D;AAAA,QACE;AAAA,QACA,CAAC,UAAU,SAAS;AAAA,QACpB;AAAA,QACA;AAAA,QACA;AAAA,MACF,EAAE,KAAK,CAAC,EAAE,OAAO,MAAM,OAAO,KAAK,CAAC;AAAA,MACpC;AAAA,QACE;AAAA,QACA,CAAC,OAAO,aAAa,MAAM,GAAG;AAAA,QAC9B;AAAA,QACA;AAAA,QACA;AAAA,MACF,EAAE,KAAK,CAAC,EAAE,OAAO,MAAM,OAAO,KAAK,CAAC;AAAA,MACpC;AAAA,QACE;AAAA,QACA;AAAA,UACE;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACC,MAAM,YAAY,KAAM;AAAA,QAC3B;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,EAAE,KAAK,CAAC,EAAE,OAAO,MAAM,OAAO,KAAK,CAAC;AAAA,IACtC,CAAC;AAED,UAAM,cAAc,OAAO,MAAM,IAAI,EAAE;AACvC,UAAM,kBACJ,cAAc,MACV,OAAO,MAAM,IAAI,EAAE,MAAM,GAAG,GAAG,EAAE,KAAK,IAAI,IAC1C,2HACA;AAEN,WAAO;AAAA,kBAAuK,MAAM;AAAA;AAAA,mDAAwD,UAAU;AAAA;AAAA;AAAA,EAAgB,mBAAmB,SAAS;AAAA;AAAA;AAAA,EAAwB,GAAG;AAAA;AAAA;AAAA,EAA6B,aAAa,qBAAqB;AAAA,EAC9X,SAAS,OAAO;AACd,aAAS,KAAK;AACd,WAAO;AAAA,EACT;AACF,CAAC;AAKM,MAAM,aAAa;AAAA,EACxB,YAEM;AACJ,UAAM,YAAY,aAAa;AAC/B,UAAM,gBAAgB,wBAAwB;AAC9C,UAAM,YAAY,cAAc;AAChC,UAAM,CAAC,WAAW,oBAAoB,aAAa,QAAQ,WAAW,IACpE,MAAM,QAAQ,IAAI;AAAA,MAChB,aAAa;AAAA,MACb,YAAY,QAAQ,QAAQ,EAAE,IAAI,sBAAsB;AAAA,MACxD,YAAY,QAAQ,QAAQ,EAAE,IAAI,eAAe;AAAA,MACjD,UAAU;AAAA,MACV,eAAe;AAAA,IACjB,CAAC;AACH,WAAO;AAAA,MACL,GAAG,cAAc;AAAA,MACjB,GAAI,qBAAqB,EAAE,mBAAmB,IAAI,CAAC;AAAA,MACnD,GAAI,YAAY,EAAE,UAAU,IAAI,CAAC;AAAA,MACjC,GAAI,YAAY,EAAE,UAAU,IAAI,CAAC;AAAA,MACjC,GAAI,cAAc,EAAE,YAAY,IAAI,CAAC;AAAA,MACrC,GAAI,SAAS,EAAE,OAAO,IAAI,CAAC;AAAA,MAC3B,GAAI,cAAc,EAAE,YAAY,IAAI,CAAC;AAAA,IACvC;AAAA,EACF;AACF;AAMO,MAAM,wBAAwB;AAAA,EACnC,iBAAmC;AACjC,QAAI;AACJ,QAAI;AACF,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,iBAAW,MAAM;AACf,wBAAgB,MAAM;AAAA,MACxB,GAAG,GAAK;AAER,YAAM,QAAQ,gBAAgB,EAAE,aAAa,MAAM;AACnD,YAAM,aAAa,OAAO;AAAA,QACxB;AAAA,UACE,MAAM;AAAA,QACR;AAAA,QACA;AAAA,UACE;AAAA,UACA,SAAS;AAAA,YACP,UAAU,CAAC;AAAA,YACX,OAAO,CAAC;AAAA,YACR,YAAY;AAAA,YACZ,gBAAgB;AAAA,YAChB,mBAAmB;AAAA,UACrB;AAAA,UACA,WAAW;AAAA,UACX,oBAAoB,CAAC;AAAA,QACvB;AAAA,MACF;AACA,YAAM,SAAS,MAAM,MAAM,UAAU;AACrC,cAAQ,OAAO;AAAA,IACjB,SAAS,OAAO;AACd,eAAS,KAAK;AACd,aAAO;AAAA,IACT;AAEA,WAAO;AAAA;AAAA,EAET,KAAK;AAAA,EACL;AACF;",
+  "names": []
+}

package/dist/cost-tracker.js

@@ -0,0 +1,76 @@
+import chalk from "chalk";
+import { useEffect } from "react";
+import { formatDuration } from "./utils/format.js";
+import {
+  getCurrentProjectConfig,
+  saveCurrentProjectConfig
+} from "./utils/config.js";
+import { SESSION_ID } from "./utils/log.js";
+const STATE = {
+  totalCost: 0,
+  totalAPIDuration: 0,
+  startTime: Date.now()
+};
+function addToTotalCost(cost, duration) {
+  STATE.totalCost += cost;
+  STATE.totalAPIDuration += duration;
+}
+function getTotalCost() {
+  return STATE.totalCost;
+}
+function getTotalDuration() {
+  return Date.now() - STATE.startTime;
+}
+function getTotalAPIDuration() {
+  return STATE.totalAPIDuration;
+}
+function formatCost(cost) {
+  return `$${cost > 0.5 ? round(cost, 100).toFixed(2) : cost.toFixed(4)}`;
+}
+function formatTotalCost() {
+  return chalk.grey(
+    `Total cost: ${formatCost(STATE.totalCost)}
+Total duration (API): ${formatDuration(STATE.totalAPIDuration)}
+Total duration (wall): ${formatDuration(getTotalDuration())}`
+  );
+}
+function useCostSummary() {
+  useEffect(() => {
+    const f = () => {
+      process.stdout.write("\n" + formatTotalCost() + "\n");
+      const projectConfig = getCurrentProjectConfig();
+      saveCurrentProjectConfig({
+        ...projectConfig,
+        lastCost: STATE.totalCost,
+        lastAPIDuration: STATE.totalAPIDuration,
+        lastDuration: getTotalDuration(),
+        lastSessionId: SESSION_ID
+      });
+    };
+    process.on("exit", f);
+    return () => {
+      process.off("exit", f);
+    };
+  }, []);
+}
+function round(number, precision) {
+  return Math.round(number * precision) / precision;
+}
+function resetStateForTests() {
+  if (process.env.NODE_ENV !== "test") {
+    throw new Error("resetStateForTests can only be called in tests");
+  }
+  STATE.startTime = Date.now();
+  STATE.totalCost = 0;
+  STATE.totalAPIDuration = 0;
+}
+export {
+  addToTotalCost,
+  formatTotalCost,
+  getTotalAPIDuration,
+  getTotalCost,
+  getTotalDuration,
+  resetStateForTests,
+  useCostSummary
+};
+//# sourceMappingURL=cost-tracker.js.map

package/dist/cost-tracker.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../src/cost-tracker.ts"],
+  "sourcesContent": ["import chalk from 'chalk'\nimport { useEffect } from 'react'\nimport { formatDuration } from './utils/format'\nimport {\n  getCurrentProjectConfig,\n  saveCurrentProjectConfig,\n} from './utils/config.js'\nimport { SESSION_ID } from './utils/log'\n\n// DO NOT ADD MORE STATE HERE OR BORIS WILL CURSE YOU\nconst STATE: {\n  totalCost: number\n  totalAPIDuration: number\n  startTime: number\n} = {\n  totalCost: 0,\n  totalAPIDuration: 0,\n  startTime: Date.now(),\n}\n\nexport function addToTotalCost(cost: number, duration: number): void {\n  STATE.totalCost += cost\n  STATE.totalAPIDuration += duration\n}\n\nexport function getTotalCost(): number {\n  return STATE.totalCost\n}\n\nexport function getTotalDuration(): number {\n  return Date.now() - STATE.startTime\n}\n\nexport function getTotalAPIDuration(): number {\n  return STATE.totalAPIDuration\n}\n\nfunction formatCost(cost: number): string {\n  return `$${cost > 0.5 ? round(cost, 100).toFixed(2) : cost.toFixed(4)}`\n}\n\nexport function formatTotalCost(): string {\n  return chalk.grey(\n    `Total cost: ${formatCost(STATE.totalCost)}\nTotal duration (API): ${formatDuration(STATE.totalAPIDuration)}\nTotal duration (wall): ${formatDuration(getTotalDuration())}`,\n  )\n}\n\nexport function useCostSummary(): void {\n  useEffect(() => {\n    const f = () => {\n      process.stdout.write('\\n' + formatTotalCost() + '\\n')\n\n      // Save last cost and duration to project config\n      const projectConfig = getCurrentProjectConfig()\n      saveCurrentProjectConfig({\n        ...projectConfig,\n        lastCost: STATE.totalCost,\n        lastAPIDuration: STATE.totalAPIDuration,\n        lastDuration: getTotalDuration(),\n        lastSessionId: SESSION_ID,\n      })\n    }\n    process.on('exit', f)\n    return () => {\n      process.off('exit', f)\n    }\n  }, [])\n}\n\nfunction round(number: number, precision: number): number {\n  return Math.round(number * precision) / precision\n}\n\n// Only used in tests\nexport function resetStateForTests(): void {\n  if (process.env.NODE_ENV !== 'test') {\n    throw new Error('resetStateForTests can only be called in tests')\n  }\n  STATE.startTime = Date.now()\n  STATE.totalCost = 0\n  STATE.totalAPIDuration = 0\n}\n"],
+  "mappings": "AAAA,OAAO,WAAW;AAClB,SAAS,iBAAiB;AAC1B,SAAS,sBAAsB;AAC/B;AAAA,EACE;AAAA,EACA;AAAA,OACK;AACP,SAAS,kBAAkB;AAG3B,MAAM,QAIF;AAAA,EACF,WAAW;AAAA,EACX,kBAAkB;AAAA,EAClB,WAAW,KAAK,IAAI;AACtB;AAEO,SAAS,eAAe,MAAc,UAAwB;AACnE,QAAM,aAAa;AACnB,QAAM,oBAAoB;AAC5B;AAEO,SAAS,eAAuB;AACrC,SAAO,MAAM;AACf;AAEO,SAAS,mBAA2B;AACzC,SAAO,KAAK,IAAI,IAAI,MAAM;AAC5B;AAEO,SAAS,sBAA8B;AAC5C,SAAO,MAAM;AACf;AAEA,SAAS,WAAW,MAAsB;AACxC,SAAO,IAAI,OAAO,MAAM,MAAM,MAAM,GAAG,EAAE,QAAQ,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;AACvE;AAEO,SAAS,kBAA0B;AACxC,SAAO,MAAM;AAAA,IACX,eAAe,WAAW,MAAM,SAAS,CAAC;AAAA,wBACtB,eAAe,MAAM,gBAAgB,CAAC;AAAA,yBACrC,eAAe,iBAAiB,CAAC,CAAC;AAAA,EACzD;AACF;AAEO,SAAS,iBAAuB;AACrC,YAAU,MAAM;AACd,UAAM,IAAI,MAAM;AACd,cAAQ,OAAO,MAAM,OAAO,gBAAgB,IAAI,IAAI;AAGpD,YAAM,gBAAgB,wBAAwB;AAC9C,+BAAyB;AAAA,QACvB,GAAG;AAAA,QACH,UAAU,MAAM;AAAA,QAChB,iBAAiB,MAAM;AAAA,QACvB,cAAc,iBAAiB;AAAA,QAC/B,eAAe;AAAA,MACjB,CAAC;AAAA,IACH;AACA,YAAQ,GAAG,QAAQ,CAAC;AACpB,WAAO,MAAM;AACX,cAAQ,IAAI,QAAQ,CAAC;AAAA,IACvB;AAAA,EACF,GAAG,CAAC,CAAC;AACP;AAEA,SAAS,MAAM,QAAgB,WAA2B;AACxD,SAAO,KAAK,MAAM,SAAS,SAAS,IAAI;AAC1C;AAGO,SAAS,qBAA2B;AACzC,MAAI,QAAQ,IAAI,aAAa,QAAQ;AACnC,UAAM,IAAI,MAAM,gDAAgD;AAAA,EAClE;AACA,QAAM,YAAY,KAAK,IAAI;AAC3B,QAAM,YAAY;AAClB,QAAM,mBAAmB;AAC3B;",
+  "names": []
+}