@shareai-lab/kode 1.2.0 → 2.0.2

package/dist/context.js

@@ -1,259 +0,0 @@
-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

@@ -1,7 +0,0 @@
-{
-  "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.map

@@ -1,7 +0,0 @@
-{
-  "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": []
-}