@crewx/sdk 0.9.0-rc.6 → 0.9.0-rc.60
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +1 -1
- package/dist/activity-log/builder.d.ts +55 -6
- package/dist/config/canonicalize-agent-options.d.ts +17 -0
- package/dist/config/global-settings.d.ts +19 -0
- package/dist/config/migrate-grok-mode.d.ts +6 -0
- package/dist/config/model-compat.d.ts +16 -0
- package/dist/config/models.generated.d.ts +55 -0
- package/dist/config/models.remote.d.ts +4 -0
- package/dist/config/overdrive.d.ts +31 -0
- package/dist/config/pricing.d.ts +9 -5
- package/dist/config/pricing.generated.d.ts +4 -0
- package/dist/config/pricing.types.d.ts +4 -0
- package/dist/config/timeout.config.d.ts +3 -0
- package/dist/config/wi-boards.d.ts +19 -0
- package/dist/context-budget/index.d.ts +105 -0
- package/dist/conversation/index.d.ts +1 -1
- package/dist/conversation/to-template-messages.d.ts +7 -0
- package/dist/conversation/types.d.ts +1 -0
- package/dist/desktop/command-runner.d.ts +4 -0
- package/dist/desktop/default-adapters.d.ts +4 -0
- package/dist/desktop/global-install.d.ts +19 -0
- package/dist/desktop/index.d.ts +2 -0
- package/dist/desktop/index.js +3 -0
- package/dist/desktop/install.d.ts +5 -0
- package/dist/desktop/lnk.d.ts +22 -0
- package/dist/desktop/manifest.d.ts +21 -0
- package/dist/desktop/types.d.ts +67 -0
- package/dist/desktop/version.d.ts +1 -0
- package/dist/desktop/windows-paths.d.ts +21 -0
- package/dist/desktop/wrapper.d.ts +18 -0
- package/dist/esm/desktop/index.js +3 -0
- package/dist/esm/index.js +240 -102
- package/dist/esm/plugins/index.js +150 -90
- package/dist/esm/publish/index.js +1 -0
- package/dist/esm/repository/index.js +126 -66
- package/dist/esm/testing/index.js +1 -1
- package/dist/events/types.d.ts +3 -0
- package/dist/facade/Crewx.d.ts +28 -0
- package/dist/index.browser.d.ts +6 -1
- package/dist/index.browser.js +6 -2
- package/dist/index.d.ts +37 -9
- package/dist/index.js +240 -102
- package/dist/internal/windows-spawn.d.ts +10 -0
- package/dist/migrations/0008_careful_black_bird.sql +17 -0
- package/dist/migrations/0009_hot_dark_phoenix.sql +1 -0
- package/dist/migrations/0010_analyze_stats.sql +6 -0
- package/dist/migrations/0011_far_black_tarantula.sql +1 -0
- package/dist/migrations/0012_blue_turbo.sql +23 -0
- package/dist/migrations/0013_hesitant_adam_destine.sql +15 -0
- package/dist/migrations/0014_organic_mongu.sql +73 -0
- package/dist/migrations/meta/0008_snapshot.json +1327 -0
- package/dist/migrations/meta/0009_snapshot.json +1335 -0
- package/dist/migrations/meta/0010_snapshot.json +1335 -0
- package/dist/migrations/meta/0011_snapshot.json +1342 -0
- package/dist/migrations/meta/0012_snapshot.json +1494 -0
- package/dist/migrations/meta/0013_snapshot.json +1592 -0
- package/dist/migrations/meta/0014_snapshot.json +1975 -0
- package/dist/migrations/meta/_journal.json +49 -0
- package/dist/paths.d.ts +1 -0
- package/dist/plugin/types.d.ts +1 -0
- package/dist/plugins/index.js +150 -90
- package/dist/provider/acp/adapters/claude.d.ts +0 -1
- package/dist/provider/acp/connection.d.ts +4 -0
- package/dist/provider/acp/meta.d.ts +15 -0
- package/dist/provider/bridge.d.ts +16 -1
- package/dist/provider/classify-task-error.d.ts +2 -0
- package/dist/provider/cli/adapter.types.d.ts +6 -0
- package/dist/provider/cli/adapters/cli-knob.util.d.ts +6 -0
- package/dist/provider/cli/adapters/grok.d.ts +2 -0
- package/dist/provider/cli/adapters/index.d.ts +1 -0
- package/dist/provider/cli/index.d.ts +1 -1
- package/dist/provider/cli/meta.d.ts +8 -1
- package/dist/provider/errors.d.ts +1 -0
- package/dist/provider/idle-watchdog.d.ts +19 -0
- package/dist/provider/models.d.ts +26 -24
- package/dist/provider/order.d.ts +1 -1
- package/dist/provider/parse-usage.d.ts +6 -0
- package/dist/provider/provider-selection.d.ts +8 -0
- package/dist/provider/redact-technical-message.d.ts +1 -0
- package/dist/provider/task-error.types.d.ts +16 -0
- package/dist/publish/auth-boundary.d.ts +22 -0
- package/dist/publish/auth-json.d.ts +11 -0
- package/dist/publish/constants.d.ts +6 -0
- package/dist/publish/index.d.ts +30 -0
- package/dist/publish/index.js +1 -0
- package/dist/publish/manifest.d.ts +17 -0
- package/dist/publish/marketplace-api.constants.d.ts +1 -0
- package/dist/publish/pack.d.ts +2 -0
- package/dist/publish/scan.d.ts +9 -0
- package/dist/publish/secrets.d.ts +6 -0
- package/dist/publish/upload.d.ts +18 -0
- package/dist/repository/agent-suggestion.repository.d.ts +34 -0
- package/dist/repository/index.d.ts +7 -1
- package/dist/repository/index.js +126 -66
- package/dist/repository/notification.repository.d.ts +58 -0
- package/dist/repository/task.repository.d.ts +94 -0
- package/dist/repository/thread.repository.d.ts +16 -2
- package/dist/repository/usage-report.repository.d.ts +27 -0
- package/dist/schema/agent-suggestions.d.ts +199 -0
- package/dist/schema/index.d.ts +5 -0
- package/dist/schema/notifications.d.ts +303 -0
- package/dist/schema/ui-action-receipts.d.ts +303 -0
- package/dist/schema/ui-instances.d.ts +267 -0
- package/dist/schema/usage-reports.d.ts +252 -0
- package/dist/testing/index.d.ts +1 -0
- package/dist/testing/index.js +1 -1
- package/dist/types/index.d.ts +448 -3
- package/dist/ui-card/index.d.ts +2 -0
- package/dist/ui-card/legacy-card-parser.d.ts +11 -0
- package/dist/ui-instance/definition.d.ts +6 -0
- package/dist/ui-instance/errors.d.ts +17 -0
- package/dist/ui-instance/index.d.ts +6 -0
- package/dist/ui-instance/json.d.ts +5 -0
- package/dist/ui-instance/runtime.d.ts +13 -0
- package/dist/ui-instance/store.d.ts +65 -0
- package/dist/ui-instance/types.d.ts +200 -0
- package/dist/utils/crewx-home.d.ts +1 -0
- package/dist/utils/id.d.ts +1 -1
- package/dist/utils/timestamp.d.ts +1 -0
- package/dist/wi/create.d.ts +48 -0
- package/dist/wi/select.d.ts +20 -0
- package/package.json +19 -3
- package/templates/agents/default.yaml +175 -117
- package/templates/documents/crewx-manual.md +138 -138
- package/templates/workflows/wi-default-flow.yaml +65 -0
- package/templates/workflows/wi-plan-flow.yaml +140 -0
package/dist/repository/index.js
CHANGED
|
@@ -1,14 +1,47 @@
|
|
|
1
|
-
'use strict';var
|
|
1
|
+
'use strict';var xe=require('fs'),it=require('path'),on=require('os'),drizzleOrm=require('drizzle-orm'),crypto=require('crypto'),sqliteCore=require('drizzle-orm/sqlite-core');function _interopDefault(e){return e&&e.__esModule?e:{default:e}}function _interopNamespace(e){if(e&&e.__esModule)return e;var n=Object.create(null);if(e){Object.keys(e).forEach(function(k){if(k!=='default'){var d=Object.getOwnPropertyDescriptor(e,k);Object.defineProperty(n,k,d.get?d:{enumerable:true,get:function(){return e[k]}});}})}n.default=e;return Object.freeze(n)}var xe__default=/*#__PURE__*/_interopDefault(xe);var it__namespace=/*#__PURE__*/_interopNamespace(it);var on__default=/*#__PURE__*/_interopDefault(on);var It=(i=>typeof require<"u"?require:typeof Proxy<"u"?new Proxy(i,{get:(t,e)=>(typeof require<"u"?require:t)[e]}):i)(function(i){if(typeof require<"u")return require.apply(this,arguments);throw Error('Dynamic require of "'+i+'" is not supported')});var N=class{resolveDbPath(){return process.env.CREWX_DB?process.env.CREWX_DB:process.env.CREWX_TRACES_DB?process.env.CREWX_TRACES_DB:it.join(on__default.default.homedir(),".crewx","crewx.db")}resolveDbPaths(){return [this.resolveDbPath()]}isMissingTableError(t){return t instanceof Error&&/no such table:/i.test(t.message)}dbExists(t){return xe.existsSync(t??this.resolveDbPath())}};var d=class extends Error{code;cause;constructor(t,e,n){super(e),this.name="RepositoryError",this.code=t,this.cause=n,Object.setPrototypeOf(this,new.target.prototype);}};function w(i){let t=It("better-sqlite3"),{drizzle:e}=It("drizzle-orm/better-sqlite3"),n=new t(i);return n.exec("PRAGMA journal_mode = WAL"),n.exec("PRAGMA busy_timeout = 5000"),n.exec("PRAGMA foreign_keys = ON"),n.exec("PRAGMA analysis_limit = 400"),n.exec("PRAGMA optimize"),{db:e(n),runRaw:(r,s=[])=>n.prepare(r).run(...s),close:()=>n.close()}}var ue=new Set,sn={agent_id:"TEXT",status:"TEXT DEFAULT 'running'",started_at:"TEXT",trace_id:"TEXT",parent_task_id:"TEXT",crewx_version:"TEXT",pid:"INTEGER",thread_id:"TEXT",workspace_id:"TEXT",workspace_ref:"TEXT",workspace_name:"TEXT",project_id:"TEXT",project_name:"TEXT"};function Wt(i,t){return (i.get(`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='${t}'`)?.cnt??0)>0}function an(i,t){if(t>0||!Wt(i,"tasks"))return;let e=i.all("PRAGMA table_info(tasks)"),n=new Set(e.map(r=>r.name));for(let[r,s]of Object.entries(sn))n.has(r)||i.run(`ALTER TABLE tasks ADD COLUMN ${r} ${s}`);}var dn={"0002_normalize_task_names":{workspace_name:"TEXT",project_name:"TEXT"}};function cn(i,t,e){if(!Wt(i,"__drizzle_migrations")||!Wt(i,"tasks"))return;let n=i.all(e`SELECT hash FROM __drizzle_migrations`),r=new Set(n.map(a=>a.hash)),s=JSON.parse(xe.readFileSync(it__namespace.default.join(t,"meta/_journal.json"),"utf-8"));for(let a of s.entries){let c=dn[a.tag];if(!c)continue;let u=it__namespace.default.join(t,`${a.tag}.sql`);if(!xe.existsSync(u))continue;let l=xe.readFileSync(u,"utf-8"),_=crypto.createHash("sha256").update(l).digest("hex");if(r.has(_))continue;let f=i.all("PRAGMA table_info(tasks)"),m=new Set(f.map(R=>R.name));for(let[R,L]of Object.entries(c))m.has(R)||(i.run(`ALTER TABLE tasks ADD COLUMN ${R} ${L}`),m.add(R));}}function ln(i,t,e){let n=i.all(e`SELECT hash FROM __drizzle_migrations`),r=new Set(n.map(a=>a.hash)),s=JSON.parse(xe.readFileSync(it__namespace.default.join(t,"meta/_journal.json"),"utf-8"));for(let a of s.entries){let c=it__namespace.default.join(t,`${a.tag}.sql`);if(!xe.existsSync(c))continue;let u=xe.readFileSync(c,"utf-8"),l=crypto.createHash("sha256").update(u).digest("hex");if(r.has(l))continue;let _=/ALTER\s+TABLE\s+[`"]?(\w+)[`"]?\s+ADD\s+[`"]?(\w+)[`"]?/gi,f=[],m;for(;(m=_.exec(u))!==null;)f.push({table:m[1],column:m[2]});if(f.length===0||!u.split(/-->\s*statement-breakpoint/).map(U=>U.trim()).filter(Boolean).every(U=>/^ALTER\s+TABLE\s+.+\s+ADD\s+/i.test(U)))continue;f.every(({table:U,column:F})=>i.all(`PRAGMA table_info("${U}")`).some(Q=>Q.name===F))&&i.run(e`INSERT INTO __drizzle_migrations (hash, created_at) VALUES (${l}, ${a.when})`);}}function bt(i){let{migrate:t}=It("drizzle-orm/better-sqlite3/migrator"),{sql:e}=It("drizzle-orm"),n=[it__namespace.default.join(__dirname,"../migrations"),it__namespace.default.join(__dirname,"migrations"),it__namespace.default.join(__dirname,"../../../../drizzle/migrations"),it__namespace.default.join(process.cwd(),"drizzle/migrations")],r=n.find(l=>xe.existsSync(it__namespace.default.join(l,"meta/_journal.json")));if(!r)throw new Error(`migrations folder not found. Searched:
|
|
2
2
|
${n.join(`
|
|
3
|
-
`)}`);let s=a.get(e`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='__drizzle_migrations'`),o=0;s?.cnt&&(o=a.get(e`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0),ae(a,o),s?.cnt&&(ce(a,r,e),le(a,r,e)),t(a,{migrationsFolder:r});let u=(a.get(e`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0)-o;if(u>0){let c=s?.cnt?"Database migrated":"Database initialized";console.error(`[crewx] ${c} (${u} migration${u>1?"s":""} applied).`);}}function x(a,t){Ct.has(t)||(Q(a),Ct.add(t));}var _=sqliteCore.sqliteTable("workspaces",{id:sqliteCore.text("id").primaryKey(),slug:sqliteCore.text("slug").notNull().unique(),name:sqliteCore.text("name").notNull(),workspace_path:sqliteCore.text("workspace_path"),description:sqliteCore.text("description"),is_active:sqliteCore.integer("is_active").notNull().default(1),created_at:sqliteCore.text("created_at").notNull(),updated_at:sqliteCore.text("updated_at").notNull()});var i=sqliteCore.sqliteTable("tasks",{id:sqliteCore.text("id").primaryKey(),agent_id:sqliteCore.text("agent_id").notNull(),user_id:sqliteCore.text("user_id"),prompt:sqliteCore.text("prompt").notNull(),mode:sqliteCore.text("mode").notNull().default("execute"),status:sqliteCore.text("status").notNull().default("running"),result:sqliteCore.text("result"),error:sqliteCore.text("error"),started_at:sqliteCore.text("started_at").notNull(),completed_at:sqliteCore.text("completed_at"),duration_ms:sqliteCore.integer("duration_ms"),metadata:sqliteCore.text("metadata"),workspace_id:sqliteCore.text("workspace_id"),trace_id:sqliteCore.text("trace_id"),parent_task_id:sqliteCore.text("parent_task_id"),caller_agent_id:sqliteCore.text("caller_agent_id"),model:sqliteCore.text("model"),platform:sqliteCore.text("platform").default("cli"),crewx_version:sqliteCore.text("crewx_version"),input_tokens:sqliteCore.integer("input_tokens").default(0),output_tokens:sqliteCore.integer("output_tokens").default(0),cost_usd:sqliteCore.real("cost_usd").default(0),pid:sqliteCore.integer("pid"),rendered_prompt:sqliteCore.text("rendered_prompt"),command:sqliteCore.text("command"),coding_agent_command:sqliteCore.text("coding_agent_command"),exit_code:sqliteCore.integer("exit_code"),logs:sqliteCore.text("logs"),thread_id:sqliteCore.text("thread_id"),workspace_ref:sqliteCore.text("workspace_ref"),project_id:sqliteCore.text("project_id"),project_ref:sqliteCore.text("project_ref"),cached_input_tokens:sqliteCore.integer("cached_input_tokens").default(0),run_epoch:sqliteCore.integer("run_epoch").default(0)},a=>({idx_tasks_agent_id:sqliteCore.index("idx_tasks_agent_id").on(a.agent_id),idx_tasks_status:sqliteCore.index("idx_tasks_status").on(a.status),idx_tasks_started_at:sqliteCore.index("idx_tasks_started_at").on(a.started_at),idx_tasks_trace_id:sqliteCore.index("idx_tasks_trace_id").on(a.trace_id),idx_tasks_parent_task_id:sqliteCore.index("idx_tasks_parent_task_id").on(a.parent_task_id),idx_tasks_crewx_version:sqliteCore.index("idx_tasks_crewx_version").on(a.crewx_version),idx_tasks_pid:sqliteCore.index("idx_tasks_pid").on(a.pid),idx_tasks_thread_id:sqliteCore.index("idx_tasks_thread_id").on(a.thread_id),idx_tasks_workspace_id:sqliteCore.index("idx_tasks_workspace_id").on(a.workspace_id),idx_tasks_workspace_ref:sqliteCore.index("idx_tasks_workspace_ref").on(a.workspace_ref),idx_tasks_project_id:sqliteCore.index("idx_tasks_project_id").on(a.project_id),idx_tasks_ws_started:sqliteCore.index("idx_tasks_ws_started").on(a.workspace_id,a.started_at)}));var p=sqliteCore.sqliteTable("threads",{id:sqliteCore.text("id").primaryKey(),workspace_id:sqliteCore.text("workspace_id").references(()=>_.id,{onDelete:"set null"}),platform:sqliteCore.text("platform").notNull().default("cli"),title:sqliteCore.text("title"),first_message:sqliteCore.text("first_message"),last_message:sqliteCore.text("last_message"),message_count:sqliteCore.integer("message_count").notNull().default(0),created_at:sqliteCore.text("created_at").notNull(),updated_at:sqliteCore.text("updated_at").notNull(),metadata:sqliteCore.text("metadata"),title_locked:sqliteCore.integer("title_locked").notNull().default(0),pinned:sqliteCore.integer("pinned").notNull().default(0),starred:sqliteCore.integer("starred").notNull().default(0)},a=>({idx_threads_updated_at:sqliteCore.index("idx_threads_updated_at").on(a.updated_at),idx_threads_workspace_id:sqliteCore.index("idx_threads_workspace_id").on(a.workspace_id)}));var U=sqliteCore.sqliteTable("spans",{id:sqliteCore.text("id").primaryKey(),task_id:sqliteCore.text("task_id").references(()=>i.id,{onDelete:"set null"}),parent_span_id:sqliteCore.text("parent_span_id").references(()=>U.id,{onDelete:"set null"}),name:sqliteCore.text("name").notNull(),kind:sqliteCore.text("kind").notNull().default("internal"),status:sqliteCore.text("status").notNull().default("ok"),started_at:sqliteCore.text("started_at").notNull(),completed_at:sqliteCore.text("completed_at"),duration_ms:sqliteCore.integer("duration_ms"),input:sqliteCore.text("input"),output:sqliteCore.text("output"),error:sqliteCore.text("error"),attributes:sqliteCore.text("attributes")},a=>({idx_spans_task_id:sqliteCore.index("idx_spans_task_id").on(a.task_id),idx_spans_parent_span_id:sqliteCore.index("idx_spans_parent_span_id").on(a.parent_span_id)}));var $=sqliteCore.sqliteTable("tool_calls",{id:sqliteCore.text("id").primaryKey(),task_id:sqliteCore.text("task_id").references(()=>i.id,{onDelete:"cascade"}),session_id:sqliteCore.text("session_id"),tool_name:sqliteCore.text("tool_name").notNull(),files:sqliteCore.text("files"),input:sqliteCore.text("input"),output:sqliteCore.text("output"),duration_ms:sqliteCore.integer("duration_ms"),timestamp:sqliteCore.text("timestamp").notNull()},a=>({idx_tool_calls_task_id:sqliteCore.index("idx_tool_calls_task_id").on(a.task_id),idx_tool_calls_tool_name:sqliteCore.index("idx_tool_calls_tool_name").on(a.tool_name),idx_tool_calls_timestamp:sqliteCore.index("idx_tool_calls_timestamp").on(a.timestamp)}));var L=sqliteCore.sqliteTable("thread_boxes",{id:sqliteCore.text("id").primaryKey(),thread_id:sqliteCore.text("thread_id").notNull().references(()=>p.id,{onDelete:"cascade"}),seq:sqliteCore.integer("seq").notNull(),first_task_id:sqliteCore.text("first_task_id").notNull(),mid_task_id:sqliteCore.text("mid_task_id").notNull(),last_task_id:sqliteCore.text("last_task_id").notNull(),task_count:sqliteCore.integer("task_count").notNull(),summary:sqliteCore.text("summary"),source_tokens:sqliteCore.integer("source_tokens").notNull(),summary_tokens:sqliteCore.integer("summary_tokens"),created_at:sqliteCore.text("created_at").notNull()},a=>({idx_thread_boxes_thread_id:sqliteCore.index("idx_thread_boxes_thread_id").on(a.thread_id),idx_thread_boxes_seq:sqliteCore.index("idx_thread_boxes_seq").on(a.thread_id,a.seq),uniq_thread_boxes_thread_seq:sqliteCore.unique().on(a.thread_id,a.seq)}));var Z=sqliteCore.sqliteTable("request_logs",{id:sqliteCore.text("id").primaryKey(),path:sqliteCore.text("path").notNull(),method:sqliteCore.text("method").notNull(),status_code:sqliteCore.integer("status_code").notNull(),duration_ms:sqliteCore.integer("duration_ms").notNull(),ip:sqliteCore.text("ip"),request_headers:sqliteCore.text("request_headers"),response_headers:sqliteCore.text("response_headers"),request_body:sqliteCore.text("request_body"),response_body:sqliteCore.text("response_body"),query:sqliteCore.text("query"),user_id:sqliteCore.text("user_id"),project_id:sqliteCore.text("project_id"),partition_key:sqliteCore.text("partition_key").notNull(),timestamp:sqliteCore.text("timestamp").notNull().default(drizzleOrm.sql`(datetime('now'))`),metadata:sqliteCore.text("metadata")},a=>({idx_request_logs_timestamp:sqliteCore.index("idx_request_logs_timestamp").on(a.timestamp),idx_request_logs_path:sqliteCore.index("idx_request_logs_path").on(a.path),idx_request_logs_status_code:sqliteCore.index("idx_request_logs_status_code").on(a.status_code),idx_request_logs_partition_key:sqliteCore.index("idx_request_logs_partition_key").on(a.partition_key)}));var T=sqliteCore.sqliteTable("usage_limit_snapshots",{id:sqliteCore.text("id").primaryKey(),provider:sqliteCore.text("provider").notNull(),account_ref:sqliteCore.text("account_ref").notNull().default("default"),limit_window:sqliteCore.text("limit_window").notNull(),bucket_start:sqliteCore.text("bucket_start").notNull(),captured_at:sqliteCore.text("captured_at").notNull(),used_percent:sqliteCore.integer("used_percent").notNull(),remaining_percent:sqliteCore.integer("remaining_percent").notNull(),resets_at:sqliteCore.text("resets_at"),source:sqliteCore.text("source").notNull(),metadata:sqliteCore.text("metadata")},a=>({uniq_usage_limit_snapshots_bucket:sqliteCore.uniqueIndex("uniq_usage_limit_snapshots_bucket").on(a.provider,a.account_ref,a.limit_window,a.bucket_start),idx_usage_limit_snapshots_window_bucket:sqliteCore.index("idx_usage_limit_snapshots_window_bucket").on(a.limit_window,a.bucket_start),idx_usage_limit_snapshots_provider_window_bucket:sqliteCore.index("idx_usage_limit_snapshots_provider_window_bucket").on(a.provider,a.limit_window,a.bucket_start)}));function wt(a){let t=z__namespace.resolve(a);return process.platform==="win32"&&(t=t.replace(/\\/g,"/"),t=t.replace(/^([A-Z]):/,(e,n)=>`${n.toLowerCase()}:`)),t.length>1&&!/^[a-zA-Z]:\/$/.test(t)&&(t=t.replace(/\/+$/,"")),t}function Ut(a){let t=wt(a);return crypto.createHash("sha256").update(t).digest("hex")}function dt(a){return a.toLowerCase().replace(/[^a-z0-9-]/g,"-").replace(/-{2,}/g,"-").replace(/^-+|-+$/g,"")}var Et=class extends S{dbRoot;constructor(t={}){super(),this.dbRoot=t.dbRoot;}resolveDbPath(){return this.dbRoot?z.join(this.dbRoot,".crewx","crewx.db"):super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=z.dirname(e);Ft.existsSync(r)||Ft.mkdirSync(r,{recursive:true});}else if(!Ft.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=k(e);if(t)try{Q(n.db);}catch(r){throw n.close(),r}return n}resolveSlug(t,e,n){let r=dt(z.basename(n)),o=`${dt(z.basename(z.dirname(n)))}-${r}`,l=[r,o];try{let u=c=>t.select({id:_.id}).from(_).where(drizzleOrm.and(drizzleOrm.eq(_.slug,c),drizzleOrm.ne(_.id,e))).limit(1).all().length>0;for(let c of l)if(!u(c))return c;for(let c=2;c<1e3;c+=1){let h=`${o}-${c}`;if(!u(h))return h}}catch{}return r}normalizeLegacySlugs(){if(!this.dbExists())return {updated:0,checked:0};let t=this.openHandle(true);try{let e=t.db.select({id:_.id,slug:_.slug}).from(_).all(),n=0;for(let r of e)if(r.slug.includes("/")){let s=dt(r.slug.replace(/\//g,"-"));t.db.update(_).set({slug:s,updated_at:new Date().toISOString()}).where(drizzleOrm.eq(_.id,r.id)).run(),n+=1;}return {updated:n,checked:e.length}}catch(e){throw new d("DB_ERROR","Failed to normalize legacy slugs",e)}finally{t.close();}}ensureRow(t,e){let{id:n,slug:r,name:s,workspacePath:o}=e,l=new Date().toISOString();t.insert(_).values({id:n,slug:r,name:s,workspace_path:o,is_active:1,created_at:l,updated_at:l}).onConflictDoNothing().run(),t.update(_).set({workspace_path:o,updated_at:l}).where(drizzleOrm.and(drizzleOrm.eq(_.id,n),drizzleOrm.isNull(_.workspace_path))).run();}registerWorkspace(t){let e=wt(t),n=this.openHandle(true);try{let r=Ut(e),s=z.basename(e),o=this.resolveSlug(n.db,r,e);return this.ensureRow(n.db,{id:r,slug:o,name:s,workspacePath:e}),{id:r,slug:o}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to register workspace",r)}finally{n.close();}}listProjects(t){if(!this.dbExists())return {rows:[],total:0};let e=this.openHandle(false);try{let n=t.isActive!==void 0?drizzleOrm.eq(_.is_active,t.isActive?1:0):void 0,r=e.db.select({count:drizzleOrm.sql`count(*)`}).from(_).where(n).get();return {rows:e.db.select().from(_).where(n).orderBy(drizzleOrm.desc(_.updated_at)).limit(t.limit).offset(t.offset).all(),total:r?.count??0}}catch(n){throw new d("DB_ERROR","Failed to list projects",n)}finally{e.close();}}findById(t){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(_).where(drizzleOrm.eq(_.id,t)).limit(1).get()??void 0}catch(n){throw new d("DB_ERROR","Failed to find workspace",n)}finally{e.close();}}findAgentsByProject(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.all(drizzleOrm.sql`SELECT DISTINCT agent_id FROM tasks WHERE workspace_id = ${t} AND agent_id IS NOT NULL ORDER BY agent_id`).map(r=>r.agent_id)}catch(n){throw new d("DB_ERROR","Failed to find agents by project",n)}finally{e.close();}}findThreadsByProject(t,e){if(!this.dbExists())return {rows:[],total:0};let n=this.openHandle(false);try{let r=n.db.get(drizzleOrm.sql`SELECT COUNT(*) as count FROM threads WHERE workspace_id = ${t}`);return {rows:n.db.all(drizzleOrm.sql`SELECT t.*,
|
|
3
|
+
`)}`);let s=i.get(e`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='__drizzle_migrations'`),a=0;s?.cnt&&(a=i.get(e`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0),an(i,a),s?.cnt&&(ln(i,r,e),cn(i,r,e)),t(i,{migrationsFolder:r});let u=(i.get(e`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0)-a;if(u>0){let l=s?.cnt?"Database migrated":"Database initialized";console.error(`[crewx] ${l} (${u} migration${u>1?"s":""} applied).`);}}function A(i,t){ue.has(t)||(bt(i),ue.add(t));}var h=sqliteCore.sqliteTable("workspaces",{id:sqliteCore.text("id").primaryKey(),slug:sqliteCore.text("slug").notNull().unique(),name:sqliteCore.text("name").notNull(),workspace_path:sqliteCore.text("workspace_path"),description:sqliteCore.text("description"),is_active:sqliteCore.integer("is_active").notNull().default(1),created_at:sqliteCore.text("created_at").notNull(),updated_at:sqliteCore.text("updated_at").notNull()});var o=sqliteCore.sqliteTable("tasks",{id:sqliteCore.text("id").primaryKey(),agent_id:sqliteCore.text("agent_id").notNull(),user_id:sqliteCore.text("user_id"),prompt:sqliteCore.text("prompt").notNull(),mode:sqliteCore.text("mode").notNull().default("execute"),status:sqliteCore.text("status").notNull().default("running"),result:sqliteCore.text("result"),error:sqliteCore.text("error"),started_at:sqliteCore.text("started_at").notNull(),completed_at:sqliteCore.text("completed_at"),duration_ms:sqliteCore.integer("duration_ms"),metadata:sqliteCore.text("metadata"),workspace_id:sqliteCore.text("workspace_id"),trace_id:sqliteCore.text("trace_id"),parent_task_id:sqliteCore.text("parent_task_id"),caller_agent_id:sqliteCore.text("caller_agent_id"),model:sqliteCore.text("model"),platform:sqliteCore.text("platform").default("cli"),crewx_version:sqliteCore.text("crewx_version"),input_tokens:sqliteCore.integer("input_tokens").default(0),output_tokens:sqliteCore.integer("output_tokens").default(0),cost_usd:sqliteCore.real("cost_usd").default(0),pid:sqliteCore.integer("pid"),rendered_prompt:sqliteCore.text("rendered_prompt"),command:sqliteCore.text("command"),coding_agent_command:sqliteCore.text("coding_agent_command"),exit_code:sqliteCore.integer("exit_code"),logs:sqliteCore.text("logs"),thread_id:sqliteCore.text("thread_id"),workspace_ref:sqliteCore.text("workspace_ref"),project_id:sqliteCore.text("project_id"),project_ref:sqliteCore.text("project_ref"),cached_input_tokens:sqliteCore.integer("cached_input_tokens").default(0),run_epoch:sqliteCore.integer("run_epoch").default(0)},i=>({idx_tasks_agent_id:sqliteCore.index("idx_tasks_agent_id").on(i.agent_id),idx_tasks_status:sqliteCore.index("idx_tasks_status").on(i.status),idx_tasks_started_at:sqliteCore.index("idx_tasks_started_at").on(i.started_at),idx_tasks_trace_id:sqliteCore.index("idx_tasks_trace_id").on(i.trace_id),idx_tasks_parent_task_id:sqliteCore.index("idx_tasks_parent_task_id").on(i.parent_task_id),idx_tasks_crewx_version:sqliteCore.index("idx_tasks_crewx_version").on(i.crewx_version),idx_tasks_pid:sqliteCore.index("idx_tasks_pid").on(i.pid),idx_tasks_thread_id:sqliteCore.index("idx_tasks_thread_id").on(i.thread_id),idx_tasks_workspace_id:sqliteCore.index("idx_tasks_workspace_id").on(i.workspace_id),idx_tasks_workspace_ref:sqliteCore.index("idx_tasks_workspace_ref").on(i.workspace_ref),idx_tasks_project_id:sqliteCore.index("idx_tasks_project_id").on(i.project_id),idx_tasks_ws_started:sqliteCore.index("idx_tasks_ws_started").on(i.workspace_id,i.started_at)}));var p=sqliteCore.sqliteTable("threads",{id:sqliteCore.text("id").primaryKey(),workspace_id:sqliteCore.text("workspace_id").references(()=>h.id,{onDelete:"set null"}),platform:sqliteCore.text("platform").notNull().default("cli"),title:sqliteCore.text("title"),first_message:sqliteCore.text("first_message"),last_message:sqliteCore.text("last_message"),message_count:sqliteCore.integer("message_count").notNull().default(0),created_at:sqliteCore.text("created_at").notNull(),updated_at:sqliteCore.text("updated_at").notNull(),metadata:sqliteCore.text("metadata"),title_locked:sqliteCore.integer("title_locked").notNull().default(0),pinned:sqliteCore.integer("pinned").notNull().default(0),starred:sqliteCore.integer("starred").notNull().default(0)},i=>({idx_threads_updated_at:sqliteCore.index("idx_threads_updated_at").on(i.updated_at),idx_threads_workspace_id:sqliteCore.index("idx_threads_workspace_id").on(i.workspace_id),idx_threads_ws_updated:sqliteCore.index("idx_threads_ws_updated").on(i.workspace_id,i.updated_at),idx_threads_title:sqliteCore.index("idx_threads_title").on(i.title)}));var dt=sqliteCore.sqliteTable("spans",{id:sqliteCore.text("id").primaryKey(),task_id:sqliteCore.text("task_id").references(()=>o.id,{onDelete:"set null"}),parent_span_id:sqliteCore.text("parent_span_id").references(()=>dt.id,{onDelete:"set null"}),name:sqliteCore.text("name").notNull(),kind:sqliteCore.text("kind").notNull().default("internal"),status:sqliteCore.text("status").notNull().default("ok"),started_at:sqliteCore.text("started_at").notNull(),completed_at:sqliteCore.text("completed_at"),duration_ms:sqliteCore.integer("duration_ms"),input:sqliteCore.text("input"),output:sqliteCore.text("output"),error:sqliteCore.text("error"),attributes:sqliteCore.text("attributes")},i=>({idx_spans_task_id:sqliteCore.index("idx_spans_task_id").on(i.task_id),idx_spans_parent_span_id:sqliteCore.index("idx_spans_parent_span_id").on(i.parent_span_id)}));var nt=sqliteCore.sqliteTable("tool_calls",{id:sqliteCore.text("id").primaryKey(),task_id:sqliteCore.text("task_id").references(()=>o.id,{onDelete:"cascade"}),session_id:sqliteCore.text("session_id"),tool_name:sqliteCore.text("tool_name").notNull(),files:sqliteCore.text("files"),input:sqliteCore.text("input"),output:sqliteCore.text("output"),duration_ms:sqliteCore.integer("duration_ms"),timestamp:sqliteCore.text("timestamp").notNull()},i=>({idx_tool_calls_task_id:sqliteCore.index("idx_tool_calls_task_id").on(i.task_id),idx_tool_calls_tool_name:sqliteCore.index("idx_tool_calls_tool_name").on(i.tool_name),idx_tool_calls_timestamp:sqliteCore.index("idx_tool_calls_timestamp").on(i.timestamp)}));var Y=sqliteCore.sqliteTable("thread_boxes",{id:sqliteCore.text("id").primaryKey(),thread_id:sqliteCore.text("thread_id").notNull().references(()=>p.id,{onDelete:"cascade"}),seq:sqliteCore.integer("seq").notNull(),first_task_id:sqliteCore.text("first_task_id").notNull(),mid_task_id:sqliteCore.text("mid_task_id").notNull(),last_task_id:sqliteCore.text("last_task_id").notNull(),task_count:sqliteCore.integer("task_count").notNull(),summary:sqliteCore.text("summary"),source_tokens:sqliteCore.integer("source_tokens").notNull(),summary_tokens:sqliteCore.integer("summary_tokens"),created_at:sqliteCore.text("created_at").notNull()},i=>({idx_thread_boxes_thread_id:sqliteCore.index("idx_thread_boxes_thread_id").on(i.thread_id),idx_thread_boxes_seq:sqliteCore.index("idx_thread_boxes_seq").on(i.thread_id,i.seq),uniq_thread_boxes_thread_seq:sqliteCore.unique().on(i.thread_id,i.seq)}));var Rt=sqliteCore.sqliteTable("request_logs",{id:sqliteCore.text("id").primaryKey(),path:sqliteCore.text("path").notNull(),method:sqliteCore.text("method").notNull(),status_code:sqliteCore.integer("status_code").notNull(),duration_ms:sqliteCore.integer("duration_ms").notNull(),ip:sqliteCore.text("ip"),request_headers:sqliteCore.text("request_headers"),response_headers:sqliteCore.text("response_headers"),request_body:sqliteCore.text("request_body"),response_body:sqliteCore.text("response_body"),query:sqliteCore.text("query"),user_id:sqliteCore.text("user_id"),project_id:sqliteCore.text("project_id"),partition_key:sqliteCore.text("partition_key").notNull(),timestamp:sqliteCore.text("timestamp").notNull().default(drizzleOrm.sql`(datetime('now'))`),metadata:sqliteCore.text("metadata")},i=>({idx_request_logs_timestamp:sqliteCore.index("idx_request_logs_timestamp").on(i.timestamp),idx_request_logs_path:sqliteCore.index("idx_request_logs_path").on(i.path),idx_request_logs_status_code:sqliteCore.index("idx_request_logs_status_code").on(i.status_code),idx_request_logs_partition_key:sqliteCore.index("idx_request_logs_partition_key").on(i.partition_key)}));var I=sqliteCore.sqliteTable("usage_limit_snapshots",{id:sqliteCore.text("id").primaryKey(),provider:sqliteCore.text("provider").notNull(),account_ref:sqliteCore.text("account_ref").notNull().default("default"),limit_window:sqliteCore.text("limit_window").notNull(),bucket_start:sqliteCore.text("bucket_start").notNull(),captured_at:sqliteCore.text("captured_at").notNull(),used_percent:sqliteCore.integer("used_percent").notNull(),remaining_percent:sqliteCore.integer("remaining_percent").notNull(),resets_at:sqliteCore.text("resets_at"),source:sqliteCore.text("source").notNull(),metadata:sqliteCore.text("metadata")},i=>({uniq_usage_limit_snapshots_bucket:sqliteCore.uniqueIndex("uniq_usage_limit_snapshots_bucket").on(i.provider,i.account_ref,i.limit_window,i.bucket_start),idx_usage_limit_snapshots_window_bucket:sqliteCore.index("idx_usage_limit_snapshots_window_bucket").on(i.limit_window,i.bucket_start),idx_usage_limit_snapshots_provider_window_bucket:sqliteCore.index("idx_usage_limit_snapshots_provider_window_bucket").on(i.provider,i.limit_window,i.bucket_start)}));var z=sqliteCore.sqliteTable("usage_reports",{id:sqliteCore.text("id").primaryKey(),workspace_id:sqliteCore.text("workspace_id").notNull(),month:sqliteCore.text("month").notNull(),generated_at:sqliteCore.text("generated_at").notNull(),source_range:sqliteCore.text("source_range").notNull(),tier:sqliteCore.text("tier").notNull(),total_tokens:sqliteCore.integer("total_tokens").notNull(),total_cost_usd:sqliteCore.real("total_cost_usd").notNull(),payload:sqliteCore.text("payload").notNull(),payload_hash:sqliteCore.text("payload_hash"),signature:sqliteCore.text("signature"),signed_at:sqliteCore.text("signed_at"),issuer:sqliteCore.text("issuer")},i=>({uniq_usage_reports_ws_month:sqliteCore.uniqueIndex("uniq_usage_reports_ws_month").on(i.workspace_id,i.month)}));var k=sqliteCore.sqliteTable("notifications",{id:sqliteCore.text("id").primaryKey(),workspace_id:sqliteCore.text("workspace_id").notNull(),agent_id:sqliteCore.text("agent_id"),task_id:sqliteCore.text("task_id"),thread_id:sqliteCore.text("thread_id"),source:sqliteCore.text("source").notNull().default("agent"),level:sqliteCore.text("level").notNull().default("info"),title:sqliteCore.text("title").notNull(),body:sqliteCore.text("body"),target_user_id:sqliteCore.text("target_user_id"),created_at:sqliteCore.text("created_at").notNull(),metadata:sqliteCore.text("metadata")},i=>({idx_notifications_ws_created:sqliteCore.index("idx_notifications_ws_created").on(i.workspace_id,drizzleOrm.sql`${i.created_at} DESC`)})),v=sqliteCore.sqliteTable("notification_reads",{notification_id:sqliteCore.text("notification_id").notNull(),user_id:sqliteCore.text("user_id").notNull(),read_at:sqliteCore.text("read_at").notNull()},i=>({pk:sqliteCore.primaryKey({columns:[i.notification_id,i.user_id]})}));var y=sqliteCore.sqliteTable("agent_suggestions",{id:sqliteCore.text("id").primaryKey(),workspace_id:sqliteCore.text("workspace_id").notNull(),agent_id:sqliteCore.text("agent_id").notNull(),task_id:sqliteCore.text("task_id"),type:sqliteCore.text("type").notNull(),status:sqliteCore.text("status").notNull().default("pending"),payload:sqliteCore.text("payload").notNull(),applied_commit_sha:sqliteCore.text("applied_commit_sha"),created_at:sqliteCore.text("created_at").notNull(),updated_at:sqliteCore.text("updated_at").notNull()},i=>({idx_agent_suggestions_ws_agent:sqliteCore.index("idx_agent_suggestions_ws_agent").on(i.workspace_id,i.agent_id),idx_agent_suggestions_ws_created:sqliteCore.index("idx_agent_suggestions_ws_created").on(i.workspace_id,i.created_at)}));sqliteCore.sqliteTable("ui_instances",{id:sqliteCore.text("id").primaryKey().notNull(),definition_id:sqliteCore.text("definition_id").notNull(),definition_version:sqliteCore.integer("definition_version").notNull(),definition_hash:sqliteCore.text("definition_hash").notNull(),workspace_id:sqliteCore.text("workspace_id").notNull(),thread_id:sqliteCore.text("thread_id").notNull(),anchor_task_id:sqliteCore.text("anchor_task_id"),domain_ref_json:sqliteCore.text("domain_ref_json").notNull(),state_json:sqliteCore.text("state_json").notNull(),revision:sqliteCore.integer("revision").notNull().default(0),status:sqliteCore.text("status").notNull().default("active"),granted_capabilities_json:sqliteCore.text("granted_capabilities_json").notNull().default("[]"),created_at:sqliteCore.integer("created_at").notNull(),updated_at:sqliteCore.integer("updated_at").notNull()},i=>({idCheck:sqliteCore.check("ui_instances_id_ck",drizzleOrm.sql`length(${i.id}) BETWEEN 5 AND 128 AND ${i.id} GLOB 'uic_*'`),definitionIdCheck:sqliteCore.check("ui_instances_definition_id_ck",drizzleOrm.sql`length(${i.definition_id}) BETWEEN 1 AND 128`),definitionVersionCheck:sqliteCore.check("ui_instances_definition_version_ck",drizzleOrm.sql`${i.definition_version} >= 1`),definitionHashCheck:sqliteCore.check("ui_instances_definition_hash_ck",drizzleOrm.sql`length(${i.definition_hash}) = 64 AND ${i.definition_hash} NOT GLOB '*[^0-9a-f]*'`),workspaceIdCheck:sqliteCore.check("ui_instances_workspace_id_ck",drizzleOrm.sql`length(${i.workspace_id}) BETWEEN 1 AND 256`),threadIdCheck:sqliteCore.check("ui_instances_thread_id_ck",drizzleOrm.sql`length(${i.thread_id}) BETWEEN 1 AND 256`),domainRefJsonCheck:sqliteCore.check("ui_instances_domain_ref_json_ck",drizzleOrm.sql`json_valid(${i.domain_ref_json}) AND json_type(${i.domain_ref_json}) = 'object'`),stateJsonCheck:sqliteCore.check("ui_instances_state_json_ck",drizzleOrm.sql`json_valid(${i.state_json}) AND json_type(${i.state_json}) = 'object'`),revisionCheck:sqliteCore.check("ui_instances_revision_ck",drizzleOrm.sql`${i.revision} >= 0`),statusCheck:sqliteCore.check("ui_instances_status_ck",drizzleOrm.sql`${i.status} IN ('active', 'completed', 'expired', 'revoked', 'unsupported')`),grantedCapabilitiesJsonCheck:sqliteCore.check("ui_instances_granted_capabilities_json_ck",drizzleOrm.sql`json_valid(${i.granted_capabilities_json}) AND json_type(${i.granted_capabilities_json}) = 'array'`),createdAtCheck:sqliteCore.check("ui_instances_created_at_ck",drizzleOrm.sql`${i.created_at} >= 0`),updatedAtCheck:sqliteCore.check("ui_instances_updated_at_ck",drizzleOrm.sql`${i.updated_at} >= ${i.created_at}`),workspaceThreadIdx:sqliteCore.index("ui_instances_workspace_thread_idx").on(i.workspace_id,i.thread_id)}));sqliteCore.sqliteTable("ui_action_receipts",{id:sqliteCore.text("id").primaryKey().notNull(),instance_id:sqliteCore.text("instance_id").notNull(),request_id:sqliteCore.text("request_id").notNull(),action_id:sqliteCore.text("action_id").notNull(),actor_json:sqliteCore.text("actor_json").notNull(),source:sqliteCore.text("source").notNull(),expected_revision:sqliteCore.integer("expected_revision").notNull(),before_revision:sqliteCore.integer("before_revision").notNull(),after_revision:sqliteCore.integer("after_revision"),status:sqliteCore.text("status").notNull(),outcome:sqliteCore.text("outcome"),input_json:sqliteCore.text("input_json").notNull(),result_json:sqliteCore.text("result_json"),error_code:sqliteCore.text("error_code"),requested_at:sqliteCore.integer("requested_at").notNull(),settled_at:sqliteCore.integer("settled_at")},i=>({idCheck:sqliteCore.check("ui_action_receipts_id_ck",drizzleOrm.sql`length(${i.id}) BETWEEN 5 AND 128`),instanceIdCheck:sqliteCore.check("ui_action_receipts_instance_id_ck",drizzleOrm.sql`length(${i.instance_id}) BETWEEN 1 AND 128`),requestIdCheck:sqliteCore.check("ui_action_receipts_request_id_ck",drizzleOrm.sql`length(${i.request_id}) BETWEEN 1 AND 128`),actionIdCheck:sqliteCore.check("ui_action_receipts_action_id_ck",drizzleOrm.sql`length(${i.action_id}) BETWEEN 1 AND 128`),actorJsonCheck:sqliteCore.check("ui_action_receipts_actor_json_ck",drizzleOrm.sql`json_valid(${i.actor_json}) AND json_type(${i.actor_json}) = 'object'`),sourceCheck:sqliteCore.check("ui_action_receipts_source_ck",drizzleOrm.sql`${i.source} IN ('ui_host', 'cli', 'system')`),expectedRevisionCheck:sqliteCore.check("ui_action_receipts_expected_revision_ck",drizzleOrm.sql`${i.expected_revision} >= 0`),beforeRevisionCheck:sqliteCore.check("ui_action_receipts_before_revision_ck",drizzleOrm.sql`${i.before_revision} >= 0`),afterRevisionCheck:sqliteCore.check("ui_action_receipts_after_revision_ck",drizzleOrm.sql`${i.after_revision} IS NULL OR ${i.after_revision} >= 0`),statusCheck:sqliteCore.check("ui_action_receipts_status_ck",drizzleOrm.sql`${i.status} IN ('requested', 'settled')`),outcomeCheck:sqliteCore.check("ui_action_receipts_outcome_ck",drizzleOrm.sql`${i.outcome} IS NULL OR ${i.outcome} IN ('succeeded', 'rejected', 'failed', 'conflicted', 'expired')`),inputJsonCheck:sqliteCore.check("ui_action_receipts_input_json_ck",drizzleOrm.sql`json_valid(${i.input_json}) AND json_type(${i.input_json}) = 'object'`),resultJsonCheck:sqliteCore.check("ui_action_receipts_result_json_ck",drizzleOrm.sql`${i.result_json} IS NULL OR (json_valid(${i.result_json}) AND json_type(${i.result_json}) = 'object')`),requestedAtCheck:sqliteCore.check("ui_action_receipts_requested_at_ck",drizzleOrm.sql`${i.requested_at} >= 0`),settledAtCheck:sqliteCore.check("ui_action_receipts_settled_at_ck",drizzleOrm.sql`${i.settled_at} IS NULL OR ${i.settled_at} >= ${i.requested_at}`),statusTransitionCheck:sqliteCore.check("ui_action_receipts_status_transition_ck",drizzleOrm.sql`(
|
|
4
|
+
(${i.status} = 'requested' AND ${i.after_revision} IS NULL AND ${i.outcome} IS NULL AND ${i.settled_at} IS NULL)
|
|
5
|
+
OR
|
|
6
|
+
(${i.status} = 'settled' AND ${i.after_revision} IS NOT NULL AND ${i.outcome} IS NOT NULL AND ${i.settled_at} IS NOT NULL)
|
|
7
|
+
)`),instanceRequestUnique:sqliteCore.uniqueIndex("ui_action_receipts_instance_request_uq").on(i.instance_id,i.request_id),instanceIdIdx:sqliteCore.index("ui_action_receipts_instance_id_idx").on(i.instance_id),requestedAtIdx:sqliteCore.index("ui_action_receipts_requested_at_idx").on(i.requested_at)}));function Yt(i){let t=it__namespace.resolve(i);return process.platform==="win32"&&(t=t.replace(/\\/g,"/"),t=t.replace(/^([A-Z]):/,(e,n)=>`${n.toLowerCase()}:`)),t.length>1&&!/^[a-zA-Z]:\/$/.test(t)&&(t=t.replace(/\/+$/,"")),t}function Ee(i){let t=Yt(i);return crypto.createHash("sha256").update(t).digest("hex")}function Ht(i){return i.toLowerCase().replace(/[^a-z0-9-]/g,"-").replace(/-{2,}/g,"-").replace(/^-+|-+$/g,"")}var Vt=class extends N{dbRoot;constructor(t={}){super(),this.dbRoot=t.dbRoot;}resolveDbPath(){return this.dbRoot?it.join(this.dbRoot,".crewx","crewx.db"):super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{bt(n.db);}catch(r){throw n.close(),r}return n}resolveSlug(t,e,n){let r=Ht(it.basename(n)),a=`${Ht(it.basename(it.dirname(n)))}-${r}`,c=[r,a];try{let u=l=>t.select({id:h.id}).from(h).where(drizzleOrm.and(drizzleOrm.eq(h.slug,l),drizzleOrm.ne(h.id,e))).limit(1).all().length>0;for(let l of c)if(!u(l))return l;for(let l=2;l<1e3;l+=1){let _=`${a}-${l}`;if(!u(_))return _}}catch{}return r}normalizeLegacySlugs(){if(!this.dbExists())return {updated:0,checked:0};let t=this.openHandle(true);try{let e=t.db.select({id:h.id,slug:h.slug}).from(h).all(),n=0;for(let r of e)if(r.slug.includes("/")){let s=Ht(r.slug.replace(/\//g,"-"));t.db.update(h).set({slug:s,updated_at:new Date().toISOString()}).where(drizzleOrm.eq(h.id,r.id)).run(),n+=1;}return {updated:n,checked:e.length}}catch(e){throw new d("DB_ERROR","Failed to normalize legacy slugs",e)}finally{t.close();}}ensureRow(t,e){let{id:n,slug:r,name:s,workspacePath:a}=e,c=new Date().toISOString();t.insert(h).values({id:n,slug:r,name:s,workspace_path:a,is_active:1,created_at:c,updated_at:c}).onConflictDoNothing().run(),t.update(h).set({workspace_path:a,updated_at:c}).where(drizzleOrm.and(drizzleOrm.eq(h.id,n),drizzleOrm.isNull(h.workspace_path))).run();}registerWorkspace(t){let e=Yt(t),n=this.openHandle(true);try{let r=Ee(e),s=it.basename(e),a=this.resolveSlug(n.db,r,e);return this.ensureRow(n.db,{id:r,slug:a,name:s,workspacePath:e}),{id:r,slug:a}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to register workspace",r)}finally{n.close();}}listProjects(t){if(!this.dbExists())return {rows:[],total:0};let e=this.openHandle(false);try{let n=t.isActive!==void 0?drizzleOrm.eq(h.is_active,t.isActive?1:0):void 0,r=e.db.select({count:drizzleOrm.sql`count(*)`}).from(h).where(n).get();return {rows:e.db.select().from(h).where(n).orderBy(drizzleOrm.desc(h.updated_at)).limit(t.limit).offset(t.offset).all(),total:r?.count??0}}catch(n){throw new d("DB_ERROR","Failed to list projects",n)}finally{e.close();}}findById(t){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(h).where(drizzleOrm.eq(h.id,t)).limit(1).get()??void 0}catch(n){throw new d("DB_ERROR","Failed to find workspace",n)}finally{e.close();}}findAgentsByProject(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.all(drizzleOrm.sql`SELECT DISTINCT agent_id FROM tasks WHERE workspace_id = ${t} AND agent_id IS NOT NULL ORDER BY agent_id`).map(r=>r.agent_id)}catch(n){throw new d("DB_ERROR","Failed to find agents by project",n)}finally{e.close();}}findThreadsByProject(t,e){if(!this.dbExists())return {rows:[],total:0};let n=this.openHandle(false);try{let r=n.db.get(drizzleOrm.sql`SELECT COUNT(*) as count FROM threads WHERE workspace_id = ${t}`);return {rows:n.db.all(drizzleOrm.sql`SELECT t.*,
|
|
4
8
|
(SELECT agent_id FROM tasks tk WHERE tk.thread_id = t.id AND tk.agent_id IS NOT NULL ORDER BY tk.started_at ASC LIMIT 1) AS agent_id
|
|
5
9
|
FROM threads t
|
|
6
10
|
WHERE t.workspace_id = ${t}
|
|
7
11
|
ORDER BY t.updated_at DESC
|
|
8
|
-
LIMIT ${e.limit} OFFSET ${e.offset}`),total:r?.count??0}}catch(r){throw new d("DB_ERROR","Failed to find threads by project",r)}finally{n.close();}}findBySlug(t){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(
|
|
12
|
+
LIMIT ${e.limit} OFFSET ${e.offset}`),total:r?.count??0}}catch(r){throw new d("DB_ERROR","Failed to find threads by project",r)}finally{n.close();}}findBySlug(t){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(h).where(drizzleOrm.eq(h.slug,t)).limit(1).get()??void 0}catch(n){throw new d("DB_ERROR","Failed to find workspace by slug",n)}finally{e.close();}}slugExists(t,e){if(!this.dbExists())return false;let n=this.openHandle(false);try{let r=e?drizzleOrm.and(drizzleOrm.eq(h.slug,t),drizzleOrm.ne(h.id,e)):drizzleOrm.eq(h.slug,t);return !!n.db.select({id:h.id}).from(h).where(r).limit(1).get()}catch(r){throw new d("DB_ERROR","Failed to check slug",r)}finally{n.close();}}insert(t,e,n,r){let s=this.openHandle(true);try{let a=new Date().toISOString();s.db.insert(h).values({id:t,slug:e,name:n,workspace_path:r,is_active:1,created_at:a,updated_at:a}).run();let c=s.db.select().from(h).where(drizzleOrm.eq(h.id,t)).limit(1).get();if(!c)throw new d("DB_ERROR","Insert did not return a row");return c}catch(a){throw a instanceof d?a:new d("DB_ERROR","Failed to insert workspace",a)}finally{s.close();}}update(t,e,n){let r=this.openHandle(true);try{r.runRaw(`UPDATE workspaces SET ${e.join(", ")} WHERE id = ?`,n);let s=r.db.select().from(h).where(drizzleOrm.eq(h.id,t)).limit(1).get();if(!s)throw new d("NOT_FOUND",`Workspace ${t} not found`);return s}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to update workspace",s)}finally{r.close();}}cleanupOrphanWorkspaces(){if(!this.dbExists())return {softDeleted:0,checked:0};let t=this.openHandle(true);try{let e=t.db.select({id:h.id,workspace_path:h.workspace_path}).from(h).where(drizzleOrm.and(drizzleOrm.eq(h.is_active,1),drizzleOrm.isNotNull(h.workspace_path))).all(),n=0;for(let r of e){let s=r.workspace_path;xe.existsSync(it.join(s,"crewx.yaml"))||xe.existsSync(it.join(s,"crewx.yml"))||(t.db.update(h).set({is_active:0,updated_at:new Date().toISOString()}).where(drizzleOrm.eq(h.id,r.id)).run(),n+=1);}return {softDeleted:n,checked:e.length}}catch(e){throw new d("DB_ERROR","Failed to cleanup orphan workspaces",e)}finally{t.close();}}delete(t){let e=this.openHandle(true);try{e.db.run(drizzleOrm.sql`UPDATE threads SET workspace_id = NULL WHERE workspace_id = ${t}`),e.db.delete(h).where(drizzleOrm.eq(h.id,t)).run();}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to delete workspace",n)}finally{e.close();}}};var Ne=[h,o,p,dt,nt,Y,Rt];function Kn(){return it__namespace.default.join(on__default.default.homedir(),".crewx","crewx.db")}function Gn(i){if(!xe__default.default.existsSync(i))return null;let t=Math.floor(Date.now()/1e3),e=`${i}.bak-${t}`;return xe__default.default.copyFileSync(i,e),e}function De(i){let t=i.all("SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%'");return new Set(t.map(e=>e.name))}function Yn(i){return i==null||typeof i=="object"&&"queryChunks"in i?null:typeof i=="number"?String(i):typeof i=="boolean"?i?"1":"0":typeof i=="string"?`'${i.replace(/'/g,"''")}'`:String(i)}function Jn(i){let t=i.getSQLType(),e=`"${i.name}" ${t}`,n=Yn(i.default);return n!==null&&(e+=` DEFAULT ${n}`),i.notNull&&(e+=" NOT NULL"),e}function Xn(i,t){let e=t?.dbPath??Kn(),n=t?.force??false,r=t?.dryRun??false,s=r?null:Gn(e);if(n&&!r)try{i.run("DELETE FROM __drizzle_migrations");}catch{}let a=De(i);if(!r)try{bt(i);}catch(f){let m=f instanceof Error?`${f.message} ${f.cause?.message??""}`:"";if(!n||!m.includes("duplicate column"))throw f}let c,u;r?(c=Ne.map(m=>sqliteCore.getTableConfig(m).name).filter(m=>!a.has(m)),u=a):(u=De(i),c=[...u].filter(f=>!a.has(f)));let l=[],_=[];for(let f of Ne){let m=sqliteCore.getTableConfig(f),R=m.name;if(!u.has(R))continue;let L=i.all(`PRAGMA table_info("${R}")`),U=new Set(L.map(D=>D.name)),F=new Set(m.columns.map(D=>D.name));for(let D of m.columns)if(!U.has(D.name)){if(!r){let Q=Jn(D);i.run(`ALTER TABLE "${R}" ADD COLUMN ${Q}`);}l.push({table:R,column:D.name});}for(let D of L)F.has(D.name)||_.push(`${R}.${D.name} exists in DB but not in schema (untouched)`);}return {created:c,altered:l,warnings:_,backupPath:s}}function Ae(i){let t=i.toLowerCase().trim();t=t.replace(/\[[^\]]*\]$/g,"");let e=t.lastIndexOf("/");return e>=0&&(t=t.slice(e+1)),t=t.replace(/-\d{8}$/,""),t}var _t="2026-05-09",or="0.8.9-rc.13",pt=10,ft=parseInt(or.split("rc.")[1]);function Oe(){let i=drizzleOrm.sql`
|
|
13
|
+
CASE
|
|
14
|
+
WHEN ${o.model} LIKE 'claude-%' OR ${o.model} IN ('opus', 'sonnet', 'haiku', 'opus[1m]', 'sonnet[1m]') THEN 'claude'
|
|
15
|
+
WHEN ${o.model} LIKE 'gpt-%' OR ${o.model} LIKE 'codex-%' THEN 'codex'
|
|
16
|
+
WHEN ${o.model} LIKE 'gemini-%' THEN 'gemini'
|
|
17
|
+
WHEN ${o.model} LIKE 'grok-%' THEN 'grok'
|
|
18
|
+
WHEN ${o.model} LIKE 'zai-%' OR ${o.model} LIKE 'openrouter/z-ai/%' THEN 'opencode'
|
|
19
|
+
WHEN ${o.model} LIKE 'minimax/%' THEN 'minimax'
|
|
20
|
+
WHEN ${o.model} LIKE 'qwen%' THEN 'qwen'
|
|
21
|
+
ELSE 'unknown'
|
|
22
|
+
END
|
|
23
|
+
`,t=drizzleOrm.sql`json_extract(${o.metadata}, '$.provider')`,e=drizzleOrm.sql`trim(${t})`,n=drizzleOrm.sql`substr(${e}, 1, instr(${e}, '/') - 1)`,r=drizzleOrm.sql`
|
|
24
|
+
CASE
|
|
25
|
+
WHEN instr(${e}, '/') > 0
|
|
26
|
+
AND ${n} IN ('cli', 'acp', 'api', 'plugin', 'remote')
|
|
27
|
+
THEN substr(${e}, instr(${e}, '/') + 1)
|
|
28
|
+
ELSE ${e}
|
|
29
|
+
END
|
|
30
|
+
`;return drizzleOrm.sql`
|
|
31
|
+
CASE
|
|
32
|
+
WHEN ${o.metadata} IS NULL THEN ${i}
|
|
33
|
+
WHEN json_valid(${o.metadata}) = 0 THEN 'unknown'
|
|
34
|
+
WHEN json_type(${o.metadata}, '$') IS NOT 'object' THEN 'unknown'
|
|
35
|
+
WHEN json_type(${o.metadata}, '$.provider') IS NULL THEN ${i}
|
|
36
|
+
WHEN json_type(${o.metadata}, '$.provider') IS NOT 'text' THEN 'unknown'
|
|
37
|
+
WHEN ${e} = '' THEN 'unknown'
|
|
38
|
+
WHEN trim(${r}) = '' THEN 'unknown'
|
|
39
|
+
ELSE ${r}
|
|
40
|
+
END
|
|
41
|
+
`}var sr=new Set(["workflow","mcp"]),ve=600*1e3,Zt=class extends N{dbPath;pidNullObservedSince=new Map;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=it.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{A(n.db,e);}catch(r){throw n.close(),r}return n}startTask(t){let e=this.openHandle(true);try{e.db.insert(o).values({id:t.id,agent_id:t.agentId,prompt:t.prompt,mode:t.mode,status:t.status,started_at:t.startedAt,pid:t.pid??null,parent_task_id:t.parentTaskId??null,caller_agent_id:t.callerAgentId??null,trace_id:t.traceId??null,command:t.command??null,metadata:t.metadata??null,workspace_id:t.workspaceId??null,platform:t.platform??"cli",crewx_version:t.crewxVersion??null,thread_id:t.threadId??null,model:t.model??null,rendered_prompt:t.renderedPrompt??null,coding_agent_command:t.codingAgentCommand??null}).onConflictDoUpdate({target:o.id,set:{pid:drizzleOrm.sql`COALESCE(excluded.pid, ${o.pid})`},setWhere:drizzleOrm.eq(o.status,"running")}).run();}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to start task",n)}finally{e.close();}}finishTask(t){let e=this.openHandle(true);try{let n=t.runEpoch??null;e.runRaw(`UPDATE tasks SET status=?, result=?, error=?, completed_at=?, duration_ms=?,
|
|
9
42
|
exit_code=?, input_tokens=?, output_tokens=?, cached_input_tokens=?, cost_usd=?,
|
|
10
43
|
model=COALESCE(?, model)
|
|
11
|
-
WHERE id=? AND status='running' AND COALESCE(run_epoch, 0) = COALESCE(?, 0)`,[t.status,t.result??null,t.error??null,t.completedAt,t.durationMs??null,t.exitCode??null,t.inputTokens??0,t.outputTokens??0,t.cachedInputTokens??0,t.costUsd??0,t.model??null,t.id,n]);}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to finish task",n)}finally{e.close();}}appendLog(t,e){let n=this.openHandle(true);try{n.db.transaction(r=>{let s=r.select({logs:
|
|
44
|
+
WHERE id=? AND status='running' AND COALESCE(run_epoch, 0) = COALESCE(?, 0)`,[t.status,t.result??null,t.error??null,t.completedAt,t.durationMs??null,t.exitCode??null,t.inputTokens??0,t.outputTokens??0,t.cachedInputTokens??0,t.costUsd??0,t.model??null,t.id,n]);}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to finish task",n)}finally{e.close();}}appendLog(t,e){let n=this.openHandle(true);try{n.db.transaction(r=>{let s=r.select({logs:o.logs}).from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get(),a=s?.logs?JSON.parse(s.logs):[];a.push(e),r.update(o).set({logs:JSON.stringify(a)}).where(drizzleOrm.eq(o.id,t)).run();},{behavior:"immediate"});}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to append log",r)}finally{n.close();}}getRunningTasks(){if(!this.dbExists())return [];let t=this.openHandle(false);try{return t.db.select().from(o).where(drizzleOrm.eq(o.status,"running")).orderBy(drizzleOrm.desc(o.started_at)).all()}catch(e){throw new d("DB_ERROR","Failed to get running tasks",e)}finally{t.close();}}getAllTasks(){if(!this.dbExists())return [];let t=this.openHandle(false);try{return t.db.select().from(o).orderBy(drizzleOrm.desc(o.started_at)).limit(100).all()}catch(e){throw new d("DB_ERROR","Failed to get all tasks",e)}finally{t.close();}}getTask(t){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get()??void 0}catch(n){throw new d("DB_ERROR","Failed to get task",n)}finally{e.close();}}killTask(t){if(!this.dbExists())return {killed:false};let e=this.openHandle(true);try{let n=e.db.select({id:o.id,status:o.status,pid:o.pid}).from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get();if(!n||n.status!=="running")return {killed:!1};if(n.pid)try{process.kill(n.pid,"SIGTERM");}catch{}return e.db.update(o).set({status:"failed",error:"Killed by user",completed_at:new Date().toISOString()}).where(drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.status,"running"))).run(),{killed:!0,pid:n.pid??void 0}}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to kill task",n)}finally{e.close();}}reapOrphanedTasks(){if(!this.dbExists())return 0;let t=this.openHandle(true);try{let e=t.db.select({id:o.id,pid:o.pid,platform:o.platform,run_epoch:o.run_epoch}).from(o).where(drizzleOrm.eq(o.status,"running")).all(),n=Date.now(),r=new Set,s=0;for(let a of e){if(a.pid){let l=!1;try{process.kill(a.pid,0),l=!0;}catch{}l||(t.db.update(o).set({status:"failed",error:"Reaped: process not found (orphaned task)",completed_at:new Date().toISOString()}).where(drizzleOrm.and(drizzleOrm.eq(o.id,a.id),drizzleOrm.eq(o.status,"running"))).run(),s++);continue}if(sr.has(a.platform??"cli"))continue;let c=`${a.id}:${a.run_epoch??0}`;r.add(c);let u=this.pidNullObservedSince.get(c);if(u===void 0){this.pidNullObservedSince.set(c,n);continue}n-u>=ve&&(t.db.update(o).set({status:"failed",error:"Reaped: pid never recorded after resume (stale running_instruction)",completed_at:new Date().toISOString()}).where(drizzleOrm.and(drizzleOrm.eq(o.id,a.id),drizzleOrm.eq(o.status,"running"))).run(),s++,this.pidNullObservedSince.delete(c));}for(let a of Array.from(this.pidNullObservedSince.keys()))r.has(a)||this.pidNullObservedSince.delete(a);return s}finally{t.close();}}reapRunningWorkflowTasks(t=ve){if(!this.dbExists())return 0;let e=this.openHandle(true);try{let n=new Date(Date.now()-t).toISOString();return e.db.update(o).set({status:"failed",error:"Reaped: stale workflow run (finally not executed)",completed_at:new Date().toISOString()}).where(drizzleOrm.and(drizzleOrm.eq(o.status,"running"),drizzleOrm.eq(o.platform,"workflow"),drizzleOrm.isNull(o.pid),drizzleOrm.lt(o.started_at,n),drizzleOrm.or(drizzleOrm.isNull(o.metadata),drizzleOrm.and(drizzleOrm.notLike(o.metadata,'%"kind":"workflow_card"%'),drizzleOrm.notLike(o.metadata,'%"kind":"workflow_ui_instance"%'))))).run().changes??0}finally{e.close();}}findTaskStatus(t,e){let n=this.resolveDbPaths();for(let r of n){if(!xe.existsSync(r))continue;let s=w(r);try{let a=e?drizzleOrm.eq(o.workspace_id,e):void 0,c=a?drizzleOrm.and(drizzleOrm.eq(o.id,t),a):drizzleOrm.eq(o.id,t),u=s.db.select().from(o).where(c).limit(1).get()??void 0;if(!u){let l=drizzleOrm.or(drizzleOrm.eq(o.thread_id,t),drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.like(o.command,`%--thread=${t}%`))),_=a?drizzleOrm.and(l,a):l;u=s.db.select().from(o).where(_).orderBy(drizzleOrm.desc(o.started_at)).limit(1).get()??void 0;}if(u)return u}catch(a){throw new d("DB_ERROR","Failed to find task status",a)}finally{s.close();}}}findChildTasks(t,e){let n=this.resolveDbPaths(),r=new Set,s=[];for(let a of n){if(!xe.existsSync(a))continue;let c=w(a);try{let u=e?drizzleOrm.and(drizzleOrm.eq(o.parent_task_id,t),drizzleOrm.eq(o.workspace_id,e)):drizzleOrm.eq(o.parent_task_id,t),l=c.db.select().from(o).where(u).orderBy(drizzleOrm.asc(o.started_at)).all();for(let _ of l)r.has(_.id)||(r.add(_.id),s.push(_));}catch(u){throw new d("DB_ERROR","Failed to find child tasks",u)}finally{c.close();}}return s}getWorkspaceUsageSummary(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.all(t?drizzleOrm.sql`
|
|
12
45
|
SELECT
|
|
13
46
|
COALESCE(workspace_id, 'unknown') AS workspace_id,
|
|
14
47
|
COALESCE(SUM(input_tokens), 0) AS input_tokens,
|
|
@@ -29,7 +62,7 @@ ${n.join(`
|
|
|
29
62
|
FROM tasks
|
|
30
63
|
GROUP BY workspace_id
|
|
31
64
|
ORDER BY (COALESCE(SUM(input_tokens), 0) + COALESCE(SUM(output_tokens), 0)) DESC
|
|
32
|
-
`)}catch(n){throw new d("DB_ERROR","Failed to get workspace usage summary",n)}finally{e.close();}}getThreadTokenUsage(t,e){let n=this.resolveDbPaths(),r=new Set,s=0,
|
|
65
|
+
`)}catch(n){throw new d("DB_ERROR","Failed to get workspace usage summary",n)}finally{e.close();}}getThreadTokenUsage(t,e){let n=this.resolveDbPaths(),r=new Set,s=0,a=0,c=0;for(let u of n){if(!xe.existsSync(u))continue;let l=w(u);try{let _=drizzleOrm.or(drizzleOrm.eq(o.thread_id,t),drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.like(o.command,`%--thread=${t}%`))),f=e?drizzleOrm.and(_,drizzleOrm.eq(o.workspace_id,e)):_,m=l.db.select({id:o.id,input_tokens:o.input_tokens,output_tokens:o.output_tokens,cost_usd:o.cost_usd}).from(o).where(f).all();for(let R of m)r.has(R.id)||(r.add(R.id),s+=R.input_tokens??0,a+=R.output_tokens??0,c+=R.cost_usd??0);}catch(_){throw new d("DB_ERROR","Failed to get thread token usage",_)}finally{l.close();}}return {inputTokens:s,outputTokens:a,costUsd:c}}findTasksByThread(t,e){let n=this.resolveDbPaths(),r=new Set,s=[];for(let a of n){if(!xe.existsSync(a))continue;let c=w(a);try{let u=drizzleOrm.or(drizzleOrm.eq(o.thread_id,t),drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.like(o.command,`%--thread=${t}%`))),l=e?drizzleOrm.and(u,drizzleOrm.eq(o.workspace_id,e)):u,_=c.db.select().from(o).where(l).orderBy(drizzleOrm.asc(o.started_at)).all();for(let f of _)r.has(f.id)||(r.add(f.id),s.push(f));}catch(u){throw new d("DB_ERROR","Failed to find tasks by thread",u)}finally{c.close();}}return s}parseWorkflowCardMetadata(t){if(!t)return {};try{let e=JSON.parse(t);return {kind:typeof e.kind=="string"?e.kind:void 0,workflowRunId:typeof e.workflowRunId=="string"?e.workflowRunId:void 0}}catch{return {}}}extractThreadIdFromCommand(t){return t?/--thread=(\S+)/.exec(t)?.[1]??null:null}findRunningWorkflowParentTasks(){if(!this.dbExists())return [];let t=[];for(let e of this.resolveDbPaths()){if(!xe.existsSync(e))continue;let n=w(e);try{t.push(...n.db.select({id:o.id,agent_id:o.agent_id,thread_id:o.thread_id,workspace_id:o.workspace_id,platform:o.platform,status:o.status,error:o.error,completed_at:o.completed_at,duration_ms:o.duration_ms,started_at:o.started_at,metadata:o.metadata}).from(o).where(drizzleOrm.and(drizzleOrm.eq(o.platform,"workflow"),drizzleOrm.or(drizzleOrm.like(o.metadata,'%"kind":"workflow_card"%'),drizzleOrm.like(o.metadata,'%"kind":"workflow_ui_instance"%')))).orderBy(drizzleOrm.desc(o.started_at)).all());}catch(r){throw new d("DB_ERROR","Failed to find workflow parent tasks",r)}finally{n.close();}}return t}batchFetchWorkflowCards(t,e){let n=new Map;if(t.length===0)return n;let r=new Set(t),s=new Map,a=e?drizzleOrm.eq(o.workspace_id,e):void 0,c=(u,l)=>{if(!u)return;let _=s.get(u);if(_&&l.started_at<=_)return;let f=this.parseWorkflowCardMetadata(l.metadata);f.kind!=="workflow_card"&&f.kind!=="workflow_ui_instance"||(s.set(u,l.started_at),n.set(u,{taskId:l.id,workflowRunId:f.workflowRunId,raw:l.result??void 0}));};for(let u of this.resolveDbPaths()){if(!xe.existsSync(u))continue;let l=w(u);try{let _=drizzleOrm.and(drizzleOrm.inArray(o.thread_id,t),drizzleOrm.or(drizzleOrm.like(o.metadata,'%"kind":"workflow_card"%'),drizzleOrm.like(o.metadata,'%"kind":"workflow_ui_instance"%'))),f=l.db.select({id:o.id,thread_id:o.thread_id,metadata:o.metadata,result:o.result,started_at:o.started_at}).from(o).where(a?drizzleOrm.and(_,a):_).all();for(let L of f)c(L.thread_id,L);let m=drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.or(drizzleOrm.like(o.metadata,'%"kind":"workflow_card"%'),drizzleOrm.like(o.metadata,'%"kind":"workflow_ui_instance"%'))),R=l.db.select({id:o.id,command:o.command,metadata:o.metadata,result:o.result,started_at:o.started_at}).from(o).where(a?drizzleOrm.and(m,a):m).all();for(let L of R){let U=this.extractThreadIdFromCommand(L.command);!U||!r.has(U)||c(U,L);}}catch(_){throw new d("DB_ERROR","Failed to batch fetch workflow cards",_)}finally{l.close();}}return n}findAllTasks(t){if(!this.dbExists())return {rows:[],total:0};let e=this.openHandle(false);try{let n=[];t.workspaceId&&n.push(drizzleOrm.eq(o.workspace_id,t.workspaceId));let r=t.agents&&t.agents.length>0?t.agents:t.agentId?[t.agentId]:null;r&&n.push(drizzleOrm.inArray(o.agent_id,r));let s=t.statuses&&t.statuses.length>0?t.statuses:t.status?[t.status]:null;s&&n.push(drizzleOrm.inArray(o.status,s));let a=t.q??t.search;a&&n.push(drizzleOrm.like(o.prompt,`%${a}%`)),t.from&&n.push(drizzleOrm.gte(o.started_at,t.from)),t.to&&n.push(drizzleOrm.lt(o.started_at,t.to));let c=n.length>0?drizzleOrm.and(...n):void 0,u=e.db.select({count:drizzleOrm.sql`count(*)`}).from(o).where(c).get(),l=(t.sortDir??"DESC")==="ASC"?drizzleOrm.asc(o.started_at):drizzleOrm.desc(o.started_at);return {rows:e.db.select().from(o).where(c).orderBy(l).limit(t.limit).offset(t.offset).all(),total:u?.count??0}}catch(n){throw new d("DB_ERROR","Failed to find all tasks",n)}finally{e.close();}}getAgentUsage(t,e,n){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.all(n?drizzleOrm.sql`
|
|
33
66
|
SELECT
|
|
34
67
|
t.agent_id,
|
|
35
68
|
t.workspace_id,
|
|
@@ -37,11 +70,11 @@ ${n.join(`
|
|
|
37
70
|
COALESCE(SUM(
|
|
38
71
|
COALESCE(t.input_tokens, 0)
|
|
39
72
|
+ CASE
|
|
40
|
-
WHEN t.started_at >= ${
|
|
73
|
+
WHEN t.started_at >= ${_t}
|
|
41
74
|
AND (
|
|
42
75
|
t.crewx_version IS NULL
|
|
43
76
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
44
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
77
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${pt}) AS INTEGER) < ${ft})
|
|
45
78
|
)
|
|
46
79
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
47
80
|
ELSE 0
|
|
@@ -60,11 +93,11 @@ ${n.join(`
|
|
|
60
93
|
COALESCE(SUM(
|
|
61
94
|
COALESCE(t.input_tokens, 0)
|
|
62
95
|
+ CASE
|
|
63
|
-
WHEN t.started_at >= ${
|
|
96
|
+
WHEN t.started_at >= ${_t}
|
|
64
97
|
AND (
|
|
65
98
|
t.crewx_version IS NULL
|
|
66
99
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
67
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
100
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${pt}) AS INTEGER) < ${ft})
|
|
68
101
|
)
|
|
69
102
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
70
103
|
ELSE 0
|
|
@@ -80,11 +113,11 @@ ${n.join(`
|
|
|
80
113
|
COALESCE(SUM(
|
|
81
114
|
COALESCE(t.input_tokens, 0)
|
|
82
115
|
+ CASE
|
|
83
|
-
WHEN t.started_at >= ${
|
|
116
|
+
WHEN t.started_at >= ${_t}
|
|
84
117
|
AND (
|
|
85
118
|
t.crewx_version IS NULL
|
|
86
119
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
87
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
120
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${pt}) AS INTEGER) < ${ft})
|
|
88
121
|
)
|
|
89
122
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
90
123
|
ELSE 0
|
|
@@ -102,11 +135,11 @@ ${n.join(`
|
|
|
102
135
|
COALESCE(SUM(
|
|
103
136
|
COALESCE(t.input_tokens, 0)
|
|
104
137
|
+ CASE
|
|
105
|
-
WHEN t.started_at >= ${
|
|
138
|
+
WHEN t.started_at >= ${_t}
|
|
106
139
|
AND (
|
|
107
140
|
t.crewx_version IS NULL
|
|
108
141
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
109
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
142
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${pt}) AS INTEGER) < ${ft})
|
|
110
143
|
)
|
|
111
144
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
112
145
|
ELSE 0
|
|
@@ -114,18 +147,18 @@ ${n.join(`
|
|
|
114
147
|
), 0)
|
|
115
148
|
+ COALESCE(SUM(t.output_tokens), 0)
|
|
116
149
|
) DESC
|
|
117
|
-
`).map(
|
|
150
|
+
`).map(a=>({agentId:a.agent_id,workspaceId:a.workspace_id??null,totalTasks:a.total_tasks,inputTokens:a.input_tokens,outputTokens:a.output_tokens,cachedInputTokens:a.cached_input_tokens,costUsd:a.cost_usd,totalTokens:a.input_tokens+a.output_tokens}))}catch(s){throw new d("DB_ERROR","Failed to get agent usage",s)}finally{r.close();}}getAgentUsageTrendRaw(t,e,n){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.all(n?drizzleOrm.sql`
|
|
118
151
|
SELECT
|
|
119
152
|
date(t.started_at) AS date,
|
|
120
153
|
t.agent_id,
|
|
121
154
|
COALESCE(SUM(
|
|
122
155
|
COALESCE(t.input_tokens, 0)
|
|
123
156
|
+ CASE
|
|
124
|
-
WHEN t.started_at >= ${
|
|
157
|
+
WHEN t.started_at >= ${_t}
|
|
125
158
|
AND (
|
|
126
159
|
t.crewx_version IS NULL
|
|
127
160
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
128
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
161
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${pt}) AS INTEGER) < ${ft})
|
|
129
162
|
)
|
|
130
163
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
131
164
|
ELSE 0
|
|
@@ -148,11 +181,11 @@ ${n.join(`
|
|
|
148
181
|
COALESCE(SUM(
|
|
149
182
|
COALESCE(t.input_tokens, 0)
|
|
150
183
|
+ CASE
|
|
151
|
-
WHEN t.started_at >= ${
|
|
184
|
+
WHEN t.started_at >= ${_t}
|
|
152
185
|
AND (
|
|
153
186
|
t.crewx_version IS NULL
|
|
154
187
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
155
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
188
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${pt}) AS INTEGER) < ${ft})
|
|
156
189
|
)
|
|
157
190
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
158
191
|
ELSE 0
|
|
@@ -167,51 +200,74 @@ ${n.join(`
|
|
|
167
200
|
AND t.started_at < ${e}
|
|
168
201
|
GROUP BY date(t.started_at), t.agent_id
|
|
169
202
|
ORDER BY date(t.started_at) ASC
|
|
170
|
-
`).map(
|
|
171
|
-
|
|
172
|
-
WHEN ${i.model} LIKE 'claude-%' OR ${i.model} IN ('opus', 'sonnet', 'haiku', 'opus[1m]', 'sonnet[1m]') THEN 'claude'
|
|
173
|
-
WHEN ${i.model} LIKE 'gpt-%' OR ${i.model} LIKE 'codex-%' THEN 'codex'
|
|
174
|
-
WHEN ${i.model} LIKE 'gemini-%' THEN 'gemini'
|
|
175
|
-
WHEN ${i.model} LIKE 'zai-%' OR ${i.model} LIKE 'openrouter/z-ai/%' THEN 'opencode'
|
|
176
|
-
WHEN ${i.model} LIKE 'minimax/%' THEN 'minimax'
|
|
177
|
-
WHEN ${i.model} LIKE 'qwen%' THEN 'qwen'
|
|
178
|
-
ELSE 'unknown'
|
|
179
|
-
END
|
|
180
|
-
`,o=drizzleOrm.sql`
|
|
181
|
-
COALESCE(${i.input_tokens}, 0)
|
|
203
|
+
`).map(a=>({date:a.date,agentId:a.agent_id,inputTokens:a.input_tokens,outputTokens:a.output_tokens,cachedInputTokens:a.cached_input_tokens,costUsd:a.cost_usd,totalTokens:a.input_tokens+a.output_tokens}))}catch(s){throw new d("DB_ERROR","Failed to get agent usage trend",s)}finally{r.close();}}findTaskForStop(t,e){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(o).where(drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.workspace_id,e))).limit(1).get()??void 0}catch(r){throw new d("DB_ERROR","Failed to find task for stop",r)}finally{n.close();}}findLatestTaskByMetadata(t){if(!this.dbExists())return null;let e=this.openHandle(false);try{return e.db.select({id:o.id,thread_id:o.thread_id,status:o.status,result:o.result,error:o.error,started_at:o.started_at,completed_at:o.completed_at,metadata:o.metadata}).from(o).where(drizzleOrm.and(drizzleOrm.eq(o.agent_id,t.agentId),drizzleOrm.eq(o.workspace_id,t.workspaceId),drizzleOrm.sql`${o.metadata} IS NOT NULL AND json_valid(${o.metadata}) AND json_extract(${o.metadata}, '$.' || ${t.metaKey}) = ${t.metaValue}`)).orderBy(drizzleOrm.desc(o.started_at)).limit(1).get()??null}catch(n){throw new d("DB_ERROR","Failed to find latest task by metadata",n)}finally{e.close();}}markTaskFailed(t,e,n){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=new Date().toISOString(),a=n?drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.status,"running"),drizzleOrm.eq(o.workspace_id,n)):drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.status,"running"));r.db.update(o).set({status:"failed",error:e,completed_at:s}).where(a).run();}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to mark task failed",s)}finally{r.close();}}findTasksByPromptHint(t,e){let n=this.resolveDbPaths(),r=new Set,s=[];for(let a of n){if(!xe.existsSync(a))continue;let c=w(a);try{let u=e?drizzleOrm.and(drizzleOrm.like(o.prompt,`%${t}%`),drizzleOrm.eq(o.workspace_id,e)):drizzleOrm.like(o.prompt,`%${t}%`),l=c.db.select().from(o).where(u).orderBy(drizzleOrm.asc(o.started_at)).all();for(let _ of l)r.has(_.id)||(r.add(_.id),s.push(_));}catch(u){throw new d("DB_ERROR","Failed to find tasks by prompt hint",u)}finally{c.close();}}return s}getProviderUsage(t,e,n){if(!this.dbExists())return [];let r=this.openHandle(false);try{let s=Oe(),a=drizzleOrm.sql`
|
|
204
|
+
COALESCE(${o.input_tokens}, 0)
|
|
182
205
|
+ CASE
|
|
183
|
-
WHEN ${
|
|
206
|
+
WHEN ${o.started_at} >= ${_t}
|
|
184
207
|
AND (
|
|
185
|
-
${
|
|
186
|
-
OR (${
|
|
187
|
-
OR (${
|
|
208
|
+
${o.crewx_version} IS NULL
|
|
209
|
+
OR (${o.crewx_version} LIKE '0.8.%' AND ${o.crewx_version} NOT LIKE '0.8.9%')
|
|
210
|
+
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${pt}) AS INTEGER) < ${ft})
|
|
188
211
|
)
|
|
189
|
-
THEN COALESCE(${
|
|
212
|
+
THEN COALESCE(${o.cached_input_tokens}, 0)
|
|
190
213
|
ELSE 0
|
|
191
214
|
END
|
|
192
|
-
`,
|
|
215
|
+
`,c=n?drizzleOrm.sql`WHERE ${o.status} IN ('completed', 'success') AND ${o.started_at} >= ${t} AND ${o.started_at} < ${e} AND ${o.workspace_id} = ${n}`:drizzleOrm.sql`WHERE ${o.status} IN ('completed', 'success') AND ${o.started_at} >= ${t} AND ${o.started_at} < ${e}`;return r.db.all(drizzleOrm.sql`
|
|
193
216
|
SELECT
|
|
194
217
|
${s} AS provider,
|
|
195
218
|
COUNT(*) AS total_tasks,
|
|
196
|
-
COALESCE(SUM(${
|
|
197
|
-
COALESCE(SUM(${
|
|
198
|
-
COALESCE(SUM(${
|
|
199
|
-
COALESCE(SUM(${
|
|
200
|
-
COALESCE(SUM(${
|
|
201
|
-
MAX(${
|
|
202
|
-
FROM ${
|
|
203
|
-
${
|
|
204
|
-
GROUP BY
|
|
205
|
-
ORDER BY (COALESCE(SUM(${
|
|
206
|
-
`).map(
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
|
|
214
|
-
|
|
219
|
+
COALESCE(SUM(${a}), 0) AS input_tokens,
|
|
220
|
+
COALESCE(SUM(${o.output_tokens}), 0) AS output_tokens,
|
|
221
|
+
COALESCE(SUM(${o.cached_input_tokens}), 0) AS cached_input_tokens,
|
|
222
|
+
COALESCE(SUM(${o.cost_usd}), 0) AS cost_usd,
|
|
223
|
+
COALESCE(SUM(${o.duration_ms}), 0) AS active_duration_ms,
|
|
224
|
+
MAX(${o.completed_at}) AS last_active_at
|
|
225
|
+
FROM ${o}
|
|
226
|
+
${c}
|
|
227
|
+
GROUP BY ${s}
|
|
228
|
+
ORDER BY (COALESCE(SUM(${a}), 0) + COALESCE(SUM(${o.output_tokens}), 0)) DESC
|
|
229
|
+
`).map(l=>({provider:l.provider,totalTasks:l.total_tasks,inputTokens:l.input_tokens,outputTokens:l.output_tokens,cachedInputTokens:l.cached_input_tokens,costUsd:l.cost_usd,totalTokens:l.input_tokens+l.output_tokens,activeDurationMs:l.active_duration_ms??0,lastActiveAt:l.last_active_at??null}))}catch(s){throw new d("DB_ERROR","Failed to get provider usage",s)}finally{r.close();}}getModelUsage(t,e,n){if(!this.dbExists())return [];let r=this.openHandle(false);try{let s=Oe(),a=drizzleOrm.sql`
|
|
230
|
+
COALESCE(${o.input_tokens}, 0)
|
|
231
|
+
+ CASE
|
|
232
|
+
WHEN ${o.started_at} >= ${_t}
|
|
233
|
+
AND (
|
|
234
|
+
${o.crewx_version} IS NULL
|
|
235
|
+
OR (${o.crewx_version} LIKE '0.8.%' AND ${o.crewx_version} NOT LIKE '0.8.9%')
|
|
236
|
+
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${pt}) AS INTEGER) < ${ft})
|
|
237
|
+
)
|
|
238
|
+
THEN COALESCE(${o.cached_input_tokens}, 0)
|
|
239
|
+
ELSE 0
|
|
240
|
+
END
|
|
241
|
+
`,c=n?drizzleOrm.sql`WHERE ${o.status} IN ('completed', 'success') AND ${o.started_at} >= ${t} AND ${o.started_at} < ${e} AND ${o.workspace_id} = ${n}`:drizzleOrm.sql`WHERE ${o.status} IN ('completed', 'success') AND ${o.started_at} >= ${t} AND ${o.started_at} < ${e}`,u=r.db.all(drizzleOrm.sql`
|
|
242
|
+
SELECT
|
|
243
|
+
${o.model} AS model,
|
|
244
|
+
${s} AS provider,
|
|
245
|
+
COUNT(*) AS total_tasks,
|
|
246
|
+
COALESCE(SUM(${a}), 0) AS input_tokens,
|
|
247
|
+
COALESCE(SUM(${o.output_tokens}), 0) AS output_tokens,
|
|
248
|
+
COALESCE(SUM(${o.cached_input_tokens}), 0) AS cached_input_tokens,
|
|
249
|
+
COALESCE(SUM(${o.cost_usd}), 0) AS cost_usd,
|
|
250
|
+
COALESCE(SUM(${o.duration_ms}), 0) AS active_duration_ms,
|
|
251
|
+
MAX(${o.completed_at}) AS last_active_at
|
|
252
|
+
FROM ${o}
|
|
253
|
+
${c}
|
|
254
|
+
GROUP BY ${o.model}, ${s}
|
|
255
|
+
`),l=new Map;for(let _ of u){let f=Ae(_.model??""),m=JSON.stringify([_.provider,f]),R=l.get(m);R?(R.totalTasks+=_.total_tasks,R.inputTokens+=_.input_tokens,R.outputTokens+=_.output_tokens,R.cachedInputTokens+=_.cached_input_tokens,R.costUsd+=_.cost_usd,R.totalTokens+=_.input_tokens+_.output_tokens,R.activeDurationMs+=_.active_duration_ms??0,_.last_active_at&&(!R.lastActiveAt||_.last_active_at>R.lastActiveAt)&&(R.lastActiveAt=_.last_active_at)):l.set(m,{model:f,provider:_.provider,totalTasks:_.total_tasks,inputTokens:_.input_tokens,outputTokens:_.output_tokens,cachedInputTokens:_.cached_input_tokens,costUsd:_.cost_usd,totalTokens:_.input_tokens+_.output_tokens,activeDurationMs:_.active_duration_ms??0,lastActiveAt:_.last_active_at??null});}return Array.from(l.values()).sort((_,f)=>f.costUsd-_.costUsd)}catch(s){throw new d("DB_ERROR","Failed to get model usage",s)}finally{r.close();}}getTasksForReport(t,e,n){if(!this.dbExists())return [];let r=this.openHandle(false);try{let s=n?drizzleOrm.sql`WHERE ${o.started_at} >= ${t} AND ${o.started_at} < ${e} AND ${o.workspace_id} = ${n}`:drizzleOrm.sql`WHERE ${o.started_at} >= ${t} AND ${o.started_at} < ${e}`;return r.db.all(drizzleOrm.sql`
|
|
256
|
+
SELECT
|
|
257
|
+
${o.thread_id} AS thread_id,
|
|
258
|
+
${o.agent_id} AS agent_id,
|
|
259
|
+
${o.started_at} AS started_at,
|
|
260
|
+
${o.status} AS status,
|
|
261
|
+
${o.error} AS error,
|
|
262
|
+
${o.exit_code} AS exit_code
|
|
263
|
+
FROM ${o}
|
|
264
|
+
${s}
|
|
265
|
+
ORDER BY ${o.started_at} ASC
|
|
266
|
+
`).map(c=>({threadId:c.thread_id,agentId:c.agent_id,startedAt:c.started_at,status:c.status,error:c.error,exitCode:c.exit_code}))}catch(s){throw new d("DB_ERROR","Failed to get tasks for report",s)}finally{r.close();}}findTaskEvidence(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{let n=[];t.workspaceId&&n.push(drizzleOrm.eq(o.workspace_id,t.workspaceId)),t.agents&&t.agents.length>0&&n.push(drizzleOrm.inArray(o.agent_id,t.agents)),t.statuses&&t.statuses.length>0&&n.push(drizzleOrm.inArray(o.status,t.statuses)),t.from&&n.push(drizzleOrm.gte(o.started_at,t.from)),t.to&&n.push(drizzleOrm.lt(o.started_at,t.to));let r=n.length>0?drizzleOrm.and(...n):void 0,s=(t.order??"DESC")==="ASC"?drizzleOrm.asc(o.started_at):drizzleOrm.desc(o.started_at),a={id:o.id,agent_id:o.agent_id,status:o.status,prompt:o.prompt,started_at:o.started_at,completed_at:o.completed_at,thread_id:o.thread_id};return t.limit?e.db.select(a).from(o).where(r).orderBy(s).limit(t.limit).all():e.db.select(a).from(o).where(r).orderBy(s).all()}catch(n){throw new d("DB_ERROR","Failed to find task evidence",n)}finally{e.close();}}findThreadChains(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{let n=t.maxThreads??8,r=t.maxTurnsPerThread??4,s=[drizzleOrm.isNotNull(o.thread_id),drizzleOrm.isNull(o.parent_task_id),drizzleOrm.isNull(o.caller_agent_id),drizzleOrm.or(drizzleOrm.isNull(o.platform),drizzleOrm.ne(o.platform,"workflow")),drizzleOrm.or(drizzleOrm.isNull(o.metadata),drizzleOrm.and(drizzleOrm.notLike(o.metadata,'%"kind":"workflow_card"%'),drizzleOrm.notLike(o.metadata,'%"kind":"workflow_ui_instance"%'))),drizzleOrm.or(drizzleOrm.isNull(o.metadata),drizzleOrm.notLike(o.metadata,'%"retriedFromTaskId"%'))];t.workspaceId&&s.push(drizzleOrm.eq(o.workspace_id,t.workspaceId)),t.from&&s.push(drizzleOrm.gte(o.started_at,t.from)),t.to&&s.push(drizzleOrm.lt(o.started_at,t.to));let a=e.db.select({id:o.id,agent_id:o.agent_id,status:o.status,prompt:o.prompt,started_at:o.started_at,completed_at:o.completed_at,thread_id:o.thread_id}).from(o).where(drizzleOrm.and(...s)).orderBy(drizzleOrm.asc(o.started_at),drizzleOrm.asc(o.id)).all(),c=new Map;for(let l of a){if(!l.thread_id)continue;let _={id:l.id,agentId:l.agent_id,status:l.status,prompt:l.prompt,startedAt:l.started_at,completedAt:l.completed_at},f=c.get(l.thread_id);f?f.push(_):c.set(l.thread_id,[_]);}let u=[];for(let[l,_]of c){let f=_.filter(m=>m.agentId===t.targetAgentId).length;f<2||u.push({threadId:l,targetAgentTurnCount:f,turns:_});}return u.sort((l,_)=>{let f=l.turns[l.turns.length-1].startedAt,m=_.turns[_.turns.length-1].startedAt;return f<m?1:f>m?-1:0}),u.slice(0,n).map(l=>({threadId:l.threadId,targetAgentTurnCount:l.targetAgentTurnCount,turns:l.turns.slice(-r)}))}catch(n){throw new d("DB_ERROR","Failed to find thread chains",n)}finally{e.close();}}};var te=class extends N{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=it.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{A(n.db,e);}catch(r){throw n.close(),r}return n}validateWorkspaceId(t,e){return t.db.select({id:h.id}).from(h).where(drizzleOrm.eq(h.id,e)).limit(1).get()?e:null}topLevelTaskPredicateSql(t="child"){return drizzleOrm.sql.raw(`NOT EXISTS (
|
|
267
|
+
SELECT 1 FROM tasks parent
|
|
268
|
+
WHERE parent.id = ${t}.parent_task_id
|
|
269
|
+
AND parent.thread_id = ${t}.thread_id
|
|
270
|
+
)`)}findAllThreads(t){let e=this.resolveDbPaths(),n=new Set,r=[];for(let s of e){if(!xe.existsSync(s))continue;let a=w(s);try{let c=t?drizzleOrm.eq(p.workspace_id,t):void 0,u=a.db.select().from(p).where(c).orderBy(drizzleOrm.desc(p.updated_at)).all();for(let l of u)n.has(l.id)||(n.add(l.id),r.push(l));}catch(c){throw new d("DB_ERROR","Failed to find all threads",c)}finally{a.close();}}return r}findThreadsByIdsOrTitles(t,e){if(t.length===0)return [];let n=this.resolveDbPaths(),r=new Set,s=[];for(let a of n){if(!xe.existsSync(a))continue;let c=w(a);try{let u=drizzleOrm.or(drizzleOrm.inArray(p.id,t),drizzleOrm.inArray(p.title,t)),l=e?drizzleOrm.and(u,drizzleOrm.eq(p.workspace_id,e)):u,_=c.db.select().from(p).where(l).orderBy(drizzleOrm.desc(p.updated_at)).all();for(let f of _)r.has(f.id)||(r.add(f.id),s.push(f));}catch(u){throw new d("DB_ERROR","Failed to find threads by ids or titles",u)}finally{c.close();}}return s}findThreadById(t,e){let n=this.resolveDbPaths();for(let r of n){if(!xe.existsSync(r))continue;let s=w(r);try{let a=drizzleOrm.eq(p.id,t),c=e?drizzleOrm.and(a,drizzleOrm.eq(p.workspace_id,e)):a,u=s.db.select().from(p).where(c).limit(1).get()??void 0;if(u)return u}catch(a){throw new d("DB_ERROR","Failed to find thread by id",a)}finally{s.close();}}}threadExists(t,e){let n=this.resolveDbPaths();for(let r of n){if(!xe.existsSync(r))continue;let s=w(r);try{let a=drizzleOrm.eq(p.id,t),c=e?drizzleOrm.and(a,drizzleOrm.eq(p.workspace_id,e)):a;if(s.db.select({id:p.id}).from(p).where(c).limit(1).get())return !0}catch(a){throw new d("DB_ERROR","Failed to check thread existence",a)}finally{s.close();}}return false}aggregateTaskStats(t,e){let n=this.resolveDbPaths(),r=0,s=0,a=0,c=0,u=0,l=new Set;for(let _ of n){if(!xe.existsSync(_))continue;let f=w(_);try{let m=f.db.get(drizzleOrm.sql`
|
|
215
271
|
SELECT
|
|
216
272
|
count(*) AS cnt,
|
|
217
273
|
COALESCE(SUM(child.input_tokens), 0) AS total_input,
|
|
@@ -222,13 +278,13 @@ ${n.join(`
|
|
|
222
278
|
WHERE child.thread_id = ${t}
|
|
223
279
|
AND ${this.topLevelTaskPredicateSql()}
|
|
224
280
|
${e?drizzleOrm.sql`AND child.workspace_id = ${e}`:drizzleOrm.sql``}
|
|
225
|
-
`);
|
|
281
|
+
`);m&&(r+=m.cnt,s+=m.total_input,a+=m.total_output,c+=m.total_cached,u+=m.total_cost);let R=f.db.all(drizzleOrm.sql`
|
|
226
282
|
SELECT DISTINCT child.agent_id FROM tasks child
|
|
227
283
|
WHERE child.thread_id = ${t}
|
|
228
284
|
AND child.agent_id IS NOT NULL AND child.agent_id != ''
|
|
229
285
|
AND ${this.topLevelTaskPredicateSql()}
|
|
230
286
|
${e?drizzleOrm.sql`AND child.workspace_id = ${e}`:drizzleOrm.sql``}
|
|
231
|
-
`);for(let
|
|
287
|
+
`);for(let L of R)l.add(L.agent_id);}catch(m){throw new d("DB_ERROR","Failed to aggregate task stats",m)}finally{f.close();}}return {taskCount:r,inputTokens:s,outputTokens:a,cachedInputTokens:c,costUsd:u,agentIds:Array.from(l)}}findTopLevelTasks(t,e,n){let r=this.resolveDbPaths(),s=new Set,a=[];for(let c of r){if(!xe.existsSync(c))continue;let u=w(c);try{let l;n!==void 0?l=u.db.all(drizzleOrm.sql`
|
|
232
288
|
SELECT * FROM (
|
|
233
289
|
SELECT child.*,
|
|
234
290
|
ROW_NUMBER() OVER (PARTITION BY child.agent_id ORDER BY child.started_at DESC) AS rn
|
|
@@ -239,19 +295,23 @@ ${n.join(`
|
|
|
239
295
|
) ranked
|
|
240
296
|
WHERE rn <= ${n}
|
|
241
297
|
ORDER BY started_at ASC
|
|
242
|
-
`):
|
|
298
|
+
`):l=u.db.all(drizzleOrm.sql`
|
|
243
299
|
SELECT child.* FROM tasks child
|
|
244
300
|
WHERE child.thread_id = ${t}
|
|
245
301
|
AND ${this.topLevelTaskPredicateSql()}
|
|
246
302
|
${e?drizzleOrm.sql`AND child.workspace_id = ${e}`:drizzleOrm.sql``}
|
|
247
303
|
ORDER BY child.started_at ASC
|
|
248
|
-
`);for(let
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
252
|
-
|
|
253
|
-
|
|
254
|
-
|
|
304
|
+
`);for(let _ of l)s.has(_.id)||(s.add(_.id),a.push(_));}catch(l){throw new d("DB_ERROR","Failed to find top-level tasks",l)}finally{u.close();}}if(n!==void 0&&r.length>1){let c=new Map;for(let u of a){let l=u.agent_id??"";c.has(l)||c.set(l,[]),c.get(l).push(u);}a=[];for(let u of c.values())u.sort((l,_)=>(_.started_at??"").localeCompare(l.started_at??"")),a.push(...u.slice(0,n));a.sort((u,l)=>(u.started_at??"").localeCompare(l.started_at??""));}return a}findAllTasks(t,e){let n=this.resolveDbPaths(),r=new Set,s=[];for(let a of n){if(!xe.existsSync(a))continue;let c=w(a);try{let u=drizzleOrm.eq(o.thread_id,t),l=e?drizzleOrm.and(u,drizzleOrm.eq(o.workspace_id,e)):u,_=c.db.select().from(o).where(l).orderBy(drizzleOrm.asc(o.started_at)).all();for(let f of _)r.has(f.id)||(r.add(f.id),s.push(f));}catch(u){throw new d("DB_ERROR","Failed to find all tasks for thread",u)}finally{c.close();}}return s}findTaskById(t,e,n){let r=this.resolveDbPaths();for(let s of r){if(!xe.existsSync(s))continue;let a=w(s);try{let c=drizzleOrm.and(drizzleOrm.eq(o.id,e),drizzleOrm.eq(o.thread_id,t)),u=n?drizzleOrm.and(c,drizzleOrm.eq(o.workspace_id,n)):c,l=a.db.select().from(o).where(u).limit(1).get();if(!l)continue;let _=a.db.select().from(o).where(drizzleOrm.eq(o.parent_task_id,l.id)).orderBy(drizzleOrm.asc(o.started_at)).all();return {task:l,children:_}}catch(c){throw new d("DB_ERROR","Failed to find task by id",c)}finally{a.close();}}}batchFetchTasksQuery(t,e){return drizzleOrm.sql`
|
|
305
|
+
SELECT child.id, child.thread_id, child.agent_id, child.status,
|
|
306
|
+
child.parent_task_id, child.started_at, child.completed_at,
|
|
307
|
+
child.duration_ms, child.input_tokens, child.output_tokens,
|
|
308
|
+
child.cost_usd, child.error
|
|
309
|
+
FROM tasks child INDEXED BY idx_tasks_thread_id
|
|
310
|
+
WHERE child.thread_id IN (${drizzleOrm.sql.join(t.map(n=>drizzleOrm.sql`${n}`),drizzleOrm.sql`, `)})
|
|
311
|
+
AND ${this.topLevelTaskPredicateSql()}
|
|
312
|
+
${e?drizzleOrm.sql`AND child.workspace_id = ${e}`:drizzleOrm.sql``}
|
|
313
|
+
ORDER BY child.started_at ASC
|
|
314
|
+
`}batchFetchTasks(t,e){let n=new Map;if(t.length===0)return n;let r=this.resolveDbPaths();for(let s of r){if(!xe.existsSync(s))continue;let a=w(s);try{let c=a.db.all(this.batchFetchTasksQuery(t,e));for(let u of c){let l=u.thread_id;n.has(l)||n.set(l,[]),n.get(l).push(u);}}catch(c){throw new d("DB_ERROR","Failed to batch fetch tasks",c)}finally{a.close();}}return n}explainBatchFetchTasksPlan(t,e){let n=this.resolveDbPaths();for(let r of n){if(!xe.existsSync(r))continue;let s=w(r);try{return s.db.all(drizzleOrm.sql`EXPLAIN QUERY PLAN ${this.batchFetchTasksQuery(t,e)}`).map(c=>c.detail)}finally{s.close();}}return []}updateThreadTitle(t,e,n){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=drizzleOrm.eq(p.id,t),a=n?drizzleOrm.and(s,drizzleOrm.eq(p.workspace_id,n)):s;if(!r.db.select({id:p.id}).from(p).where(a).limit(1).get())return;r.db.update(p).set({title:e,title_locked:1,updated_at:new Date().toISOString()}).where(drizzleOrm.eq(p.id,t)).run();}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to update thread title",s)}finally{r.close();}}upsertThread(t,e){let n=this.openHandle(true);try{let r=e.workspaceId?this.validateWorkspaceId(n,e.workspaceId):null,s=new Date().toISOString();if(n.db.select({id:p.id,message_count:p.message_count}).from(p).where(drizzleOrm.eq(p.id,t)).limit(1).get()){let c={updated_at:s};e.title!==void 0&&(c.title=e.title),e.titleLocked!==void 0&&(c.title_locked=e.titleLocked?1:0),n.db.update(p).set(c).where(drizzleOrm.eq(p.id,t)).run();}else n.db.insert(p).values({id:t,platform:e.platform,workspace_id:r,title:e.title??null,title_locked:e.titleLocked?1:0,message_count:0,created_at:s,updated_at:s}).run();}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to upsert thread",r)}finally{n.close();}}ensureThread(t,e,n){let r=this.openHandle(true);try{let s=n?this.validateWorkspaceId(r,n):null,a=r.db.select({id:p.id,platform:p.platform,workspace_id:p.workspace_id}).from(p).where(drizzleOrm.eq(p.id,t)).limit(1).get();if(a){s&&!a.workspace_id&&r.db.update(p).set({workspace_id:s}).where(drizzleOrm.eq(p.id,t)).run();return}let c=new Date().toISOString();r.db.insert(p).values({id:t,platform:e,workspace_id:s,message_count:0,created_at:c,updated_at:c}).run();}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to ensure thread",s)}finally{r.close();}}saveUserMessage(t,e,n){if(!this.dbExists())return {firstMessage:false};let r=this.openHandle(true);try{let s=new Date().toISOString();return {firstMessage:r.db.transaction(c=>{let l=c.select({message_count:p.message_count}).from(p).where(drizzleOrm.eq(p.id,t)).limit(1).get()?.message_count===0;return c.run(drizzleOrm.sql`
|
|
255
315
|
UPDATE threads
|
|
256
316
|
SET first_message = COALESCE(first_message, ${e}),
|
|
257
317
|
title = CASE WHEN title_locked = 0 AND title IS NULL THEN substr(${e}, 1, 60) ELSE title END,
|
|
@@ -259,4 +319,4 @@ ${n.join(`
|
|
|
259
319
|
message_count = message_count + 1,
|
|
260
320
|
updated_at = ${s}
|
|
261
321
|
WHERE id = ${t}
|
|
262
|
-
`),c},{behavior:"immediate"})}}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to save user message",s)}finally{r.close();}}saveAssistantMessage(t,e,n){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=new Date().toISOString();r.db.update(p).set({last_message:e,updated_at:s}).where(drizzleOrm.eq(p.id,t)).run();}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to save assistant message",s)}finally{r.close();}}updateThread(t,e){if(!this.dbExists())return;let n=this.openHandle(true);try{let r={updated_at:new Date().toISOString()};e.title!==void 0&&(r.title=e.title,r.title_locked=1),e.titleLocked!==void 0&&(r.title_locked=e.titleLocked?1:0),n.db.update(p).set(r).where(drizzleOrm.eq(p.id,t)).run();}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to update thread",r)}finally{n.close();}}togglePin(t,e){let n=this.openHandle(true);try{let r=e?drizzleOrm.and(drizzleOrm.eq(p.id,t),drizzleOrm.eq(p.workspace_id,e)):drizzleOrm.eq(p.id,t),s=n.db.select({pinned:p.pinned,metadata:p.metadata}).from(p).where(r).get();if(!s)return null;let o=s.pinned?0:1,l=s.metadata?JSON.parse(s.metadata):{};if(o){let u=e?drizzleOrm.and(drizzleOrm.eq(p.pinned,1),drizzleOrm.eq(p.workspace_id,e)):drizzleOrm.eq(p.pinned,1),c=n.db.select({metadata:p.metadata}).from(p).where(u).all(),h=0;for(let m of c){let g=m.metadata?JSON.parse(m.metadata):{};typeof g.pinOrder=="number"&&g.pinOrder>h&&(h=g.pinOrder);}l.pinOrder=h+1;}else delete l.pinOrder;return n.db.update(p).set({pinned:o,metadata:Object.keys(l).length>0?JSON.stringify(l):null}).where(r).run(),{pinned:!!o}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle pin",r)}finally{n.close();}}reorderPins(t,e){let n=this.openHandle(true);try{for(let r=0;r<t.length;r++){let s=e?drizzleOrm.and(drizzleOrm.eq(p.id,t[r]),drizzleOrm.eq(p.workspace_id,e)):drizzleOrm.eq(p.id,t[r]),o=n.db.select({metadata:p.metadata}).from(p).where(s).get();if(!o)continue;let l=o.metadata?JSON.parse(o.metadata):{};l.pinOrder=r+1,n.db.update(p).set({metadata:JSON.stringify(l)}).where(s).run();}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to reorder pins",r)}finally{n.close();}}toggleStar(t,e){let n=this.openHandle(true);try{let r=e?drizzleOrm.and(drizzleOrm.eq(p.id,t),drizzleOrm.eq(p.workspace_id,e)):drizzleOrm.eq(p.id,t),s=n.db.select({starred:p.starred}).from(p).where(r).get();if(!s)return null;let o=s.starred?0:1;return n.db.update(p).set({starred:o}).where(r).run(),{starred:!!o}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle star",r)}finally{n.close();}}};var xt=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=z.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=z.dirname(e);Ft.existsSync(r)||Ft.mkdirSync(r,{recursive:true});}else if(!Ft.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=k(e);if(t)try{x(n.db,e);}catch(r){throw n.close(),r}return n}insertSpan(t){let e=this.openHandle(true);try{e.db.insert(U).values(t).run();}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to insert span",n)}finally{e.close();}}findByTaskId(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.select().from(U).where(drizzleOrm.eq(U.task_id,t)).all()}catch(n){throw new d("DB_ERROR","Failed to find spans by task id",n)}finally{e.close();}}findById(t){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(U).where(drizzleOrm.eq(U.id,t)).limit(1).get()??void 0}catch(n){throw new d("DB_ERROR","Failed to find span by id",n)}finally{e.close();}}};var Dt=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=z.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=z.dirname(e);Ft.existsSync(r)||Ft.mkdirSync(r,{recursive:true});}else if(!Ft.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=k(e);if(t)try{x(n.db,e);}catch(r){throw n.close(),r}return n}insertToolCall(t){let e=this.openHandle(true);try{e.db.insert($).values(t).run();}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to insert tool call",n)}finally{e.close();}}findByTaskId(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.select().from($).where(drizzleOrm.eq($.task_id,t)).all()}catch(n){throw new d("DB_ERROR","Failed to find tool calls by task id",n)}finally{e.close();}}aggregateByName(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.select({toolName:$.tool_name,count:drizzleOrm.sql`count(*)`}).from($).where(drizzleOrm.eq($.task_id,t)).groupBy($.tool_name).all().map(n=>({toolName:n.toolName,count:n.count}))}catch(n){throw new d("DB_ERROR","Failed to aggregate tool calls by name",n)}finally{e.close();}}};var At=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=z.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=z.dirname(e);Ft.existsSync(r)||Ft.mkdirSync(r,{recursive:true});}else if(!Ft.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=k(e);if(t)try{x(n.db,e);}catch(r){throw n.close(),r}return n}ensureThreadExists(t,e){if(!t.db.select({id:p.id}).from(p).where(drizzleOrm.eq(p.id,e)).limit(1).get())throw new d("NOT_FOUND",`Thread not found: ${e}`)}findByThreadId(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return this.ensureThreadExists(e,t),e.db.select().from(L).where(drizzleOrm.eq(L.thread_id,t)).orderBy(drizzleOrm.asc(L.seq)).all()}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to find boxes by thread id",n)}finally{e.close();}}findById(t,e){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(L).where(drizzleOrm.and(drizzleOrm.eq(L.id,e),drizzleOrm.eq(L.thread_id,t))).limit(1).get()??void 0}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to find box by id",r)}finally{n.close();}}insert(t){let e=this.openHandle(true);try{this.ensureThreadExists(e,t.threadId);try{e.db.insert(L).values({id:t.id,thread_id:t.threadId,seq:t.seq,first_task_id:t.first_task_id,mid_task_id:t.mid_task_id,last_task_id:t.last_task_id,task_count:t.task_count,summary:t.summary??null,source_tokens:t.source_tokens,summary_tokens:t.summary_tokens??null,created_at:t.created_at}).run();}catch(r){throw r instanceof Error&&/UNIQUE constraint failed/i.test(r.message)?new d("CONFLICT",`Duplicate seq ${String(t.seq)} for thread ${t.threadId}`,r):r}let n=e.db.select().from(L).where(drizzleOrm.eq(L.id,t.id)).limit(1).get();if(!n)throw new d("DB_ERROR","Insert did not return a row");return n}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to insert thread box",n)}finally{e.close();}}};var Ot=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=z.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=z.dirname(e);Ft.existsSync(r)||Ft.mkdirSync(r,{recursive:true});}else if(!Ft.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=k(e);if(t)try{x(n.db,e);}catch(r){throw n.close(),r}return n}bulkInsert(t){if(t.length===0)return;let e=this.openHandle(true);try{let n=new Date().toISOString();e.db.transaction(r=>{for(let s of t)r.insert(Z).values({id:crypto.randomUUID(),path:s.path,method:s.method,status_code:s.statusCode,duration_ms:s.durationMs,ip:s.ip??null,request_headers:s.requestHeaders??null,response_headers:s.responseHeaders??null,request_body:s.requestBody??null,response_body:s.responseBody??null,query:s.query??null,user_id:s.userId??null,project_id:s.projectId??null,partition_key:s.partitionKey??"default",timestamp:n,metadata:s.metadata??null}).run();});}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to bulk insert request logs",n)}finally{e.close();}}findRecent(t=100){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.select().from(Z).orderBy(drizzleOrm.desc(Z.timestamp)).limit(t).all()}catch(n){throw new d("DB_ERROR","Failed to find recent request logs",n)}finally{e.close();}}};function ne(a){return new Date(Math.floor(a/3e5)*3e5).toISOString()}var Nt=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=z.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=z.dirname(e);Ft.existsSync(r)||Ft.mkdirSync(r,{recursive:true});}else if(!Ft.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=k(e);if(t)try{x(n.db,e);}catch(r){throw n.close(),r}return n}upsert(t){let e=this.openHandle(true);try{let n=ne(t.bucketStartMs),r=t.accountRef??"default";e.db.insert(T).values({id:t.id,provider:t.provider,account_ref:r,limit_window:t.limitWindow,bucket_start:n,captured_at:t.capturedAt,used_percent:t.usedPercent,remaining_percent:t.remainingPercent,resets_at:t.resetsAt??null,source:t.source,metadata:t.metadata??null}).onConflictDoUpdate({target:[T.provider,T.account_ref,T.limit_window,T.bucket_start],set:{captured_at:t.capturedAt,used_percent:t.usedPercent,remaining_percent:t.remainingPercent,resets_at:t.resetsAt??null,metadata:t.metadata??null}}).run();}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to upsert usage limit snapshot",n)}finally{e.close();}}getTrends(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{let n=e.db.select({provider:T.provider,bucket_start:T.bucket_start,used_percent:T.used_percent,remaining_percent:T.remaining_percent,resets_at:T.resets_at}).from(T).where(drizzleOrm.and(drizzleOrm.eq(T.limit_window,t.window),drizzleOrm.gte(T.bucket_start,t.fromIso),drizzleOrm.lte(T.bucket_start,t.toIso))).orderBy(drizzleOrm.asc(T.bucket_start)).all(),r=new Map;for(let s of n)r.has(s.provider)||r.set(s.provider,[]),r.get(s.provider).push({bucketStart:s.bucket_start,usedPercent:s.used_percent,remainingPercent:s.remaining_percent,resetsAt:s.resets_at??null});return Array.from(r.entries()).map(([s,o])=>({provider:s,points:o}))}catch(n){throw new d("DB_ERROR","Failed to get usage limit trends",n)}finally{e.close();}}};exports.BaseSqliteRepository=S;exports.RepositoryError=d;exports.RequestLogRepository=Ot;exports.SpanRepository=xt;exports.TaskRepository=St;exports.ThreadBoxRepository=At;exports.ThreadRepository=Tt;exports.ToolCallRepository=Dt;exports.UsageLimitSnapshotRepository=Nt;exports.WorkspaceRepository=Et;exports.calcBucketStart=ne;exports.openDrizzleDb=k;exports.pushSchema=ze;exports.runMigrations=Q;exports.runMigrationsOnce=x;exports.tasks=i;
|
|
322
|
+
`),l},{behavior:"immediate"})}}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to save user message",s)}finally{r.close();}}saveAssistantMessage(t,e,n){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=new Date().toISOString();r.db.update(p).set({last_message:e,updated_at:s}).where(drizzleOrm.eq(p.id,t)).run();}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to save assistant message",s)}finally{r.close();}}updateThread(t,e){if(!this.dbExists())return;let n=this.openHandle(true);try{let r={updated_at:new Date().toISOString()};e.title!==void 0&&(r.title=e.title,r.title_locked=1),e.titleLocked!==void 0&&(r.title_locked=e.titleLocked?1:0),n.db.update(p).set(r).where(drizzleOrm.eq(p.id,t)).run();}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to update thread",r)}finally{n.close();}}togglePin(t,e){let n=this.openHandle(true);try{let r=e?drizzleOrm.and(drizzleOrm.eq(p.id,t),drizzleOrm.eq(p.workspace_id,e)):drizzleOrm.eq(p.id,t),s=n.db.select({pinned:p.pinned,metadata:p.metadata}).from(p).where(r).get();if(!s)return null;let a=s.pinned?0:1,c=s.metadata?JSON.parse(s.metadata):{};if(a){let u=e?drizzleOrm.and(drizzleOrm.eq(p.pinned,1),drizzleOrm.eq(p.workspace_id,e)):drizzleOrm.eq(p.pinned,1),l=n.db.select({metadata:p.metadata}).from(p).where(u).all(),_=null;for(let f of l){let m=f.metadata?JSON.parse(f.metadata):{};typeof m.pinOrder=="number"&&(_===null||m.pinOrder<_)&&(_=m.pinOrder);}c.pinOrder=_===null?0:_-1;}else delete c.pinOrder;return n.db.update(p).set({pinned:a,metadata:Object.keys(c).length>0?JSON.stringify(c):null}).where(r).run(),{pinned:!!a}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle pin",r)}finally{n.close();}}reorderPins(t,e){let n=this.openHandle(true);try{for(let r=0;r<t.length;r++){let s=e?drizzleOrm.and(drizzleOrm.eq(p.id,t[r]),drizzleOrm.eq(p.workspace_id,e)):drizzleOrm.eq(p.id,t[r]),a=n.db.select({metadata:p.metadata}).from(p).where(s).get();if(!a)continue;let c=a.metadata?JSON.parse(a.metadata):{};c.pinOrder=r+1,n.db.update(p).set({metadata:JSON.stringify(c)}).where(s).run();}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to reorder pins",r)}finally{n.close();}}toggleStar(t,e){let n=this.openHandle(true);try{let r=e?drizzleOrm.and(drizzleOrm.eq(p.id,t),drizzleOrm.eq(p.workspace_id,e)):drizzleOrm.eq(p.id,t),s=n.db.select({starred:p.starred}).from(p).where(r).get();if(!s)return null;let a=s.starred?0:1;return n.db.update(p).set({starred:a}).where(r).run(),{starred:!!a}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle star",r)}finally{n.close();}}resolveOverdriveForRequest(t,e,n,r){let a=this.openHandle(true);try{let c=e?drizzleOrm.and(drizzleOrm.eq(p.id,t),drizzleOrm.eq(p.workspace_id,e)):drizzleOrm.eq(p.id,t);return a.db.transaction(l=>{let _=l.select({metadata:p.metadata}).from(p).where(c).get();if(!_)return null;let f=_.metadata?JSON.parse(_.metadata):{},m=f.overdrive??{},R=r?.defaultTurns??m.defaultTurns??3,L=r?.updatedBy??"ui",U=new Date().toISOString(),F,D,Q,St;switch(n){case "enable-count":{let xt=r?.turns??R,Tt=r?.consumeCurrent?Math.max(xt-1,0):xt;F=Tt>0?"count":"off",D=Tt,Q=!0,St="count";break}case "enable-latch":{F="latch",D=0,Q=!0,St="latch";break}case "disable":{F="off",D=0,Q=!1;break}default:{let xt=m.state??"off",Tt=typeof m.remaining=="number"?m.remaining:0;if(xt==="count"&&Tt>0){let le=Tt-1;le<=0?(F="off",D=0):(F="count",D=le),Q=!0,St="count";}else xt==="latch"?(F="latch",D=0,Q=!0,St="latch"):(F="off",D=0,Q=!1);break}}return f.overdrive={state:F,remaining:D,defaultTurns:R,updatedAt:U,updatedBy:L},l.update(p).set({metadata:JSON.stringify(f),updated_at:U}).where(c).run(),{applied:Q,appliedMode:St,state:F,remaining:D}},{behavior:"immediate"})}catch(c){throw c instanceof d?c:new d("DB_ERROR","Failed to resolve overdrive",c)}finally{a.close();}}};var ee=class extends N{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=it.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{A(n.db,e);}catch(r){throw n.close(),r}return n}insertSpan(t){let e=this.openHandle(true);try{e.db.insert(dt).values(t).run();}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to insert span",n)}finally{e.close();}}findByTaskId(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.select().from(dt).where(drizzleOrm.eq(dt.task_id,t)).all()}catch(n){throw new d("DB_ERROR","Failed to find spans by task id",n)}finally{e.close();}}findById(t){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(dt).where(drizzleOrm.eq(dt.id,t)).limit(1).get()??void 0}catch(n){throw new d("DB_ERROR","Failed to find span by id",n)}finally{e.close();}}};var ne=class extends N{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=it.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{A(n.db,e);}catch(r){throw n.close(),r}return n}insertToolCall(t){let e=this.openHandle(true);try{e.db.insert(nt).values(t).run();}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to insert tool call",n)}finally{e.close();}}findByTaskId(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.select().from(nt).where(drizzleOrm.eq(nt.task_id,t)).all()}catch(n){throw new d("DB_ERROR","Failed to find tool calls by task id",n)}finally{e.close();}}aggregateByName(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.select({toolName:nt.tool_name,count:drizzleOrm.sql`count(*)`}).from(nt).where(drizzleOrm.eq(nt.task_id,t)).groupBy(nt.tool_name).all().map(n=>({toolName:n.toolName,count:n.count}))}catch(n){throw new d("DB_ERROR","Failed to aggregate tool calls by name",n)}finally{e.close();}}};var re=class extends N{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=it.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{A(n.db,e);}catch(r){throw n.close(),r}return n}ensureThreadExists(t,e){if(!t.db.select({id:p.id}).from(p).where(drizzleOrm.eq(p.id,e)).limit(1).get())throw new d("NOT_FOUND",`Thread not found: ${e}`)}findByThreadId(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return this.ensureThreadExists(e,t),e.db.select().from(Y).where(drizzleOrm.eq(Y.thread_id,t)).orderBy(drizzleOrm.asc(Y.seq)).all()}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to find boxes by thread id",n)}finally{e.close();}}findById(t,e){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(Y).where(drizzleOrm.and(drizzleOrm.eq(Y.id,e),drizzleOrm.eq(Y.thread_id,t))).limit(1).get()??void 0}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to find box by id",r)}finally{n.close();}}insert(t){let e=this.openHandle(true);try{this.ensureThreadExists(e,t.threadId);try{e.db.insert(Y).values({id:t.id,thread_id:t.threadId,seq:t.seq,first_task_id:t.first_task_id,mid_task_id:t.mid_task_id,last_task_id:t.last_task_id,task_count:t.task_count,summary:t.summary??null,source_tokens:t.source_tokens,summary_tokens:t.summary_tokens??null,created_at:t.created_at}).run();}catch(r){throw r instanceof Error&&/UNIQUE constraint failed/i.test(r.message)?new d("CONFLICT",`Duplicate seq ${String(t.seq)} for thread ${t.threadId}`,r):r}let n=e.db.select().from(Y).where(drizzleOrm.eq(Y.id,t.id)).limit(1).get();if(!n)throw new d("DB_ERROR","Insert did not return a row");return n}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to insert thread box",n)}finally{e.close();}}};var oe=class extends N{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=it.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{A(n.db,e);}catch(r){throw n.close(),r}return n}bulkInsert(t){if(t.length===0)return;let e=this.openHandle(true);try{let n=new Date().toISOString();e.db.transaction(r=>{for(let s of t)r.insert(Rt).values({id:crypto.randomUUID(),path:s.path,method:s.method,status_code:s.statusCode,duration_ms:s.durationMs,ip:s.ip??null,request_headers:s.requestHeaders??null,response_headers:s.responseHeaders??null,request_body:s.requestBody??null,response_body:s.responseBody??null,query:s.query??null,user_id:s.userId??null,project_id:s.projectId??null,partition_key:s.partitionKey??"default",timestamp:n,metadata:s.metadata??null}).run();});}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to bulk insert request logs",n)}finally{e.close();}}findRecent(t=100){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.select().from(Rt).orderBy(drizzleOrm.desc(Rt.timestamp)).limit(t).all()}catch(n){throw new d("DB_ERROR","Failed to find recent request logs",n)}finally{e.close();}}};function ze(i){return new Date(Math.floor(i/3e5)*3e5).toISOString()}var se=class extends N{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=it.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{A(n.db,e);}catch(r){throw n.close(),r}return n}upsert(t){let e=this.openHandle(true);try{let n=ze(t.bucketStartMs),r=t.accountRef??"default";e.db.insert(I).values({id:t.id,provider:t.provider,account_ref:r,limit_window:t.limitWindow,bucket_start:n,captured_at:t.capturedAt,used_percent:t.usedPercent,remaining_percent:t.remainingPercent,resets_at:t.resetsAt??null,source:t.source,metadata:t.metadata??null}).onConflictDoUpdate({target:[I.provider,I.account_ref,I.limit_window,I.bucket_start],set:{captured_at:t.capturedAt,used_percent:t.usedPercent,remaining_percent:t.remainingPercent,resets_at:t.resetsAt??null,metadata:t.metadata??null}}).run();}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to upsert usage limit snapshot",n)}finally{e.close();}}getTrends(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{let n=e.db.select({provider:I.provider,bucket_start:I.bucket_start,used_percent:I.used_percent,remaining_percent:I.remaining_percent,resets_at:I.resets_at}).from(I).where(drizzleOrm.and(drizzleOrm.eq(I.limit_window,t.window),drizzleOrm.gte(I.bucket_start,t.fromIso),drizzleOrm.lte(I.bucket_start,t.toIso))).orderBy(drizzleOrm.asc(I.bucket_start)).all(),r=new Map;for(let s of n)r.has(s.provider)||r.set(s.provider,[]),r.get(s.provider).push({bucketStart:s.bucket_start,usedPercent:s.used_percent,remainingPercent:s.remaining_percent,resetsAt:s.resets_at??null});return Array.from(r.entries()).map(([s,a])=>({provider:s,points:a}))}catch(n){throw new d("DB_ERROR","Failed to get usage limit trends",n)}finally{e.close();}}};var ie=class extends N{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=it.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{A(n.db,e);}catch(r){throw n.close(),r}return n}publish(t){let e=this.openHandle(true);try{e.db.insert(z).values({id:t.id,workspace_id:t.workspaceId,month:t.month,generated_at:t.generatedAt,source_range:t.sourceRange,tier:t.tier,total_tokens:t.totalTokens,total_cost_usd:t.totalCostUsd,payload:t.payload}).onConflictDoUpdate({target:[z.workspace_id,z.month],set:{generated_at:t.generatedAt,source_range:t.sourceRange,tier:t.tier,total_tokens:t.totalTokens,total_cost_usd:t.totalCostUsd,payload:t.payload}}).run();let n=e.db.select().from(z).where(drizzleOrm.and(drizzleOrm.eq(z.workspace_id,t.workspaceId),drizzleOrm.eq(z.month,t.month))).limit(1).get();if(!n)throw new d("DB_ERROR","Failed to read back published usage report");return n}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to publish usage report",n)}finally{e.close();}}findByMonth(t,e){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(z).where(drizzleOrm.and(drizzleOrm.eq(z.workspace_id,t),drizzleOrm.eq(z.month,e))).limit(1).get()??void 0}catch(r){throw new d("DB_ERROR","Failed to find usage report",r)}finally{n.close();}}listByWorkspace(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.select().from(z).where(drizzleOrm.eq(z.workspace_id,t)).orderBy(drizzleOrm.desc(z.month)).all()}catch(n){throw new d("DB_ERROR","Failed to list usage reports",n)}finally{e.close();}}};var Ke="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";function Mr(i){let t=crypto.randomBytes(i*2),e="";for(let n=0;n<t.length&&e.length<i;n++){let r=t[n];r<248&&(e+=Ke[r%62]);}for(;e.length<i;)e+=Ke[crypto.randomBytes(1)[0]%62];return e}function jt(i){return `${i}_${Mr(8)}`}var Ve=500,Qe=50,ae=class extends N{dbPath;cap;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=it.join(t.dbRoot,".crewx","crewx.db")),this.cap=t.cap??Ve;}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{A(n.db,e);}catch(r){throw n.close(),r}return n}create(t){let e=this.openHandle(true);try{let n=jt("ntf"),r=new Date().toISOString();return e.db.insert(k).values({id:n,workspace_id:t.workspaceId,agent_id:t.agentId??null,task_id:t.taskId??null,thread_id:t.threadId??null,source:t.source??"agent",level:t.level??"info",title:t.title,body:t.body??null,target_user_id:t.targetUserId??null,created_at:r,metadata:t.metadata??null}).run(),this.enforceCap(e,t.workspaceId),{id:n}}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to create notification",n)}finally{e.close();}}enforceCap(t,e){let s=(t.db.select({count:drizzleOrm.sql`count(*)`}).from(k).where(drizzleOrm.eq(k.workspace_id,e)).get()?.count??0)-this.cap;if(s<=0)return;let c=t.db.select({id:k.id}).from(k).innerJoin(v,drizzleOrm.eq(v.notification_id,k.id)).where(drizzleOrm.eq(k.workspace_id,e)).groupBy(k.id).orderBy(drizzleOrm.asc(k.created_at)).limit(s).all().map(u=>u.id);c.length!==0&&(t.db.delete(v).where(drizzleOrm.inArray(v.notification_id,c)).run(),t.db.delete(k).where(drizzleOrm.inArray(k.id,c)).run());}list(t){if(!this.dbExists())return [];let e=t.limit??Qe,n=this.openHandle(false);try{return n.db.select({id:k.id,workspace_id:k.workspace_id,agent_id:k.agent_id,task_id:k.task_id,thread_id:k.thread_id,source:k.source,level:k.level,title:k.title,body:k.body,target_user_id:k.target_user_id,created_at:k.created_at,metadata:k.metadata,read_at:v.read_at}).from(k).leftJoin(v,drizzleOrm.and(drizzleOrm.eq(v.notification_id,k.id),drizzleOrm.eq(v.user_id,t.userId))).where(drizzleOrm.eq(k.workspace_id,t.workspaceId)).orderBy(drizzleOrm.desc(k.created_at)).limit(e).all().map(s=>({...s,read_at:s.read_at??null,read:s.read_at!=null}))}catch(r){throw new d("DB_ERROR","Failed to list notifications",r)}finally{n.close();}}unreadCount(t){if(!this.dbExists())return 0;let e=this.openHandle(false);try{return e.db.select({count:drizzleOrm.sql`count(*)`}).from(k).leftJoin(v,drizzleOrm.and(drizzleOrm.eq(v.notification_id,k.id),drizzleOrm.eq(v.user_id,t.userId))).where(drizzleOrm.and(drizzleOrm.eq(k.workspace_id,t.workspaceId),drizzleOrm.isNull(v.notification_id))).get()?.count??0}catch(n){throw new d("DB_ERROR","Failed to count unread notifications",n)}finally{e.close();}}markRead(t){let e=this.openHandle(true);try{if(!e.db.select({id:k.id}).from(k).where(drizzleOrm.and(drizzleOrm.eq(k.id,t.notificationId),drizzleOrm.eq(k.workspace_id,t.workspaceId))).get())throw new d("NOT_FOUND","Notification not found in workspace");return e.db.insert(v).values({notification_id:t.notificationId,user_id:t.userId,read_at:new Date().toISOString()}).onConflictDoNothing().run().changes>0}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to mark notification read",n)}finally{e.close();}}markAllRead(t){let e=this.openHandle(true);try{let n=e.db.select({id:k.id}).from(k).leftJoin(v,drizzleOrm.and(drizzleOrm.eq(v.notification_id,k.id),drizzleOrm.eq(v.user_id,t.userId))).where(drizzleOrm.and(drizzleOrm.eq(k.workspace_id,t.workspaceId),drizzleOrm.isNull(v.notification_id))).all();if(n.length===0)return 0;let r=new Date().toISOString();return e.db.transaction(s=>{for(let a of n)s.insert(v).values({notification_id:a.id,user_id:t.userId,read_at:r}).onConflictDoNothing().run();}),n.length}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to mark all notifications read",n)}finally{e.close();}}};var en=50,ce=class extends N{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=it.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let r=it.dirname(e);xe.existsSync(r)||xe.mkdirSync(r,{recursive:true});}else if(!xe.existsSync(e))throw new d("NOT_FOUND","Database not found");let n=w(e);if(t)try{A(n.db,e);}catch(r){throw n.close(),r}return n}create(t){let e=this.openHandle(true);try{let n=new Date().toISOString(),r=[],s=t.map(a=>{let c=jt("sug");return r.push(c),{id:c,workspace_id:a.workspaceId,agent_id:a.agentId,task_id:a.taskId??null,type:a.type,status:a.status??"pending",payload:a.payload,applied_commit_sha:a.appliedCommitSha??null,created_at:n,updated_at:n}});return s.length>0&&e.db.insert(y).values(s).run(),{ids:r}}catch(n){throw n instanceof d?n:new d("DB_ERROR","Failed to create agent suggestions",n)}finally{e.close();}}findLatestSet(t,e){if(!this.dbExists())return [];let n=this.openHandle(false);try{let r=n.db.select({task_id:y.task_id}).from(y).where(drizzleOrm.and(drizzleOrm.eq(y.workspace_id,t),drizzleOrm.eq(y.agent_id,e))).orderBy(drizzleOrm.desc(y.created_at)).limit(1).get();if(!r)return [];let s=r.task_id===null?drizzleOrm.isNull(y.task_id):drizzleOrm.eq(y.task_id,r.task_id);return n.db.select().from(y).where(drizzleOrm.and(drizzleOrm.eq(y.workspace_id,t),drizzleOrm.eq(y.agent_id,e),s)).orderBy(drizzleOrm.asc(y.created_at)).all()}catch(r){throw new d("DB_ERROR","Failed to find latest agent suggestion set",r)}finally{n.close();}}findSetByTaskId(t,e,n){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(y).where(drizzleOrm.and(drizzleOrm.eq(y.workspace_id,t),drizzleOrm.eq(y.agent_id,e),drizzleOrm.eq(y.task_id,n))).orderBy(drizzleOrm.asc(y.created_at)).all()}catch(s){throw new d("DB_ERROR","Failed to find agent suggestion set by task id",s)}finally{r.close();}}findById(t,e,n){if(!this.dbExists())return null;let r=this.openHandle(false);try{return r.db.select().from(y).where(drizzleOrm.and(drizzleOrm.eq(y.workspace_id,t),drizzleOrm.eq(y.agent_id,e),drizzleOrm.eq(y.id,n))).get()??null}catch(s){throw new d("DB_ERROR","Failed to find agent suggestion by id",s)}finally{r.close();}}updateStatus(t,e,n){let r=this.openHandle(true);try{let s={status:e,updated_at:new Date().toISOString()};return n!==void 0&&(s.applied_commit_sha=n),r.db.update(y).set(s).where(drizzleOrm.eq(y.id,t)).run().changes>0}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to update agent suggestion status",s)}finally{r.close();}}listByAgent(t,e,n=en){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(y).where(drizzleOrm.and(drizzleOrm.eq(y.workspace_id,t),drizzleOrm.eq(y.agent_id,e))).orderBy(drizzleOrm.desc(y.created_at)).limit(n).all()}catch(s){throw new d("DB_ERROR","Failed to list agent suggestions",s)}finally{r.close();}}findDismissed(t,e,n=20){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(y).where(drizzleOrm.and(drizzleOrm.eq(y.workspace_id,t),drizzleOrm.eq(y.agent_id,e),drizzleOrm.eq(y.status,"dismissed"))).orderBy(drizzleOrm.desc(y.updated_at)).limit(n).all()}catch(s){throw new d("DB_ERROR","Failed to find dismissed agent suggestions",s)}finally{r.close();}}};exports.AgentSuggestionRepository=ce;exports.BaseSqliteRepository=N;exports.DEFAULT_AGENT_SUGGESTION_LIST_LIMIT=en;exports.DEFAULT_NOTIFICATION_CAP=Ve;exports.DEFAULT_NOTIFICATION_LIST_LIMIT=Qe;exports.NotificationRepository=ae;exports.RepositoryError=d;exports.RequestLogRepository=oe;exports.SpanRepository=ee;exports.TaskRepository=Zt;exports.ThreadBoxRepository=re;exports.ThreadRepository=te;exports.ToolCallRepository=ne;exports.UsageLimitSnapshotRepository=se;exports.UsageReportRepository=ie;exports.WorkspaceRepository=Vt;exports.calcBucketStart=ze;exports.openDrizzleDb=w;exports.pushSchema=Xn;exports.runMigrations=bt;exports.runMigrationsOnce=A;exports.tasks=o;
|