@crewx/sdk 0.9.0-rc.35 → 0.9.0-rc.37
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.
|
@@ -1,14 +1,14 @@
|
|
|
1
|
-
import*as
|
|
1
|
+
import*as Z from'path';import Z__default,{join,dirname,basename}from'path';import {fileURLToPath}from'url';import me,{existsSync,mkdirSync,readFileSync}from'fs';import Ze from'os';import {sql,and,eq,ne,isNull,desc,isNotNull,lt,or,like,asc,inArray,gte,notLike,lte}from'drizzle-orm';import {randomUUID,createHash,randomBytes}from'crypto';import {sqliteTable,text,integer,real,index,unique,uniqueIndex,primaryKey,getTableConfig}from'drizzle-orm/sqlite-core';var Tt=(a=>typeof require<"u"?require:typeof Proxy<"u"?new Proxy(a,{get:(t,n)=>(typeof require<"u"?require:t)[n]}):a)(function(a){if(typeof require<"u")return require.apply(this,arguments);throw Error('Dynamic require of "'+a+'" is not supported')});var Xe=()=>fileURLToPath(import.meta.url),Je=()=>Z__default.dirname(Xe()),R=Je();var T=class{resolveDbPath(){return process.env.CREWX_DB?process.env.CREWX_DB:process.env.CREWX_TRACES_DB?process.env.CREWX_TRACES_DB:join(Ze.homedir(),".crewx","crewx.db")}resolveDbPaths(){return [this.resolveDbPath()]}isMissingTableError(t){return t instanceof Error&&/no such table:/i.test(t.message)}dbExists(t){return existsSync(t??this.resolveDbPath())}};var d=class extends Error{code;cause;constructor(t,n,e){super(n),this.name="RepositoryError",this.code=t,this.cause=e,Object.setPrototypeOf(this,new.target.prototype);}};function w(a){let t=Tt("better-sqlite3"),{drizzle:n}=Tt("drizzle-orm/better-sqlite3"),e=new t(a);return e.exec("PRAGMA journal_mode = WAL"),e.exec("PRAGMA busy_timeout = 5000"),e.exec("PRAGMA foreign_keys = ON"),e.exec("PRAGMA analysis_limit = 400"),e.exec("PRAGMA optimize"),{db:n(e),runRaw:(r,s=[])=>e.prepare(r).run(...s),close:()=>e.close()}}var re=new Set,tn={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 Bt(a,t){return (a.get(`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='${t}'`)?.cnt??0)>0}function en(a,t){if(t>0||!Bt(a,"tasks"))return;let n=a.all("PRAGMA table_info(tasks)"),e=new Set(n.map(r=>r.name));for(let[r,s]of Object.entries(tn))e.has(r)||a.run(`ALTER TABLE tasks ADD COLUMN ${r} ${s}`);}var nn={"0002_normalize_task_names":{workspace_name:"TEXT",project_name:"TEXT"}};function rn(a,t,n){if(!Bt(a,"__drizzle_migrations")||!Bt(a,"tasks"))return;let e=a.all(n`SELECT hash FROM __drizzle_migrations`),r=new Set(e.map(i=>i.hash)),s=JSON.parse(readFileSync(Z__default.join(t,"meta/_journal.json"),"utf-8"));for(let i of s.entries){let l=nn[i.tag];if(!l)continue;let u=Z__default.join(t,`${i.tag}.sql`);if(!existsSync(u))continue;let c=readFileSync(u,"utf-8"),p=createHash("sha256").update(c).digest("hex");if(r.has(p))continue;let h=a.all("PRAGMA table_info(tasks)"),f=new Set(h.map(x=>x.name));for(let[x,P]of Object.entries(l))f.has(x)||(a.run(`ALTER TABLE tasks ADD COLUMN ${x} ${P}`),f.add(x));}}function on(a,t,n){let e=a.all(n`SELECT hash FROM __drizzle_migrations`),r=new Set(e.map(i=>i.hash)),s=JSON.parse(readFileSync(Z__default.join(t,"meta/_journal.json"),"utf-8"));for(let i of s.entries){let l=Z__default.join(t,`${i.tag}.sql`);if(!existsSync(l))continue;let u=readFileSync(l,"utf-8"),c=createHash("sha256").update(u).digest("hex");if(r.has(c))continue;let p=/ALTER\s+TABLE\s+[`"]?(\w+)[`"]?\s+ADD\s+[`"]?(\w+)[`"]?/gi,h=[],f;for(;(f=p.exec(u))!==null;)h.push({table:f[1],column:f[2]});if(h.length===0||!u.split(/-->\s*statement-breakpoint/).map($=>$.trim()).filter(Boolean).every($=>/^ALTER\s+TABLE\s+.+\s+ADD\s+/i.test($)))continue;h.every(({table:$,column:M})=>a.all(`PRAGMA table_info("${$}")`).some(Y=>Y.name===M))&&a.run(n`INSERT INTO __drizzle_migrations (hash, created_at) VALUES (${c}, ${i.when})`);}}function _t(a){let{migrate:t}=Tt("drizzle-orm/better-sqlite3/migrator"),{sql:n}=Tt("drizzle-orm"),e=[Z__default.join(R,"../migrations"),Z__default.join(R,"migrations"),Z__default.join(R,"../../../../drizzle/migrations"),Z__default.join(process.cwd(),"drizzle/migrations")],r=e.find(c=>existsSync(Z__default.join(c,"meta/_journal.json")));if(!r)throw new Error(`migrations folder not found. Searched:
|
|
2
2
|
${e.join(`
|
|
3
|
-
`)}`);let s=a.get(n`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='__drizzle_migrations'`),i=0;s?.cnt&&(i=a.get(n`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0),tn(a,i),s?.cnt&&(rn(a,r,n),nn(a,r,n)),t(a,{migrationsFolder:r});let u=(a.get(n`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0)-i;if(u>0){let c=s?.cnt?"Database migrated":"Database initialized";console.error(`[crewx] ${c} (${u} migration${u>1?"s":""} applied).`);}}function A(a,t){ne.has(t)||(_t(a),ne.add(t));}var m=sqliteTable("workspaces",{id:text("id").primaryKey(),slug:text("slug").notNull().unique(),name:text("name").notNull(),workspace_path:text("workspace_path"),description:text("description"),is_active:integer("is_active").notNull().default(1),created_at:text("created_at").notNull(),updated_at:text("updated_at").notNull()});var o=sqliteTable("tasks",{id:text("id").primaryKey(),agent_id:text("agent_id").notNull(),user_id:text("user_id"),prompt:text("prompt").notNull(),mode:text("mode").notNull().default("execute"),status:text("status").notNull().default("running"),result:text("result"),error:text("error"),started_at:text("started_at").notNull(),completed_at:text("completed_at"),duration_ms:integer("duration_ms"),metadata:text("metadata"),workspace_id:text("workspace_id"),trace_id:text("trace_id"),parent_task_id:text("parent_task_id"),caller_agent_id:text("caller_agent_id"),model:text("model"),platform:text("platform").default("cli"),crewx_version:text("crewx_version"),input_tokens:integer("input_tokens").default(0),output_tokens:integer("output_tokens").default(0),cost_usd:real("cost_usd").default(0),pid:integer("pid"),rendered_prompt:text("rendered_prompt"),command:text("command"),coding_agent_command:text("coding_agent_command"),exit_code:integer("exit_code"),logs:text("logs"),thread_id:text("thread_id"),workspace_ref:text("workspace_ref"),project_id:text("project_id"),project_ref:text("project_ref"),cached_input_tokens:integer("cached_input_tokens").default(0),run_epoch:integer("run_epoch").default(0)},a=>({idx_tasks_agent_id:index("idx_tasks_agent_id").on(a.agent_id),idx_tasks_status:index("idx_tasks_status").on(a.status),idx_tasks_started_at:index("idx_tasks_started_at").on(a.started_at),idx_tasks_trace_id:index("idx_tasks_trace_id").on(a.trace_id),idx_tasks_parent_task_id:index("idx_tasks_parent_task_id").on(a.parent_task_id),idx_tasks_crewx_version:index("idx_tasks_crewx_version").on(a.crewx_version),idx_tasks_pid:index("idx_tasks_pid").on(a.pid),idx_tasks_thread_id:index("idx_tasks_thread_id").on(a.thread_id),idx_tasks_workspace_id:index("idx_tasks_workspace_id").on(a.workspace_id),idx_tasks_workspace_ref:index("idx_tasks_workspace_ref").on(a.workspace_ref),idx_tasks_project_id:index("idx_tasks_project_id").on(a.project_id),idx_tasks_ws_started:index("idx_tasks_ws_started").on(a.workspace_id,a.started_at)}));var _=sqliteTable("threads",{id:text("id").primaryKey(),workspace_id:text("workspace_id").references(()=>m.id,{onDelete:"set null"}),platform:text("platform").notNull().default("cli"),title:text("title"),first_message:text("first_message"),last_message:text("last_message"),message_count:integer("message_count").notNull().default(0),created_at:text("created_at").notNull(),updated_at:text("updated_at").notNull(),metadata:text("metadata"),title_locked:integer("title_locked").notNull().default(0),pinned:integer("pinned").notNull().default(0),starred:integer("starred").notNull().default(0)},a=>({idx_threads_updated_at:index("idx_threads_updated_at").on(a.updated_at),idx_threads_workspace_id:index("idx_threads_workspace_id").on(a.workspace_id),idx_threads_ws_updated:index("idx_threads_ws_updated").on(a.workspace_id,a.updated_at),idx_threads_title:index("idx_threads_title").on(a.title)}));var tt=sqliteTable("spans",{id:text("id").primaryKey(),task_id:text("task_id").references(()=>o.id,{onDelete:"set null"}),parent_span_id:text("parent_span_id").references(()=>tt.id,{onDelete:"set null"}),name:text("name").notNull(),kind:text("kind").notNull().default("internal"),status:text("status").notNull().default("ok"),started_at:text("started_at").notNull(),completed_at:text("completed_at"),duration_ms:integer("duration_ms"),input:text("input"),output:text("output"),error:text("error"),attributes:text("attributes")},a=>({idx_spans_task_id:index("idx_spans_task_id").on(a.task_id),idx_spans_parent_span_id:index("idx_spans_parent_span_id").on(a.parent_span_id)}));var X=sqliteTable("tool_calls",{id:text("id").primaryKey(),task_id:text("task_id").references(()=>o.id,{onDelete:"cascade"}),session_id:text("session_id"),tool_name:text("tool_name").notNull(),files:text("files"),input:text("input"),output:text("output"),duration_ms:integer("duration_ms"),timestamp:text("timestamp").notNull()},a=>({idx_tool_calls_task_id:index("idx_tool_calls_task_id").on(a.task_id),idx_tool_calls_tool_name:index("idx_tool_calls_tool_name").on(a.tool_name),idx_tool_calls_timestamp:index("idx_tool_calls_timestamp").on(a.timestamp)}));var j=sqliteTable("thread_boxes",{id:text("id").primaryKey(),thread_id:text("thread_id").notNull().references(()=>_.id,{onDelete:"cascade"}),seq:integer("seq").notNull(),first_task_id:text("first_task_id").notNull(),mid_task_id:text("mid_task_id").notNull(),last_task_id:text("last_task_id").notNull(),task_count:integer("task_count").notNull(),summary:text("summary"),source_tokens:integer("source_tokens").notNull(),summary_tokens:integer("summary_tokens"),created_at:text("created_at").notNull()},a=>({idx_thread_boxes_thread_id:index("idx_thread_boxes_thread_id").on(a.thread_id),idx_thread_boxes_seq:index("idx_thread_boxes_seq").on(a.thread_id,a.seq),uniq_thread_boxes_thread_seq:unique().on(a.thread_id,a.seq)}));var ht=sqliteTable("request_logs",{id:text("id").primaryKey(),path:text("path").notNull(),method:text("method").notNull(),status_code:integer("status_code").notNull(),duration_ms:integer("duration_ms").notNull(),ip:text("ip"),request_headers:text("request_headers"),response_headers:text("response_headers"),request_body:text("request_body"),response_body:text("response_body"),query:text("query"),user_id:text("user_id"),project_id:text("project_id"),partition_key:text("partition_key").notNull(),timestamp:text("timestamp").notNull().default(sql`(datetime('now'))`),metadata:text("metadata")},a=>({idx_request_logs_timestamp:index("idx_request_logs_timestamp").on(a.timestamp),idx_request_logs_path:index("idx_request_logs_path").on(a.path),idx_request_logs_status_code:index("idx_request_logs_status_code").on(a.status_code),idx_request_logs_partition_key:index("idx_request_logs_partition_key").on(a.partition_key)}));var I=sqliteTable("usage_limit_snapshots",{id:text("id").primaryKey(),provider:text("provider").notNull(),account_ref:text("account_ref").notNull().default("default"),limit_window:text("limit_window").notNull(),bucket_start:text("bucket_start").notNull(),captured_at:text("captured_at").notNull(),used_percent:integer("used_percent").notNull(),remaining_percent:integer("remaining_percent").notNull(),resets_at:text("resets_at"),source:text("source").notNull(),metadata:text("metadata")},a=>({uniq_usage_limit_snapshots_bucket:uniqueIndex("uniq_usage_limit_snapshots_bucket").on(a.provider,a.account_ref,a.limit_window,a.bucket_start),idx_usage_limit_snapshots_window_bucket:index("idx_usage_limit_snapshots_window_bucket").on(a.limit_window,a.bucket_start),idx_usage_limit_snapshots_provider_window_bucket:index("idx_usage_limit_snapshots_provider_window_bucket").on(a.provider,a.limit_window,a.bucket_start)}));var M=sqliteTable("usage_reports",{id:text("id").primaryKey(),workspace_id:text("workspace_id").notNull(),month:text("month").notNull(),generated_at:text("generated_at").notNull(),source_range:text("source_range").notNull(),tier:text("tier").notNull(),total_tokens:integer("total_tokens").notNull(),total_cost_usd:real("total_cost_usd").notNull(),payload:text("payload").notNull(),payload_hash:text("payload_hash"),signature:text("signature"),signed_at:text("signed_at"),issuer:text("issuer")},a=>({uniq_usage_reports_ws_month:uniqueIndex("uniq_usage_reports_ws_month").on(a.workspace_id,a.month)}));var k=sqliteTable("notifications",{id:text("id").primaryKey(),workspace_id:text("workspace_id").notNull(),agent_id:text("agent_id"),task_id:text("task_id"),thread_id:text("thread_id"),source:text("source").notNull().default("agent"),level:text("level").notNull().default("info"),title:text("title").notNull(),body:text("body"),target_user_id:text("target_user_id"),created_at:text("created_at").notNull(),metadata:text("metadata")},a=>({idx_notifications_ws_created:index("idx_notifications_ws_created").on(a.workspace_id,sql`${a.created_at} DESC`)})),C=sqliteTable("notification_reads",{notification_id:text("notification_id").notNull(),user_id:text("user_id").notNull(),read_at:text("read_at").notNull()},a=>({pk:primaryKey({columns:[a.notification_id,a.user_id]})}));var v=sqliteTable("agent_suggestions",{id:text("id").primaryKey(),workspace_id:text("workspace_id").notNull(),agent_id:text("agent_id").notNull(),task_id:text("task_id"),type:text("type").notNull(),status:text("status").notNull().default("pending"),payload:text("payload").notNull(),applied_commit_sha:text("applied_commit_sha"),created_at:text("created_at").notNull(),updated_at:text("updated_at").notNull()},a=>({idx_agent_suggestions_ws_agent:index("idx_agent_suggestions_ws_agent").on(a.workspace_id,a.agent_id),idx_agent_suggestions_ws_created:index("idx_agent_suggestions_ws_created").on(a.workspace_id,a.created_at)}));function Ht(a){let t=Q.resolve(a);return process.platform==="win32"&&(t=t.replace(/\\/g,"/"),t=t.replace(/^([A-Z]):/,(n,e)=>`${e.toLowerCase()}:`)),t.length>1&&!/^[a-zA-Z]:\/$/.test(t)&&(t=t.replace(/\/+$/,"")),t}function pe(a){let t=Ht(a);return createHash("sha256").update(t).digest("hex")}function Ct(a){return a.toLowerCase().replace(/[^a-z0-9-]/g,"-").replace(/-{2,}/g,"-").replace(/^-+|-+$/g,"")}var Ft=class extends x{dbRoot;constructor(t={}){super(),this.dbRoot=t.dbRoot;}resolveDbPath(){return this.dbRoot?join(this.dbRoot,".crewx","crewx.db"):super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{_t(e.db);}catch(r){throw e.close(),r}return e}resolveSlug(t,n,e){let r=Ct(basename(e)),i=`${Ct(basename(dirname(e)))}-${r}`,l=[r,i];try{let u=c=>t.select({id:m.id}).from(m).where(and(eq(m.slug,c),ne$1(m.id,n))).limit(1).all().length>0;for(let c of l)if(!u(c))return c;for(let c=2;c<1e3;c+=1){let p=`${i}-${c}`;if(!u(p))return p}}catch{}return r}normalizeLegacySlugs(){if(!this.dbExists())return {updated:0,checked:0};let t=this.openHandle(true);try{let n=t.db.select({id:m.id,slug:m.slug}).from(m).all(),e=0;for(let r of n)if(r.slug.includes("/")){let s=Ct(r.slug.replace(/\//g,"-"));t.db.update(m).set({slug:s,updated_at:new Date().toISOString()}).where(eq(m.id,r.id)).run(),e+=1;}return {updated:e,checked:n.length}}catch(n){throw new d("DB_ERROR","Failed to normalize legacy slugs",n)}finally{t.close();}}ensureRow(t,n){let{id:e,slug:r,name:s,workspacePath:i}=n,l=new Date().toISOString();t.insert(m).values({id:e,slug:r,name:s,workspace_path:i,is_active:1,created_at:l,updated_at:l}).onConflictDoNothing().run(),t.update(m).set({workspace_path:i,updated_at:l}).where(and(eq(m.id,e),isNull(m.workspace_path))).run();}registerWorkspace(t){let n=Ht(t),e=this.openHandle(true);try{let r=pe(n),s=basename(n),i=this.resolveSlug(e.db,r,n);return this.ensureRow(e.db,{id:r,slug:i,name:s,workspacePath:n}),{id:r,slug:i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to register workspace",r)}finally{e.close();}}listProjects(t){if(!this.dbExists())return {rows:[],total:0};let n=this.openHandle(false);try{let e=t.isActive!==void 0?eq(m.is_active,t.isActive?1:0):void 0,r=n.db.select({count:sql`count(*)`}).from(m).where(e).get();return {rows:n.db.select().from(m).where(e).orderBy(desc(m.updated_at)).limit(t.limit).offset(t.offset).all(),total:r?.count??0}}catch(e){throw new d("DB_ERROR","Failed to list projects",e)}finally{n.close();}}findById(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(m).where(eq(m.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find workspace",e)}finally{n.close();}}findAgentsByProject(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.all(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(e){throw new d("DB_ERROR","Failed to find agents by project",e)}finally{n.close();}}findThreadsByProject(t,n){if(!this.dbExists())return {rows:[],total:0};let e=this.openHandle(false);try{let r=e.db.get(sql`SELECT COUNT(*) as count FROM threads WHERE workspace_id = ${t}`);return {rows:e.db.all(sql`SELECT t.*,
|
|
3
|
+
`)}`);let s=a.get(n`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='__drizzle_migrations'`),i=0;s?.cnt&&(i=a.get(n`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0),en(a,i),s?.cnt&&(on(a,r,n),rn(a,r,n)),t(a,{migrationsFolder:r});let u=(a.get(n`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0)-i;if(u>0){let c=s?.cnt?"Database migrated":"Database initialized";console.error(`[crewx] ${c} (${u} migration${u>1?"s":""} applied).`);}}function O(a,t){re.has(t)||(_t(a),re.add(t));}var m=sqliteTable("workspaces",{id:text("id").primaryKey(),slug:text("slug").notNull().unique(),name:text("name").notNull(),workspace_path:text("workspace_path"),description:text("description"),is_active:integer("is_active").notNull().default(1),created_at:text("created_at").notNull(),updated_at:text("updated_at").notNull()});var o=sqliteTable("tasks",{id:text("id").primaryKey(),agent_id:text("agent_id").notNull(),user_id:text("user_id"),prompt:text("prompt").notNull(),mode:text("mode").notNull().default("execute"),status:text("status").notNull().default("running"),result:text("result"),error:text("error"),started_at:text("started_at").notNull(),completed_at:text("completed_at"),duration_ms:integer("duration_ms"),metadata:text("metadata"),workspace_id:text("workspace_id"),trace_id:text("trace_id"),parent_task_id:text("parent_task_id"),caller_agent_id:text("caller_agent_id"),model:text("model"),platform:text("platform").default("cli"),crewx_version:text("crewx_version"),input_tokens:integer("input_tokens").default(0),output_tokens:integer("output_tokens").default(0),cost_usd:real("cost_usd").default(0),pid:integer("pid"),rendered_prompt:text("rendered_prompt"),command:text("command"),coding_agent_command:text("coding_agent_command"),exit_code:integer("exit_code"),logs:text("logs"),thread_id:text("thread_id"),workspace_ref:text("workspace_ref"),project_id:text("project_id"),project_ref:text("project_ref"),cached_input_tokens:integer("cached_input_tokens").default(0),run_epoch:integer("run_epoch").default(0)},a=>({idx_tasks_agent_id:index("idx_tasks_agent_id").on(a.agent_id),idx_tasks_status:index("idx_tasks_status").on(a.status),idx_tasks_started_at:index("idx_tasks_started_at").on(a.started_at),idx_tasks_trace_id:index("idx_tasks_trace_id").on(a.trace_id),idx_tasks_parent_task_id:index("idx_tasks_parent_task_id").on(a.parent_task_id),idx_tasks_crewx_version:index("idx_tasks_crewx_version").on(a.crewx_version),idx_tasks_pid:index("idx_tasks_pid").on(a.pid),idx_tasks_thread_id:index("idx_tasks_thread_id").on(a.thread_id),idx_tasks_workspace_id:index("idx_tasks_workspace_id").on(a.workspace_id),idx_tasks_workspace_ref:index("idx_tasks_workspace_ref").on(a.workspace_ref),idx_tasks_project_id:index("idx_tasks_project_id").on(a.project_id),idx_tasks_ws_started:index("idx_tasks_ws_started").on(a.workspace_id,a.started_at)}));var _=sqliteTable("threads",{id:text("id").primaryKey(),workspace_id:text("workspace_id").references(()=>m.id,{onDelete:"set null"}),platform:text("platform").notNull().default("cli"),title:text("title"),first_message:text("first_message"),last_message:text("last_message"),message_count:integer("message_count").notNull().default(0),created_at:text("created_at").notNull(),updated_at:text("updated_at").notNull(),metadata:text("metadata"),title_locked:integer("title_locked").notNull().default(0),pinned:integer("pinned").notNull().default(0),starred:integer("starred").notNull().default(0)},a=>({idx_threads_updated_at:index("idx_threads_updated_at").on(a.updated_at),idx_threads_workspace_id:index("idx_threads_workspace_id").on(a.workspace_id),idx_threads_ws_updated:index("idx_threads_ws_updated").on(a.workspace_id,a.updated_at),idx_threads_title:index("idx_threads_title").on(a.title)}));var et=sqliteTable("spans",{id:text("id").primaryKey(),task_id:text("task_id").references(()=>o.id,{onDelete:"set null"}),parent_span_id:text("parent_span_id").references(()=>et.id,{onDelete:"set null"}),name:text("name").notNull(),kind:text("kind").notNull().default("internal"),status:text("status").notNull().default("ok"),started_at:text("started_at").notNull(),completed_at:text("completed_at"),duration_ms:integer("duration_ms"),input:text("input"),output:text("output"),error:text("error"),attributes:text("attributes")},a=>({idx_spans_task_id:index("idx_spans_task_id").on(a.task_id),idx_spans_parent_span_id:index("idx_spans_parent_span_id").on(a.parent_span_id)}));var J=sqliteTable("tool_calls",{id:text("id").primaryKey(),task_id:text("task_id").references(()=>o.id,{onDelete:"cascade"}),session_id:text("session_id"),tool_name:text("tool_name").notNull(),files:text("files"),input:text("input"),output:text("output"),duration_ms:integer("duration_ms"),timestamp:text("timestamp").notNull()},a=>({idx_tool_calls_task_id:index("idx_tool_calls_task_id").on(a.task_id),idx_tool_calls_tool_name:index("idx_tool_calls_tool_name").on(a.tool_name),idx_tool_calls_timestamp:index("idx_tool_calls_timestamp").on(a.timestamp)}));var W=sqliteTable("thread_boxes",{id:text("id").primaryKey(),thread_id:text("thread_id").notNull().references(()=>_.id,{onDelete:"cascade"}),seq:integer("seq").notNull(),first_task_id:text("first_task_id").notNull(),mid_task_id:text("mid_task_id").notNull(),last_task_id:text("last_task_id").notNull(),task_count:integer("task_count").notNull(),summary:text("summary"),source_tokens:integer("source_tokens").notNull(),summary_tokens:integer("summary_tokens"),created_at:text("created_at").notNull()},a=>({idx_thread_boxes_thread_id:index("idx_thread_boxes_thread_id").on(a.thread_id),idx_thread_boxes_seq:index("idx_thread_boxes_seq").on(a.thread_id,a.seq),uniq_thread_boxes_thread_seq:unique().on(a.thread_id,a.seq)}));var ht=sqliteTable("request_logs",{id:text("id").primaryKey(),path:text("path").notNull(),method:text("method").notNull(),status_code:integer("status_code").notNull(),duration_ms:integer("duration_ms").notNull(),ip:text("ip"),request_headers:text("request_headers"),response_headers:text("response_headers"),request_body:text("request_body"),response_body:text("response_body"),query:text("query"),user_id:text("user_id"),project_id:text("project_id"),partition_key:text("partition_key").notNull(),timestamp:text("timestamp").notNull().default(sql`(datetime('now'))`),metadata:text("metadata")},a=>({idx_request_logs_timestamp:index("idx_request_logs_timestamp").on(a.timestamp),idx_request_logs_path:index("idx_request_logs_path").on(a.path),idx_request_logs_status_code:index("idx_request_logs_status_code").on(a.status_code),idx_request_logs_partition_key:index("idx_request_logs_partition_key").on(a.partition_key)}));var I=sqliteTable("usage_limit_snapshots",{id:text("id").primaryKey(),provider:text("provider").notNull(),account_ref:text("account_ref").notNull().default("default"),limit_window:text("limit_window").notNull(),bucket_start:text("bucket_start").notNull(),captured_at:text("captured_at").notNull(),used_percent:integer("used_percent").notNull(),remaining_percent:integer("remaining_percent").notNull(),resets_at:text("resets_at"),source:text("source").notNull(),metadata:text("metadata")},a=>({uniq_usage_limit_snapshots_bucket:uniqueIndex("uniq_usage_limit_snapshots_bucket").on(a.provider,a.account_ref,a.limit_window,a.bucket_start),idx_usage_limit_snapshots_window_bucket:index("idx_usage_limit_snapshots_window_bucket").on(a.limit_window,a.bucket_start),idx_usage_limit_snapshots_provider_window_bucket:index("idx_usage_limit_snapshots_provider_window_bucket").on(a.provider,a.limit_window,a.bucket_start)}));var F=sqliteTable("usage_reports",{id:text("id").primaryKey(),workspace_id:text("workspace_id").notNull(),month:text("month").notNull(),generated_at:text("generated_at").notNull(),source_range:text("source_range").notNull(),tier:text("tier").notNull(),total_tokens:integer("total_tokens").notNull(),total_cost_usd:real("total_cost_usd").notNull(),payload:text("payload").notNull(),payload_hash:text("payload_hash"),signature:text("signature"),signed_at:text("signed_at"),issuer:text("issuer")},a=>({uniq_usage_reports_ws_month:uniqueIndex("uniq_usage_reports_ws_month").on(a.workspace_id,a.month)}));var k=sqliteTable("notifications",{id:text("id").primaryKey(),workspace_id:text("workspace_id").notNull(),agent_id:text("agent_id"),task_id:text("task_id"),thread_id:text("thread_id"),source:text("source").notNull().default("agent"),level:text("level").notNull().default("info"),title:text("title").notNull(),body:text("body"),target_user_id:text("target_user_id"),created_at:text("created_at").notNull(),metadata:text("metadata")},a=>({idx_notifications_ws_created:index("idx_notifications_ws_created").on(a.workspace_id,sql`${a.created_at} DESC`)})),v=sqliteTable("notification_reads",{notification_id:text("notification_id").notNull(),user_id:text("user_id").notNull(),read_at:text("read_at").notNull()},a=>({pk:primaryKey({columns:[a.notification_id,a.user_id]})}));var y=sqliteTable("agent_suggestions",{id:text("id").primaryKey(),workspace_id:text("workspace_id").notNull(),agent_id:text("agent_id").notNull(),task_id:text("task_id"),type:text("type").notNull(),status:text("status").notNull().default("pending"),payload:text("payload").notNull(),applied_commit_sha:text("applied_commit_sha"),created_at:text("created_at").notNull(),updated_at:text("updated_at").notNull()},a=>({idx_agent_suggestions_ws_agent:index("idx_agent_suggestions_ws_agent").on(a.workspace_id,a.agent_id),idx_agent_suggestions_ws_created:index("idx_agent_suggestions_ws_created").on(a.workspace_id,a.created_at)}));function Mt(a){let t=Z.resolve(a);return process.platform==="win32"&&(t=t.replace(/\\/g,"/"),t=t.replace(/^([A-Z]):/,(n,e)=>`${e.toLowerCase()}:`)),t.length>1&&!/^[a-zA-Z]:\/$/.test(t)&&(t=t.replace(/\/+$/,"")),t}function _e(a){let t=Mt(a);return createHash("sha256").update(t).digest("hex")}function vt(a){return a.toLowerCase().replace(/[^a-z0-9-]/g,"-").replace(/-{2,}/g,"-").replace(/^-+|-+$/g,"")}var qt=class extends T{dbRoot;constructor(t={}){super(),this.dbRoot=t.dbRoot;}resolveDbPath(){return this.dbRoot?join(this.dbRoot,".crewx","crewx.db"):super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{_t(e.db);}catch(r){throw e.close(),r}return e}resolveSlug(t,n,e){let r=vt(basename(e)),i=`${vt(basename(dirname(e)))}-${r}`,l=[r,i];try{let u=c=>t.select({id:m.id}).from(m).where(and(eq(m.slug,c),ne(m.id,n))).limit(1).all().length>0;for(let c of l)if(!u(c))return c;for(let c=2;c<1e3;c+=1){let p=`${i}-${c}`;if(!u(p))return p}}catch{}return r}normalizeLegacySlugs(){if(!this.dbExists())return {updated:0,checked:0};let t=this.openHandle(true);try{let n=t.db.select({id:m.id,slug:m.slug}).from(m).all(),e=0;for(let r of n)if(r.slug.includes("/")){let s=vt(r.slug.replace(/\//g,"-"));t.db.update(m).set({slug:s,updated_at:new Date().toISOString()}).where(eq(m.id,r.id)).run(),e+=1;}return {updated:e,checked:n.length}}catch(n){throw new d("DB_ERROR","Failed to normalize legacy slugs",n)}finally{t.close();}}ensureRow(t,n){let{id:e,slug:r,name:s,workspacePath:i}=n,l=new Date().toISOString();t.insert(m).values({id:e,slug:r,name:s,workspace_path:i,is_active:1,created_at:l,updated_at:l}).onConflictDoNothing().run(),t.update(m).set({workspace_path:i,updated_at:l}).where(and(eq(m.id,e),isNull(m.workspace_path))).run();}registerWorkspace(t){let n=Mt(t),e=this.openHandle(true);try{let r=_e(n),s=basename(n),i=this.resolveSlug(e.db,r,n);return this.ensureRow(e.db,{id:r,slug:i,name:s,workspacePath:n}),{id:r,slug:i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to register workspace",r)}finally{e.close();}}listProjects(t){if(!this.dbExists())return {rows:[],total:0};let n=this.openHandle(false);try{let e=t.isActive!==void 0?eq(m.is_active,t.isActive?1:0):void 0,r=n.db.select({count:sql`count(*)`}).from(m).where(e).get();return {rows:n.db.select().from(m).where(e).orderBy(desc(m.updated_at)).limit(t.limit).offset(t.offset).all(),total:r?.count??0}}catch(e){throw new d("DB_ERROR","Failed to list projects",e)}finally{n.close();}}findById(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(m).where(eq(m.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find workspace",e)}finally{n.close();}}findAgentsByProject(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.all(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(e){throw new d("DB_ERROR","Failed to find agents by project",e)}finally{n.close();}}findThreadsByProject(t,n){if(!this.dbExists())return {rows:[],total:0};let e=this.openHandle(false);try{let r=e.db.get(sql`SELECT COUNT(*) as count FROM threads WHERE workspace_id = ${t}`);return {rows:e.db.all(sql`SELECT t.*,
|
|
4
4
|
(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
5
|
FROM threads t
|
|
6
6
|
WHERE t.workspace_id = ${t}
|
|
7
7
|
ORDER BY t.updated_at DESC
|
|
8
|
-
LIMIT ${n.limit} OFFSET ${n.offset}`),total:r?.count??0}}catch(r){throw new d("DB_ERROR","Failed to find threads by project",r)}finally{e.close();}}findBySlug(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(m).where(eq(m.slug,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find workspace by slug",e)}finally{n.close();}}slugExists(t,n){if(!this.dbExists())return false;let e=this.openHandle(false);try{let r=n?and(eq(m.slug,t),ne
|
|
8
|
+
LIMIT ${n.limit} OFFSET ${n.offset}`),total:r?.count??0}}catch(r){throw new d("DB_ERROR","Failed to find threads by project",r)}finally{e.close();}}findBySlug(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(m).where(eq(m.slug,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find workspace by slug",e)}finally{n.close();}}slugExists(t,n){if(!this.dbExists())return false;let e=this.openHandle(false);try{let r=n?and(eq(m.slug,t),ne(m.id,n)):eq(m.slug,t);return !!e.db.select({id:m.id}).from(m).where(r).limit(1).get()}catch(r){throw new d("DB_ERROR","Failed to check slug",r)}finally{e.close();}}insert(t,n,e,r){let s=this.openHandle(true);try{let i=new Date().toISOString();s.db.insert(m).values({id:t,slug:n,name:e,workspace_path:r,is_active:1,created_at:i,updated_at:i}).run();let l=s.db.select().from(m).where(eq(m.id,t)).limit(1).get();if(!l)throw new d("DB_ERROR","Insert did not return a row");return l}catch(i){throw i instanceof d?i:new d("DB_ERROR","Failed to insert workspace",i)}finally{s.close();}}update(t,n,e){let r=this.openHandle(true);try{r.runRaw(`UPDATE workspaces SET ${n.join(", ")} WHERE id = ?`,e);let s=r.db.select().from(m).where(eq(m.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 n=t.db.select({id:m.id,workspace_path:m.workspace_path}).from(m).where(and(eq(m.is_active,1),isNotNull(m.workspace_path))).all(),e=0;for(let r of n){let s=r.workspace_path;existsSync(join(s,"crewx.yaml"))||existsSync(join(s,"crewx.yml"))||(t.db.update(m).set({is_active:0,updated_at:new Date().toISOString()}).where(eq(m.id,r.id)).run(),e+=1);}return {softDeleted:e,checked:n.length}}catch(n){throw new d("DB_ERROR","Failed to cleanup orphan workspaces",n)}finally{t.close();}}delete(t){let n=this.openHandle(true);try{n.db.run(sql`UPDATE threads SET workspace_id = NULL WHERE workspace_id = ${t}`),n.db.delete(m).where(eq(m.id,t)).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to delete workspace",e)}finally{n.close();}}};var we=[m,o,_,et,J,W,ht];function Bn(){return Z__default.join(Ze.homedir(),".crewx","crewx.db")}function Un(a){if(!me.existsSync(a))return null;let t=Math.floor(Date.now()/1e3),n=`${a}.bak-${t}`;return me.copyFileSync(a,n),n}function ke(a){let t=a.all("SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%'");return new Set(t.map(n=>n.name))}function Hn(a){return a==null||typeof a=="object"&&"queryChunks"in a?null:typeof a=="number"?String(a):typeof a=="boolean"?a?"1":"0":typeof a=="string"?`'${a.replace(/'/g,"''")}'`:String(a)}function Mn(a){let t=a.getSQLType(),n=`"${a.name}" ${t}`,e=Hn(a.default);return e!==null&&(n+=` DEFAULT ${e}`),a.notNull&&(n+=" NOT NULL"),n}function Fn(a,t){let n=t?.dbPath??Bn(),e=t?.force??false,r=t?.dryRun??false,s=r?null:Un(n);if(e&&!r)try{a.run("DELETE FROM __drizzle_migrations");}catch{}let i=ke(a);if(!r)try{_t(a);}catch(h){let f=h instanceof Error?`${h.message} ${h.cause?.message??""}`:"";if(!e||!f.includes("duplicate column"))throw h}let l,u;r?(l=we.map(f=>getTableConfig(f).name).filter(f=>!i.has(f)),u=i):(u=ke(a),l=[...u].filter(h=>!i.has(h)));let c=[],p=[];for(let h of we){let f=getTableConfig(h),x=f.name;if(!u.has(x))continue;let P=a.all(`PRAGMA table_info("${x}")`),$=new Set(P.map(A=>A.name)),M=new Set(f.columns.map(A=>A.name));for(let A of f.columns)if(!$.has(A.name)){if(!r){let Y=Mn(A);a.run(`ALTER TABLE "${x}" ADD COLUMN ${Y}`);}c.push({table:x,column:A.name});}for(let A of P)M.has(A.name)||p.push(`${x}.${A.name} exists in DB but not in schema (untouched)`);}return {created:l,altered:c,warnings:p,backupPath:s}}function be(a){let t=a.toLowerCase().trim();t=t.replace(/\[[^\]]*\]$/g,"");let n=t.lastIndexOf("/");return n>=0&&(t=t.slice(n+1)),t=t.replace(/-\d{8}$/,""),t}var it="2026-05-09",Yn="0.8.9-rc.13",at=10,dt=parseInt(Yn.split("rc.")[1]),Xn=new Set(["workflow","mcp"]),Ee=600*1e3,Wt=class extends T{dbPath;pidNullObservedSince=new Map;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{O(e.db,n);}catch(r){throw e.close(),r}return e}startTask(t){let n=this.openHandle(true);try{n.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:sql`COALESCE(excluded.pid, ${o.pid})`},setWhere:eq(o.status,"running")}).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to start task",e)}finally{n.close();}}finishTask(t){let n=this.openHandle(true);try{let e=t.runEpoch??null;n.runRaw(`UPDATE tasks SET status=?, result=?, error=?, completed_at=?, duration_ms=?,
|
|
9
9
|
exit_code=?, input_tokens=?, output_tokens=?, cached_input_tokens=?, cost_usd=?,
|
|
10
10
|
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,e]);}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to finish task",e)}finally{n.close();}}appendLog(t,n){let e=this.openHandle(true);try{e.db.transaction(r=>{let s=r.select({logs:o.logs}).from(o).where(eq(o.id,t)).limit(1).get(),i=s?.logs?JSON.parse(s.logs):[];i.push(n),r.update(o).set({logs:JSON.stringify(i)}).where(eq(o.id,t)).run();},{behavior:"immediate"});}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to append log",r)}finally{e.close();}}getRunningTasks(){if(!this.dbExists())return [];let t=this.openHandle(false);try{return t.db.select().from(o).where(eq(o.status,"running")).orderBy(desc(o.started_at)).all()}catch(n){throw new d("DB_ERROR","Failed to get running tasks",n)}finally{t.close();}}getAllTasks(){if(!this.dbExists())return [];let t=this.openHandle(false);try{return t.db.select().from(o).orderBy(desc(o.started_at)).limit(100).all()}catch(n){throw new d("DB_ERROR","Failed to get all tasks",n)}finally{t.close();}}getTask(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(o).where(eq(o.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to get task",e)}finally{n.close();}}killTask(t){if(!this.dbExists())return {killed:false};let n=this.openHandle(true);try{let e=n.db.select({id:o.id,status:o.status,pid:o.pid}).from(o).where(eq(o.id,t)).limit(1).get();if(!e||e.status!=="running")return {killed:!1};if(e.pid)try{process.kill(e.pid,"SIGTERM");}catch{}return n.db.update(o).set({status:"failed",error:"Killed by user",completed_at:new Date().toISOString()}).where(and(eq(o.id,t),eq(o.status,"running"))).run(),{killed:!0,pid:e.pid??void 0}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to kill task",e)}finally{n.close();}}reapOrphanedTasks(){if(!this.dbExists())return 0;let t=this.openHandle(true);try{let n=t.db.select({id:o.id,pid:o.pid,platform:o.platform,run_epoch:o.run_epoch}).from(o).where(eq(o.status,"running")).all(),e=Date.now(),r=new Set,s=0;for(let i of n){if(i.pid){let c=!1;try{process.kill(i.pid,0),c=!0;}catch{}c||(t.db.update(o).set({status:"failed",error:"Reaped: process not found (orphaned task)",completed_at:new Date().toISOString()}).where(and(eq(o.id,i.id),eq(o.status,"running"))).run(),s++);continue}if(
|
|
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,e]);}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to finish task",e)}finally{n.close();}}appendLog(t,n){let e=this.openHandle(true);try{e.db.transaction(r=>{let s=r.select({logs:o.logs}).from(o).where(eq(o.id,t)).limit(1).get(),i=s?.logs?JSON.parse(s.logs):[];i.push(n),r.update(o).set({logs:JSON.stringify(i)}).where(eq(o.id,t)).run();},{behavior:"immediate"});}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to append log",r)}finally{e.close();}}getRunningTasks(){if(!this.dbExists())return [];let t=this.openHandle(false);try{return t.db.select().from(o).where(eq(o.status,"running")).orderBy(desc(o.started_at)).all()}catch(n){throw new d("DB_ERROR","Failed to get running tasks",n)}finally{t.close();}}getAllTasks(){if(!this.dbExists())return [];let t=this.openHandle(false);try{return t.db.select().from(o).orderBy(desc(o.started_at)).limit(100).all()}catch(n){throw new d("DB_ERROR","Failed to get all tasks",n)}finally{t.close();}}getTask(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(o).where(eq(o.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to get task",e)}finally{n.close();}}killTask(t){if(!this.dbExists())return {killed:false};let n=this.openHandle(true);try{let e=n.db.select({id:o.id,status:o.status,pid:o.pid}).from(o).where(eq(o.id,t)).limit(1).get();if(!e||e.status!=="running")return {killed:!1};if(e.pid)try{process.kill(e.pid,"SIGTERM");}catch{}return n.db.update(o).set({status:"failed",error:"Killed by user",completed_at:new Date().toISOString()}).where(and(eq(o.id,t),eq(o.status,"running"))).run(),{killed:!0,pid:e.pid??void 0}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to kill task",e)}finally{n.close();}}reapOrphanedTasks(){if(!this.dbExists())return 0;let t=this.openHandle(true);try{let n=t.db.select({id:o.id,pid:o.pid,platform:o.platform,run_epoch:o.run_epoch}).from(o).where(eq(o.status,"running")).all(),e=Date.now(),r=new Set,s=0;for(let i of n){if(i.pid){let c=!1;try{process.kill(i.pid,0),c=!0;}catch{}c||(t.db.update(o).set({status:"failed",error:"Reaped: process not found (orphaned task)",completed_at:new Date().toISOString()}).where(and(eq(o.id,i.id),eq(o.status,"running"))).run(),s++);continue}if(Xn.has(i.platform??"cli"))continue;let l=`${i.id}:${i.run_epoch??0}`;r.add(l);let u=this.pidNullObservedSince.get(l);if(u===void 0){this.pidNullObservedSince.set(l,e);continue}e-u>=Ee&&(t.db.update(o).set({status:"failed",error:"Reaped: pid never recorded after resume (stale running_instruction)",completed_at:new Date().toISOString()}).where(and(eq(o.id,i.id),eq(o.status,"running"))).run(),s++,this.pidNullObservedSince.delete(l));}for(let i of Array.from(this.pidNullObservedSince.keys()))r.has(i)||this.pidNullObservedSince.delete(i);return s}finally{t.close();}}reapRunningWorkflowTasks(t=Ee){if(!this.dbExists())return 0;let n=this.openHandle(true);try{let e=new Date(Date.now()-t).toISOString();return n.db.update(o).set({status:"failed",error:"Reaped: stale workflow run (finally not executed)",completed_at:new Date().toISOString()}).where(and(eq(o.status,"running"),eq(o.platform,"workflow"),isNull(o.pid),lt(o.started_at,e))).run().changes??0}finally{n.close();}}findTaskStatus(t,n){let e=this.resolveDbPaths();for(let r of e){if(!existsSync(r))continue;let s=w(r);try{let i=n?eq(o.workspace_id,n):void 0,l=i?and(eq(o.id,t),i):eq(o.id,t),u=s.db.select().from(o).where(l).limit(1).get()??void 0;if(!u){let c=or(eq(o.thread_id,t),and(isNull(o.thread_id),like(o.command,`%--thread=${t}%`))),p=i?and(c,i):c;u=s.db.select().from(o).where(p).orderBy(desc(o.started_at)).limit(1).get()??void 0;}if(u)return u}catch(i){throw new d("DB_ERROR","Failed to find task status",i)}finally{s.close();}}}findChildTasks(t,n){let e=this.resolveDbPaths(),r=new Set,s=[];for(let i of e){if(!existsSync(i))continue;let l=w(i);try{let u=n?and(eq(o.parent_task_id,t),eq(o.workspace_id,n)):eq(o.parent_task_id,t),c=l.db.select().from(o).where(u).orderBy(asc(o.started_at)).all();for(let p of c)r.has(p.id)||(r.add(p.id),s.push(p));}catch(u){throw new d("DB_ERROR","Failed to find child tasks",u)}finally{l.close();}}return s}getWorkspaceUsageSummary(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.all(t?sql`
|
|
12
12
|
SELECT
|
|
13
13
|
COALESCE(workspace_id, 'unknown') AS workspace_id,
|
|
14
14
|
COALESCE(SUM(input_tokens), 0) AS input_tokens,
|
|
@@ -29,7 +29,7 @@ ${e.join(`
|
|
|
29
29
|
FROM tasks
|
|
30
30
|
GROUP BY workspace_id
|
|
31
31
|
ORDER BY (COALESCE(SUM(input_tokens), 0) + COALESCE(SUM(output_tokens), 0)) DESC
|
|
32
|
-
`)}catch(e){throw new d("DB_ERROR","Failed to get workspace usage summary",e)}finally{n.close();}}getThreadTokenUsage(t,n){let e=this.resolveDbPaths(),r=new Set,s=0,i=0,l=0;for(let u of e){if(!existsSync(u))continue;let c=w(u);try{let p=or(eq(o.thread_id,t),and(isNull(o.thread_id),like(o.command,`%--thread=${t}%`))),h=n?and(p,eq(o.workspace_id,n)):p,f=c.db.select({id:o.id,input_tokens:o.input_tokens,output_tokens:o.output_tokens,cost_usd:o.cost_usd}).from(o).where(h).all();for(let
|
|
32
|
+
`)}catch(e){throw new d("DB_ERROR","Failed to get workspace usage summary",e)}finally{n.close();}}getThreadTokenUsage(t,n){let e=this.resolveDbPaths(),r=new Set,s=0,i=0,l=0;for(let u of e){if(!existsSync(u))continue;let c=w(u);try{let p=or(eq(o.thread_id,t),and(isNull(o.thread_id),like(o.command,`%--thread=${t}%`))),h=n?and(p,eq(o.workspace_id,n)):p,f=c.db.select({id:o.id,input_tokens:o.input_tokens,output_tokens:o.output_tokens,cost_usd:o.cost_usd}).from(o).where(h).all();for(let x of f)r.has(x.id)||(r.add(x.id),s+=x.input_tokens??0,i+=x.output_tokens??0,l+=x.cost_usd??0);}catch(p){throw new d("DB_ERROR","Failed to get thread token usage",p)}finally{c.close();}}return {inputTokens:s,outputTokens:i,costUsd:l}}findTasksByThread(t,n){let e=this.resolveDbPaths(),r=new Set,s=[];for(let i of e){if(!existsSync(i))continue;let l=w(i);try{let u=or(eq(o.thread_id,t),and(isNull(o.thread_id),like(o.command,`%--thread=${t}%`))),c=n?and(u,eq(o.workspace_id,n)):u,p=l.db.select().from(o).where(c).orderBy(asc(o.started_at)).all();for(let h of p)r.has(h.id)||(r.add(h.id),s.push(h));}catch(u){throw new d("DB_ERROR","Failed to find tasks by thread",u)}finally{l.close();}}return s}parseWorkflowCardMetadata(t){if(!t)return {};try{let n=JSON.parse(t);return {kind:typeof n.kind=="string"?n.kind:void 0,workflowRunId:typeof n.workflowRunId=="string"?n.workflowRunId:void 0}}catch{return {}}}extractThreadIdFromCommand(t){return t?/--thread=(\S+)/.exec(t)?.[1]??null:null}batchFetchWorkflowCards(t,n){let e=new Map;if(t.length===0)return e;let r=new Set(t),s=new Map,i=n?eq(o.workspace_id,n):void 0,l=(u,c)=>{if(!u)return;let p=s.get(u);if(p&&c.started_at<=p)return;let h=this.parseWorkflowCardMetadata(c.metadata);h.kind==="workflow_card"&&(s.set(u,c.started_at),e.set(u,{taskId:c.id,workflowRunId:h.workflowRunId,raw:c.result??void 0}));};for(let u of this.resolveDbPaths()){if(!existsSync(u))continue;let c=w(u);try{let p=and(inArray(o.thread_id,t),like(o.metadata,'%"kind":"workflow_card"%')),h=c.db.select({id:o.id,thread_id:o.thread_id,metadata:o.metadata,result:o.result,started_at:o.started_at}).from(o).where(i?and(p,i):p).all();for(let P of h)l(P.thread_id,P);let f=and(isNull(o.thread_id),like(o.metadata,'%"kind":"workflow_card"%')),x=c.db.select({id:o.id,command:o.command,metadata:o.metadata,result:o.result,started_at:o.started_at}).from(o).where(i?and(f,i):f).all();for(let P of x){let $=this.extractThreadIdFromCommand(P.command);!$||!r.has($)||l($,P);}}catch(p){throw new d("DB_ERROR","Failed to batch fetch workflow cards",p)}finally{c.close();}}return e}findAllTasks(t){if(!this.dbExists())return {rows:[],total:0};let n=this.openHandle(false);try{let e=[];t.workspaceId&&e.push(eq(o.workspace_id,t.workspaceId));let r=t.agents&&t.agents.length>0?t.agents:t.agentId?[t.agentId]:null;r&&e.push(inArray(o.agent_id,r));let s=t.statuses&&t.statuses.length>0?t.statuses:t.status?[t.status]:null;s&&e.push(inArray(o.status,s));let i=t.q??t.search;i&&e.push(like(o.prompt,`%${i}%`)),t.from&&e.push(gte(o.started_at,t.from)),t.to&&e.push(lt(o.started_at,t.to));let l=e.length>0?and(...e):void 0,u=n.db.select({count:sql`count(*)`}).from(o).where(l).get(),c=(t.sortDir??"DESC")==="ASC"?asc(o.started_at):desc(o.started_at);return {rows:n.db.select().from(o).where(l).orderBy(c).limit(t.limit).offset(t.offset).all(),total:u?.count??0}}catch(e){throw new d("DB_ERROR","Failed to find all tasks",e)}finally{n.close();}}getAgentUsage(t,n,e){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.all(e?sql`
|
|
33
33
|
SELECT
|
|
34
34
|
t.agent_id,
|
|
35
35
|
t.workspace_id,
|
|
@@ -37,11 +37,11 @@ ${e.join(`
|
|
|
37
37
|
COALESCE(SUM(
|
|
38
38
|
COALESCE(t.input_tokens, 0)
|
|
39
39
|
+ CASE
|
|
40
|
-
WHEN t.started_at >= ${
|
|
40
|
+
WHEN t.started_at >= ${it}
|
|
41
41
|
AND (
|
|
42
42
|
t.crewx_version IS NULL
|
|
43
43
|
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, ${
|
|
44
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${at}) AS INTEGER) < ${dt})
|
|
45
45
|
)
|
|
46
46
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
47
47
|
ELSE 0
|
|
@@ -60,11 +60,11 @@ ${e.join(`
|
|
|
60
60
|
COALESCE(SUM(
|
|
61
61
|
COALESCE(t.input_tokens, 0)
|
|
62
62
|
+ CASE
|
|
63
|
-
WHEN t.started_at >= ${
|
|
63
|
+
WHEN t.started_at >= ${it}
|
|
64
64
|
AND (
|
|
65
65
|
t.crewx_version IS NULL
|
|
66
66
|
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, ${
|
|
67
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${at}) AS INTEGER) < ${dt})
|
|
68
68
|
)
|
|
69
69
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
70
70
|
ELSE 0
|
|
@@ -80,11 +80,11 @@ ${e.join(`
|
|
|
80
80
|
COALESCE(SUM(
|
|
81
81
|
COALESCE(t.input_tokens, 0)
|
|
82
82
|
+ CASE
|
|
83
|
-
WHEN t.started_at >= ${
|
|
83
|
+
WHEN t.started_at >= ${it}
|
|
84
84
|
AND (
|
|
85
85
|
t.crewx_version IS NULL
|
|
86
86
|
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, ${
|
|
87
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${at}) AS INTEGER) < ${dt})
|
|
88
88
|
)
|
|
89
89
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
90
90
|
ELSE 0
|
|
@@ -102,11 +102,11 @@ ${e.join(`
|
|
|
102
102
|
COALESCE(SUM(
|
|
103
103
|
COALESCE(t.input_tokens, 0)
|
|
104
104
|
+ CASE
|
|
105
|
-
WHEN t.started_at >= ${
|
|
105
|
+
WHEN t.started_at >= ${it}
|
|
106
106
|
AND (
|
|
107
107
|
t.crewx_version IS NULL
|
|
108
108
|
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, ${
|
|
109
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${at}) AS INTEGER) < ${dt})
|
|
110
110
|
)
|
|
111
111
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
112
112
|
ELSE 0
|
|
@@ -121,11 +121,11 @@ ${e.join(`
|
|
|
121
121
|
COALESCE(SUM(
|
|
122
122
|
COALESCE(t.input_tokens, 0)
|
|
123
123
|
+ CASE
|
|
124
|
-
WHEN t.started_at >= ${
|
|
124
|
+
WHEN t.started_at >= ${it}
|
|
125
125
|
AND (
|
|
126
126
|
t.crewx_version IS NULL
|
|
127
127
|
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, ${
|
|
128
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${at}) AS INTEGER) < ${dt})
|
|
129
129
|
)
|
|
130
130
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
131
131
|
ELSE 0
|
|
@@ -148,11 +148,11 @@ ${e.join(`
|
|
|
148
148
|
COALESCE(SUM(
|
|
149
149
|
COALESCE(t.input_tokens, 0)
|
|
150
150
|
+ CASE
|
|
151
|
-
WHEN t.started_at >= ${
|
|
151
|
+
WHEN t.started_at >= ${it}
|
|
152
152
|
AND (
|
|
153
153
|
t.crewx_version IS NULL
|
|
154
154
|
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, ${
|
|
155
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${at}) AS INTEGER) < ${dt})
|
|
156
156
|
)
|
|
157
157
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
158
158
|
ELSE 0
|
|
@@ -180,11 +180,11 @@ ${e.join(`
|
|
|
180
180
|
`,i=sql`
|
|
181
181
|
COALESCE(${o.input_tokens}, 0)
|
|
182
182
|
+ CASE
|
|
183
|
-
WHEN ${o.started_at} >= ${
|
|
183
|
+
WHEN ${o.started_at} >= ${it}
|
|
184
184
|
AND (
|
|
185
185
|
${o.crewx_version} IS NULL
|
|
186
186
|
OR (${o.crewx_version} LIKE '0.8.%' AND ${o.crewx_version} NOT LIKE '0.8.9%')
|
|
187
|
-
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${
|
|
187
|
+
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${at}) AS INTEGER) < ${dt})
|
|
188
188
|
)
|
|
189
189
|
THEN COALESCE(${o.cached_input_tokens}, 0)
|
|
190
190
|
ELSE 0
|
|
@@ -216,11 +216,11 @@ ${e.join(`
|
|
|
216
216
|
`,i=sql`
|
|
217
217
|
COALESCE(${o.input_tokens}, 0)
|
|
218
218
|
+ CASE
|
|
219
|
-
WHEN ${o.started_at} >= ${
|
|
219
|
+
WHEN ${o.started_at} >= ${it}
|
|
220
220
|
AND (
|
|
221
221
|
${o.crewx_version} IS NULL
|
|
222
222
|
OR (${o.crewx_version} LIKE '0.8.%' AND ${o.crewx_version} NOT LIKE '0.8.9%')
|
|
223
|
-
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${
|
|
223
|
+
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${at}) AS INTEGER) < ${dt})
|
|
224
224
|
)
|
|
225
225
|
THEN COALESCE(${o.cached_input_tokens}, 0)
|
|
226
226
|
ELSE 0
|
|
@@ -239,7 +239,7 @@ ${e.join(`
|
|
|
239
239
|
FROM ${o}
|
|
240
240
|
${l}
|
|
241
241
|
GROUP BY ${o.model}
|
|
242
|
-
`),c=new Map;for(let p of u){let h=
|
|
242
|
+
`),c=new Map;for(let p of u){let h=be(p.model??""),f=c.get(h);f?(f.totalTasks+=p.total_tasks,f.inputTokens+=p.input_tokens,f.outputTokens+=p.output_tokens,f.cachedInputTokens+=p.cached_input_tokens,f.costUsd+=p.cost_usd,f.totalTokens+=p.input_tokens+p.output_tokens,f.activeDurationMs+=p.active_duration_ms??0,p.last_active_at&&(!f.lastActiveAt||p.last_active_at>f.lastActiveAt)&&(f.lastActiveAt=p.last_active_at)):c.set(h,{model:h,provider:p.provider,totalTasks:p.total_tasks,inputTokens:p.input_tokens,outputTokens:p.output_tokens,cachedInputTokens:p.cached_input_tokens,costUsd:p.cost_usd,totalTokens:p.input_tokens+p.output_tokens,activeDurationMs:p.active_duration_ms??0,lastActiveAt:p.last_active_at??null});}return Array.from(c.values()).sort((p,h)=>h.costUsd-p.costUsd)}catch(s){throw new d("DB_ERROR","Failed to get model usage",s)}finally{r.close();}}getTasksForReport(t,n,e){if(!this.dbExists())return [];let r=this.openHandle(false);try{let s=e?sql`WHERE ${o.started_at} >= ${t} AND ${o.started_at} < ${n} AND ${o.workspace_id} = ${e}`:sql`WHERE ${o.started_at} >= ${t} AND ${o.started_at} < ${n}`;return r.db.all(sql`
|
|
243
243
|
SELECT
|
|
244
244
|
${o.thread_id} AS thread_id,
|
|
245
245
|
${o.agent_id} AS agent_id,
|
|
@@ -250,7 +250,7 @@ ${e.join(`
|
|
|
250
250
|
FROM ${o}
|
|
251
251
|
${s}
|
|
252
252
|
ORDER BY ${o.started_at} ASC
|
|
253
|
-
`).map(l=>({threadId:l.thread_id,agentId:l.agent_id,startedAt:l.started_at,status:l.status,error:l.error,exitCode:l.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 n=this.openHandle(false);try{let e=[];t.workspaceId&&e.push(eq(o.workspace_id,t.workspaceId)),t.agents&&t.agents.length>0&&e.push(inArray(o.agent_id,t.agents)),t.statuses&&t.statuses.length>0&&e.push(inArray(o.status,t.statuses)),t.from&&e.push(gte(o.started_at,t.from)),t.to&&e.push(lt(o.started_at,t.to));let r=e.length>0?and(...e):void 0,s=(t.order??"DESC")==="ASC"?asc(o.started_at):desc(o.started_at),i={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?n.db.select(i).from(o).where(r).orderBy(s).limit(t.limit).all():n.db.select(i).from(o).where(r).orderBy(s).all()}catch(e){throw new d("DB_ERROR","Failed to find task evidence",e)}finally{n.close();}}findThreadChains(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{let e=t.maxThreads??8,r=t.maxTurnsPerThread??4,s=[isNotNull(o.thread_id),isNull(o.parent_task_id),isNull(o.caller_agent_id),or(isNull(o.platform),ne
|
|
253
|
+
`).map(l=>({threadId:l.thread_id,agentId:l.agent_id,startedAt:l.started_at,status:l.status,error:l.error,exitCode:l.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 n=this.openHandle(false);try{let e=[];t.workspaceId&&e.push(eq(o.workspace_id,t.workspaceId)),t.agents&&t.agents.length>0&&e.push(inArray(o.agent_id,t.agents)),t.statuses&&t.statuses.length>0&&e.push(inArray(o.status,t.statuses)),t.from&&e.push(gte(o.started_at,t.from)),t.to&&e.push(lt(o.started_at,t.to));let r=e.length>0?and(...e):void 0,s=(t.order??"DESC")==="ASC"?asc(o.started_at):desc(o.started_at),i={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?n.db.select(i).from(o).where(r).orderBy(s).limit(t.limit).all():n.db.select(i).from(o).where(r).orderBy(s).all()}catch(e){throw new d("DB_ERROR","Failed to find task evidence",e)}finally{n.close();}}findThreadChains(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{let e=t.maxThreads??8,r=t.maxTurnsPerThread??4,s=[isNotNull(o.thread_id),isNull(o.parent_task_id),isNull(o.caller_agent_id),or(isNull(o.platform),ne(o.platform,"workflow")),or(isNull(o.metadata),notLike(o.metadata,'%"kind":"workflow_card"%')),or(isNull(o.metadata),notLike(o.metadata,'%"retriedFromTaskId"%'))];t.workspaceId&&s.push(eq(o.workspace_id,t.workspaceId)),t.from&&s.push(gte(o.started_at,t.from)),t.to&&s.push(lt(o.started_at,t.to));let i=n.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(and(...s)).orderBy(asc(o.started_at),asc(o.id)).all(),l=new Map;for(let c of i){if(!c.thread_id)continue;let p={id:c.id,agentId:c.agent_id,status:c.status,prompt:c.prompt,startedAt:c.started_at,completedAt:c.completed_at},h=l.get(c.thread_id);h?h.push(p):l.set(c.thread_id,[p]);}let u=[];for(let[c,p]of l){let h=p.filter(f=>f.agentId===t.targetAgentId).length;h<2||u.push({threadId:c,targetAgentTurnCount:h,turns:p});}return u.sort((c,p)=>{let h=c.turns[c.turns.length-1].startedAt,f=p.turns[p.turns.length-1].startedAt;return h<f?1:h>f?-1:0}),u.slice(0,e).map(c=>({threadId:c.threadId,targetAgentTurnCount:c.targetAgentTurnCount,turns:c.turns.slice(-r)}))}catch(e){throw new d("DB_ERROR","Failed to find thread chains",e)}finally{n.close();}}};var Kt=class extends T{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{O(e.db,n);}catch(r){throw e.close(),r}return e}validateWorkspaceId(t,n){return t.db.select({id:m.id}).from(m).where(eq(m.id,n)).limit(1).get()?n:null}topLevelTaskPredicateSql(t="child"){return sql.raw(`NOT EXISTS (
|
|
254
254
|
SELECT 1 FROM tasks parent
|
|
255
255
|
WHERE parent.id = ${t}.parent_task_id
|
|
256
256
|
AND parent.thread_id = ${t}.thread_id
|
|
@@ -265,13 +265,13 @@ ${e.join(`
|
|
|
265
265
|
WHERE child.thread_id = ${t}
|
|
266
266
|
AND ${this.topLevelTaskPredicateSql()}
|
|
267
267
|
${n?sql`AND child.workspace_id = ${n}`:sql``}
|
|
268
|
-
`);f&&(r+=f.cnt,s+=f.total_input,i+=f.total_output,l+=f.total_cached,u+=f.total_cost);let
|
|
268
|
+
`);f&&(r+=f.cnt,s+=f.total_input,i+=f.total_output,l+=f.total_cached,u+=f.total_cost);let x=h.db.all(sql`
|
|
269
269
|
SELECT DISTINCT child.agent_id FROM tasks child
|
|
270
270
|
WHERE child.thread_id = ${t}
|
|
271
271
|
AND child.agent_id IS NOT NULL AND child.agent_id != ''
|
|
272
272
|
AND ${this.topLevelTaskPredicateSql()}
|
|
273
273
|
${n?sql`AND child.workspace_id = ${n}`:sql``}
|
|
274
|
-
`);for(let P of
|
|
274
|
+
`);for(let P of x)c.add(P.agent_id);}catch(f){throw new d("DB_ERROR","Failed to aggregate task stats",f)}finally{h.close();}}return {taskCount:r,inputTokens:s,outputTokens:i,cachedInputTokens:l,costUsd:u,agentIds:Array.from(c)}}findTopLevelTasks(t,n,e){let r=this.resolveDbPaths(),s=new Set,i=[];for(let l of r){if(!existsSync(l))continue;let u=w(l);try{let c;e!==void 0?c=u.db.all(sql`
|
|
275
275
|
SELECT * FROM (
|
|
276
276
|
SELECT child.*,
|
|
277
277
|
ROW_NUMBER() OVER (PARTITION BY child.agent_id ORDER BY child.started_at DESC) AS rn
|
|
@@ -306,4 +306,4 @@ ${e.join(`
|
|
|
306
306
|
message_count = message_count + 1,
|
|
307
307
|
updated_at = ${s}
|
|
308
308
|
WHERE id = ${t}
|
|
309
|
-
`),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,n,e){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=new Date().toISOString();r.db.update(_).set({last_message:n,updated_at:s}).where(eq(_.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,n){if(!this.dbExists())return;let e=this.openHandle(true);try{let r={updated_at:new Date().toISOString()};n.title!==void 0&&(r.title=n.title,r.title_locked=1),n.titleLocked!==void 0&&(r.title_locked=n.titleLocked?1:0),e.db.update(_).set(r).where(eq(_.id,t)).run();}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to update thread",r)}finally{e.close();}}togglePin(t,n){let e=this.openHandle(true);try{let r=n?and(eq(_.id,t),eq(_.workspace_id,n)):eq(_.id,t),s=e.db.select({pinned:_.pinned,metadata:_.metadata}).from(_).where(r).get();if(!s)return null;let i=s.pinned?0:1,l=s.metadata?JSON.parse(s.metadata):{};if(i){let u=n?and(eq(_.pinned,1),eq(_.workspace_id,n)):eq(_.pinned,1),c=e.db.select({metadata:_.metadata}).from(_).where(u).all(),p=null;for(let h of c){let f=h.metadata?JSON.parse(h.metadata):{};typeof f.pinOrder=="number"&&(p===null||f.pinOrder<p)&&(p=f.pinOrder);}l.pinOrder=p===null?0:p-1;}else delete l.pinOrder;return e.db.update(_).set({pinned:i,metadata:Object.keys(l).length>0?JSON.stringify(l):null}).where(r).run(),{pinned:!!i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle pin",r)}finally{e.close();}}reorderPins(t,n){let e=this.openHandle(true);try{for(let r=0;r<t.length;r++){let s=n?and(eq(_.id,t[r]),eq(_.workspace_id,n)):eq(_.id,t[r]),i=e.db.select({metadata:_.metadata}).from(_).where(s).get();if(!i)continue;let l=i.metadata?JSON.parse(i.metadata):{};l.pinOrder=r+1,e.db.update(_).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{e.close();}}toggleStar(t,n){let e=this.openHandle(true);try{let r=n?and(eq(_.id,t),eq(_.workspace_id,n)):eq(_.id,t),s=e.db.select({starred:_.starred}).from(_).where(r).get();if(!s)return null;let i=s.starred?0:1;return e.db.update(_).set({starred:i}).where(r).run(),{starred:!!i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle star",r)}finally{e.close();}}resolveOverdriveForRequest(t,n,e,r){let i=this.openHandle(true);try{let l=n?and(eq(_.id,t),eq(_.workspace_id,n)):eq(_.id,t);return i.db.transaction(c=>{let p=c.select({metadata:_.metadata}).from(_).where(l).get();if(!p)return null;let h=p.metadata?JSON.parse(p.metadata):{},f=h.overdrive??{},S=r?.defaultTurns??f.defaultTurns??3,P=r?.updatedBy??"ui",$=new Date().toISOString(),H,N,G,wt;switch(e){case "enable-count":{let kt=r?.turns??S,bt=r?.consumeCurrent?Math.max(kt-1,0):kt;H=bt>0?"count":"off",N=bt,G=!0,wt="count";break}case "enable-latch":{H="latch",N=0,G=!0,wt="latch";break}case "disable":{H="off",N=0,G=!1;break}default:{let kt=f.state??"off",bt=typeof f.remaining=="number"?f.remaining:0;if(kt==="count"&&bt>0){let ee=bt-1;ee<=0?(H="off",N=0):(H="count",N=ee),G=!0,wt="count";}else kt==="latch"?(H="latch",N=0,G=!0,wt="latch"):(H="off",N=0,G=!1);break}}return h.overdrive={state:H,remaining:N,defaultTurns:S,updatedAt:$,updatedBy:P},c.update(_).set({metadata:JSON.stringify(h),updated_at:$}).where(l).run(),{applied:G,appliedMode:wt,state:H,remaining:N}},{behavior:"immediate"})}catch(l){throw l instanceof d?l:new d("DB_ERROR","Failed to resolve overdrive",l)}finally{i.close();}}};var Kt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}insertSpan(t){let n=this.openHandle(true);try{n.db.insert(tt).values(t).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert span",e)}finally{n.close();}}findByTaskId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(tt).where(eq(tt.task_id,t)).all()}catch(e){throw new d("DB_ERROR","Failed to find spans by task id",e)}finally{n.close();}}findById(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(tt).where(eq(tt.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find span by id",e)}finally{n.close();}}};var Gt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}insertToolCall(t){let n=this.openHandle(true);try{n.db.insert(X).values(t).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert tool call",e)}finally{n.close();}}findByTaskId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(X).where(eq(X.task_id,t)).all()}catch(e){throw new d("DB_ERROR","Failed to find tool calls by task id",e)}finally{n.close();}}aggregateByName(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select({toolName:X.tool_name,count:sql`count(*)`}).from(X).where(eq(X.task_id,t)).groupBy(X.tool_name).all().map(e=>({toolName:e.toolName,count:e.count}))}catch(e){throw new d("DB_ERROR","Failed to aggregate tool calls by name",e)}finally{n.close();}}};var Yt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}ensureThreadExists(t,n){if(!t.db.select({id:_.id}).from(_).where(eq(_.id,n)).limit(1).get())throw new d("NOT_FOUND",`Thread not found: ${n}`)}findByThreadId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return this.ensureThreadExists(n,t),n.db.select().from(j).where(eq(j.thread_id,t)).orderBy(asc(j.seq)).all()}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to find boxes by thread id",e)}finally{n.close();}}findById(t,n){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(j).where(and(eq(j.id,n),eq(j.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{e.close();}}insert(t){let n=this.openHandle(true);try{this.ensureThreadExists(n,t.threadId);try{n.db.insert(j).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 e=n.db.select().from(j).where(eq(j.id,t.id)).limit(1).get();if(!e)throw new d("DB_ERROR","Insert did not return a row");return e}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert thread box",e)}finally{n.close();}}};var Xt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}bulkInsert(t){if(t.length===0)return;let n=this.openHandle(true);try{let e=new Date().toISOString();n.db.transaction(r=>{for(let s of t)r.insert(ht).values({id: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:e,metadata:s.metadata??null}).run();});}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to bulk insert request logs",e)}finally{n.close();}}findRecent(t=100){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(ht).orderBy(desc(ht.timestamp)).limit(t).all()}catch(e){throw new d("DB_ERROR","Failed to find recent request logs",e)}finally{n.close();}}};function ve(a){return new Date(Math.floor(a/3e5)*3e5).toISOString()}var Jt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}upsert(t){let n=this.openHandle(true);try{let e=ve(t.bucketStartMs),r=t.accountRef??"default";n.db.insert(I).values({id:t.id,provider:t.provider,account_ref:r,limit_window:t.limitWindow,bucket_start:e,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(e){throw e instanceof d?e:new d("DB_ERROR","Failed to upsert usage limit snapshot",e)}finally{n.close();}}getTrends(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{let e=n.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(and(eq(I.limit_window,t.window),gte(I.bucket_start,t.fromIso),lte(I.bucket_start,t.toIso))).orderBy(asc(I.bucket_start)).all(),r=new Map;for(let s of e)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,i])=>({provider:s,points:i}))}catch(e){throw new d("DB_ERROR","Failed to get usage limit trends",e)}finally{n.close();}}};var Vt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}publish(t){let n=this.openHandle(true);try{n.db.insert(M).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:[M.workspace_id,M.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 e=n.db.select().from(M).where(and(eq(M.workspace_id,t.workspaceId),eq(M.month,t.month))).limit(1).get();if(!e)throw new d("DB_ERROR","Failed to read back published usage report");return e}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to publish usage report",e)}finally{n.close();}}findByMonth(t,n){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(M).where(and(eq(M.workspace_id,t),eq(M.month,n))).limit(1).get()??void 0}catch(r){throw new d("DB_ERROR","Failed to find usage report",r)}finally{e.close();}}listByWorkspace(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(M).where(eq(M.workspace_id,t)).orderBy(desc(M.month)).all()}catch(e){throw new d("DB_ERROR","Failed to list usage reports",e)}finally{n.close();}}};var $e="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";function Nr(a){let t=randomBytes(a*2),n="";for(let e=0;e<t.length&&n.length<a;e++){let r=t[e];r<248&&(n+=$e[r%62]);}for(;n.length<a;)n+=$e[randomBytes(1)[0]%62];return n}function Lt(a){return `${a}_${Nr(8)}`}var ze=500,Fe=50,Qt=class extends x{dbPath;cap;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db")),this.cap=t.cap??ze;}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}create(t){let n=this.openHandle(true);try{let e=Lt("ntf"),r=new Date().toISOString();return n.db.insert(k).values({id:e,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(n,t.workspaceId),{id:e}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to create notification",e)}finally{n.close();}}enforceCap(t,n){let s=(t.db.select({count:sql`count(*)`}).from(k).where(eq(k.workspace_id,n)).get()?.count??0)-this.cap;if(s<=0)return;let l=t.db.select({id:k.id}).from(k).innerJoin(C,eq(C.notification_id,k.id)).where(eq(k.workspace_id,n)).groupBy(k.id).orderBy(asc(k.created_at)).limit(s).all().map(u=>u.id);l.length!==0&&(t.db.delete(C).where(inArray(C.notification_id,l)).run(),t.db.delete(k).where(inArray(k.id,l)).run());}list(t){if(!this.dbExists())return [];let n=t.limit??Fe,e=this.openHandle(false);try{return e.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:C.read_at}).from(k).leftJoin(C,and(eq(C.notification_id,k.id),eq(C.user_id,t.userId))).where(eq(k.workspace_id,t.workspaceId)).orderBy(desc(k.created_at)).limit(n).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{e.close();}}unreadCount(t){if(!this.dbExists())return 0;let n=this.openHandle(false);try{return n.db.select({count:sql`count(*)`}).from(k).leftJoin(C,and(eq(C.notification_id,k.id),eq(C.user_id,t.userId))).where(and(eq(k.workspace_id,t.workspaceId),isNull(C.notification_id))).get()?.count??0}catch(e){throw new d("DB_ERROR","Failed to count unread notifications",e)}finally{n.close();}}markRead(t){let n=this.openHandle(true);try{if(!n.db.select({id:k.id}).from(k).where(and(eq(k.id,t.notificationId),eq(k.workspace_id,t.workspaceId))).get())throw new d("NOT_FOUND","Notification not found in workspace");return n.db.insert(C).values({notification_id:t.notificationId,user_id:t.userId,read_at:new Date().toISOString()}).onConflictDoNothing().run().changes>0}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to mark notification read",e)}finally{n.close();}}markAllRead(t){let n=this.openHandle(true);try{let e=n.db.select({id:k.id}).from(k).leftJoin(C,and(eq(C.notification_id,k.id),eq(C.user_id,t.userId))).where(and(eq(k.workspace_id,t.workspaceId),isNull(C.notification_id))).all();if(e.length===0)return 0;let r=new Date().toISOString();return n.db.transaction(s=>{for(let i of e)s.insert(C).values({notification_id:i.id,user_id:t.userId,read_at:r}).onConflictDoNothing().run();}),e.length}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to mark all notifications read",e)}finally{n.close();}}};var We=50,te=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}create(t){let n=this.openHandle(true);try{let e=new Date().toISOString(),r=[],s=t.map(i=>{let l=Lt("sug");return r.push(l),{id:l,workspace_id:i.workspaceId,agent_id:i.agentId,task_id:i.taskId??null,type:i.type,status:i.status??"pending",payload:i.payload,applied_commit_sha:i.appliedCommitSha??null,created_at:e,updated_at:e}});return s.length>0&&n.db.insert(v).values(s).run(),{ids:r}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to create agent suggestions",e)}finally{n.close();}}findLatestSet(t,n){if(!this.dbExists())return [];let e=this.openHandle(false);try{let r=e.db.select({task_id:v.task_id}).from(v).where(and(eq(v.workspace_id,t),eq(v.agent_id,n))).orderBy(desc(v.created_at)).limit(1).get();if(!r)return [];let s=r.task_id===null?isNull(v.task_id):eq(v.task_id,r.task_id);return e.db.select().from(v).where(and(eq(v.workspace_id,t),eq(v.agent_id,n),s)).orderBy(asc(v.created_at)).all()}catch(r){throw new d("DB_ERROR","Failed to find latest agent suggestion set",r)}finally{e.close();}}updateStatus(t,n,e){let r=this.openHandle(true);try{let s={status:n,updated_at:new Date().toISOString()};return e!==void 0&&(s.applied_commit_sha=e),r.db.update(v).set(s).where(eq(v.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,n,e=We){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(v).where(and(eq(v.workspace_id,t),eq(v.agent_id,n))).orderBy(desc(v.created_at)).limit(e).all()}catch(s){throw new d("DB_ERROR","Failed to list agent suggestions",s)}finally{r.close();}}};export{te as AgentSuggestionRepository,x as BaseSqliteRepository,We as DEFAULT_AGENT_SUGGESTION_LIST_LIMIT,ze as DEFAULT_NOTIFICATION_CAP,Fe as DEFAULT_NOTIFICATION_LIST_LIMIT,Qt as NotificationRepository,d as RepositoryError,Xt as RequestLogRepository,Kt as SpanRepository,jt as TaskRepository,Yt as ThreadBoxRepository,Wt as ThreadRepository,Gt as ToolCallRepository,Jt as UsageLimitSnapshotRepository,Vt as UsageReportRepository,Ft as WorkspaceRepository,ve as calcBucketStart,w as openDrizzleDb,Mn as pushSchema,_t as runMigrations,A as runMigrationsOnce,o as tasks};
|
|
309
|
+
`),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,n,e){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=new Date().toISOString();r.db.update(_).set({last_message:n,updated_at:s}).where(eq(_.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,n){if(!this.dbExists())return;let e=this.openHandle(true);try{let r={updated_at:new Date().toISOString()};n.title!==void 0&&(r.title=n.title,r.title_locked=1),n.titleLocked!==void 0&&(r.title_locked=n.titleLocked?1:0),e.db.update(_).set(r).where(eq(_.id,t)).run();}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to update thread",r)}finally{e.close();}}togglePin(t,n){let e=this.openHandle(true);try{let r=n?and(eq(_.id,t),eq(_.workspace_id,n)):eq(_.id,t),s=e.db.select({pinned:_.pinned,metadata:_.metadata}).from(_).where(r).get();if(!s)return null;let i=s.pinned?0:1,l=s.metadata?JSON.parse(s.metadata):{};if(i){let u=n?and(eq(_.pinned,1),eq(_.workspace_id,n)):eq(_.pinned,1),c=e.db.select({metadata:_.metadata}).from(_).where(u).all(),p=null;for(let h of c){let f=h.metadata?JSON.parse(h.metadata):{};typeof f.pinOrder=="number"&&(p===null||f.pinOrder<p)&&(p=f.pinOrder);}l.pinOrder=p===null?0:p-1;}else delete l.pinOrder;return e.db.update(_).set({pinned:i,metadata:Object.keys(l).length>0?JSON.stringify(l):null}).where(r).run(),{pinned:!!i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle pin",r)}finally{e.close();}}reorderPins(t,n){let e=this.openHandle(true);try{for(let r=0;r<t.length;r++){let s=n?and(eq(_.id,t[r]),eq(_.workspace_id,n)):eq(_.id,t[r]),i=e.db.select({metadata:_.metadata}).from(_).where(s).get();if(!i)continue;let l=i.metadata?JSON.parse(i.metadata):{};l.pinOrder=r+1,e.db.update(_).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{e.close();}}toggleStar(t,n){let e=this.openHandle(true);try{let r=n?and(eq(_.id,t),eq(_.workspace_id,n)):eq(_.id,t),s=e.db.select({starred:_.starred}).from(_).where(r).get();if(!s)return null;let i=s.starred?0:1;return e.db.update(_).set({starred:i}).where(r).run(),{starred:!!i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle star",r)}finally{e.close();}}resolveOverdriveForRequest(t,n,e,r){let i=this.openHandle(true);try{let l=n?and(eq(_.id,t),eq(_.workspace_id,n)):eq(_.id,t);return i.db.transaction(c=>{let p=c.select({metadata:_.metadata}).from(_).where(l).get();if(!p)return null;let h=p.metadata?JSON.parse(p.metadata):{},f=h.overdrive??{},x=r?.defaultTurns??f.defaultTurns??3,P=r?.updatedBy??"ui",$=new Date().toISOString(),M,A,Y,wt;switch(e){case "enable-count":{let kt=r?.turns??x,bt=r?.consumeCurrent?Math.max(kt-1,0):kt;M=bt>0?"count":"off",A=bt,Y=!0,wt="count";break}case "enable-latch":{M="latch",A=0,Y=!0,wt="latch";break}case "disable":{M="off",A=0,Y=!1;break}default:{let kt=f.state??"off",bt=typeof f.remaining=="number"?f.remaining:0;if(kt==="count"&&bt>0){let ne=bt-1;ne<=0?(M="off",A=0):(M="count",A=ne),Y=!0,wt="count";}else kt==="latch"?(M="latch",A=0,Y=!0,wt="latch"):(M="off",A=0,Y=!1);break}}return h.overdrive={state:M,remaining:A,defaultTurns:x,updatedAt:$,updatedBy:P},c.update(_).set({metadata:JSON.stringify(h),updated_at:$}).where(l).run(),{applied:Y,appliedMode:wt,state:M,remaining:A}},{behavior:"immediate"})}catch(l){throw l instanceof d?l:new d("DB_ERROR","Failed to resolve overdrive",l)}finally{i.close();}}};var Gt=class extends T{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{O(e.db,n);}catch(r){throw e.close(),r}return e}insertSpan(t){let n=this.openHandle(true);try{n.db.insert(et).values(t).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert span",e)}finally{n.close();}}findByTaskId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(et).where(eq(et.task_id,t)).all()}catch(e){throw new d("DB_ERROR","Failed to find spans by task id",e)}finally{n.close();}}findById(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(et).where(eq(et.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find span by id",e)}finally{n.close();}}};var Yt=class extends T{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{O(e.db,n);}catch(r){throw e.close(),r}return e}insertToolCall(t){let n=this.openHandle(true);try{n.db.insert(J).values(t).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert tool call",e)}finally{n.close();}}findByTaskId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(J).where(eq(J.task_id,t)).all()}catch(e){throw new d("DB_ERROR","Failed to find tool calls by task id",e)}finally{n.close();}}aggregateByName(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select({toolName:J.tool_name,count:sql`count(*)`}).from(J).where(eq(J.task_id,t)).groupBy(J.tool_name).all().map(e=>({toolName:e.toolName,count:e.count}))}catch(e){throw new d("DB_ERROR","Failed to aggregate tool calls by name",e)}finally{n.close();}}};var Xt=class extends T{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{O(e.db,n);}catch(r){throw e.close(),r}return e}ensureThreadExists(t,n){if(!t.db.select({id:_.id}).from(_).where(eq(_.id,n)).limit(1).get())throw new d("NOT_FOUND",`Thread not found: ${n}`)}findByThreadId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return this.ensureThreadExists(n,t),n.db.select().from(W).where(eq(W.thread_id,t)).orderBy(asc(W.seq)).all()}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to find boxes by thread id",e)}finally{n.close();}}findById(t,n){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(W).where(and(eq(W.id,n),eq(W.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{e.close();}}insert(t){let n=this.openHandle(true);try{this.ensureThreadExists(n,t.threadId);try{n.db.insert(W).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 e=n.db.select().from(W).where(eq(W.id,t.id)).limit(1).get();if(!e)throw new d("DB_ERROR","Insert did not return a row");return e}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert thread box",e)}finally{n.close();}}};var Jt=class extends T{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{O(e.db,n);}catch(r){throw e.close(),r}return e}bulkInsert(t){if(t.length===0)return;let n=this.openHandle(true);try{let e=new Date().toISOString();n.db.transaction(r=>{for(let s of t)r.insert(ht).values({id: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:e,metadata:s.metadata??null}).run();});}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to bulk insert request logs",e)}finally{n.close();}}findRecent(t=100){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(ht).orderBy(desc(ht.timestamp)).limit(t).all()}catch(e){throw new d("DB_ERROR","Failed to find recent request logs",e)}finally{n.close();}}};function Ie(a){return new Date(Math.floor(a/3e5)*3e5).toISOString()}var Vt=class extends T{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{O(e.db,n);}catch(r){throw e.close(),r}return e}upsert(t){let n=this.openHandle(true);try{let e=Ie(t.bucketStartMs),r=t.accountRef??"default";n.db.insert(I).values({id:t.id,provider:t.provider,account_ref:r,limit_window:t.limitWindow,bucket_start:e,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(e){throw e instanceof d?e:new d("DB_ERROR","Failed to upsert usage limit snapshot",e)}finally{n.close();}}getTrends(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{let e=n.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(and(eq(I.limit_window,t.window),gte(I.bucket_start,t.fromIso),lte(I.bucket_start,t.toIso))).orderBy(asc(I.bucket_start)).all(),r=new Map;for(let s of e)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,i])=>({provider:s,points:i}))}catch(e){throw new d("DB_ERROR","Failed to get usage limit trends",e)}finally{n.close();}}};var Qt=class extends T{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{O(e.db,n);}catch(r){throw e.close(),r}return e}publish(t){let n=this.openHandle(true);try{n.db.insert(F).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:[F.workspace_id,F.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 e=n.db.select().from(F).where(and(eq(F.workspace_id,t.workspaceId),eq(F.month,t.month))).limit(1).get();if(!e)throw new d("DB_ERROR","Failed to read back published usage report");return e}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to publish usage report",e)}finally{n.close();}}findByMonth(t,n){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(F).where(and(eq(F.workspace_id,t),eq(F.month,n))).limit(1).get()??void 0}catch(r){throw new d("DB_ERROR","Failed to find usage report",r)}finally{e.close();}}listByWorkspace(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(F).where(eq(F.workspace_id,t)).orderBy(desc(F.month)).all()}catch(e){throw new d("DB_ERROR","Failed to list usage reports",e)}finally{n.close();}}};var Be="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";function Ar(a){let t=randomBytes(a*2),n="";for(let e=0;e<t.length&&n.length<a;e++){let r=t[e];r<248&&(n+=Be[r%62]);}for(;n.length<a;)n+=Be[randomBytes(1)[0]%62];return n}function $t(a){return `${a}_${Ar(8)}`}var ze=500,qe=50,Zt=class extends T{dbPath;cap;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db")),this.cap=t.cap??ze;}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{O(e.db,n);}catch(r){throw e.close(),r}return e}create(t){let n=this.openHandle(true);try{let e=$t("ntf"),r=new Date().toISOString();return n.db.insert(k).values({id:e,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(n,t.workspaceId),{id:e}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to create notification",e)}finally{n.close();}}enforceCap(t,n){let s=(t.db.select({count:sql`count(*)`}).from(k).where(eq(k.workspace_id,n)).get()?.count??0)-this.cap;if(s<=0)return;let l=t.db.select({id:k.id}).from(k).innerJoin(v,eq(v.notification_id,k.id)).where(eq(k.workspace_id,n)).groupBy(k.id).orderBy(asc(k.created_at)).limit(s).all().map(u=>u.id);l.length!==0&&(t.db.delete(v).where(inArray(v.notification_id,l)).run(),t.db.delete(k).where(inArray(k.id,l)).run());}list(t){if(!this.dbExists())return [];let n=t.limit??qe,e=this.openHandle(false);try{return e.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,and(eq(v.notification_id,k.id),eq(v.user_id,t.userId))).where(eq(k.workspace_id,t.workspaceId)).orderBy(desc(k.created_at)).limit(n).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{e.close();}}unreadCount(t){if(!this.dbExists())return 0;let n=this.openHandle(false);try{return n.db.select({count:sql`count(*)`}).from(k).leftJoin(v,and(eq(v.notification_id,k.id),eq(v.user_id,t.userId))).where(and(eq(k.workspace_id,t.workspaceId),isNull(v.notification_id))).get()?.count??0}catch(e){throw new d("DB_ERROR","Failed to count unread notifications",e)}finally{n.close();}}markRead(t){let n=this.openHandle(true);try{if(!n.db.select({id:k.id}).from(k).where(and(eq(k.id,t.notificationId),eq(k.workspace_id,t.workspaceId))).get())throw new d("NOT_FOUND","Notification not found in workspace");return n.db.insert(v).values({notification_id:t.notificationId,user_id:t.userId,read_at:new Date().toISOString()}).onConflictDoNothing().run().changes>0}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to mark notification read",e)}finally{n.close();}}markAllRead(t){let n=this.openHandle(true);try{let e=n.db.select({id:k.id}).from(k).leftJoin(v,and(eq(v.notification_id,k.id),eq(v.user_id,t.userId))).where(and(eq(k.workspace_id,t.workspaceId),isNull(v.notification_id))).all();if(e.length===0)return 0;let r=new Date().toISOString();return n.db.transaction(s=>{for(let i of e)s.insert(v).values({notification_id:i.id,user_id:t.userId,read_at:r}).onConflictDoNothing().run();}),e.length}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to mark all notifications read",e)}finally{n.close();}}};var Ke=50,ee=class extends T{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=dirname(n);existsSync(r)||mkdirSync(r,{recursive:true});}else if(!existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{O(e.db,n);}catch(r){throw e.close(),r}return e}create(t){let n=this.openHandle(true);try{let e=new Date().toISOString(),r=[],s=t.map(i=>{let l=$t("sug");return r.push(l),{id:l,workspace_id:i.workspaceId,agent_id:i.agentId,task_id:i.taskId??null,type:i.type,status:i.status??"pending",payload:i.payload,applied_commit_sha:i.appliedCommitSha??null,created_at:e,updated_at:e}});return s.length>0&&n.db.insert(y).values(s).run(),{ids:r}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to create agent suggestions",e)}finally{n.close();}}findLatestSet(t,n){if(!this.dbExists())return [];let e=this.openHandle(false);try{let r=e.db.select({task_id:y.task_id}).from(y).where(and(eq(y.workspace_id,t),eq(y.agent_id,n))).orderBy(desc(y.created_at)).limit(1).get();if(!r)return [];let s=r.task_id===null?isNull(y.task_id):eq(y.task_id,r.task_id);return e.db.select().from(y).where(and(eq(y.workspace_id,t),eq(y.agent_id,n),s)).orderBy(asc(y.created_at)).all()}catch(r){throw new d("DB_ERROR","Failed to find latest agent suggestion set",r)}finally{e.close();}}findSetByTaskId(t,n,e){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(y).where(and(eq(y.workspace_id,t),eq(y.agent_id,n),eq(y.task_id,e))).orderBy(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();}}updateStatus(t,n,e){let r=this.openHandle(true);try{let s={status:n,updated_at:new Date().toISOString()};return e!==void 0&&(s.applied_commit_sha=e),r.db.update(y).set(s).where(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,n,e=Ke){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(y).where(and(eq(y.workspace_id,t),eq(y.agent_id,n))).orderBy(desc(y.created_at)).limit(e).all()}catch(s){throw new d("DB_ERROR","Failed to list agent suggestions",s)}finally{r.close();}}findDismissed(t,n,e=20){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(y).where(and(eq(y.workspace_id,t),eq(y.agent_id,n),eq(y.status,"dismissed"))).orderBy(desc(y.updated_at)).limit(e).all()}catch(s){throw new d("DB_ERROR","Failed to find dismissed agent suggestions",s)}finally{r.close();}}};export{ee as AgentSuggestionRepository,T as BaseSqliteRepository,Ke as DEFAULT_AGENT_SUGGESTION_LIST_LIMIT,ze as DEFAULT_NOTIFICATION_CAP,qe as DEFAULT_NOTIFICATION_LIST_LIMIT,Zt as NotificationRepository,d as RepositoryError,Jt as RequestLogRepository,Gt as SpanRepository,Wt as TaskRepository,Xt as ThreadBoxRepository,Kt as ThreadRepository,Yt as ToolCallRepository,Vt as UsageLimitSnapshotRepository,Qt as UsageReportRepository,qt as WorkspaceRepository,Ie as calcBucketStart,w as openDrizzleDb,Fn as pushSchema,_t as runMigrations,O as runMigrationsOnce,o as tasks};
|
|
@@ -26,6 +26,8 @@ export declare class AgentSuggestionRepository extends BaseSqliteRepository {
|
|
|
26
26
|
ids: string[];
|
|
27
27
|
};
|
|
28
28
|
findLatestSet(workspaceId: string, agentId: string): AgentSuggestionRow[];
|
|
29
|
+
findSetByTaskId(workspaceId: string, agentId: string, taskId: string): AgentSuggestionRow[];
|
|
29
30
|
updateStatus(id: string, status: SuggestionStatus, appliedCommitSha?: string | null): boolean;
|
|
30
31
|
listByAgent(workspaceId: string, agentId: string, limit?: number): AgentSuggestionRow[];
|
|
32
|
+
findDismissed(workspaceId: string, agentId: string, limit?: number): AgentSuggestionRow[];
|
|
31
33
|
}
|
package/dist/repository/index.js
CHANGED
|
@@ -1,14 +1,14 @@
|
|
|
1
|
-
'use strict';var
|
|
1
|
+
'use strict';var fe=require('fs'),Q=require('path'),Ye=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 fe__default=/*#__PURE__*/_interopDefault(fe);var Q__namespace=/*#__PURE__*/_interopNamespace(Q);var Ye__default=/*#__PURE__*/_interopDefault(Ye);var xt=(a=>typeof require<"u"?require:typeof Proxy<"u"?new Proxy(a,{get:(t,n)=>(typeof require<"u"?require:t)[n]}):a)(function(a){if(typeof require<"u")return require.apply(this,arguments);throw Error('Dynamic require of "'+a+'" is not supported')});var x=class{resolveDbPath(){return process.env.CREWX_DB?process.env.CREWX_DB:process.env.CREWX_TRACES_DB?process.env.CREWX_TRACES_DB:Q.join(Ye__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 fe.existsSync(t??this.resolveDbPath())}};var d=class extends Error{code;cause;constructor(t,n,e){super(n),this.name="RepositoryError",this.code=t,this.cause=e,Object.setPrototypeOf(this,new.target.prototype);}};function w(a){let t=xt("better-sqlite3"),{drizzle:n}=xt("drizzle-orm/better-sqlite3"),e=new t(a);return e.exec("PRAGMA journal_mode = WAL"),e.exec("PRAGMA busy_timeout = 5000"),e.exec("PRAGMA foreign_keys = ON"),e.exec("PRAGMA analysis_limit = 400"),e.exec("PRAGMA optimize"),{db:n(e),runRaw:(r,s=[])=>e.prepare(r).run(...s),close:()=>e.close()}}var ne=new Set,Xe={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 $t(a,t){return (a.get(`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='${t}'`)?.cnt??0)>0}function Je(a,t){if(t>0||!$t(a,"tasks"))return;let n=a.all("PRAGMA table_info(tasks)"),e=new Set(n.map(r=>r.name));for(let[r,s]of Object.entries(Xe))e.has(r)||a.run(`ALTER TABLE tasks ADD COLUMN ${r} ${s}`);}var Ve={"0002_normalize_task_names":{workspace_name:"TEXT",project_name:"TEXT"}};function Qe(a,t,n){if(!$t(a,"__drizzle_migrations")||!$t(a,"tasks"))return;let e=a.all(n`SELECT hash FROM __drizzle_migrations`),r=new Set(e.map(i=>i.hash)),s=JSON.parse(fe.readFileSync(Q__namespace.default.join(t,"meta/_journal.json"),"utf-8"));for(let i of s.entries){let l=Ve[i.tag];if(!l)continue;let u=Q__namespace.default.join(t,`${i.tag}.sql`);if(!fe.existsSync(u))continue;let c=fe.readFileSync(u,"utf-8"),p=crypto.createHash("sha256").update(c).digest("hex");if(r.has(p))continue;let h=a.all("PRAGMA table_info(tasks)"),f=new Set(h.map(S=>S.name));for(let[S,I]of Object.entries(l))f.has(S)||(a.run(`ALTER TABLE tasks ADD COLUMN ${S} ${I}`),f.add(S));}}function Ze(a,t,n){let e=a.all(n`SELECT hash FROM __drizzle_migrations`),r=new Set(e.map(i=>i.hash)),s=JSON.parse(fe.readFileSync(Q__namespace.default.join(t,"meta/_journal.json"),"utf-8"));for(let i of s.entries){let l=Q__namespace.default.join(t,`${i.tag}.sql`);if(!fe.existsSync(l))continue;let u=fe.readFileSync(l,"utf-8"),c=crypto.createHash("sha256").update(u).digest("hex");if(r.has(c))continue;let p=/ALTER\s+TABLE\s+[`"]?(\w+)[`"]?\s+ADD\s+[`"]?(\w+)[`"]?/gi,h=[],f;for(;(f=p.exec(u))!==null;)h.push({table:f[1],column:f[2]});if(h.length===0||!u.split(/-->\s*statement-breakpoint/).map(L=>L.trim()).filter(Boolean).every(L=>/^ALTER\s+TABLE\s+.+\s+ADD\s+/i.test(L)))continue;h.every(({table:L,column:H})=>a.all(`PRAGMA table_info("${L}")`).some(G=>G.name===H))&&a.run(n`INSERT INTO __drizzle_migrations (hash, created_at) VALUES (${c}, ${i.when})`);}}function pt(a){let{migrate:t}=xt("drizzle-orm/better-sqlite3/migrator"),{sql:n}=xt("drizzle-orm"),e=[Q__namespace.default.join(__dirname,"../migrations"),Q__namespace.default.join(__dirname,"migrations"),Q__namespace.default.join(__dirname,"../../../../drizzle/migrations"),Q__namespace.default.join(process.cwd(),"drizzle/migrations")],r=e.find(c=>fe.existsSync(Q__namespace.default.join(c,"meta/_journal.json")));if(!r)throw new Error(`migrations folder not found. Searched:
|
|
2
2
|
${e.join(`
|
|
3
|
-
`)}`);let s=a.get(n`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='__drizzle_migrations'`),i=0;s?.cnt&&(i=a.get(n`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0),Xe(a,i),s?.cnt&&(Qe(a,r,n),Ve(a,r,n)),t(a,{migrationsFolder:r});let u=(a.get(n`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0)-i;if(u>0){let c=s?.cnt?"Database migrated":"Database initialized";console.error(`[crewx] ${c} (${u} migration${u>1?"s":""} applied).`);}}function N(a,t){ee.has(t)||(pt(a),ee.add(t));}var m=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)},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 _=sqliteCore.sqliteTable("threads",{id:sqliteCore.text("id").primaryKey(),workspace_id:sqliteCore.text("workspace_id").references(()=>m.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),idx_threads_ws_updated:sqliteCore.index("idx_threads_ws_updated").on(a.workspace_id,a.updated_at),idx_threads_title:sqliteCore.index("idx_threads_title").on(a.title)}));var Z=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(()=>Z.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 Y=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()},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 q=sqliteCore.sqliteTable("thread_boxes",{id:sqliteCore.text("id").primaryKey(),thread_id:sqliteCore.text("thread_id").notNull().references(()=>_.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 _t=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 v=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)}));var H=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")},a=>({uniq_usage_reports_ws_month:sqliteCore.uniqueIndex("uniq_usage_reports_ws_month").on(a.workspace_id,a.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")},a=>({idx_notifications_ws_created:sqliteCore.index("idx_notifications_ws_created").on(a.workspace_id,drizzleOrm.sql`${a.created_at} DESC`)})),O=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()},a=>({pk:sqliteCore.primaryKey({columns:[a.notification_id,a.user_id]})}));var C=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()},a=>({idx_agent_suggestions_ws_agent:sqliteCore.index("idx_agent_suggestions_ws_agent").on(a.workspace_id,a.agent_id),idx_agent_suggestions_ws_created:sqliteCore.index("idx_agent_suggestions_ws_created").on(a.workspace_id,a.created_at)}));function Ut(a){let t=V__namespace.resolve(a);return process.platform==="win32"&&(t=t.replace(/\\/g,"/"),t=t.replace(/^([A-Z]):/,(n,e)=>`${e.toLowerCase()}:`)),t.length>1&&!/^[a-zA-Z]:\/$/.test(t)&&(t=t.replace(/\/+$/,"")),t}function ue(a){let t=Ut(a);return crypto.createHash("sha256").update(t).digest("hex")}function Ot(a){return a.toLowerCase().replace(/[^a-z0-9-]/g,"-").replace(/-{2,}/g,"-").replace(/^-+|-+$/g,"")}var zt=class extends S{dbRoot;constructor(t={}){super(),this.dbRoot=t.dbRoot;}resolveDbPath(){return this.dbRoot?V.join(this.dbRoot,".crewx","crewx.db"):super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=V.dirname(n);he.existsSync(r)||he.mkdirSync(r,{recursive:true});}else if(!he.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{pt(e.db);}catch(r){throw e.close(),r}return e}resolveSlug(t,n,e){let r=Ot(V.basename(e)),i=`${Ot(V.basename(V.dirname(e)))}-${r}`,l=[r,i];try{let u=c=>t.select({id:m.id}).from(m).where(drizzleOrm.and(drizzleOrm.eq(m.slug,c),drizzleOrm.ne(m.id,n))).limit(1).all().length>0;for(let c of l)if(!u(c))return c;for(let c=2;c<1e3;c+=1){let p=`${i}-${c}`;if(!u(p))return p}}catch{}return r}normalizeLegacySlugs(){if(!this.dbExists())return {updated:0,checked:0};let t=this.openHandle(true);try{let n=t.db.select({id:m.id,slug:m.slug}).from(m).all(),e=0;for(let r of n)if(r.slug.includes("/")){let s=Ot(r.slug.replace(/\//g,"-"));t.db.update(m).set({slug:s,updated_at:new Date().toISOString()}).where(drizzleOrm.eq(m.id,r.id)).run(),e+=1;}return {updated:e,checked:n.length}}catch(n){throw new d("DB_ERROR","Failed to normalize legacy slugs",n)}finally{t.close();}}ensureRow(t,n){let{id:e,slug:r,name:s,workspacePath:i}=n,l=new Date().toISOString();t.insert(m).values({id:e,slug:r,name:s,workspace_path:i,is_active:1,created_at:l,updated_at:l}).onConflictDoNothing().run(),t.update(m).set({workspace_path:i,updated_at:l}).where(drizzleOrm.and(drizzleOrm.eq(m.id,e),drizzleOrm.isNull(m.workspace_path))).run();}registerWorkspace(t){let n=Ut(t),e=this.openHandle(true);try{let r=ue(n),s=V.basename(n),i=this.resolveSlug(e.db,r,n);return this.ensureRow(e.db,{id:r,slug:i,name:s,workspacePath:n}),{id:r,slug:i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to register workspace",r)}finally{e.close();}}listProjects(t){if(!this.dbExists())return {rows:[],total:0};let n=this.openHandle(false);try{let e=t.isActive!==void 0?drizzleOrm.eq(m.is_active,t.isActive?1:0):void 0,r=n.db.select({count:drizzleOrm.sql`count(*)`}).from(m).where(e).get();return {rows:n.db.select().from(m).where(e).orderBy(drizzleOrm.desc(m.updated_at)).limit(t.limit).offset(t.offset).all(),total:r?.count??0}}catch(e){throw new d("DB_ERROR","Failed to list projects",e)}finally{n.close();}}findById(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(m).where(drizzleOrm.eq(m.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find workspace",e)}finally{n.close();}}findAgentsByProject(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.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(e){throw new d("DB_ERROR","Failed to find agents by project",e)}finally{n.close();}}findThreadsByProject(t,n){if(!this.dbExists())return {rows:[],total:0};let e=this.openHandle(false);try{let r=e.db.get(drizzleOrm.sql`SELECT COUNT(*) as count FROM threads WHERE workspace_id = ${t}`);return {rows:e.db.all(drizzleOrm.sql`SELECT t.*,
|
|
3
|
+
`)}`);let s=a.get(n`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='__drizzle_migrations'`),i=0;s?.cnt&&(i=a.get(n`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0),Je(a,i),s?.cnt&&(Ze(a,r,n),Qe(a,r,n)),t(a,{migrationsFolder:r});let u=(a.get(n`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0)-i;if(u>0){let c=s?.cnt?"Database migrated":"Database initialized";console.error(`[crewx] ${c} (${u} migration${u>1?"s":""} applied).`);}}function A(a,t){ne.has(t)||(pt(a),ne.add(t));}var m=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)},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 _=sqliteCore.sqliteTable("threads",{id:sqliteCore.text("id").primaryKey(),workspace_id:sqliteCore.text("workspace_id").references(()=>m.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),idx_threads_ws_updated:sqliteCore.index("idx_threads_ws_updated").on(a.workspace_id,a.updated_at),idx_threads_title:sqliteCore.index("idx_threads_title").on(a.title)}));var tt=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(()=>tt.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 X=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()},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 j=sqliteCore.sqliteTable("thread_boxes",{id:sqliteCore.text("id").primaryKey(),thread_id:sqliteCore.text("thread_id").notNull().references(()=>_.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 _t=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 v=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)}));var M=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")},a=>({uniq_usage_reports_ws_month:sqliteCore.uniqueIndex("uniq_usage_reports_ws_month").on(a.workspace_id,a.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")},a=>({idx_notifications_ws_created:sqliteCore.index("idx_notifications_ws_created").on(a.workspace_id,drizzleOrm.sql`${a.created_at} DESC`)})),C=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()},a=>({pk:sqliteCore.primaryKey({columns:[a.notification_id,a.user_id]})}));var E=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()},a=>({idx_agent_suggestions_ws_agent:sqliteCore.index("idx_agent_suggestions_ws_agent").on(a.workspace_id,a.agent_id),idx_agent_suggestions_ws_created:sqliteCore.index("idx_agent_suggestions_ws_created").on(a.workspace_id,a.created_at)}));function Ht(a){let t=Q__namespace.resolve(a);return process.platform==="win32"&&(t=t.replace(/\\/g,"/"),t=t.replace(/^([A-Z]):/,(n,e)=>`${e.toLowerCase()}:`)),t.length>1&&!/^[a-zA-Z]:\/$/.test(t)&&(t=t.replace(/\/+$/,"")),t}function pe(a){let t=Ht(a);return crypto.createHash("sha256").update(t).digest("hex")}function Ct(a){return a.toLowerCase().replace(/[^a-z0-9-]/g,"-").replace(/-{2,}/g,"-").replace(/^-+|-+$/g,"")}var Ft=class extends x{dbRoot;constructor(t={}){super(),this.dbRoot=t.dbRoot;}resolveDbPath(){return this.dbRoot?Q.join(this.dbRoot,".crewx","crewx.db"):super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{pt(e.db);}catch(r){throw e.close(),r}return e}resolveSlug(t,n,e){let r=Ct(Q.basename(e)),i=`${Ct(Q.basename(Q.dirname(e)))}-${r}`,l=[r,i];try{let u=c=>t.select({id:m.id}).from(m).where(drizzleOrm.and(drizzleOrm.eq(m.slug,c),drizzleOrm.ne(m.id,n))).limit(1).all().length>0;for(let c of l)if(!u(c))return c;for(let c=2;c<1e3;c+=1){let p=`${i}-${c}`;if(!u(p))return p}}catch{}return r}normalizeLegacySlugs(){if(!this.dbExists())return {updated:0,checked:0};let t=this.openHandle(true);try{let n=t.db.select({id:m.id,slug:m.slug}).from(m).all(),e=0;for(let r of n)if(r.slug.includes("/")){let s=Ct(r.slug.replace(/\//g,"-"));t.db.update(m).set({slug:s,updated_at:new Date().toISOString()}).where(drizzleOrm.eq(m.id,r.id)).run(),e+=1;}return {updated:e,checked:n.length}}catch(n){throw new d("DB_ERROR","Failed to normalize legacy slugs",n)}finally{t.close();}}ensureRow(t,n){let{id:e,slug:r,name:s,workspacePath:i}=n,l=new Date().toISOString();t.insert(m).values({id:e,slug:r,name:s,workspace_path:i,is_active:1,created_at:l,updated_at:l}).onConflictDoNothing().run(),t.update(m).set({workspace_path:i,updated_at:l}).where(drizzleOrm.and(drizzleOrm.eq(m.id,e),drizzleOrm.isNull(m.workspace_path))).run();}registerWorkspace(t){let n=Ht(t),e=this.openHandle(true);try{let r=pe(n),s=Q.basename(n),i=this.resolveSlug(e.db,r,n);return this.ensureRow(e.db,{id:r,slug:i,name:s,workspacePath:n}),{id:r,slug:i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to register workspace",r)}finally{e.close();}}listProjects(t){if(!this.dbExists())return {rows:[],total:0};let n=this.openHandle(false);try{let e=t.isActive!==void 0?drizzleOrm.eq(m.is_active,t.isActive?1:0):void 0,r=n.db.select({count:drizzleOrm.sql`count(*)`}).from(m).where(e).get();return {rows:n.db.select().from(m).where(e).orderBy(drizzleOrm.desc(m.updated_at)).limit(t.limit).offset(t.offset).all(),total:r?.count??0}}catch(e){throw new d("DB_ERROR","Failed to list projects",e)}finally{n.close();}}findById(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(m).where(drizzleOrm.eq(m.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find workspace",e)}finally{n.close();}}findAgentsByProject(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.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(e){throw new d("DB_ERROR","Failed to find agents by project",e)}finally{n.close();}}findThreadsByProject(t,n){if(!this.dbExists())return {rows:[],total:0};let e=this.openHandle(false);try{let r=e.db.get(drizzleOrm.sql`SELECT COUNT(*) as count FROM threads WHERE workspace_id = ${t}`);return {rows:e.db.all(drizzleOrm.sql`SELECT t.*,
|
|
4
4
|
(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
5
|
FROM threads t
|
|
6
6
|
WHERE t.workspace_id = ${t}
|
|
7
7
|
ORDER BY t.updated_at DESC
|
|
8
|
-
LIMIT ${n.limit} OFFSET ${n.offset}`),total:r?.count??0}}catch(r){throw new d("DB_ERROR","Failed to find threads by project",r)}finally{e.close();}}findBySlug(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(m).where(drizzleOrm.eq(m.slug,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find workspace by slug",e)}finally{n.close();}}slugExists(t,n){if(!this.dbExists())return false;let e=this.openHandle(false);try{let r=n?drizzleOrm.and(drizzleOrm.eq(m.slug,t),drizzleOrm.ne(m.id,n)):drizzleOrm.eq(m.slug,t);return !!e.db.select({id:m.id}).from(m).where(r).limit(1).get()}catch(r){throw new d("DB_ERROR","Failed to check slug",r)}finally{e.close();}}insert(t,n,e,r){let s=this.openHandle(true);try{let i=new Date().toISOString();s.db.insert(m).values({id:t,slug:n,name:e,workspace_path:r,is_active:1,created_at:i,updated_at:i}).run();let l=s.db.select().from(m).where(drizzleOrm.eq(m.id,t)).limit(1).get();if(!l)throw new d("DB_ERROR","Insert did not return a row");return l}catch(i){throw i instanceof d?i:new d("DB_ERROR","Failed to insert workspace",i)}finally{s.close();}}update(t,n,e){let r=this.openHandle(true);try{r.runRaw(`UPDATE workspaces SET ${n.join(", ")} WHERE id = ?`,e);let s=r.db.select().from(m).where(drizzleOrm.eq(m.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 n=t.db.select({id:m.id,workspace_path:m.workspace_path}).from(m).where(drizzleOrm.and(drizzleOrm.eq(m.is_active,1),drizzleOrm.isNotNull(m.workspace_path))).all(),e=0;for(let r of n){let s=r.workspace_path;
|
|
8
|
+
LIMIT ${n.limit} OFFSET ${n.offset}`),total:r?.count??0}}catch(r){throw new d("DB_ERROR","Failed to find threads by project",r)}finally{e.close();}}findBySlug(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(m).where(drizzleOrm.eq(m.slug,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find workspace by slug",e)}finally{n.close();}}slugExists(t,n){if(!this.dbExists())return false;let e=this.openHandle(false);try{let r=n?drizzleOrm.and(drizzleOrm.eq(m.slug,t),drizzleOrm.ne(m.id,n)):drizzleOrm.eq(m.slug,t);return !!e.db.select({id:m.id}).from(m).where(r).limit(1).get()}catch(r){throw new d("DB_ERROR","Failed to check slug",r)}finally{e.close();}}insert(t,n,e,r){let s=this.openHandle(true);try{let i=new Date().toISOString();s.db.insert(m).values({id:t,slug:n,name:e,workspace_path:r,is_active:1,created_at:i,updated_at:i}).run();let l=s.db.select().from(m).where(drizzleOrm.eq(m.id,t)).limit(1).get();if(!l)throw new d("DB_ERROR","Insert did not return a row");return l}catch(i){throw i instanceof d?i:new d("DB_ERROR","Failed to insert workspace",i)}finally{s.close();}}update(t,n,e){let r=this.openHandle(true);try{r.runRaw(`UPDATE workspaces SET ${n.join(", ")} WHERE id = ?`,e);let s=r.db.select().from(m).where(drizzleOrm.eq(m.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 n=t.db.select({id:m.id,workspace_path:m.workspace_path}).from(m).where(drizzleOrm.and(drizzleOrm.eq(m.is_active,1),drizzleOrm.isNotNull(m.workspace_path))).all(),e=0;for(let r of n){let s=r.workspace_path;fe.existsSync(Q.join(s,"crewx.yaml"))||fe.existsSync(Q.join(s,"crewx.yml"))||(t.db.update(m).set({is_active:0,updated_at:new Date().toISOString()}).where(drizzleOrm.eq(m.id,r.id)).run(),e+=1);}return {softDeleted:e,checked:n.length}}catch(n){throw new d("DB_ERROR","Failed to cleanup orphan workspaces",n)}finally{t.close();}}delete(t){let n=this.openHandle(true);try{n.db.run(drizzleOrm.sql`UPDATE threads SET workspace_id = NULL WHERE workspace_id = ${t}`),n.db.delete(m).where(drizzleOrm.eq(m.id,t)).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to delete workspace",e)}finally{n.close();}}};var ge=[m,o,_,tt,X,j,_t];function vn(){return Q__namespace.default.join(Ye__default.default.homedir(),".crewx","crewx.db")}function In(a){if(!fe__default.default.existsSync(a))return null;let t=Math.floor(Date.now()/1e3),n=`${a}.bak-${t}`;return fe__default.default.copyFileSync(a,n),n}function we(a){let t=a.all("SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%'");return new Set(t.map(n=>n.name))}function Pn(a){return a==null||typeof a=="object"&&"queryChunks"in a?null:typeof a=="number"?String(a):typeof a=="boolean"?a?"1":"0":typeof a=="string"?`'${a.replace(/'/g,"''")}'`:String(a)}function Ln(a){let t=a.getSQLType(),n=`"${a.name}" ${t}`,e=Pn(a.default);return e!==null&&(n+=` DEFAULT ${e}`),a.notNull&&(n+=" NOT NULL"),n}function $n(a,t){let n=t?.dbPath??vn(),e=t?.force??false,r=t?.dryRun??false,s=r?null:In(n);if(e&&!r)try{a.run("DELETE FROM __drizzle_migrations");}catch{}let i=we(a);if(!r)try{pt(a);}catch(h){let f=h instanceof Error?`${h.message} ${h.cause?.message??""}`:"";if(!e||!f.includes("duplicate column"))throw h}let l,u;r?(l=ge.map(f=>sqliteCore.getTableConfig(f).name).filter(f=>!i.has(f)),u=i):(u=we(a),l=[...u].filter(h=>!i.has(h)));let c=[],p=[];for(let h of ge){let f=sqliteCore.getTableConfig(h),S=f.name;if(!u.has(S))continue;let I=a.all(`PRAGMA table_info("${S}")`),L=new Set(I.map(N=>N.name)),H=new Set(f.columns.map(N=>N.name));for(let N of f.columns)if(!L.has(N.name)){if(!r){let G=Ln(N);a.run(`ALTER TABLE "${S}" ADD COLUMN ${G}`);}c.push({table:S,column:N.name});}for(let N of I)H.has(N.name)||p.push(`${S}.${N.name} exists in DB but not in schema (untouched)`);}return {created:l,altered:c,warnings:p,backupPath:s}}function ke(a){let t=a.toLowerCase().trim();t=t.replace(/\[[^\]]*\]$/g,"");let n=t.lastIndexOf("/");return n>=0&&(t=t.slice(n+1)),t=t.replace(/-\d{8}$/,""),t}var st="2026-05-09",qn="0.8.9-rc.13",it=10,at=parseInt(qn.split("rc.")[1]),jn=new Set(["workflow","mcp"]),Re=600*1e3,jt=class extends x{dbPath;pidNullObservedSince=new Map;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=Q.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}startTask(t){let n=this.openHandle(true);try{n.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(e){throw e instanceof d?e:new d("DB_ERROR","Failed to start task",e)}finally{n.close();}}finishTask(t){let n=this.openHandle(true);try{let e=t.runEpoch??null;n.runRaw(`UPDATE tasks SET status=?, result=?, error=?, completed_at=?, duration_ms=?,
|
|
9
9
|
exit_code=?, input_tokens=?, output_tokens=?, cached_input_tokens=?, cost_usd=?,
|
|
10
10
|
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,e]);}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to finish task",e)}finally{n.close();}}appendLog(t,n){let e=this.openHandle(true);try{e.db.transaction(r=>{let s=r.select({logs:o.logs}).from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get(),i=s?.logs?JSON.parse(s.logs):[];i.push(n),r.update(o).set({logs:JSON.stringify(i)}).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{e.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(n){throw new d("DB_ERROR","Failed to get running tasks",n)}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(n){throw new d("DB_ERROR","Failed to get all tasks",n)}finally{t.close();}}getTask(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to get task",e)}finally{n.close();}}killTask(t){if(!this.dbExists())return {killed:false};let n=this.openHandle(true);try{let e=n.db.select({id:o.id,status:o.status,pid:o.pid}).from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get();if(!e||e.status!=="running")return {killed:!1};if(e.pid)try{process.kill(e.pid,"SIGTERM");}catch{}return n.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:e.pid??void 0}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to kill task",e)}finally{n.close();}}reapOrphanedTasks(){if(!this.dbExists())return 0;let t=this.openHandle(true);try{let n=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(),e=Date.now(),r=new Set,s=0;for(let i of n){if(i.pid){let c=!1;try{process.kill(i.pid,0),c=!0;}catch{}c||(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,i.id),drizzleOrm.eq(o.status,"running"))).run(),s++);continue}if(
|
|
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,e]);}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to finish task",e)}finally{n.close();}}appendLog(t,n){let e=this.openHandle(true);try{e.db.transaction(r=>{let s=r.select({logs:o.logs}).from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get(),i=s?.logs?JSON.parse(s.logs):[];i.push(n),r.update(o).set({logs:JSON.stringify(i)}).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{e.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(n){throw new d("DB_ERROR","Failed to get running tasks",n)}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(n){throw new d("DB_ERROR","Failed to get all tasks",n)}finally{t.close();}}getTask(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to get task",e)}finally{n.close();}}killTask(t){if(!this.dbExists())return {killed:false};let n=this.openHandle(true);try{let e=n.db.select({id:o.id,status:o.status,pid:o.pid}).from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get();if(!e||e.status!=="running")return {killed:!1};if(e.pid)try{process.kill(e.pid,"SIGTERM");}catch{}return n.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:e.pid??void 0}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to kill task",e)}finally{n.close();}}reapOrphanedTasks(){if(!this.dbExists())return 0;let t=this.openHandle(true);try{let n=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(),e=Date.now(),r=new Set,s=0;for(let i of n){if(i.pid){let c=!1;try{process.kill(i.pid,0),c=!0;}catch{}c||(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,i.id),drizzleOrm.eq(o.status,"running"))).run(),s++);continue}if(jn.has(i.platform??"cli"))continue;let l=`${i.id}:${i.run_epoch??0}`;r.add(l);let u=this.pidNullObservedSince.get(l);if(u===void 0){this.pidNullObservedSince.set(l,e);continue}e-u>=Re&&(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,i.id),drizzleOrm.eq(o.status,"running"))).run(),s++,this.pidNullObservedSince.delete(l));}for(let i of Array.from(this.pidNullObservedSince.keys()))r.has(i)||this.pidNullObservedSince.delete(i);return s}finally{t.close();}}reapRunningWorkflowTasks(t=Re){if(!this.dbExists())return 0;let n=this.openHandle(true);try{let e=new Date(Date.now()-t).toISOString();return n.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,e))).run().changes??0}finally{n.close();}}findTaskStatus(t,n){let e=this.resolveDbPaths();for(let r of e){if(!fe.existsSync(r))continue;let s=w(r);try{let i=n?drizzleOrm.eq(o.workspace_id,n):void 0,l=i?drizzleOrm.and(drizzleOrm.eq(o.id,t),i):drizzleOrm.eq(o.id,t),u=s.db.select().from(o).where(l).limit(1).get()??void 0;if(!u){let c=drizzleOrm.or(drizzleOrm.eq(o.thread_id,t),drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.like(o.command,`%--thread=${t}%`))),p=i?drizzleOrm.and(c,i):c;u=s.db.select().from(o).where(p).orderBy(drizzleOrm.desc(o.started_at)).limit(1).get()??void 0;}if(u)return u}catch(i){throw new d("DB_ERROR","Failed to find task status",i)}finally{s.close();}}}findChildTasks(t,n){let e=this.resolveDbPaths(),r=new Set,s=[];for(let i of e){if(!fe.existsSync(i))continue;let l=w(i);try{let u=n?drizzleOrm.and(drizzleOrm.eq(o.parent_task_id,t),drizzleOrm.eq(o.workspace_id,n)):drizzleOrm.eq(o.parent_task_id,t),c=l.db.select().from(o).where(u).orderBy(drizzleOrm.asc(o.started_at)).all();for(let p of c)r.has(p.id)||(r.add(p.id),s.push(p));}catch(u){throw new d("DB_ERROR","Failed to find child tasks",u)}finally{l.close();}}return s}getWorkspaceUsageSummary(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.all(t?drizzleOrm.sql`
|
|
12
12
|
SELECT
|
|
13
13
|
COALESCE(workspace_id, 'unknown') AS workspace_id,
|
|
14
14
|
COALESCE(SUM(input_tokens), 0) AS input_tokens,
|
|
@@ -29,7 +29,7 @@ ${e.join(`
|
|
|
29
29
|
FROM tasks
|
|
30
30
|
GROUP BY workspace_id
|
|
31
31
|
ORDER BY (COALESCE(SUM(input_tokens), 0) + COALESCE(SUM(output_tokens), 0)) DESC
|
|
32
|
-
`)}catch(e){throw new d("DB_ERROR","Failed to get workspace usage summary",e)}finally{n.close();}}getThreadTokenUsage(t,n){let e=this.resolveDbPaths(),r=new Set,s=0,i=0,l=0;for(let u of e){if(!
|
|
32
|
+
`)}catch(e){throw new d("DB_ERROR","Failed to get workspace usage summary",e)}finally{n.close();}}getThreadTokenUsage(t,n){let e=this.resolveDbPaths(),r=new Set,s=0,i=0,l=0;for(let u of e){if(!fe.existsSync(u))continue;let c=w(u);try{let p=drizzleOrm.or(drizzleOrm.eq(o.thread_id,t),drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.like(o.command,`%--thread=${t}%`))),h=n?drizzleOrm.and(p,drizzleOrm.eq(o.workspace_id,n)):p,f=c.db.select({id:o.id,input_tokens:o.input_tokens,output_tokens:o.output_tokens,cost_usd:o.cost_usd}).from(o).where(h).all();for(let S of f)r.has(S.id)||(r.add(S.id),s+=S.input_tokens??0,i+=S.output_tokens??0,l+=S.cost_usd??0);}catch(p){throw new d("DB_ERROR","Failed to get thread token usage",p)}finally{c.close();}}return {inputTokens:s,outputTokens:i,costUsd:l}}findTasksByThread(t,n){let e=this.resolveDbPaths(),r=new Set,s=[];for(let i of e){if(!fe.existsSync(i))continue;let l=w(i);try{let u=drizzleOrm.or(drizzleOrm.eq(o.thread_id,t),drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.like(o.command,`%--thread=${t}%`))),c=n?drizzleOrm.and(u,drizzleOrm.eq(o.workspace_id,n)):u,p=l.db.select().from(o).where(c).orderBy(drizzleOrm.asc(o.started_at)).all();for(let h of p)r.has(h.id)||(r.add(h.id),s.push(h));}catch(u){throw new d("DB_ERROR","Failed to find tasks by thread",u)}finally{l.close();}}return s}parseWorkflowCardMetadata(t){if(!t)return {};try{let n=JSON.parse(t);return {kind:typeof n.kind=="string"?n.kind:void 0,workflowRunId:typeof n.workflowRunId=="string"?n.workflowRunId:void 0}}catch{return {}}}extractThreadIdFromCommand(t){return t?/--thread=(\S+)/.exec(t)?.[1]??null:null}batchFetchWorkflowCards(t,n){let e=new Map;if(t.length===0)return e;let r=new Set(t),s=new Map,i=n?drizzleOrm.eq(o.workspace_id,n):void 0,l=(u,c)=>{if(!u)return;let p=s.get(u);if(p&&c.started_at<=p)return;let h=this.parseWorkflowCardMetadata(c.metadata);h.kind==="workflow_card"&&(s.set(u,c.started_at),e.set(u,{taskId:c.id,workflowRunId:h.workflowRunId,raw:c.result??void 0}));};for(let u of this.resolveDbPaths()){if(!fe.existsSync(u))continue;let c=w(u);try{let p=drizzleOrm.and(drizzleOrm.inArray(o.thread_id,t),drizzleOrm.like(o.metadata,'%"kind":"workflow_card"%')),h=c.db.select({id:o.id,thread_id:o.thread_id,metadata:o.metadata,result:o.result,started_at:o.started_at}).from(o).where(i?drizzleOrm.and(p,i):p).all();for(let I of h)l(I.thread_id,I);let f=drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.like(o.metadata,'%"kind":"workflow_card"%')),S=c.db.select({id:o.id,command:o.command,metadata:o.metadata,result:o.result,started_at:o.started_at}).from(o).where(i?drizzleOrm.and(f,i):f).all();for(let I of S){let L=this.extractThreadIdFromCommand(I.command);!L||!r.has(L)||l(L,I);}}catch(p){throw new d("DB_ERROR","Failed to batch fetch workflow cards",p)}finally{c.close();}}return e}findAllTasks(t){if(!this.dbExists())return {rows:[],total:0};let n=this.openHandle(false);try{let e=[];t.workspaceId&&e.push(drizzleOrm.eq(o.workspace_id,t.workspaceId));let r=t.agents&&t.agents.length>0?t.agents:t.agentId?[t.agentId]:null;r&&e.push(drizzleOrm.inArray(o.agent_id,r));let s=t.statuses&&t.statuses.length>0?t.statuses:t.status?[t.status]:null;s&&e.push(drizzleOrm.inArray(o.status,s));let i=t.q??t.search;i&&e.push(drizzleOrm.like(o.prompt,`%${i}%`)),t.from&&e.push(drizzleOrm.gte(o.started_at,t.from)),t.to&&e.push(drizzleOrm.lt(o.started_at,t.to));let l=e.length>0?drizzleOrm.and(...e):void 0,u=n.db.select({count:drizzleOrm.sql`count(*)`}).from(o).where(l).get(),c=(t.sortDir??"DESC")==="ASC"?drizzleOrm.asc(o.started_at):drizzleOrm.desc(o.started_at);return {rows:n.db.select().from(o).where(l).orderBy(c).limit(t.limit).offset(t.offset).all(),total:u?.count??0}}catch(e){throw new d("DB_ERROR","Failed to find all tasks",e)}finally{n.close();}}getAgentUsage(t,n,e){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.all(e?drizzleOrm.sql`
|
|
33
33
|
SELECT
|
|
34
34
|
t.agent_id,
|
|
35
35
|
t.workspace_id,
|
|
@@ -37,11 +37,11 @@ ${e.join(`
|
|
|
37
37
|
COALESCE(SUM(
|
|
38
38
|
COALESCE(t.input_tokens, 0)
|
|
39
39
|
+ CASE
|
|
40
|
-
WHEN t.started_at >= ${
|
|
40
|
+
WHEN t.started_at >= ${st}
|
|
41
41
|
AND (
|
|
42
42
|
t.crewx_version IS NULL
|
|
43
43
|
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, ${
|
|
44
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${it}) AS INTEGER) < ${at})
|
|
45
45
|
)
|
|
46
46
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
47
47
|
ELSE 0
|
|
@@ -60,11 +60,11 @@ ${e.join(`
|
|
|
60
60
|
COALESCE(SUM(
|
|
61
61
|
COALESCE(t.input_tokens, 0)
|
|
62
62
|
+ CASE
|
|
63
|
-
WHEN t.started_at >= ${
|
|
63
|
+
WHEN t.started_at >= ${st}
|
|
64
64
|
AND (
|
|
65
65
|
t.crewx_version IS NULL
|
|
66
66
|
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, ${
|
|
67
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${it}) AS INTEGER) < ${at})
|
|
68
68
|
)
|
|
69
69
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
70
70
|
ELSE 0
|
|
@@ -80,11 +80,11 @@ ${e.join(`
|
|
|
80
80
|
COALESCE(SUM(
|
|
81
81
|
COALESCE(t.input_tokens, 0)
|
|
82
82
|
+ CASE
|
|
83
|
-
WHEN t.started_at >= ${
|
|
83
|
+
WHEN t.started_at >= ${st}
|
|
84
84
|
AND (
|
|
85
85
|
t.crewx_version IS NULL
|
|
86
86
|
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, ${
|
|
87
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${it}) AS INTEGER) < ${at})
|
|
88
88
|
)
|
|
89
89
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
90
90
|
ELSE 0
|
|
@@ -102,11 +102,11 @@ ${e.join(`
|
|
|
102
102
|
COALESCE(SUM(
|
|
103
103
|
COALESCE(t.input_tokens, 0)
|
|
104
104
|
+ CASE
|
|
105
|
-
WHEN t.started_at >= ${
|
|
105
|
+
WHEN t.started_at >= ${st}
|
|
106
106
|
AND (
|
|
107
107
|
t.crewx_version IS NULL
|
|
108
108
|
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, ${
|
|
109
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${it}) AS INTEGER) < ${at})
|
|
110
110
|
)
|
|
111
111
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
112
112
|
ELSE 0
|
|
@@ -121,11 +121,11 @@ ${e.join(`
|
|
|
121
121
|
COALESCE(SUM(
|
|
122
122
|
COALESCE(t.input_tokens, 0)
|
|
123
123
|
+ CASE
|
|
124
|
-
WHEN t.started_at >= ${
|
|
124
|
+
WHEN t.started_at >= ${st}
|
|
125
125
|
AND (
|
|
126
126
|
t.crewx_version IS NULL
|
|
127
127
|
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, ${
|
|
128
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${it}) AS INTEGER) < ${at})
|
|
129
129
|
)
|
|
130
130
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
131
131
|
ELSE 0
|
|
@@ -148,11 +148,11 @@ ${e.join(`
|
|
|
148
148
|
COALESCE(SUM(
|
|
149
149
|
COALESCE(t.input_tokens, 0)
|
|
150
150
|
+ CASE
|
|
151
|
-
WHEN t.started_at >= ${
|
|
151
|
+
WHEN t.started_at >= ${st}
|
|
152
152
|
AND (
|
|
153
153
|
t.crewx_version IS NULL
|
|
154
154
|
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, ${
|
|
155
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${it}) AS INTEGER) < ${at})
|
|
156
156
|
)
|
|
157
157
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
158
158
|
ELSE 0
|
|
@@ -167,7 +167,7 @@ ${e.join(`
|
|
|
167
167
|
AND t.started_at < ${n}
|
|
168
168
|
GROUP BY date(t.started_at), t.agent_id
|
|
169
169
|
ORDER BY date(t.started_at) ASC
|
|
170
|
-
`).map(i=>({date:i.date,agentId:i.agent_id,inputTokens:i.input_tokens,outputTokens:i.output_tokens,cachedInputTokens:i.cached_input_tokens,costUsd:i.cost_usd,totalTokens:i.input_tokens+i.output_tokens}))}catch(s){throw new d("DB_ERROR","Failed to get agent usage trend",s)}finally{r.close();}}findTaskForStop(t,n){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(o).where(drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.workspace_id,n))).limit(1).get()??void 0}catch(r){throw new d("DB_ERROR","Failed to find task for stop",r)}finally{e.close();}}findLatestTaskByMetadata(t){if(!this.dbExists())return null;let n=this.openHandle(false);try{return n.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}).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(e){throw new d("DB_ERROR","Failed to find latest task by metadata",e)}finally{n.close();}}markTaskFailed(t,n,e){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=new Date().toISOString(),i=e?drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.status,"running"),drizzleOrm.eq(o.workspace_id,e)):drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.status,"running"));r.db.update(o).set({status:"failed",error:n,completed_at:s}).where(i).run();}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to mark task failed",s)}finally{r.close();}}findTasksByPromptHint(t,n){let e=this.resolveDbPaths(),r=new Set,s=[];for(let i of e){if(!
|
|
170
|
+
`).map(i=>({date:i.date,agentId:i.agent_id,inputTokens:i.input_tokens,outputTokens:i.output_tokens,cachedInputTokens:i.cached_input_tokens,costUsd:i.cost_usd,totalTokens:i.input_tokens+i.output_tokens}))}catch(s){throw new d("DB_ERROR","Failed to get agent usage trend",s)}finally{r.close();}}findTaskForStop(t,n){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(o).where(drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.workspace_id,n))).limit(1).get()??void 0}catch(r){throw new d("DB_ERROR","Failed to find task for stop",r)}finally{e.close();}}findLatestTaskByMetadata(t){if(!this.dbExists())return null;let n=this.openHandle(false);try{return n.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}).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(e){throw new d("DB_ERROR","Failed to find latest task by metadata",e)}finally{n.close();}}markTaskFailed(t,n,e){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=new Date().toISOString(),i=e?drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.status,"running"),drizzleOrm.eq(o.workspace_id,e)):drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.status,"running"));r.db.update(o).set({status:"failed",error:n,completed_at:s}).where(i).run();}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to mark task failed",s)}finally{r.close();}}findTasksByPromptHint(t,n){let e=this.resolveDbPaths(),r=new Set,s=[];for(let i of e){if(!fe.existsSync(i))continue;let l=w(i);try{let u=n?drizzleOrm.and(drizzleOrm.like(o.prompt,`%${t}%`),drizzleOrm.eq(o.workspace_id,n)):drizzleOrm.like(o.prompt,`%${t}%`),c=l.db.select().from(o).where(u).orderBy(drizzleOrm.asc(o.started_at)).all();for(let p of c)r.has(p.id)||(r.add(p.id),s.push(p));}catch(u){throw new d("DB_ERROR","Failed to find tasks by prompt hint",u)}finally{l.close();}}return s}getProviderUsage(t,n,e){if(!this.dbExists())return [];let r=this.openHandle(false);try{let s=drizzleOrm.sql`
|
|
171
171
|
CASE
|
|
172
172
|
WHEN ${o.model} LIKE 'claude-%' OR ${o.model} IN ('opus', 'sonnet', 'haiku', 'opus[1m]', 'sonnet[1m]') THEN 'claude'
|
|
173
173
|
WHEN ${o.model} LIKE 'gpt-%' OR ${o.model} LIKE 'codex-%' THEN 'codex'
|
|
@@ -180,11 +180,11 @@ ${e.join(`
|
|
|
180
180
|
`,i=drizzleOrm.sql`
|
|
181
181
|
COALESCE(${o.input_tokens}, 0)
|
|
182
182
|
+ CASE
|
|
183
|
-
WHEN ${o.started_at} >= ${
|
|
183
|
+
WHEN ${o.started_at} >= ${st}
|
|
184
184
|
AND (
|
|
185
185
|
${o.crewx_version} IS NULL
|
|
186
186
|
OR (${o.crewx_version} LIKE '0.8.%' AND ${o.crewx_version} NOT LIKE '0.8.9%')
|
|
187
|
-
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${
|
|
187
|
+
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${it}) AS INTEGER) < ${at})
|
|
188
188
|
)
|
|
189
189
|
THEN COALESCE(${o.cached_input_tokens}, 0)
|
|
190
190
|
ELSE 0
|
|
@@ -216,11 +216,11 @@ ${e.join(`
|
|
|
216
216
|
`,i=drizzleOrm.sql`
|
|
217
217
|
COALESCE(${o.input_tokens}, 0)
|
|
218
218
|
+ CASE
|
|
219
|
-
WHEN ${o.started_at} >= ${
|
|
219
|
+
WHEN ${o.started_at} >= ${st}
|
|
220
220
|
AND (
|
|
221
221
|
${o.crewx_version} IS NULL
|
|
222
222
|
OR (${o.crewx_version} LIKE '0.8.%' AND ${o.crewx_version} NOT LIKE '0.8.9%')
|
|
223
|
-
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${
|
|
223
|
+
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${it}) AS INTEGER) < ${at})
|
|
224
224
|
)
|
|
225
225
|
THEN COALESCE(${o.cached_input_tokens}, 0)
|
|
226
226
|
ELSE 0
|
|
@@ -239,7 +239,7 @@ ${e.join(`
|
|
|
239
239
|
FROM ${o}
|
|
240
240
|
${l}
|
|
241
241
|
GROUP BY ${o.model}
|
|
242
|
-
`),c=new Map;for(let p of u){let h=
|
|
242
|
+
`),c=new Map;for(let p of u){let h=ke(p.model??""),f=c.get(h);f?(f.totalTasks+=p.total_tasks,f.inputTokens+=p.input_tokens,f.outputTokens+=p.output_tokens,f.cachedInputTokens+=p.cached_input_tokens,f.costUsd+=p.cost_usd,f.totalTokens+=p.input_tokens+p.output_tokens,f.activeDurationMs+=p.active_duration_ms??0,p.last_active_at&&(!f.lastActiveAt||p.last_active_at>f.lastActiveAt)&&(f.lastActiveAt=p.last_active_at)):c.set(h,{model:h,provider:p.provider,totalTasks:p.total_tasks,inputTokens:p.input_tokens,outputTokens:p.output_tokens,cachedInputTokens:p.cached_input_tokens,costUsd:p.cost_usd,totalTokens:p.input_tokens+p.output_tokens,activeDurationMs:p.active_duration_ms??0,lastActiveAt:p.last_active_at??null});}return Array.from(c.values()).sort((p,h)=>h.costUsd-p.costUsd)}catch(s){throw new d("DB_ERROR","Failed to get model usage",s)}finally{r.close();}}getTasksForReport(t,n,e){if(!this.dbExists())return [];let r=this.openHandle(false);try{let s=e?drizzleOrm.sql`WHERE ${o.started_at} >= ${t} AND ${o.started_at} < ${n} AND ${o.workspace_id} = ${e}`:drizzleOrm.sql`WHERE ${o.started_at} >= ${t} AND ${o.started_at} < ${n}`;return r.db.all(drizzleOrm.sql`
|
|
243
243
|
SELECT
|
|
244
244
|
${o.thread_id} AS thread_id,
|
|
245
245
|
${o.agent_id} AS agent_id,
|
|
@@ -250,11 +250,11 @@ ${e.join(`
|
|
|
250
250
|
FROM ${o}
|
|
251
251
|
${s}
|
|
252
252
|
ORDER BY ${o.started_at} ASC
|
|
253
|
-
`).map(l=>({threadId:l.thread_id,agentId:l.agent_id,startedAt:l.started_at,status:l.status,error:l.error,exitCode:l.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 n=this.openHandle(false);try{let e=[];t.workspaceId&&e.push(drizzleOrm.eq(o.workspace_id,t.workspaceId)),t.agents&&t.agents.length>0&&e.push(drizzleOrm.inArray(o.agent_id,t.agents)),t.statuses&&t.statuses.length>0&&e.push(drizzleOrm.inArray(o.status,t.statuses)),t.from&&e.push(drizzleOrm.gte(o.started_at,t.from)),t.to&&e.push(drizzleOrm.lt(o.started_at,t.to));let r=e.length>0?drizzleOrm.and(...e):void 0,s=(t.order??"DESC")==="ASC"?drizzleOrm.asc(o.started_at):drizzleOrm.desc(o.started_at),i={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?n.db.select(i).from(o).where(r).orderBy(s).limit(t.limit).all():n.db.select(i).from(o).where(r).orderBy(s).all()}catch(e){throw new d("DB_ERROR","Failed to find task evidence",e)}finally{n.close();}}findThreadChains(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{let e=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.notLike(o.metadata,'%"kind":"workflow_card"%')),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 i=n.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(),l=new Map;for(let c of i){if(!c.thread_id)continue;let p={id:c.id,agentId:c.agent_id,status:c.status,prompt:c.prompt,startedAt:c.started_at,completedAt:c.completed_at},h=l.get(c.thread_id);h?h.push(p):l.set(c.thread_id,[p]);}let u=[];for(let[c,p]of l){let h=p.filter(f=>f.agentId===t.targetAgentId).length;h<2||u.push({threadId:c,targetAgentTurnCount:h,turns:p});}return u.sort((c,p)=>{let h=c.turns[c.turns.length-1].startedAt,f=p.turns[p.turns.length-1].startedAt;return h<f?1:h>f?-1:0}),u.slice(0,e).map(c=>({threadId:c.threadId,targetAgentTurnCount:c.targetAgentTurnCount,turns:c.turns.slice(-r)}))}catch(e){throw new d("DB_ERROR","Failed to find thread chains",e)}finally{n.close();}}};var
|
|
253
|
+
`).map(l=>({threadId:l.thread_id,agentId:l.agent_id,startedAt:l.started_at,status:l.status,error:l.error,exitCode:l.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 n=this.openHandle(false);try{let e=[];t.workspaceId&&e.push(drizzleOrm.eq(o.workspace_id,t.workspaceId)),t.agents&&t.agents.length>0&&e.push(drizzleOrm.inArray(o.agent_id,t.agents)),t.statuses&&t.statuses.length>0&&e.push(drizzleOrm.inArray(o.status,t.statuses)),t.from&&e.push(drizzleOrm.gte(o.started_at,t.from)),t.to&&e.push(drizzleOrm.lt(o.started_at,t.to));let r=e.length>0?drizzleOrm.and(...e):void 0,s=(t.order??"DESC")==="ASC"?drizzleOrm.asc(o.started_at):drizzleOrm.desc(o.started_at),i={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?n.db.select(i).from(o).where(r).orderBy(s).limit(t.limit).all():n.db.select(i).from(o).where(r).orderBy(s).all()}catch(e){throw new d("DB_ERROR","Failed to find task evidence",e)}finally{n.close();}}findThreadChains(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{let e=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.notLike(o.metadata,'%"kind":"workflow_card"%')),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 i=n.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(),l=new Map;for(let c of i){if(!c.thread_id)continue;let p={id:c.id,agentId:c.agent_id,status:c.status,prompt:c.prompt,startedAt:c.started_at,completedAt:c.completed_at},h=l.get(c.thread_id);h?h.push(p):l.set(c.thread_id,[p]);}let u=[];for(let[c,p]of l){let h=p.filter(f=>f.agentId===t.targetAgentId).length;h<2||u.push({threadId:c,targetAgentTurnCount:h,turns:p});}return u.sort((c,p)=>{let h=c.turns[c.turns.length-1].startedAt,f=p.turns[p.turns.length-1].startedAt;return h<f?1:h>f?-1:0}),u.slice(0,e).map(c=>({threadId:c.threadId,targetAgentTurnCount:c.targetAgentTurnCount,turns:c.turns.slice(-r)}))}catch(e){throw new d("DB_ERROR","Failed to find thread chains",e)}finally{n.close();}}};var Wt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=Q.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}validateWorkspaceId(t,n){return t.db.select({id:m.id}).from(m).where(drizzleOrm.eq(m.id,n)).limit(1).get()?n:null}topLevelTaskPredicateSql(t="child"){return drizzleOrm.sql.raw(`NOT EXISTS (
|
|
254
254
|
SELECT 1 FROM tasks parent
|
|
255
255
|
WHERE parent.id = ${t}.parent_task_id
|
|
256
256
|
AND parent.thread_id = ${t}.thread_id
|
|
257
|
-
)`)}findAllThreads(t){let n=this.resolveDbPaths(),e=new Set,r=[];for(let s of n){if(!
|
|
257
|
+
)`)}findAllThreads(t){let n=this.resolveDbPaths(),e=new Set,r=[];for(let s of n){if(!fe.existsSync(s))continue;let i=w(s);try{let l=t?drizzleOrm.eq(_.workspace_id,t):void 0,u=i.db.select().from(_).where(l).orderBy(drizzleOrm.desc(_.updated_at)).all();for(let c of u)e.has(c.id)||(e.add(c.id),r.push(c));}catch(l){throw new d("DB_ERROR","Failed to find all threads",l)}finally{i.close();}}return r}findThreadsByIdsOrTitles(t,n){if(t.length===0)return [];let e=this.resolveDbPaths(),r=new Set,s=[];for(let i of e){if(!fe.existsSync(i))continue;let l=w(i);try{let u=drizzleOrm.or(drizzleOrm.inArray(_.id,t),drizzleOrm.inArray(_.title,t)),c=n?drizzleOrm.and(u,drizzleOrm.eq(_.workspace_id,n)):u,p=l.db.select().from(_).where(c).orderBy(drizzleOrm.desc(_.updated_at)).all();for(let h of p)r.has(h.id)||(r.add(h.id),s.push(h));}catch(u){throw new d("DB_ERROR","Failed to find threads by ids or titles",u)}finally{l.close();}}return s}findThreadById(t,n){let e=this.resolveDbPaths();for(let r of e){if(!fe.existsSync(r))continue;let s=w(r);try{let i=drizzleOrm.eq(_.id,t),l=n?drizzleOrm.and(i,drizzleOrm.eq(_.workspace_id,n)):i,u=s.db.select().from(_).where(l).limit(1).get()??void 0;if(u)return u}catch(i){throw new d("DB_ERROR","Failed to find thread by id",i)}finally{s.close();}}}threadExists(t,n){let e=this.resolveDbPaths();for(let r of e){if(!fe.existsSync(r))continue;let s=w(r);try{let i=drizzleOrm.eq(_.id,t),l=n?drizzleOrm.and(i,drizzleOrm.eq(_.workspace_id,n)):i;if(s.db.select({id:_.id}).from(_).where(l).limit(1).get())return !0}catch(i){throw new d("DB_ERROR","Failed to check thread existence",i)}finally{s.close();}}return false}aggregateTaskStats(t,n){let e=this.resolveDbPaths(),r=0,s=0,i=0,l=0,u=0,c=new Set;for(let p of e){if(!fe.existsSync(p))continue;let h=w(p);try{let f=h.db.get(drizzleOrm.sql`
|
|
258
258
|
SELECT
|
|
259
259
|
count(*) AS cnt,
|
|
260
260
|
COALESCE(SUM(child.input_tokens), 0) AS total_input,
|
|
@@ -265,13 +265,13 @@ ${e.join(`
|
|
|
265
265
|
WHERE child.thread_id = ${t}
|
|
266
266
|
AND ${this.topLevelTaskPredicateSql()}
|
|
267
267
|
${n?drizzleOrm.sql`AND child.workspace_id = ${n}`:drizzleOrm.sql``}
|
|
268
|
-
`);f&&(r+=f.cnt,s+=f.total_input,i+=f.total_output,l+=f.total_cached,u+=f.total_cost);let
|
|
268
|
+
`);f&&(r+=f.cnt,s+=f.total_input,i+=f.total_output,l+=f.total_cached,u+=f.total_cost);let S=h.db.all(drizzleOrm.sql`
|
|
269
269
|
SELECT DISTINCT child.agent_id FROM tasks child
|
|
270
270
|
WHERE child.thread_id = ${t}
|
|
271
271
|
AND child.agent_id IS NOT NULL AND child.agent_id != ''
|
|
272
272
|
AND ${this.topLevelTaskPredicateSql()}
|
|
273
273
|
${n?drizzleOrm.sql`AND child.workspace_id = ${n}`:drizzleOrm.sql``}
|
|
274
|
-
`);for(let I of
|
|
274
|
+
`);for(let I of S)c.add(I.agent_id);}catch(f){throw new d("DB_ERROR","Failed to aggregate task stats",f)}finally{h.close();}}return {taskCount:r,inputTokens:s,outputTokens:i,cachedInputTokens:l,costUsd:u,agentIds:Array.from(c)}}findTopLevelTasks(t,n,e){let r=this.resolveDbPaths(),s=new Set,i=[];for(let l of r){if(!fe.existsSync(l))continue;let u=w(l);try{let c;e!==void 0?c=u.db.all(drizzleOrm.sql`
|
|
275
275
|
SELECT * FROM (
|
|
276
276
|
SELECT child.*,
|
|
277
277
|
ROW_NUMBER() OVER (PARTITION BY child.agent_id ORDER BY child.started_at DESC) AS rn
|
|
@@ -288,7 +288,7 @@ ${e.join(`
|
|
|
288
288
|
AND ${this.topLevelTaskPredicateSql()}
|
|
289
289
|
${n?drizzleOrm.sql`AND child.workspace_id = ${n}`:drizzleOrm.sql``}
|
|
290
290
|
ORDER BY child.started_at ASC
|
|
291
|
-
`);for(let p of c)s.has(p.id)||(s.add(p.id),i.push(p));}catch(c){throw new d("DB_ERROR","Failed to find top-level tasks",c)}finally{u.close();}}if(e!==void 0&&r.length>1){let l=new Map;for(let u of i){let c=u.agent_id??"";l.has(c)||l.set(c,[]),l.get(c).push(u);}i=[];for(let u of l.values())u.sort((c,p)=>(p.started_at??"").localeCompare(c.started_at??"")),i.push(...u.slice(0,e));i.sort((u,c)=>(u.started_at??"").localeCompare(c.started_at??""));}return i}findAllTasks(t,n){let e=this.resolveDbPaths(),r=new Set,s=[];for(let i of e){if(!
|
|
291
|
+
`);for(let p of c)s.has(p.id)||(s.add(p.id),i.push(p));}catch(c){throw new d("DB_ERROR","Failed to find top-level tasks",c)}finally{u.close();}}if(e!==void 0&&r.length>1){let l=new Map;for(let u of i){let c=u.agent_id??"";l.has(c)||l.set(c,[]),l.get(c).push(u);}i=[];for(let u of l.values())u.sort((c,p)=>(p.started_at??"").localeCompare(c.started_at??"")),i.push(...u.slice(0,e));i.sort((u,c)=>(u.started_at??"").localeCompare(c.started_at??""));}return i}findAllTasks(t,n){let e=this.resolveDbPaths(),r=new Set,s=[];for(let i of e){if(!fe.existsSync(i))continue;let l=w(i);try{let u=drizzleOrm.eq(o.thread_id,t),c=n?drizzleOrm.and(u,drizzleOrm.eq(o.workspace_id,n)):u,p=l.db.select().from(o).where(c).orderBy(drizzleOrm.asc(o.started_at)).all();for(let h of p)r.has(h.id)||(r.add(h.id),s.push(h));}catch(u){throw new d("DB_ERROR","Failed to find all tasks for thread",u)}finally{l.close();}}return s}findTaskById(t,n,e){let r=this.resolveDbPaths();for(let s of r){if(!fe.existsSync(s))continue;let i=w(s);try{let l=drizzleOrm.and(drizzleOrm.eq(o.id,n),drizzleOrm.eq(o.thread_id,t)),u=e?drizzleOrm.and(l,drizzleOrm.eq(o.workspace_id,e)):l,c=i.db.select().from(o).where(u).limit(1).get();if(!c)continue;let p=i.db.select().from(o).where(drizzleOrm.eq(o.parent_task_id,c.id)).orderBy(drizzleOrm.asc(o.started_at)).all();return {task:c,children:p}}catch(l){throw new d("DB_ERROR","Failed to find task by id",l)}finally{i.close();}}}batchFetchTasksQuery(t,n){return drizzleOrm.sql`
|
|
292
292
|
SELECT child.id, child.thread_id, child.agent_id, child.status,
|
|
293
293
|
child.parent_task_id, child.started_at, child.completed_at,
|
|
294
294
|
child.duration_ms, child.input_tokens, child.output_tokens,
|
|
@@ -298,7 +298,7 @@ ${e.join(`
|
|
|
298
298
|
AND ${this.topLevelTaskPredicateSql()}
|
|
299
299
|
${n?drizzleOrm.sql`AND child.workspace_id = ${n}`:drizzleOrm.sql``}
|
|
300
300
|
ORDER BY child.started_at ASC
|
|
301
|
-
`}batchFetchTasks(t,n){let e=new Map;if(t.length===0)return e;let r=this.resolveDbPaths();for(let s of r){if(!
|
|
301
|
+
`}batchFetchTasks(t,n){let e=new Map;if(t.length===0)return e;let r=this.resolveDbPaths();for(let s of r){if(!fe.existsSync(s))continue;let i=w(s);try{let l=i.db.all(this.batchFetchTasksQuery(t,n));for(let u of l){let c=u.thread_id;e.has(c)||e.set(c,[]),e.get(c).push(u);}}catch(l){throw new d("DB_ERROR","Failed to batch fetch tasks",l)}finally{i.close();}}return e}explainBatchFetchTasksPlan(t,n){let e=this.resolveDbPaths();for(let r of e){if(!fe.existsSync(r))continue;let s=w(r);try{return s.db.all(drizzleOrm.sql`EXPLAIN QUERY PLAN ${this.batchFetchTasksQuery(t,n)}`).map(l=>l.detail)}finally{s.close();}}return []}updateThreadTitle(t,n,e){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=drizzleOrm.eq(_.id,t),i=e?drizzleOrm.and(s,drizzleOrm.eq(_.workspace_id,e)):s;if(!r.db.select({id:_.id}).from(_).where(i).limit(1).get())return;r.db.update(_).set({title:n,title_locked:1,updated_at:new Date().toISOString()}).where(drizzleOrm.eq(_.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,n){let e=this.openHandle(true);try{let r=n.workspaceId?this.validateWorkspaceId(e,n.workspaceId):null,s=new Date().toISOString();if(e.db.select({id:_.id,message_count:_.message_count}).from(_).where(drizzleOrm.eq(_.id,t)).limit(1).get()){let l={updated_at:s};n.title!==void 0&&(l.title=n.title),n.titleLocked!==void 0&&(l.title_locked=n.titleLocked?1:0),e.db.update(_).set(l).where(drizzleOrm.eq(_.id,t)).run();}else e.db.insert(_).values({id:t,platform:n.platform,workspace_id:r,title:n.title??null,title_locked:n.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{e.close();}}ensureThread(t,n,e){let r=this.openHandle(true);try{let s=e?this.validateWorkspaceId(r,e):null,i=r.db.select({id:_.id,platform:_.platform,workspace_id:_.workspace_id}).from(_).where(drizzleOrm.eq(_.id,t)).limit(1).get();if(i){s&&!i.workspace_id&&r.db.update(_).set({workspace_id:s}).where(drizzleOrm.eq(_.id,t)).run();return}let l=new Date().toISOString();r.db.insert(_).values({id:t,platform:n,workspace_id:s,message_count:0,created_at:l,updated_at:l}).run();}catch(s){throw s instanceof d?s:new d("DB_ERROR","Failed to ensure thread",s)}finally{r.close();}}saveUserMessage(t,n,e){if(!this.dbExists())return {firstMessage:false};let r=this.openHandle(true);try{let s=new Date().toISOString();return {firstMessage:r.db.transaction(l=>{let c=l.select({message_count:_.message_count}).from(_).where(drizzleOrm.eq(_.id,t)).limit(1).get()?.message_count===0;return l.run(drizzleOrm.sql`
|
|
302
302
|
UPDATE threads
|
|
303
303
|
SET first_message = COALESCE(first_message, ${n}),
|
|
304
304
|
title = CASE WHEN title_locked = 0 AND title IS NULL THEN substr(${n}, 1, 60) ELSE title END,
|
|
@@ -306,4 +306,4 @@ ${e.join(`
|
|
|
306
306
|
message_count = message_count + 1,
|
|
307
307
|
updated_at = ${s}
|
|
308
308
|
WHERE id = ${t}
|
|
309
|
-
`),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,n,e){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=new Date().toISOString();r.db.update(_).set({last_message:n,updated_at:s}).where(drizzleOrm.eq(_.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,n){if(!this.dbExists())return;let e=this.openHandle(true);try{let r={updated_at:new Date().toISOString()};n.title!==void 0&&(r.title=n.title,r.title_locked=1),n.titleLocked!==void 0&&(r.title_locked=n.titleLocked?1:0),e.db.update(_).set(r).where(drizzleOrm.eq(_.id,t)).run();}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to update thread",r)}finally{e.close();}}togglePin(t,n){let e=this.openHandle(true);try{let r=n?drizzleOrm.and(drizzleOrm.eq(_.id,t),drizzleOrm.eq(_.workspace_id,n)):drizzleOrm.eq(_.id,t),s=e.db.select({pinned:_.pinned,metadata:_.metadata}).from(_).where(r).get();if(!s)return null;let i=s.pinned?0:1,l=s.metadata?JSON.parse(s.metadata):{};if(i){let u=n?drizzleOrm.and(drizzleOrm.eq(_.pinned,1),drizzleOrm.eq(_.workspace_id,n)):drizzleOrm.eq(_.pinned,1),c=e.db.select({metadata:_.metadata}).from(_).where(u).all(),p=null;for(let h of c){let f=h.metadata?JSON.parse(h.metadata):{};typeof f.pinOrder=="number"&&(p===null||f.pinOrder<p)&&(p=f.pinOrder);}l.pinOrder=p===null?0:p-1;}else delete l.pinOrder;return e.db.update(_).set({pinned:i,metadata:Object.keys(l).length>0?JSON.stringify(l):null}).where(r).run(),{pinned:!!i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle pin",r)}finally{e.close();}}reorderPins(t,n){let e=this.openHandle(true);try{for(let r=0;r<t.length;r++){let s=n?drizzleOrm.and(drizzleOrm.eq(_.id,t[r]),drizzleOrm.eq(_.workspace_id,n)):drizzleOrm.eq(_.id,t[r]),i=e.db.select({metadata:_.metadata}).from(_).where(s).get();if(!i)continue;let l=i.metadata?JSON.parse(i.metadata):{};l.pinOrder=r+1,e.db.update(_).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{e.close();}}toggleStar(t,n){let e=this.openHandle(true);try{let r=n?drizzleOrm.and(drizzleOrm.eq(_.id,t),drizzleOrm.eq(_.workspace_id,n)):drizzleOrm.eq(_.id,t),s=e.db.select({starred:_.starred}).from(_).where(r).get();if(!s)return null;let i=s.starred?0:1;return e.db.update(_).set({starred:i}).where(r).run(),{starred:!!i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle star",r)}finally{e.close();}}resolveOverdriveForRequest(t,n,e,r){let i=this.openHandle(true);try{let l=n?drizzleOrm.and(drizzleOrm.eq(_.id,t),drizzleOrm.eq(_.workspace_id,n)):drizzleOrm.eq(_.id,t);return i.db.transaction(c=>{let p=c.select({metadata:_.metadata}).from(_).where(l).get();if(!p)return null;let h=p.metadata?JSON.parse(p.metadata):{},f=h.overdrive??{},y=r?.defaultTurns??f.defaultTurns??3,I=r?.updatedBy??"ui",L=new Date().toISOString(),U,D,K,gt;switch(e){case "enable-count":{let wt=r?.turns??y,kt=r?.consumeCurrent?Math.max(wt-1,0):wt;U=kt>0?"count":"off",D=kt,K=!0,gt="count";break}case "enable-latch":{U="latch",D=0,K=!0,gt="latch";break}case "disable":{U="off",D=0,K=!1;break}default:{let wt=f.state??"off",kt=typeof f.remaining=="number"?f.remaining:0;if(wt==="count"&&kt>0){let te=kt-1;te<=0?(U="off",D=0):(U="count",D=te),K=!0,gt="count";}else wt==="latch"?(U="latch",D=0,K=!0,gt="latch"):(U="off",D=0,K=!1);break}}return h.overdrive={state:U,remaining:D,defaultTurns:y,updatedAt:L,updatedBy:I},c.update(_).set({metadata:JSON.stringify(h),updated_at:L}).where(l).run(),{applied:K,appliedMode:gt,state:U,remaining:D}},{behavior:"immediate"})}catch(l){throw l instanceof d?l:new d("DB_ERROR","Failed to resolve overdrive",l)}finally{i.close();}}};var Wt=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=V.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=V.dirname(n);he.existsSync(r)||he.mkdirSync(r,{recursive:true});}else if(!he.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{N(e.db,n);}catch(r){throw e.close(),r}return e}insertSpan(t){let n=this.openHandle(true);try{n.db.insert(Z).values(t).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert span",e)}finally{n.close();}}findByTaskId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(Z).where(drizzleOrm.eq(Z.task_id,t)).all()}catch(e){throw new d("DB_ERROR","Failed to find spans by task id",e)}finally{n.close();}}findById(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(Z).where(drizzleOrm.eq(Z.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find span by id",e)}finally{n.close();}}};var Kt=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=V.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=V.dirname(n);he.existsSync(r)||he.mkdirSync(r,{recursive:true});}else if(!he.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{N(e.db,n);}catch(r){throw e.close(),r}return e}insertToolCall(t){let n=this.openHandle(true);try{n.db.insert(Y).values(t).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert tool call",e)}finally{n.close();}}findByTaskId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(Y).where(drizzleOrm.eq(Y.task_id,t)).all()}catch(e){throw new d("DB_ERROR","Failed to find tool calls by task id",e)}finally{n.close();}}aggregateByName(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select({toolName:Y.tool_name,count:drizzleOrm.sql`count(*)`}).from(Y).where(drizzleOrm.eq(Y.task_id,t)).groupBy(Y.tool_name).all().map(e=>({toolName:e.toolName,count:e.count}))}catch(e){throw new d("DB_ERROR","Failed to aggregate tool calls by name",e)}finally{n.close();}}};var Gt=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=V.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=V.dirname(n);he.existsSync(r)||he.mkdirSync(r,{recursive:true});}else if(!he.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{N(e.db,n);}catch(r){throw e.close(),r}return e}ensureThreadExists(t,n){if(!t.db.select({id:_.id}).from(_).where(drizzleOrm.eq(_.id,n)).limit(1).get())throw new d("NOT_FOUND",`Thread not found: ${n}`)}findByThreadId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return this.ensureThreadExists(n,t),n.db.select().from(q).where(drizzleOrm.eq(q.thread_id,t)).orderBy(drizzleOrm.asc(q.seq)).all()}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to find boxes by thread id",e)}finally{n.close();}}findById(t,n){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(q).where(drizzleOrm.and(drizzleOrm.eq(q.id,n),drizzleOrm.eq(q.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{e.close();}}insert(t){let n=this.openHandle(true);try{this.ensureThreadExists(n,t.threadId);try{n.db.insert(q).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 e=n.db.select().from(q).where(drizzleOrm.eq(q.id,t.id)).limit(1).get();if(!e)throw new d("DB_ERROR","Insert did not return a row");return e}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert thread box",e)}finally{n.close();}}};var Yt=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=V.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=V.dirname(n);he.existsSync(r)||he.mkdirSync(r,{recursive:true});}else if(!he.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{N(e.db,n);}catch(r){throw e.close(),r}return e}bulkInsert(t){if(t.length===0)return;let n=this.openHandle(true);try{let e=new Date().toISOString();n.db.transaction(r=>{for(let s of t)r.insert(_t).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:e,metadata:s.metadata??null}).run();});}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to bulk insert request logs",e)}finally{n.close();}}findRecent(t=100){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(_t).orderBy(drizzleOrm.desc(_t.timestamp)).limit(t).all()}catch(e){throw new d("DB_ERROR","Failed to find recent request logs",e)}finally{n.close();}}};function Ce(a){return new Date(Math.floor(a/3e5)*3e5).toISOString()}var Xt=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=V.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=V.dirname(n);he.existsSync(r)||he.mkdirSync(r,{recursive:true});}else if(!he.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{N(e.db,n);}catch(r){throw e.close(),r}return e}upsert(t){let n=this.openHandle(true);try{let e=Ce(t.bucketStartMs),r=t.accountRef??"default";n.db.insert(v).values({id:t.id,provider:t.provider,account_ref:r,limit_window:t.limitWindow,bucket_start:e,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:[v.provider,v.account_ref,v.limit_window,v.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(e){throw e instanceof d?e:new d("DB_ERROR","Failed to upsert usage limit snapshot",e)}finally{n.close();}}getTrends(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{let e=n.db.select({provider:v.provider,bucket_start:v.bucket_start,used_percent:v.used_percent,remaining_percent:v.remaining_percent,resets_at:v.resets_at}).from(v).where(drizzleOrm.and(drizzleOrm.eq(v.limit_window,t.window),drizzleOrm.gte(v.bucket_start,t.fromIso),drizzleOrm.lte(v.bucket_start,t.toIso))).orderBy(drizzleOrm.asc(v.bucket_start)).all(),r=new Map;for(let s of e)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,i])=>({provider:s,points:i}))}catch(e){throw new d("DB_ERROR","Failed to get usage limit trends",e)}finally{n.close();}}};var Jt=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=V.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=V.dirname(n);he.existsSync(r)||he.mkdirSync(r,{recursive:true});}else if(!he.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{N(e.db,n);}catch(r){throw e.close(),r}return e}publish(t){let n=this.openHandle(true);try{n.db.insert(H).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:[H.workspace_id,H.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 e=n.db.select().from(H).where(drizzleOrm.and(drizzleOrm.eq(H.workspace_id,t.workspaceId),drizzleOrm.eq(H.month,t.month))).limit(1).get();if(!e)throw new d("DB_ERROR","Failed to read back published usage report");return e}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to publish usage report",e)}finally{n.close();}}findByMonth(t,n){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(H).where(drizzleOrm.and(drizzleOrm.eq(H.workspace_id,t),drizzleOrm.eq(H.month,n))).limit(1).get()??void 0}catch(r){throw new d("DB_ERROR","Failed to find usage report",r)}finally{e.close();}}listByWorkspace(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(H).where(drizzleOrm.eq(H.workspace_id,t)).orderBy(drizzleOrm.desc(H.month)).all()}catch(e){throw new d("DB_ERROR","Failed to list usage reports",e)}finally{n.close();}}};var Le="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";function yr(a){let t=crypto.randomBytes(a*2),n="";for(let e=0;e<t.length&&n.length<a;e++){let r=t[e];r<248&&(n+=Le[r%62]);}for(;n.length<a;)n+=Le[crypto.randomBytes(1)[0]%62];return n}function Pt(a){return `${a}_${yr(8)}`}var Me=500,ze=50,Vt=class extends S{dbPath;cap;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=V.join(t.dbRoot,".crewx","crewx.db")),this.cap=t.cap??Me;}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=V.dirname(n);he.existsSync(r)||he.mkdirSync(r,{recursive:true});}else if(!he.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{N(e.db,n);}catch(r){throw e.close(),r}return e}create(t){let n=this.openHandle(true);try{let e=Pt("ntf"),r=new Date().toISOString();return n.db.insert(k).values({id:e,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(n,t.workspaceId),{id:e}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to create notification",e)}finally{n.close();}}enforceCap(t,n){let s=(t.db.select({count:drizzleOrm.sql`count(*)`}).from(k).where(drizzleOrm.eq(k.workspace_id,n)).get()?.count??0)-this.cap;if(s<=0)return;let l=t.db.select({id:k.id}).from(k).innerJoin(O,drizzleOrm.eq(O.notification_id,k.id)).where(drizzleOrm.eq(k.workspace_id,n)).groupBy(k.id).orderBy(drizzleOrm.asc(k.created_at)).limit(s).all().map(u=>u.id);l.length!==0&&(t.db.delete(O).where(drizzleOrm.inArray(O.notification_id,l)).run(),t.db.delete(k).where(drizzleOrm.inArray(k.id,l)).run());}list(t){if(!this.dbExists())return [];let n=t.limit??ze,e=this.openHandle(false);try{return e.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:O.read_at}).from(k).leftJoin(O,drizzleOrm.and(drizzleOrm.eq(O.notification_id,k.id),drizzleOrm.eq(O.user_id,t.userId))).where(drizzleOrm.eq(k.workspace_id,t.workspaceId)).orderBy(drizzleOrm.desc(k.created_at)).limit(n).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{e.close();}}unreadCount(t){if(!this.dbExists())return 0;let n=this.openHandle(false);try{return n.db.select({count:drizzleOrm.sql`count(*)`}).from(k).leftJoin(O,drizzleOrm.and(drizzleOrm.eq(O.notification_id,k.id),drizzleOrm.eq(O.user_id,t.userId))).where(drizzleOrm.and(drizzleOrm.eq(k.workspace_id,t.workspaceId),drizzleOrm.isNull(O.notification_id))).get()?.count??0}catch(e){throw new d("DB_ERROR","Failed to count unread notifications",e)}finally{n.close();}}markRead(t){let n=this.openHandle(true);try{if(!n.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 n.db.insert(O).values({notification_id:t.notificationId,user_id:t.userId,read_at:new Date().toISOString()}).onConflictDoNothing().run().changes>0}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to mark notification read",e)}finally{n.close();}}markAllRead(t){let n=this.openHandle(true);try{let e=n.db.select({id:k.id}).from(k).leftJoin(O,drizzleOrm.and(drizzleOrm.eq(O.notification_id,k.id),drizzleOrm.eq(O.user_id,t.userId))).where(drizzleOrm.and(drizzleOrm.eq(k.workspace_id,t.workspaceId),drizzleOrm.isNull(O.notification_id))).all();if(e.length===0)return 0;let r=new Date().toISOString();return n.db.transaction(s=>{for(let i of e)s.insert(O).values({notification_id:i.id,user_id:t.userId,read_at:r}).onConflictDoNothing().run();}),e.length}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to mark all notifications read",e)}finally{n.close();}}};var je=50,Zt=class extends S{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=V.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=V.dirname(n);he.existsSync(r)||he.mkdirSync(r,{recursive:true});}else if(!he.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{N(e.db,n);}catch(r){throw e.close(),r}return e}create(t){let n=this.openHandle(true);try{let e=new Date().toISOString(),r=[],s=t.map(i=>{let l=Pt("sug");return r.push(l),{id:l,workspace_id:i.workspaceId,agent_id:i.agentId,task_id:i.taskId??null,type:i.type,status:i.status??"pending",payload:i.payload,applied_commit_sha:i.appliedCommitSha??null,created_at:e,updated_at:e}});return s.length>0&&n.db.insert(C).values(s).run(),{ids:r}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to create agent suggestions",e)}finally{n.close();}}findLatestSet(t,n){if(!this.dbExists())return [];let e=this.openHandle(false);try{let r=e.db.select({task_id:C.task_id}).from(C).where(drizzleOrm.and(drizzleOrm.eq(C.workspace_id,t),drizzleOrm.eq(C.agent_id,n))).orderBy(drizzleOrm.desc(C.created_at)).limit(1).get();if(!r)return [];let s=r.task_id===null?drizzleOrm.isNull(C.task_id):drizzleOrm.eq(C.task_id,r.task_id);return e.db.select().from(C).where(drizzleOrm.and(drizzleOrm.eq(C.workspace_id,t),drizzleOrm.eq(C.agent_id,n),s)).orderBy(drizzleOrm.asc(C.created_at)).all()}catch(r){throw new d("DB_ERROR","Failed to find latest agent suggestion set",r)}finally{e.close();}}updateStatus(t,n,e){let r=this.openHandle(true);try{let s={status:n,updated_at:new Date().toISOString()};return e!==void 0&&(s.applied_commit_sha=e),r.db.update(C).set(s).where(drizzleOrm.eq(C.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,n,e=je){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(C).where(drizzleOrm.and(drizzleOrm.eq(C.workspace_id,t),drizzleOrm.eq(C.agent_id,n))).orderBy(drizzleOrm.desc(C.created_at)).limit(e).all()}catch(s){throw new d("DB_ERROR","Failed to list agent suggestions",s)}finally{r.close();}}};exports.AgentSuggestionRepository=Zt;exports.BaseSqliteRepository=S;exports.DEFAULT_AGENT_SUGGESTION_LIST_LIMIT=je;exports.DEFAULT_NOTIFICATION_CAP=Me;exports.DEFAULT_NOTIFICATION_LIST_LIMIT=ze;exports.NotificationRepository=Vt;exports.RepositoryError=d;exports.RequestLogRepository=Yt;exports.SpanRepository=Wt;exports.TaskRepository=qt;exports.ThreadBoxRepository=Gt;exports.ThreadRepository=jt;exports.ToolCallRepository=Kt;exports.UsageLimitSnapshotRepository=Xt;exports.UsageReportRepository=Jt;exports.WorkspaceRepository=zt;exports.calcBucketStart=Ce;exports.openDrizzleDb=w;exports.pushSchema=Ln;exports.runMigrations=pt;exports.runMigrationsOnce=N;exports.tasks=o;
|
|
309
|
+
`),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,n,e){if(!this.dbExists())return;let r=this.openHandle(true);try{let s=new Date().toISOString();r.db.update(_).set({last_message:n,updated_at:s}).where(drizzleOrm.eq(_.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,n){if(!this.dbExists())return;let e=this.openHandle(true);try{let r={updated_at:new Date().toISOString()};n.title!==void 0&&(r.title=n.title,r.title_locked=1),n.titleLocked!==void 0&&(r.title_locked=n.titleLocked?1:0),e.db.update(_).set(r).where(drizzleOrm.eq(_.id,t)).run();}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to update thread",r)}finally{e.close();}}togglePin(t,n){let e=this.openHandle(true);try{let r=n?drizzleOrm.and(drizzleOrm.eq(_.id,t),drizzleOrm.eq(_.workspace_id,n)):drizzleOrm.eq(_.id,t),s=e.db.select({pinned:_.pinned,metadata:_.metadata}).from(_).where(r).get();if(!s)return null;let i=s.pinned?0:1,l=s.metadata?JSON.parse(s.metadata):{};if(i){let u=n?drizzleOrm.and(drizzleOrm.eq(_.pinned,1),drizzleOrm.eq(_.workspace_id,n)):drizzleOrm.eq(_.pinned,1),c=e.db.select({metadata:_.metadata}).from(_).where(u).all(),p=null;for(let h of c){let f=h.metadata?JSON.parse(h.metadata):{};typeof f.pinOrder=="number"&&(p===null||f.pinOrder<p)&&(p=f.pinOrder);}l.pinOrder=p===null?0:p-1;}else delete l.pinOrder;return e.db.update(_).set({pinned:i,metadata:Object.keys(l).length>0?JSON.stringify(l):null}).where(r).run(),{pinned:!!i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle pin",r)}finally{e.close();}}reorderPins(t,n){let e=this.openHandle(true);try{for(let r=0;r<t.length;r++){let s=n?drizzleOrm.and(drizzleOrm.eq(_.id,t[r]),drizzleOrm.eq(_.workspace_id,n)):drizzleOrm.eq(_.id,t[r]),i=e.db.select({metadata:_.metadata}).from(_).where(s).get();if(!i)continue;let l=i.metadata?JSON.parse(i.metadata):{};l.pinOrder=r+1,e.db.update(_).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{e.close();}}toggleStar(t,n){let e=this.openHandle(true);try{let r=n?drizzleOrm.and(drizzleOrm.eq(_.id,t),drizzleOrm.eq(_.workspace_id,n)):drizzleOrm.eq(_.id,t),s=e.db.select({starred:_.starred}).from(_).where(r).get();if(!s)return null;let i=s.starred?0:1;return e.db.update(_).set({starred:i}).where(r).run(),{starred:!!i}}catch(r){throw r instanceof d?r:new d("DB_ERROR","Failed to toggle star",r)}finally{e.close();}}resolveOverdriveForRequest(t,n,e,r){let i=this.openHandle(true);try{let l=n?drizzleOrm.and(drizzleOrm.eq(_.id,t),drizzleOrm.eq(_.workspace_id,n)):drizzleOrm.eq(_.id,t);return i.db.transaction(c=>{let p=c.select({metadata:_.metadata}).from(_).where(l).get();if(!p)return null;let h=p.metadata?JSON.parse(p.metadata):{},f=h.overdrive??{},S=r?.defaultTurns??f.defaultTurns??3,I=r?.updatedBy??"ui",L=new Date().toISOString(),H,N,G,gt;switch(e){case "enable-count":{let wt=r?.turns??S,kt=r?.consumeCurrent?Math.max(wt-1,0):wt;H=kt>0?"count":"off",N=kt,G=!0,gt="count";break}case "enable-latch":{H="latch",N=0,G=!0,gt="latch";break}case "disable":{H="off",N=0,G=!1;break}default:{let wt=f.state??"off",kt=typeof f.remaining=="number"?f.remaining:0;if(wt==="count"&&kt>0){let ee=kt-1;ee<=0?(H="off",N=0):(H="count",N=ee),G=!0,gt="count";}else wt==="latch"?(H="latch",N=0,G=!0,gt="latch"):(H="off",N=0,G=!1);break}}return h.overdrive={state:H,remaining:N,defaultTurns:S,updatedAt:L,updatedBy:I},c.update(_).set({metadata:JSON.stringify(h),updated_at:L}).where(l).run(),{applied:G,appliedMode:gt,state:H,remaining:N}},{behavior:"immediate"})}catch(l){throw l instanceof d?l:new d("DB_ERROR","Failed to resolve overdrive",l)}finally{i.close();}}};var Kt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=Q.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}insertSpan(t){let n=this.openHandle(true);try{n.db.insert(tt).values(t).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert span",e)}finally{n.close();}}findByTaskId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(tt).where(drizzleOrm.eq(tt.task_id,t)).all()}catch(e){throw new d("DB_ERROR","Failed to find spans by task id",e)}finally{n.close();}}findById(t){if(!this.dbExists())return;let n=this.openHandle(false);try{return n.db.select().from(tt).where(drizzleOrm.eq(tt.id,t)).limit(1).get()??void 0}catch(e){throw new d("DB_ERROR","Failed to find span by id",e)}finally{n.close();}}};var Gt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=Q.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}insertToolCall(t){let n=this.openHandle(true);try{n.db.insert(X).values(t).run();}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert tool call",e)}finally{n.close();}}findByTaskId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(X).where(drizzleOrm.eq(X.task_id,t)).all()}catch(e){throw new d("DB_ERROR","Failed to find tool calls by task id",e)}finally{n.close();}}aggregateByName(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select({toolName:X.tool_name,count:drizzleOrm.sql`count(*)`}).from(X).where(drizzleOrm.eq(X.task_id,t)).groupBy(X.tool_name).all().map(e=>({toolName:e.toolName,count:e.count}))}catch(e){throw new d("DB_ERROR","Failed to aggregate tool calls by name",e)}finally{n.close();}}};var Yt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=Q.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}ensureThreadExists(t,n){if(!t.db.select({id:_.id}).from(_).where(drizzleOrm.eq(_.id,n)).limit(1).get())throw new d("NOT_FOUND",`Thread not found: ${n}`)}findByThreadId(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return this.ensureThreadExists(n,t),n.db.select().from(j).where(drizzleOrm.eq(j.thread_id,t)).orderBy(drizzleOrm.asc(j.seq)).all()}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to find boxes by thread id",e)}finally{n.close();}}findById(t,n){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(j).where(drizzleOrm.and(drizzleOrm.eq(j.id,n),drizzleOrm.eq(j.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{e.close();}}insert(t){let n=this.openHandle(true);try{this.ensureThreadExists(n,t.threadId);try{n.db.insert(j).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 e=n.db.select().from(j).where(drizzleOrm.eq(j.id,t.id)).limit(1).get();if(!e)throw new d("DB_ERROR","Insert did not return a row");return e}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to insert thread box",e)}finally{n.close();}}};var Xt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=Q.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}bulkInsert(t){if(t.length===0)return;let n=this.openHandle(true);try{let e=new Date().toISOString();n.db.transaction(r=>{for(let s of t)r.insert(_t).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:e,metadata:s.metadata??null}).run();});}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to bulk insert request logs",e)}finally{n.close();}}findRecent(t=100){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(_t).orderBy(drizzleOrm.desc(_t.timestamp)).limit(t).all()}catch(e){throw new d("DB_ERROR","Failed to find recent request logs",e)}finally{n.close();}}};function ve(a){return new Date(Math.floor(a/3e5)*3e5).toISOString()}var Jt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=Q.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}upsert(t){let n=this.openHandle(true);try{let e=ve(t.bucketStartMs),r=t.accountRef??"default";n.db.insert(v).values({id:t.id,provider:t.provider,account_ref:r,limit_window:t.limitWindow,bucket_start:e,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:[v.provider,v.account_ref,v.limit_window,v.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(e){throw e instanceof d?e:new d("DB_ERROR","Failed to upsert usage limit snapshot",e)}finally{n.close();}}getTrends(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{let e=n.db.select({provider:v.provider,bucket_start:v.bucket_start,used_percent:v.used_percent,remaining_percent:v.remaining_percent,resets_at:v.resets_at}).from(v).where(drizzleOrm.and(drizzleOrm.eq(v.limit_window,t.window),drizzleOrm.gte(v.bucket_start,t.fromIso),drizzleOrm.lte(v.bucket_start,t.toIso))).orderBy(drizzleOrm.asc(v.bucket_start)).all(),r=new Map;for(let s of e)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,i])=>({provider:s,points:i}))}catch(e){throw new d("DB_ERROR","Failed to get usage limit trends",e)}finally{n.close();}}};var Vt=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=Q.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}publish(t){let n=this.openHandle(true);try{n.db.insert(M).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:[M.workspace_id,M.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 e=n.db.select().from(M).where(drizzleOrm.and(drizzleOrm.eq(M.workspace_id,t.workspaceId),drizzleOrm.eq(M.month,t.month))).limit(1).get();if(!e)throw new d("DB_ERROR","Failed to read back published usage report");return e}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to publish usage report",e)}finally{n.close();}}findByMonth(t,n){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(M).where(drizzleOrm.and(drizzleOrm.eq(M.workspace_id,t),drizzleOrm.eq(M.month,n))).limit(1).get()??void 0}catch(r){throw new d("DB_ERROR","Failed to find usage report",r)}finally{e.close();}}listByWorkspace(t){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.select().from(M).where(drizzleOrm.eq(M.workspace_id,t)).orderBy(drizzleOrm.desc(M.month)).all()}catch(e){throw new d("DB_ERROR","Failed to list usage reports",e)}finally{n.close();}}};var $e="ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";function Sr(a){let t=crypto.randomBytes(a*2),n="";for(let e=0;e<t.length&&n.length<a;e++){let r=t[e];r<248&&(n+=$e[r%62]);}for(;n.length<a;)n+=$e[crypto.randomBytes(1)[0]%62];return n}function Lt(a){return `${a}_${Sr(8)}`}var ze=500,Fe=50,Qt=class extends x{dbPath;cap;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=Q.join(t.dbRoot,".crewx","crewx.db")),this.cap=t.cap??ze;}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}create(t){let n=this.openHandle(true);try{let e=Lt("ntf"),r=new Date().toISOString();return n.db.insert(k).values({id:e,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(n,t.workspaceId),{id:e}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to create notification",e)}finally{n.close();}}enforceCap(t,n){let s=(t.db.select({count:drizzleOrm.sql`count(*)`}).from(k).where(drizzleOrm.eq(k.workspace_id,n)).get()?.count??0)-this.cap;if(s<=0)return;let l=t.db.select({id:k.id}).from(k).innerJoin(C,drizzleOrm.eq(C.notification_id,k.id)).where(drizzleOrm.eq(k.workspace_id,n)).groupBy(k.id).orderBy(drizzleOrm.asc(k.created_at)).limit(s).all().map(u=>u.id);l.length!==0&&(t.db.delete(C).where(drizzleOrm.inArray(C.notification_id,l)).run(),t.db.delete(k).where(drizzleOrm.inArray(k.id,l)).run());}list(t){if(!this.dbExists())return [];let n=t.limit??Fe,e=this.openHandle(false);try{return e.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:C.read_at}).from(k).leftJoin(C,drizzleOrm.and(drizzleOrm.eq(C.notification_id,k.id),drizzleOrm.eq(C.user_id,t.userId))).where(drizzleOrm.eq(k.workspace_id,t.workspaceId)).orderBy(drizzleOrm.desc(k.created_at)).limit(n).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{e.close();}}unreadCount(t){if(!this.dbExists())return 0;let n=this.openHandle(false);try{return n.db.select({count:drizzleOrm.sql`count(*)`}).from(k).leftJoin(C,drizzleOrm.and(drizzleOrm.eq(C.notification_id,k.id),drizzleOrm.eq(C.user_id,t.userId))).where(drizzleOrm.and(drizzleOrm.eq(k.workspace_id,t.workspaceId),drizzleOrm.isNull(C.notification_id))).get()?.count??0}catch(e){throw new d("DB_ERROR","Failed to count unread notifications",e)}finally{n.close();}}markRead(t){let n=this.openHandle(true);try{if(!n.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 n.db.insert(C).values({notification_id:t.notificationId,user_id:t.userId,read_at:new Date().toISOString()}).onConflictDoNothing().run().changes>0}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to mark notification read",e)}finally{n.close();}}markAllRead(t){let n=this.openHandle(true);try{let e=n.db.select({id:k.id}).from(k).leftJoin(C,drizzleOrm.and(drizzleOrm.eq(C.notification_id,k.id),drizzleOrm.eq(C.user_id,t.userId))).where(drizzleOrm.and(drizzleOrm.eq(k.workspace_id,t.workspaceId),drizzleOrm.isNull(C.notification_id))).all();if(e.length===0)return 0;let r=new Date().toISOString();return n.db.transaction(s=>{for(let i of e)s.insert(C).values({notification_id:i.id,user_id:t.userId,read_at:r}).onConflictDoNothing().run();}),e.length}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to mark all notifications read",e)}finally{n.close();}}};var We=50,te=class extends x{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=Q.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let n=this.resolveDbPath();if(t){let r=Q.dirname(n);fe.existsSync(r)||fe.mkdirSync(r,{recursive:true});}else if(!fe.existsSync(n))throw new d("NOT_FOUND","Database not found");let e=w(n);if(t)try{A(e.db,n);}catch(r){throw e.close(),r}return e}create(t){let n=this.openHandle(true);try{let e=new Date().toISOString(),r=[],s=t.map(i=>{let l=Lt("sug");return r.push(l),{id:l,workspace_id:i.workspaceId,agent_id:i.agentId,task_id:i.taskId??null,type:i.type,status:i.status??"pending",payload:i.payload,applied_commit_sha:i.appliedCommitSha??null,created_at:e,updated_at:e}});return s.length>0&&n.db.insert(E).values(s).run(),{ids:r}}catch(e){throw e instanceof d?e:new d("DB_ERROR","Failed to create agent suggestions",e)}finally{n.close();}}findLatestSet(t,n){if(!this.dbExists())return [];let e=this.openHandle(false);try{let r=e.db.select({task_id:E.task_id}).from(E).where(drizzleOrm.and(drizzleOrm.eq(E.workspace_id,t),drizzleOrm.eq(E.agent_id,n))).orderBy(drizzleOrm.desc(E.created_at)).limit(1).get();if(!r)return [];let s=r.task_id===null?drizzleOrm.isNull(E.task_id):drizzleOrm.eq(E.task_id,r.task_id);return e.db.select().from(E).where(drizzleOrm.and(drizzleOrm.eq(E.workspace_id,t),drizzleOrm.eq(E.agent_id,n),s)).orderBy(drizzleOrm.asc(E.created_at)).all()}catch(r){throw new d("DB_ERROR","Failed to find latest agent suggestion set",r)}finally{e.close();}}findSetByTaskId(t,n,e){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(E).where(drizzleOrm.and(drizzleOrm.eq(E.workspace_id,t),drizzleOrm.eq(E.agent_id,n),drizzleOrm.eq(E.task_id,e))).orderBy(drizzleOrm.asc(E.created_at)).all()}catch(s){throw new d("DB_ERROR","Failed to find agent suggestion set by task id",s)}finally{r.close();}}updateStatus(t,n,e){let r=this.openHandle(true);try{let s={status:n,updated_at:new Date().toISOString()};return e!==void 0&&(s.applied_commit_sha=e),r.db.update(E).set(s).where(drizzleOrm.eq(E.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,n,e=We){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(E).where(drizzleOrm.and(drizzleOrm.eq(E.workspace_id,t),drizzleOrm.eq(E.agent_id,n))).orderBy(drizzleOrm.desc(E.created_at)).limit(e).all()}catch(s){throw new d("DB_ERROR","Failed to list agent suggestions",s)}finally{r.close();}}findDismissed(t,n,e=20){if(!this.dbExists())return [];let r=this.openHandle(false);try{return r.db.select().from(E).where(drizzleOrm.and(drizzleOrm.eq(E.workspace_id,t),drizzleOrm.eq(E.agent_id,n),drizzleOrm.eq(E.status,"dismissed"))).orderBy(drizzleOrm.desc(E.updated_at)).limit(e).all()}catch(s){throw new d("DB_ERROR","Failed to find dismissed agent suggestions",s)}finally{r.close();}}};exports.AgentSuggestionRepository=te;exports.BaseSqliteRepository=x;exports.DEFAULT_AGENT_SUGGESTION_LIST_LIMIT=We;exports.DEFAULT_NOTIFICATION_CAP=ze;exports.DEFAULT_NOTIFICATION_LIST_LIMIT=Fe;exports.NotificationRepository=Qt;exports.RepositoryError=d;exports.RequestLogRepository=Xt;exports.SpanRepository=Kt;exports.TaskRepository=jt;exports.ThreadBoxRepository=Yt;exports.ThreadRepository=Wt;exports.ToolCallRepository=Gt;exports.UsageLimitSnapshotRepository=Jt;exports.UsageReportRepository=Vt;exports.WorkspaceRepository=Ft;exports.calcBucketStart=ve;exports.openDrizzleDb=w;exports.pushSchema=$n;exports.runMigrations=pt;exports.runMigrationsOnce=A;exports.tasks=o;
|
package/package.json
CHANGED
|
@@ -487,9 +487,10 @@ agents:
|
|
|
487
487
|
|
|
488
488
|
## Task
|
|
489
489
|
Each request gives you one target agent's memory summary, recent failed
|
|
490
|
-
tasks, and rework rate. Locate and read that agent's actual prompt
|
|
491
|
-
|
|
492
|
-
|
|
490
|
+
tasks, and rework rate. Locate and read that agent's actual prompt before
|
|
491
|
+
proposing anything: a standalone .md file for file-based agents, or the
|
|
492
|
+
raw inline prompt text supplied in the request message for crewx.yaml
|
|
493
|
+
inline-prompt agents.
|
|
493
494
|
|
|
494
495
|
## Rules
|
|
495
496
|
- Never fabricate evidence. If you find no clear, well-supported
|
|
@@ -499,16 +500,39 @@ agents:
|
|
|
499
500
|
after review.
|
|
500
501
|
- Output ONLY the JSON specified in the request message — no prose, no
|
|
501
502
|
markdown outside the JSON block.
|
|
502
|
-
- If the target agent's prompt is
|
|
503
|
-
|
|
504
|
-
|
|
505
|
-
|
|
506
|
-
|
|
507
|
-
|
|
503
|
+
- If the target agent's prompt is inline (a crewx.yaml `inline.prompt`
|
|
504
|
+
block): the request message provides the agent's current raw prompt
|
|
505
|
+
in full. Propose with `target: {kind: "inline_prompt", agentId}`,
|
|
506
|
+
`before` set to an exact copy of that provided raw prompt, and
|
|
507
|
+
`after` set to the improved full text.
|
|
508
|
+
- If the target agent's prompt is a standalone `.md` document: propose
|
|
509
|
+
with `target: {kind: "file", file}` as before.
|
|
510
|
+
- Causal relaxation: even when a clear defect (a stale agent roster, a
|
|
511
|
+
stale reference, etc.) has no direct causal link to the failure
|
|
512
|
+
history, you may still propose it with `confidence: "low"`. Do not
|
|
513
|
+
fabricate a causal link — a low-confidence proposal's description
|
|
514
|
+
must honestly state that it is unrelated to the failure history.
|
|
508
515
|
- Even when there are zero proposals, the output must be JSON only —
|
|
509
516
|
answering in prose is not allowed. (Observed 2026-07-22:
|
|
510
517
|
tsk_wSdT7B5q / tsk_2KD2Kfs2 answered in prose and were treated as
|
|
511
518
|
PARSE_EMPTY.)
|
|
519
|
+
|
|
520
|
+
## How to write suggestions (the reader is a non-technical product owner)
|
|
521
|
+
- The person reading each suggestion is a business owner or product manager,
|
|
522
|
+
NOT a developer. Write `title` and `description` so a non-technical reader
|
|
523
|
+
instantly grasps WHAT would change and WHY it helps them.
|
|
524
|
+
- No jargon, file paths, function names, or code identifiers in the `title`.
|
|
525
|
+
In the `description`, if a technical term is unavoidable, explain it in one
|
|
526
|
+
plain phrase right after it.
|
|
527
|
+
- Lead with the plain-language benefit (what gets better), then the change
|
|
528
|
+
itself. Keep the title short and outcome-focused.
|
|
529
|
+
- Write `title` and `description` in the SAME language as the target agent's
|
|
530
|
+
own prompt, so it reads naturally to the user who sees it — do NOT force them
|
|
531
|
+
into English. (These analyst instructions are in English on purpose, for model
|
|
532
|
+
reliability; that does not change the language of your user-facing output.)
|
|
533
|
+
- This plain-language rule applies ONLY to `title` and `description`.
|
|
534
|
+
`before`/`after` are the actual prompt text and MUST stay verbatim and exact —
|
|
535
|
+
an exact copy is required for the change to apply.
|
|
512
536
|
skills:
|
|
513
537
|
include: []
|
|
514
538
|
|