duocode 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.env.example +36 -0
- package/LICENSE +21 -0
- package/README.md +52 -0
- package/dist/ast/context.d.ts +16 -0
- package/dist/ast/context.js +37 -0
- package/dist/ast/context.js.map +1 -0
- package/dist/ast/diff.d.ts +27 -0
- package/dist/ast/diff.js +44 -0
- package/dist/ast/diff.js.map +1 -0
- package/dist/ast/locks.d.ts +47 -0
- package/dist/ast/locks.js +88 -0
- package/dist/ast/locks.js.map +1 -0
- package/dist/ast/merge.d.ts +22 -0
- package/dist/ast/merge.js +120 -0
- package/dist/ast/merge.js.map +1 -0
- package/dist/ast/ownership.d.ts +31 -0
- package/dist/ast/ownership.js +111 -0
- package/dist/ast/ownership.js.map +1 -0
- package/dist/ast/parser.d.ts +44 -0
- package/dist/ast/parser.js +134 -0
- package/dist/ast/parser.js.map +1 -0
- package/dist/cli.d.ts +2 -0
- package/dist/cli.js +423 -0
- package/dist/cli.js.map +1 -0
- package/dist/commands/doctor.d.ts +5 -0
- package/dist/commands/doctor.js +63 -0
- package/dist/commands/doctor.js.map +1 -0
- package/dist/commands/duo.d.ts +9 -0
- package/dist/commands/duo.js +285 -0
- package/dist/commands/duo.js.map +1 -0
- package/dist/commands/github.d.ts +2 -0
- package/dist/commands/github.js +85 -0
- package/dist/commands/github.js.map +1 -0
- package/dist/commands/init.d.ts +2 -0
- package/dist/commands/init.js +33 -0
- package/dist/commands/init.js.map +1 -0
- package/dist/commands/negotiation.d.ts +2 -0
- package/dist/commands/negotiation.js +160 -0
- package/dist/commands/negotiation.js.map +1 -0
- package/dist/commands/repl_commands.d.ts +26 -0
- package/dist/commands/repl_commands.js +226 -0
- package/dist/commands/repl_commands.js.map +1 -0
- package/dist/commands/shell.d.ts +1 -0
- package/dist/commands/shell.js +110 -0
- package/dist/commands/shell.js.map +1 -0
- package/dist/commands/start.d.ts +2 -0
- package/dist/commands/start.js +231 -0
- package/dist/commands/start.js.map +1 -0
- package/dist/commands/task.d.ts +2 -0
- package/dist/commands/task.js +215 -0
- package/dist/commands/task.js.map +1 -0
- package/dist/config/loader.d.ts +193 -0
- package/dist/config/loader.js +106 -0
- package/dist/config/loader.js.map +1 -0
- package/dist/context/project_context.d.ts +79 -0
- package/dist/context/project_context.js +292 -0
- package/dist/context/project_context.js.map +1 -0
- package/dist/context/token_budget.d.ts +35 -0
- package/dist/context/token_budget.js +81 -0
- package/dist/context/token_budget.js.map +1 -0
- package/dist/db/queries.d.ts +121 -0
- package/dist/db/queries.js +109 -0
- package/dist/db/queries.js.map +1 -0
- package/dist/db/schema.d.ts +110 -0
- package/dist/db/schema.js +346 -0
- package/dist/db/schema.js.map +1 -0
- package/dist/duo/duo_orchestrator.d.ts +50 -0
- package/dist/duo/duo_orchestrator.js +510 -0
- package/dist/duo/duo_orchestrator.js.map +1 -0
- package/dist/duo/duo_session.d.ts +47 -0
- package/dist/duo/duo_session.js +127 -0
- package/dist/duo/duo_session.js.map +1 -0
- package/dist/duo/duo_types.d.ts +168 -0
- package/dist/duo/duo_types.js +53 -0
- package/dist/duo/duo_types.js.map +1 -0
- package/dist/duo/session_store.d.ts +71 -0
- package/dist/duo/session_store.js +177 -0
- package/dist/duo/session_store.js.map +1 -0
- package/dist/git/worktree.d.ts +21 -0
- package/dist/git/worktree.js +86 -0
- package/dist/git/worktree.js.map +1 -0
- package/dist/github/cache.d.ts +23 -0
- package/dist/github/cache.js +67 -0
- package/dist/github/cache.js.map +1 -0
- package/dist/github/issues.d.ts +17 -0
- package/dist/github/issues.js +93 -0
- package/dist/github/issues.js.map +1 -0
- package/dist/github/mcp_client.d.ts +57 -0
- package/dist/github/mcp_client.js +214 -0
- package/dist/github/mcp_client.js.map +1 -0
- package/dist/github/sync.d.ts +11 -0
- package/dist/github/sync.js +65 -0
- package/dist/github/sync.js.map +1 -0
- package/dist/github/webhook.d.ts +25 -0
- package/dist/github/webhook.js +197 -0
- package/dist/github/webhook.js.map +1 -0
- package/dist/negotiation/index.d.ts +1 -0
- package/dist/negotiation/index.js +2 -0
- package/dist/negotiation/index.js.map +1 -0
- package/dist/negotiation/protocol.d.ts +62 -0
- package/dist/negotiation/protocol.js +188 -0
- package/dist/negotiation/protocol.js.map +1 -0
- package/dist/orchestrator/complexity_scorer.d.ts +2 -0
- package/dist/orchestrator/complexity_scorer.js +79 -0
- package/dist/orchestrator/complexity_scorer.js.map +1 -0
- package/dist/orchestrator/dependency_graph.d.ts +7 -0
- package/dist/orchestrator/dependency_graph.js +73 -0
- package/dist/orchestrator/dependency_graph.js.map +1 -0
- package/dist/orchestrator/intent_parser.d.ts +11 -0
- package/dist/orchestrator/intent_parser.js +116 -0
- package/dist/orchestrator/intent_parser.js.map +1 -0
- package/dist/orchestrator/task_runner.d.ts +56 -0
- package/dist/orchestrator/task_runner.js +181 -0
- package/dist/orchestrator/task_runner.js.map +1 -0
- package/dist/orchestrator/types.d.ts +44 -0
- package/dist/orchestrator/types.js +21 -0
- package/dist/orchestrator/types.js.map +1 -0
- package/dist/providers/anthropic.d.ts +12 -0
- package/dist/providers/anthropic.js +258 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/auction.d.ts +42 -0
- package/dist/providers/auction.js +190 -0
- package/dist/providers/auction.js.map +1 -0
- package/dist/providers/base.d.ts +103 -0
- package/dist/providers/base.js +2 -0
- package/dist/providers/base.js.map +1 -0
- package/dist/providers/cost_tracker.d.ts +45 -0
- package/dist/providers/cost_tracker.js +111 -0
- package/dist/providers/cost_tracker.js.map +1 -0
- package/dist/providers/duo_pair_router.d.ts +11 -0
- package/dist/providers/duo_pair_router.js +67 -0
- package/dist/providers/duo_pair_router.js.map +1 -0
- package/dist/providers/factory.d.ts +7 -0
- package/dist/providers/factory.js +130 -0
- package/dist/providers/factory.js.map +1 -0
- package/dist/providers/grading_rubric.d.ts +37 -0
- package/dist/providers/grading_rubric.js +238 -0
- package/dist/providers/grading_rubric.js.map +1 -0
- package/dist/providers/openai.d.ts +12 -0
- package/dist/providers/openai.js +229 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/providers/openrouter.d.ts +14 -0
- package/dist/providers/openrouter.js +178 -0
- package/dist/providers/openrouter.js.map +1 -0
- package/dist/providers/performance_tracker.d.ts +21 -0
- package/dist/providers/performance_tracker.js +63 -0
- package/dist/providers/performance_tracker.js.map +1 -0
- package/dist/providers/registry_loader.d.ts +6 -0
- package/dist/providers/registry_loader.js +54 -0
- package/dist/providers/registry_loader.js.map +1 -0
- package/dist/providers/retry.d.ts +66 -0
- package/dist/providers/retry.js +203 -0
- package/dist/providers/retry.js.map +1 -0
- package/dist/providers/role_scorer.d.ts +16 -0
- package/dist/providers/role_scorer.js +16 -0
- package/dist/providers/role_scorer.js.map +1 -0
- package/dist/providers/router.d.ts +84 -0
- package/dist/providers/router.js +542 -0
- package/dist/providers/router.js.map +1 -0
- package/dist/security/credentials.d.ts +6 -0
- package/dist/security/credentials.js +16 -0
- package/dist/security/credentials.js.map +1 -0
- package/dist/setup/browser.d.ts +1 -0
- package/dist/setup/browser.js +12 -0
- package/dist/setup/browser.js.map +1 -0
- package/dist/setup/global_config.d.ts +14 -0
- package/dist/setup/global_config.js +54 -0
- package/dist/setup/global_config.js.map +1 -0
- package/dist/setup/wizard.d.ts +2 -0
- package/dist/setup/wizard.js +206 -0
- package/dist/setup/wizard.js.map +1 -0
- package/dist/tools/agent_loop.d.ts +38 -0
- package/dist/tools/agent_loop.js +72 -0
- package/dist/tools/agent_loop.js.map +1 -0
- package/dist/tools/approval.d.ts +64 -0
- package/dist/tools/approval.js +172 -0
- package/dist/tools/approval.js.map +1 -0
- package/dist/tools/checkpoint.d.ts +65 -0
- package/dist/tools/checkpoint.js +342 -0
- package/dist/tools/checkpoint.js.map +1 -0
- package/dist/tools/definitions.d.ts +13 -0
- package/dist/tools/definitions.js +103 -0
- package/dist/tools/definitions.js.map +1 -0
- package/dist/tools/diff_display.d.ts +46 -0
- package/dist/tools/diff_display.js +298 -0
- package/dist/tools/diff_display.js.map +1 -0
- package/dist/tools/executor.d.ts +12 -0
- package/dist/tools/executor.js +340 -0
- package/dist/tools/executor.js.map +1 -0
- package/dist/tools/permissions.d.ts +17 -0
- package/dist/tools/permissions.js +139 -0
- package/dist/tools/permissions.js.map +1 -0
- package/dist/tools/tool_types.d.ts +48 -0
- package/dist/tools/tool_types.js +7 -0
- package/dist/tools/tool_types.js.map +1 -0
- package/dist/ui/banner.d.ts +4 -0
- package/dist/ui/banner.js +104 -0
- package/dist/ui/banner.js.map +1 -0
- package/dist/ui/callbacks.d.ts +30 -0
- package/dist/ui/callbacks.js +132 -0
- package/dist/ui/callbacks.js.map +1 -0
- package/dist/ui/colors.d.ts +14 -0
- package/dist/ui/colors.js +28 -0
- package/dist/ui/colors.js.map +1 -0
- package/dist/ui/dashboard.d.ts +51 -0
- package/dist/ui/dashboard.js +181 -0
- package/dist/ui/dashboard.js.map +1 -0
- package/dist/ui/leaderboard.d.ts +16 -0
- package/dist/ui/leaderboard.js +43 -0
- package/dist/ui/leaderboard.js.map +1 -0
- package/dist/ui/logger.d.ts +28 -0
- package/dist/ui/logger.js +117 -0
- package/dist/ui/logger.js.map +1 -0
- package/dist/ui/progress.d.ts +16 -0
- package/dist/ui/progress.js +62 -0
- package/dist/ui/progress.js.map +1 -0
- package/dist/ui/tokenizer.d.ts +5 -0
- package/dist/ui/tokenizer.js +54 -0
- package/dist/ui/tokenizer.js.map +1 -0
- package/package.json +63 -0
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
function ensureSyncTable(db) {
|
|
2
|
+
db.exec(`
|
|
3
|
+
CREATE TABLE IF NOT EXISTS github_pr_sync (
|
|
4
|
+
task_id TEXT PRIMARY KEY,
|
|
5
|
+
pr_number INTEGER NOT NULL,
|
|
6
|
+
pr_state TEXT NOT NULL,
|
|
7
|
+
synced_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
8
|
+
)
|
|
9
|
+
`);
|
|
10
|
+
}
|
|
11
|
+
function loadRoutes(db) {
|
|
12
|
+
return db
|
|
13
|
+
.prepare("SELECT task_id, branch_name FROM worktree_routes")
|
|
14
|
+
.all();
|
|
15
|
+
}
|
|
16
|
+
function findOpenPrByBranch(pulls, branchName) {
|
|
17
|
+
return pulls.find((pr) => pr.head?.ref === branchName && pr.state === "open");
|
|
18
|
+
}
|
|
19
|
+
export async function syncWorktreesWithPullRequests(db, github, context) {
|
|
20
|
+
ensureSyncTable(db);
|
|
21
|
+
await github.validateRequiredTools([
|
|
22
|
+
"github_list_pull_requests",
|
|
23
|
+
"github_create_pull_request",
|
|
24
|
+
]);
|
|
25
|
+
const pulls = await github.listPullRequests(context.owner, context.repo, "open");
|
|
26
|
+
const typedPulls = pulls;
|
|
27
|
+
const routes = loadRoutes(db);
|
|
28
|
+
let created = 0;
|
|
29
|
+
let linked = 0;
|
|
30
|
+
for (const route of routes) {
|
|
31
|
+
const existing = findOpenPrByBranch(typedPulls, route.branch_name);
|
|
32
|
+
let prNumber = null;
|
|
33
|
+
if (existing && typeof existing.number === "number") {
|
|
34
|
+
prNumber = existing.number;
|
|
35
|
+
linked += 1;
|
|
36
|
+
}
|
|
37
|
+
else {
|
|
38
|
+
const createdPr = await github.createPullRequest({
|
|
39
|
+
owner: context.owner,
|
|
40
|
+
repo: context.repo,
|
|
41
|
+
title: `Task ${route.task_id.slice(0, 8)}: automated change`,
|
|
42
|
+
head: route.branch_name,
|
|
43
|
+
base: context.baseBranch,
|
|
44
|
+
body: `Auto-generated PR for DuoCode task ${route.task_id}.`,
|
|
45
|
+
draft: true,
|
|
46
|
+
idempotencyKey: `pr:${context.owner}/${context.repo}:${route.task_id}:${route.branch_name}`,
|
|
47
|
+
});
|
|
48
|
+
const maybeNumber = createdPr.number;
|
|
49
|
+
if (typeof maybeNumber === "number") {
|
|
50
|
+
prNumber = maybeNumber;
|
|
51
|
+
}
|
|
52
|
+
created += 1;
|
|
53
|
+
}
|
|
54
|
+
if (prNumber !== null) {
|
|
55
|
+
db.prepare(`INSERT INTO github_pr_sync (task_id, pr_number, pr_state, synced_at)
|
|
56
|
+
VALUES (?, ?, 'open', datetime('now'))
|
|
57
|
+
ON CONFLICT(task_id) DO UPDATE SET
|
|
58
|
+
pr_number = excluded.pr_number,
|
|
59
|
+
pr_state = excluded.pr_state,
|
|
60
|
+
synced_at = excluded.synced_at`).run(route.task_id, prNumber);
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
return { created, linked };
|
|
64
|
+
}
|
|
65
|
+
//# sourceMappingURL=sync.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sync.js","sourceRoot":"","sources":["../../src/github/sync.ts"],"names":[],"mappings":"AAoBA,SAAS,eAAe,CAAC,EAAqB;IAC5C,EAAE,CAAC,IAAI,CAAC;;;;;;;GAOP,CAAC,CAAC;AACL,CAAC;AAED,SAAS,UAAU,CAAC,EAAqB;IACvC,OAAO,EAAE;SACN,OAAO,CAAC,kDAAkD,CAAC;SAC3D,GAAG,EAAgB,CAAC;AACzB,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAwB,EAAE,UAAkB;IACtE,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,UAAU,IAAI,EAAE,CAAC,KAAK,KAAK,MAAM,CAAC,CAAC;AAChF,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,6BAA6B,CACjD,EAAqB,EACrB,MAAuB,EACvB,OAAoB;IAEpB,eAAe,CAAC,EAAE,CAAC,CAAC;IACpB,MAAM,MAAM,CAAC,qBAAqB,CAAC;QACjC,2BAA2B;QAC3B,4BAA4B;KAC7B,CAAC,CAAC;IAEH,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACjF,MAAM,UAAU,GAAG,KAA0B,CAAC;IAE9C,MAAM,MAAM,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC;IAC9B,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,IAAI,MAAM,GAAG,CAAC,CAAC;IAEf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,kBAAkB,CAAC,UAAU,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC;QAEnE,IAAI,QAAQ,GAAkB,IAAI,CAAC;QAEnC,IAAI,QAAQ,IAAI,OAAO,QAAQ,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YACpD,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC;YAC3B,MAAM,IAAI,CAAC,CAAC;QACd,CAAC;aAAM,CAAC;YACN,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC;gBAC/C,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,KAAK,EAAE,QAAQ,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,oBAAoB;gBAC5D,IAAI,EAAE,KAAK,CAAC,WAAW;gBACvB,IAAI,EAAE,OAAO,CAAC,UAAU;gBACxB,IAAI,EAAE,sCAAsC,KAAK,CAAC,OAAO,GAAG;gBAC5D,KAAK,EAAE,IAAI;gBACX,cAAc,EAAE,MAAM,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,IAAI,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,WAAW,EAAE;aAC5F,CAAC,CAAC;YAEH,MAAM,WAAW,GAAI,SAAiC,CAAC,MAAM,CAAC;YAC9D,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;gBACpC,QAAQ,GAAG,WAAW,CAAC;YACzB,CAAC;YAED,OAAO,IAAI,CAAC,CAAC;QACf,CAAC;QAED,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;YACtB,EAAE,CAAC,OAAO,CACR;;;;;0CAKkC,CACnC,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AAC7B,CAAC"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import express from "express";
|
|
2
|
+
import type Database from "better-sqlite3";
|
|
3
|
+
export interface WebhookServerOptions {
|
|
4
|
+
secret: string;
|
|
5
|
+
db: Database.Database;
|
|
6
|
+
onPullRequestEvent?: (payload: Record<string, unknown>) => Promise<void> | void;
|
|
7
|
+
onIssueConverted?: (taskId: string, issueNumber: number) => Promise<void> | void;
|
|
8
|
+
}
|
|
9
|
+
export declare class RateLimiter {
|
|
10
|
+
private readonly maxRequests;
|
|
11
|
+
private readonly windowMs;
|
|
12
|
+
private readonly buckets;
|
|
13
|
+
private cleanupTimer;
|
|
14
|
+
constructor(maxRequests?: number, windowMs?: number);
|
|
15
|
+
/** Start periodic cleanup of expired buckets. */
|
|
16
|
+
start(): void;
|
|
17
|
+
stop(): void;
|
|
18
|
+
/**
|
|
19
|
+
* Check whether a request from `key` is allowed.
|
|
20
|
+
* Returns remaining requests, or -1 if rate-limited.
|
|
21
|
+
*/
|
|
22
|
+
check(key: string): number;
|
|
23
|
+
private cleanup;
|
|
24
|
+
}
|
|
25
|
+
export declare function createGithubWebhookApp(options: WebhookServerOptions): express.Express;
|
|
@@ -0,0 +1,197 @@
|
|
|
1
|
+
import { createHmac, timingSafeEqual } from "node:crypto";
|
|
2
|
+
import express from "express";
|
|
3
|
+
import { convertIssueToTask } from "./issues.js";
|
|
4
|
+
import { updateTaskStatus } from "../db/schema.js";
|
|
5
|
+
export class RateLimiter {
|
|
6
|
+
maxRequests;
|
|
7
|
+
windowMs;
|
|
8
|
+
buckets = new Map();
|
|
9
|
+
cleanupTimer = null;
|
|
10
|
+
constructor(maxRequests = 60, windowMs = 60000) {
|
|
11
|
+
this.maxRequests = maxRequests;
|
|
12
|
+
this.windowMs = windowMs;
|
|
13
|
+
}
|
|
14
|
+
/** Start periodic cleanup of expired buckets. */
|
|
15
|
+
start() {
|
|
16
|
+
if (this.cleanupTimer)
|
|
17
|
+
return;
|
|
18
|
+
this.cleanupTimer = setInterval(() => this.cleanup(), this.windowMs * 2);
|
|
19
|
+
this.cleanupTimer.unref(); // Don't prevent process exit
|
|
20
|
+
}
|
|
21
|
+
stop() {
|
|
22
|
+
if (this.cleanupTimer) {
|
|
23
|
+
clearInterval(this.cleanupTimer);
|
|
24
|
+
this.cleanupTimer = null;
|
|
25
|
+
}
|
|
26
|
+
this.buckets.clear();
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* Check whether a request from `key` is allowed.
|
|
30
|
+
* Returns remaining requests, or -1 if rate-limited.
|
|
31
|
+
*/
|
|
32
|
+
check(key) {
|
|
33
|
+
const now = Date.now();
|
|
34
|
+
let bucket = this.buckets.get(key);
|
|
35
|
+
if (!bucket || now >= bucket.resetAt) {
|
|
36
|
+
bucket = { count: 0, resetAt: now + this.windowMs };
|
|
37
|
+
this.buckets.set(key, bucket);
|
|
38
|
+
}
|
|
39
|
+
bucket.count++;
|
|
40
|
+
if (bucket.count > this.maxRequests) {
|
|
41
|
+
return -1;
|
|
42
|
+
}
|
|
43
|
+
return this.maxRequests - bucket.count;
|
|
44
|
+
}
|
|
45
|
+
cleanup() {
|
|
46
|
+
const now = Date.now();
|
|
47
|
+
for (const [key, bucket] of this.buckets) {
|
|
48
|
+
if (now >= bucket.resetAt) {
|
|
49
|
+
this.buckets.delete(key);
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
// ── Webhook helpers ────────────────────────────────────────────────
|
|
55
|
+
function pullRequestStateToTaskStatus(payload) {
|
|
56
|
+
const action = String(payload.action ?? "");
|
|
57
|
+
const pr = payload.pull_request;
|
|
58
|
+
const merged = Boolean(pr?.merged);
|
|
59
|
+
if (merged || action === "closed") {
|
|
60
|
+
return "completed";
|
|
61
|
+
}
|
|
62
|
+
if (action === "opened" || action === "reopened" || action === "synchronize" || action === "ready_for_review") {
|
|
63
|
+
return "reviewing";
|
|
64
|
+
}
|
|
65
|
+
if (action === "draft") {
|
|
66
|
+
return "executing";
|
|
67
|
+
}
|
|
68
|
+
return null;
|
|
69
|
+
}
|
|
70
|
+
function maybeUpdateStatusFromPullRequest(db, payload) {
|
|
71
|
+
const pr = payload.pull_request;
|
|
72
|
+
const head = pr?.head;
|
|
73
|
+
const branch = typeof head?.ref === "string" ? head.ref : null;
|
|
74
|
+
if (!branch) {
|
|
75
|
+
return;
|
|
76
|
+
}
|
|
77
|
+
const taskLink = db
|
|
78
|
+
.prepare(`SELECT wr.task_id AS task_id
|
|
79
|
+
FROM worktree_routes wr
|
|
80
|
+
WHERE wr.branch_name = ?`)
|
|
81
|
+
.get(branch);
|
|
82
|
+
if (!taskLink) {
|
|
83
|
+
return;
|
|
84
|
+
}
|
|
85
|
+
const status = pullRequestStateToTaskStatus(payload);
|
|
86
|
+
if (status) {
|
|
87
|
+
updateTaskStatus(db, taskLink.task_id, status);
|
|
88
|
+
}
|
|
89
|
+
const prNumber = Number(pr?.number ?? payload.number);
|
|
90
|
+
if (Number.isFinite(prNumber)) {
|
|
91
|
+
const prState = typeof pr?.state === "string" ? pr.state : "open";
|
|
92
|
+
db.prepare(`INSERT INTO github_pr_sync (task_id, pr_number, pr_state, synced_at)
|
|
93
|
+
VALUES (?, ?, ?, datetime('now'))
|
|
94
|
+
ON CONFLICT(task_id) DO UPDATE SET
|
|
95
|
+
pr_number = excluded.pr_number,
|
|
96
|
+
pr_state = excluded.pr_state,
|
|
97
|
+
synced_at = excluded.synced_at`).run(taskLink.task_id, prNumber, prState);
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
function maybeUpdateStatusFromIssueComment(db, payload) {
|
|
101
|
+
const issue = payload.issue;
|
|
102
|
+
const issueNumber = Number(issue?.number);
|
|
103
|
+
if (!Number.isFinite(issueNumber)) {
|
|
104
|
+
return;
|
|
105
|
+
}
|
|
106
|
+
const link = db
|
|
107
|
+
.prepare("SELECT task_id FROM github_issue_tasks WHERE issue_number = ?")
|
|
108
|
+
.get(issueNumber);
|
|
109
|
+
if (!link) {
|
|
110
|
+
return;
|
|
111
|
+
}
|
|
112
|
+
const action = String(payload.action ?? "");
|
|
113
|
+
if (action === "created") {
|
|
114
|
+
updateTaskStatus(db, link.task_id, "reviewing");
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
function verifySignature(req, secret) {
|
|
118
|
+
const header = req.header("x-hub-signature-256");
|
|
119
|
+
if (!header || !header.startsWith("sha256=")) {
|
|
120
|
+
return false;
|
|
121
|
+
}
|
|
122
|
+
const signature = header.slice("sha256=".length);
|
|
123
|
+
const body = req.rawBody ?? Buffer.from("");
|
|
124
|
+
const digest = createHmac("sha256", secret).update(body).digest("hex");
|
|
125
|
+
try {
|
|
126
|
+
return timingSafeEqual(Buffer.from(signature, "hex"), Buffer.from(digest, "hex"));
|
|
127
|
+
}
|
|
128
|
+
catch {
|
|
129
|
+
return false;
|
|
130
|
+
}
|
|
131
|
+
}
|
|
132
|
+
function rawBodyCapture(req, _res, buffer) {
|
|
133
|
+
req.rawBody = Buffer.from(buffer);
|
|
134
|
+
}
|
|
135
|
+
function getClientIp(req) {
|
|
136
|
+
const forwarded = req.header("x-forwarded-for");
|
|
137
|
+
if (forwarded) {
|
|
138
|
+
return forwarded.split(",")[0].trim();
|
|
139
|
+
}
|
|
140
|
+
return req.ip ?? req.socket.remoteAddress ?? "unknown";
|
|
141
|
+
}
|
|
142
|
+
// ── Express app ────────────────────────────────────────────────────
|
|
143
|
+
export function createGithubWebhookApp(options) {
|
|
144
|
+
const app = express();
|
|
145
|
+
const limiter = new RateLimiter(60, 60000); // 60 requests per minute per IP
|
|
146
|
+
limiter.start();
|
|
147
|
+
app.use(express.json({ verify: rawBodyCapture }));
|
|
148
|
+
app.post("/github/webhook", async (req, res, next) => {
|
|
149
|
+
try {
|
|
150
|
+
// Rate limiting
|
|
151
|
+
const clientIp = getClientIp(req);
|
|
152
|
+
const remaining = limiter.check(clientIp);
|
|
153
|
+
if (remaining < 0) {
|
|
154
|
+
res.status(429).json({ error: "Too many requests" });
|
|
155
|
+
return;
|
|
156
|
+
}
|
|
157
|
+
res.setHeader("X-RateLimit-Remaining", String(Math.max(0, remaining)));
|
|
158
|
+
if (!verifySignature(req, options.secret)) {
|
|
159
|
+
res.status(401).json({ error: "Invalid webhook signature" });
|
|
160
|
+
return;
|
|
161
|
+
}
|
|
162
|
+
const event = req.header("x-github-event") ?? "unknown";
|
|
163
|
+
const payload = req.body;
|
|
164
|
+
if (event === "issues") {
|
|
165
|
+
const action = String(payload.action ?? "");
|
|
166
|
+
if (action === "opened" || action === "reopened") {
|
|
167
|
+
const issue = payload.issue;
|
|
168
|
+
if (issue && typeof issue.number === "number" && typeof issue.title === "string") {
|
|
169
|
+
const link = convertIssueToTask(options.db, issue);
|
|
170
|
+
if (options.onIssueConverted) {
|
|
171
|
+
await options.onIssueConverted(link.taskId, link.issueNumber);
|
|
172
|
+
}
|
|
173
|
+
}
|
|
174
|
+
}
|
|
175
|
+
}
|
|
176
|
+
if (event === "pull_request") {
|
|
177
|
+
maybeUpdateStatusFromPullRequest(options.db, payload);
|
|
178
|
+
if (options.onPullRequestEvent) {
|
|
179
|
+
await options.onPullRequestEvent(payload);
|
|
180
|
+
}
|
|
181
|
+
}
|
|
182
|
+
if (event === "issue_comment") {
|
|
183
|
+
maybeUpdateStatusFromIssueComment(options.db, payload);
|
|
184
|
+
}
|
|
185
|
+
res.status(202).json({ accepted: true, event });
|
|
186
|
+
}
|
|
187
|
+
catch (error) {
|
|
188
|
+
next(error);
|
|
189
|
+
}
|
|
190
|
+
});
|
|
191
|
+
app.use((error, _req, res, _next) => {
|
|
192
|
+
const message = error instanceof Error ? error.message : String(error);
|
|
193
|
+
res.status(500).json({ error: message });
|
|
194
|
+
});
|
|
195
|
+
return app;
|
|
196
|
+
}
|
|
197
|
+
//# sourceMappingURL=webhook.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"webhook.js","sourceRoot":"","sources":["../../src/github/webhook.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAE1D,OAAO,OAAO,MAAM,SAAS,CAAC;AAE9B,OAAO,EAAE,kBAAkB,EAAoB,MAAM,aAAa,CAAC;AACnE,OAAO,EAAE,gBAAgB,EAAmB,MAAM,iBAAiB,CAAC;AAgBpE,MAAM,OAAO,WAAW;IACL,WAAW,CAAS;IACpB,QAAQ,CAAS;IACjB,OAAO,GAAG,IAAI,GAAG,EAAsB,CAAC;IACjD,YAAY,GAA0C,IAAI,CAAC;IAEnE,YAAY,WAAW,GAAG,EAAE,EAAE,QAAQ,GAAG,KAAM;QAC7C,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,iDAAiD;IACjD,KAAK;QACH,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAC9B,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;QACzE,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC,CAAC,6BAA6B;IAC1D,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACjC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC3B,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,GAAW;QACf,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAEnC,IAAI,CAAC,MAAM,IAAI,GAAG,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACrC,MAAM,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAChC,CAAC;QAED,MAAM,CAAC,KAAK,EAAE,CAAC;QAEf,IAAI,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;YACpC,OAAO,CAAC,CAAC,CAAC;QACZ,CAAC;QAED,OAAO,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC;IACzC,CAAC;IAEO,OAAO;QACb,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACzC,IAAI,GAAG,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBAC1B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED,sEAAsE;AAEtE,SAAS,4BAA4B,CAAC,OAAgC;IACpE,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;IAC5C,MAAM,EAAE,GAAG,OAAO,CAAC,YAAmD,CAAC;IACvE,MAAM,MAAM,GAAG,OAAO,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IAEnC,IAAI,MAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;QAClC,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,UAAU,IAAI,MAAM,KAAK,aAAa,IAAI,MAAM,KAAK,kBAAkB,EAAE,CAAC;QAC9G,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;QACvB,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,gCAAgC,CAAC,EAAqB,EAAE,OAAgC;IAC/F,MAAM,EAAE,GAAG,OAAO,CAAC,YAAmD,CAAC;IACvE,MAAM,IAAI,GAAG,EAAE,EAAE,IAA2C,CAAC;IAC7D,MAAM,MAAM,GAAG,OAAO,IAAI,EAAE,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;IAC/D,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO;IACT,CAAC;IAED,MAAM,QAAQ,GAAG,EAAE;SAChB,OAAO,CACN;;gCAE0B,CAC3B;SACA,GAAG,CAAC,MAAM,CAAoC,CAAC;IAElD,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,OAAO;IACT,CAAC;IAED,MAAM,MAAM,GAAG,4BAA4B,CAAC,OAAO,CAAC,CAAC;IACrD,IAAI,MAAM,EAAE,CAAC;QACX,gBAAgB,CAAC,EAAE,EAAE,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAE,EAAE,EAAE,MAA6B,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;IAC9E,IAAI,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,OAAO,EAAE,EAAE,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;QAClE,EAAE,CAAC,OAAO,CACR;;;;;wCAKkC,CACnC,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC7C,CAAC;AACH,CAAC;AAED,SAAS,iCAAiC,CAAC,EAAqB,EAAE,OAAgC;IAChG,MAAM,KAAK,GAAG,OAAO,CAAC,KAA4C,CAAC;IACnE,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC1C,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;QAClC,OAAO;IACT,CAAC;IAED,MAAM,IAAI,GAAG,EAAE;SACZ,OAAO,CAAC,+DAA+D,CAAC;SACxE,GAAG,CAAC,WAAW,CAAoC,CAAC;IAEvD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;IACT,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;IAC5C,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,gBAAgB,CAAC,EAAE,EAAE,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAClD,CAAC;AACH,CAAC;AAMD,SAAS,eAAe,CAAC,GAAmB,EAAE,MAAc;IAC1D,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;IACjD,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC7C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACjD,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAE5C,MAAM,MAAM,GAAG,UAAU,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAEvE,IAAI,CAAC;QACH,OAAO,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;IACpF,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,GAAmB,EAAE,IAAc,EAAE,MAAc;IACzE,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,WAAW,CAAC,GAAY;IAC/B,MAAM,SAAS,GAAG,GAAG,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;IAChD,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IACxC,CAAC;IACD,OAAO,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,MAAM,CAAC,aAAa,IAAI,SAAS,CAAC;AACzD,CAAC;AAED,sEAAsE;AAEtE,MAAM,UAAU,sBAAsB,CAAC,OAA6B;IAClE,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IACtB,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,EAAE,EAAE,KAAM,CAAC,CAAC,CAAC,gCAAgC;IAC7E,OAAO,CAAC,KAAK,EAAE,CAAC;IAEhB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;IAElD,GAAG,CAAC,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,GAAmB,EAAE,GAAa,EAAE,IAAkB,EAAE,EAAE;QAC3F,IAAI,CAAC;YACH,gBAAgB;YAChB,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC;YAClC,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC1C,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;gBAClB,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;gBACrD,OAAO;YACT,CAAC;YACD,GAAG,CAAC,SAAS,CAAC,uBAAuB,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;YAEvE,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC1C,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,2BAA2B,EAAE,CAAC,CAAC;gBAC7D,OAAO;YACT,CAAC;YAED,MAAM,KAAK,GAAG,GAAG,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,SAAS,CAAC;YACxD,MAAM,OAAO,GAAG,GAAG,CAAC,IAA+B,CAAC;YAEpD,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACvB,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;gBAC5C,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,UAAU,EAAE,CAAC;oBACjD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAgC,CAAC;oBACvD,IAAI,KAAK,IAAI,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;wBACjF,MAAM,IAAI,GAAG,kBAAkB,CAAC,OAAO,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;wBACnD,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;4BAC7B,MAAM,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;wBAChE,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,KAAK,KAAK,cAAc,EAAE,CAAC;gBAC7B,gCAAgC,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;gBACtD,IAAI,OAAO,CAAC,kBAAkB,EAAE,CAAC;oBAC/B,MAAM,OAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;YAED,IAAI,KAAK,KAAK,eAAe,EAAE,CAAC;gBAC9B,iCAAiC,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACzD,CAAC;YAED,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAClD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,KAAK,CAAC,CAAC;QACd,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,GAAG,CAAC,CAAC,KAAc,EAAE,IAAa,EAAE,GAAa,EAAE,KAAmB,EAAE,EAAE;QAC5E,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,OAAO,GAAG,CAAC;AACb,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export * from "./protocol.js";
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/negotiation/index.ts"],"names":[],"mappings":"AAAA,cAAc,eAAe,CAAC"}
|
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
import { DistributedOwnershipLockManager, type OwnershipLock } from "../ast/locks.js";
|
|
2
|
+
import type { SupportedLanguage } from "../ast/parser.js";
|
|
3
|
+
export type NegotiationPhase = "detection" | "analysis" | "auto_resolving" | "debating" | "resolved" | "escalated";
|
|
4
|
+
export interface ConflictSide {
|
|
5
|
+
agentId: string;
|
|
6
|
+
source: string;
|
|
7
|
+
rationale?: string;
|
|
8
|
+
}
|
|
9
|
+
export interface NegotiationConfig {
|
|
10
|
+
consensusThreshold: number;
|
|
11
|
+
maxRounds: number;
|
|
12
|
+
lockTtlMs: number;
|
|
13
|
+
}
|
|
14
|
+
export declare const DEFAULT_NEGOTIATION_CONFIG: NegotiationConfig;
|
|
15
|
+
export interface NegotiationConflict {
|
|
16
|
+
path: string;
|
|
17
|
+
leftType: string;
|
|
18
|
+
rightType: string;
|
|
19
|
+
}
|
|
20
|
+
export interface DebateRound {
|
|
21
|
+
round: number;
|
|
22
|
+
leftScore: number;
|
|
23
|
+
rightScore: number;
|
|
24
|
+
consensus: number;
|
|
25
|
+
reasoning: string;
|
|
26
|
+
}
|
|
27
|
+
export interface NegotiationResolution {
|
|
28
|
+
method: "auto_compose" | "debate_winner" | "escalated";
|
|
29
|
+
winnerAgentId: string | null;
|
|
30
|
+
mergedSource: string;
|
|
31
|
+
confidence: number;
|
|
32
|
+
explanation: string;
|
|
33
|
+
}
|
|
34
|
+
export interface NegotiationSession {
|
|
35
|
+
id: string;
|
|
36
|
+
phase: NegotiationPhase;
|
|
37
|
+
language: SupportedLanguage;
|
|
38
|
+
filePath: string;
|
|
39
|
+
conflictCount: number;
|
|
40
|
+
conflicts: NegotiationConflict[];
|
|
41
|
+
rounds: DebateRound[];
|
|
42
|
+
locks: OwnershipLock[];
|
|
43
|
+
resolution: NegotiationResolution;
|
|
44
|
+
}
|
|
45
|
+
export declare class AstNegotiationEngine {
|
|
46
|
+
private readonly lockManager;
|
|
47
|
+
constructor(lockManager?: DistributedOwnershipLockManager);
|
|
48
|
+
negotiate(params: {
|
|
49
|
+
negotiationId?: string;
|
|
50
|
+
language: SupportedLanguage;
|
|
51
|
+
filePath: string;
|
|
52
|
+
baseSource: string;
|
|
53
|
+
left: ConflictSide;
|
|
54
|
+
right: ConflictSide;
|
|
55
|
+
config?: Partial<NegotiationConfig>;
|
|
56
|
+
}): Promise<NegotiationSession>;
|
|
57
|
+
releaseSessionLocks(session: NegotiationSession): Promise<number>;
|
|
58
|
+
private findConflicts;
|
|
59
|
+
private acquireConflictLocks;
|
|
60
|
+
private scoreSide;
|
|
61
|
+
private makeId;
|
|
62
|
+
}
|
|
@@ -0,0 +1,188 @@
|
|
|
1
|
+
import { diffAstOwnership } from "../ast/diff.js";
|
|
2
|
+
import { DistributedOwnershipLockManager, InMemoryOwnershipLockStore, } from "../ast/locks.js";
|
|
3
|
+
import { mergeAstThreeWay } from "../ast/merge.js";
|
|
4
|
+
export const DEFAULT_NEGOTIATION_CONFIG = {
|
|
5
|
+
consensusThreshold: 0.9,
|
|
6
|
+
maxRounds: 3,
|
|
7
|
+
lockTtlMs: 30000,
|
|
8
|
+
};
|
|
9
|
+
export class AstNegotiationEngine {
|
|
10
|
+
lockManager;
|
|
11
|
+
constructor(lockManager) {
|
|
12
|
+
this.lockManager =
|
|
13
|
+
lockManager ?? new DistributedOwnershipLockManager(new InMemoryOwnershipLockStore());
|
|
14
|
+
}
|
|
15
|
+
async negotiate(params) {
|
|
16
|
+
const config = {
|
|
17
|
+
...DEFAULT_NEGOTIATION_CONFIG,
|
|
18
|
+
...params.config,
|
|
19
|
+
};
|
|
20
|
+
const sessionId = params.negotiationId ?? this.makeId();
|
|
21
|
+
const leftDiff = diffAstOwnership({
|
|
22
|
+
language: params.language,
|
|
23
|
+
filePath: params.filePath,
|
|
24
|
+
beforeSource: params.baseSource,
|
|
25
|
+
afterSource: params.left.source,
|
|
26
|
+
});
|
|
27
|
+
const rightDiff = diffAstOwnership({
|
|
28
|
+
language: params.language,
|
|
29
|
+
filePath: params.filePath,
|
|
30
|
+
beforeSource: params.baseSource,
|
|
31
|
+
afterSource: params.right.source,
|
|
32
|
+
});
|
|
33
|
+
const conflicts = this.findConflicts(leftDiff.changes, rightDiff.changes);
|
|
34
|
+
const locks = await this.acquireConflictLocks(conflicts, params.filePath, config.lockTtlMs, [
|
|
35
|
+
params.left.agentId,
|
|
36
|
+
params.right.agentId,
|
|
37
|
+
]);
|
|
38
|
+
const merged = mergeAstThreeWay({
|
|
39
|
+
language: params.language,
|
|
40
|
+
filePath: params.filePath,
|
|
41
|
+
baseSource: params.baseSource,
|
|
42
|
+
leftSource: params.left.source,
|
|
43
|
+
rightSource: params.right.source,
|
|
44
|
+
});
|
|
45
|
+
if (merged.conflicts.length === 0) {
|
|
46
|
+
return {
|
|
47
|
+
id: sessionId,
|
|
48
|
+
phase: "resolved",
|
|
49
|
+
language: params.language,
|
|
50
|
+
filePath: params.filePath,
|
|
51
|
+
conflictCount: conflicts.length,
|
|
52
|
+
conflicts,
|
|
53
|
+
rounds: [],
|
|
54
|
+
locks,
|
|
55
|
+
resolution: {
|
|
56
|
+
method: "auto_compose",
|
|
57
|
+
winnerAgentId: null,
|
|
58
|
+
mergedSource: merged.mergedSource,
|
|
59
|
+
confidence: 0.95,
|
|
60
|
+
explanation: "AST merge completed without conflicting ownership nodes.",
|
|
61
|
+
},
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
const rounds = [];
|
|
65
|
+
let best = null;
|
|
66
|
+
for (let round = 1; round <= config.maxRounds; round += 1) {
|
|
67
|
+
const leftScore = this.scoreSide(params.left, leftDiff.summary, conflicts.length);
|
|
68
|
+
const rightScore = this.scoreSide(params.right, rightDiff.summary, conflicts.length);
|
|
69
|
+
const total = leftScore + rightScore;
|
|
70
|
+
const consensus = total === 0 ? 0 : Math.max(leftScore, rightScore) / total;
|
|
71
|
+
rounds.push({
|
|
72
|
+
round,
|
|
73
|
+
leftScore,
|
|
74
|
+
rightScore,
|
|
75
|
+
consensus,
|
|
76
|
+
reasoning: `Round ${round}: compared AST change impact, confidence hints, and conflict overlap.`,
|
|
77
|
+
});
|
|
78
|
+
const winner = leftScore >= rightScore ? params.left : params.right;
|
|
79
|
+
const winnerScore = Math.max(leftScore, rightScore);
|
|
80
|
+
best = {
|
|
81
|
+
agentId: winner.agentId,
|
|
82
|
+
score: winnerScore,
|
|
83
|
+
source: winner.source,
|
|
84
|
+
rationale: winner.rationale ?? "No rationale provided.",
|
|
85
|
+
};
|
|
86
|
+
if (consensus >= config.consensusThreshold) {
|
|
87
|
+
return {
|
|
88
|
+
id: sessionId,
|
|
89
|
+
phase: "resolved",
|
|
90
|
+
language: params.language,
|
|
91
|
+
filePath: params.filePath,
|
|
92
|
+
conflictCount: conflicts.length,
|
|
93
|
+
conflicts,
|
|
94
|
+
rounds,
|
|
95
|
+
locks,
|
|
96
|
+
resolution: {
|
|
97
|
+
method: "debate_winner",
|
|
98
|
+
winnerAgentId: best.agentId,
|
|
99
|
+
mergedSource: best.source,
|
|
100
|
+
confidence: consensus,
|
|
101
|
+
explanation: `Consensus reached in round ${round}. Winner: ${best.agentId}. ${best.rationale}`,
|
|
102
|
+
},
|
|
103
|
+
};
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
return {
|
|
107
|
+
id: sessionId,
|
|
108
|
+
phase: "escalated",
|
|
109
|
+
language: params.language,
|
|
110
|
+
filePath: params.filePath,
|
|
111
|
+
conflictCount: conflicts.length,
|
|
112
|
+
conflicts,
|
|
113
|
+
rounds,
|
|
114
|
+
locks,
|
|
115
|
+
resolution: {
|
|
116
|
+
method: "escalated",
|
|
117
|
+
winnerAgentId: best?.agentId ?? null,
|
|
118
|
+
mergedSource: params.baseSource,
|
|
119
|
+
confidence: rounds.length > 0 ? rounds[rounds.length - 1].consensus : 0,
|
|
120
|
+
explanation: "Consensus threshold not met. Escalation required.",
|
|
121
|
+
},
|
|
122
|
+
};
|
|
123
|
+
}
|
|
124
|
+
async releaseSessionLocks(session) {
|
|
125
|
+
let released = 0;
|
|
126
|
+
for (const lock of session.locks) {
|
|
127
|
+
const ok = await this.lockManager.release({ key: lock.key, token: lock.token });
|
|
128
|
+
if (ok) {
|
|
129
|
+
released += 1;
|
|
130
|
+
}
|
|
131
|
+
}
|
|
132
|
+
return released;
|
|
133
|
+
}
|
|
134
|
+
findConflicts(leftChanges, rightChanges) {
|
|
135
|
+
const rightByPath = new Map(rightChanges.map((change) => [change.path, change]));
|
|
136
|
+
const conflicts = [];
|
|
137
|
+
for (const left of leftChanges) {
|
|
138
|
+
if (left.type === "unchanged") {
|
|
139
|
+
continue;
|
|
140
|
+
}
|
|
141
|
+
const right = rightByPath.get(left.path);
|
|
142
|
+
if (!right || right.type === "unchanged") {
|
|
143
|
+
continue;
|
|
144
|
+
}
|
|
145
|
+
if (left.type === right.type) {
|
|
146
|
+
continue;
|
|
147
|
+
}
|
|
148
|
+
conflicts.push({
|
|
149
|
+
path: left.path,
|
|
150
|
+
leftType: left.type,
|
|
151
|
+
rightType: right.type,
|
|
152
|
+
});
|
|
153
|
+
}
|
|
154
|
+
return conflicts;
|
|
155
|
+
}
|
|
156
|
+
async acquireConflictLocks(conflicts, filePath, ttlMs, owners) {
|
|
157
|
+
const locks = [];
|
|
158
|
+
for (let idx = 0; idx < conflicts.length; idx += 1) {
|
|
159
|
+
const conflict = conflicts[idx];
|
|
160
|
+
const owner = owners[idx % owners.length];
|
|
161
|
+
const key = `${filePath}:${conflict.path}`;
|
|
162
|
+
const lock = await this.lockManager.acquire({
|
|
163
|
+
key,
|
|
164
|
+
ownerId: owner,
|
|
165
|
+
ttlMs,
|
|
166
|
+
metadata: {
|
|
167
|
+
filePath,
|
|
168
|
+
ownershipPath: conflict.path,
|
|
169
|
+
},
|
|
170
|
+
});
|
|
171
|
+
if (lock) {
|
|
172
|
+
locks.push(lock);
|
|
173
|
+
}
|
|
174
|
+
}
|
|
175
|
+
return locks;
|
|
176
|
+
}
|
|
177
|
+
scoreSide(side, summary, conflictCount) {
|
|
178
|
+
const rationaleBoost = side.rationale ? 0.15 : 0;
|
|
179
|
+
const productiveChanges = summary.added + summary.modified;
|
|
180
|
+
const destructivePenalty = summary.removed * 0.1;
|
|
181
|
+
const conflictPenalty = conflictCount * 0.08;
|
|
182
|
+
return Math.max(0.01, productiveChanges * 0.4 + summary.unchanged * 0.05 + rationaleBoost - destructivePenalty - conflictPenalty);
|
|
183
|
+
}
|
|
184
|
+
makeId() {
|
|
185
|
+
return `${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 10)}`;
|
|
186
|
+
}
|
|
187
|
+
}
|
|
188
|
+
//# sourceMappingURL=protocol.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"protocol.js","sourceRoot":"","sources":["../../src/negotiation/protocol.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EACL,+BAA+B,EAC/B,0BAA0B,GAE3B,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAuBnD,MAAM,CAAC,MAAM,0BAA0B,GAAsB;IAC3D,kBAAkB,EAAE,GAAG;IACvB,SAAS,EAAE,CAAC;IACZ,SAAS,EAAE,KAAM;CAClB,CAAC;AAoCF,MAAM,OAAO,oBAAoB;IACd,WAAW,CAAkC;IAE9D,YAAY,WAA6C;QACvD,IAAI,CAAC,WAAW;YACd,WAAW,IAAI,IAAI,+BAA+B,CAAC,IAAI,0BAA0B,EAAE,CAAC,CAAC;IACzF,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAQf;QACC,MAAM,MAAM,GAAsB;YAChC,GAAG,0BAA0B;YAC7B,GAAG,MAAM,CAAC,MAAM;SACjB,CAAC;QAEF,MAAM,SAAS,GAAG,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAExD,MAAM,QAAQ,GAAG,gBAAgB,CAAC;YAChC,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,YAAY,EAAE,MAAM,CAAC,UAAU;YAC/B,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM;SAChC,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,gBAAgB,CAAC;YACjC,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,YAAY,EAAE,MAAM,CAAC,UAAU;YAC/B,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;SACjC,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,OAAO,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC;QAC1E,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,SAAS,EAAE;YAC1F,MAAM,CAAC,IAAI,CAAC,OAAO;YACnB,MAAM,CAAC,KAAK,CAAC,OAAO;SACrB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,gBAAgB,CAAC;YAC9B,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM;YAC9B,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;SACjC,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,OAAO;gBACL,EAAE,EAAE,SAAS;gBACb,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,aAAa,EAAE,SAAS,CAAC,MAAM;gBAC/B,SAAS;gBACT,MAAM,EAAE,EAAE;gBACV,KAAK;gBACL,UAAU,EAAE;oBACV,MAAM,EAAE,cAAc;oBACtB,aAAa,EAAE,IAAI;oBACnB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,UAAU,EAAE,IAAI;oBAChB,WAAW,EAAE,0DAA0D;iBACxE;aACF,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAkB,EAAE,CAAC;QACjC,IAAI,IAAI,GAAiF,IAAI,CAAC;QAE9F,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,IAAI,MAAM,CAAC,SAAS,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YAC1D,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;YAClF,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;YAErF,MAAM,KAAK,GAAG,SAAS,GAAG,UAAU,CAAC;YACrC,MAAM,SAAS,GAAG,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,GAAG,KAAK,CAAC;YAE5E,MAAM,CAAC,IAAI,CAAC;gBACV,KAAK;gBACL,SAAS;gBACT,UAAU;gBACV,SAAS;gBACT,SAAS,EAAE,SAAS,KAAK,uEAAuE;aACjG,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,SAAS,IAAI,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YACpE,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YACpD,IAAI,GAAG;gBACL,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,WAAW;gBAClB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,wBAAwB;aACxD,CAAC;YAEF,IAAI,SAAS,IAAI,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC3C,OAAO;oBACL,EAAE,EAAE,SAAS;oBACb,KAAK,EAAE,UAAU;oBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,QAAQ,EAAE,MAAM,CAAC,QAAQ;oBACzB,aAAa,EAAE,SAAS,CAAC,MAAM;oBAC/B,SAAS;oBACT,MAAM;oBACN,KAAK;oBACL,UAAU,EAAE;wBACV,MAAM,EAAE,eAAe;wBACvB,aAAa,EAAE,IAAI,CAAC,OAAO;wBAC3B,YAAY,EAAE,IAAI,CAAC,MAAM;wBACzB,UAAU,EAAE,SAAS;wBACrB,WAAW,EAAE,8BAA8B,KAAK,aAAa,IAAI,CAAC,OAAO,KAAK,IAAI,CAAC,SAAS,EAAE;qBAC/F;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO;YACL,EAAE,EAAE,SAAS;YACb,KAAK,EAAE,WAAW;YAClB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,aAAa,EAAE,SAAS,CAAC,MAAM;YAC/B,SAAS;YACT,MAAM;YACN,KAAK;YACL,UAAU,EAAE;gBACV,MAAM,EAAE,WAAW;gBACnB,aAAa,EAAE,IAAI,EAAE,OAAO,IAAI,IAAI;gBACpC,YAAY,EAAE,MAAM,CAAC,UAAU;gBAC/B,UAAU,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBACvE,WAAW,EAAE,mDAAmD;aACjE;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,OAA2B;QACnD,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACjC,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;YAChF,IAAI,EAAE,EAAE,CAAC;gBACP,QAAQ,IAAI,CAAC,CAAC;YAChB,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,aAAa,CACnB,WAAkD,EAClD,YAAmD;QAEnD,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;QACjF,MAAM,SAAS,GAA0B,EAAE,CAAC;QAE5C,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;YAC/B,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC9B,SAAS;YACX,CAAC;YAED,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACzC,SAAS;YACX,CAAC;YAED,IAAI,IAAI,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI,EAAE,CAAC;gBAC7B,SAAS;YACX,CAAC;YAED,SAAS,CAAC,IAAI,CAAC;gBACb,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,SAAS,EAAE,KAAK,CAAC,IAAI;aACtB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,SAAgC,EAChC,QAAgB,EAChB,KAAa,EACb,MAAwB;QAExB,MAAM,KAAK,GAAoB,EAAE,CAAC;QAElC,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,SAAS,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC;YACnD,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;YAChC,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,GAAG,GAAG,GAAG,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;YAE3C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;gBAC1C,GAAG;gBACH,OAAO,EAAE,KAAK;gBACd,KAAK;gBACL,QAAQ,EAAE;oBACR,QAAQ;oBACR,aAAa,EAAE,QAAQ,CAAC,IAAI;iBAC7B;aACF,CAAC,CAAC;YAEH,IAAI,IAAI,EAAE,CAAC;gBACT,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnB,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,SAAS,CACf,IAAkB,EAClB,OAAgF,EAChF,aAAqB;QAErB,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACjD,MAAM,iBAAiB,GAAG,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC;QAC3D,MAAM,kBAAkB,GAAG,OAAO,CAAC,OAAO,GAAG,GAAG,CAAC;QACjD,MAAM,eAAe,GAAG,aAAa,GAAG,IAAI,CAAC;QAE7C,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,iBAAiB,GAAG,GAAG,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,GAAG,cAAc,GAAG,kBAAkB,GAAG,eAAe,CAAC,CAAC;IACpI,CAAC;IAEO,MAAM;QACZ,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;IACjF,CAAC;CACF"}
|
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
function scoreWorkUnitCount(count) {
|
|
2
|
+
if (count <= 1)
|
|
3
|
+
return 0;
|
|
4
|
+
if (count === 2)
|
|
5
|
+
return 1;
|
|
6
|
+
if (count <= 4)
|
|
7
|
+
return 2;
|
|
8
|
+
return 3;
|
|
9
|
+
}
|
|
10
|
+
function scoreCapabilityCount(count) {
|
|
11
|
+
if (count <= 1)
|
|
12
|
+
return 0;
|
|
13
|
+
if (count === 2)
|
|
14
|
+
return 1;
|
|
15
|
+
return 2;
|
|
16
|
+
}
|
|
17
|
+
function scoreLanguageCount(count) {
|
|
18
|
+
return count >= 2 ? 1 : 0;
|
|
19
|
+
}
|
|
20
|
+
function scoreDependencyDepth(depth) {
|
|
21
|
+
if (depth <= 1)
|
|
22
|
+
return 0;
|
|
23
|
+
if (depth === 2)
|
|
24
|
+
return 1;
|
|
25
|
+
return 2;
|
|
26
|
+
}
|
|
27
|
+
function scoreScopeIndicators(scopes) {
|
|
28
|
+
if (scopes.some((s) => s === "architecture"))
|
|
29
|
+
return 2;
|
|
30
|
+
if (scopes.some((s) => s === "multi-file"))
|
|
31
|
+
return 1;
|
|
32
|
+
return 0;
|
|
33
|
+
}
|
|
34
|
+
function toTier(score) {
|
|
35
|
+
if (score <= 3)
|
|
36
|
+
return "fast";
|
|
37
|
+
if (score <= 6)
|
|
38
|
+
return "strong";
|
|
39
|
+
return "frontier";
|
|
40
|
+
}
|
|
41
|
+
function buildExplanation(intent, scopes) {
|
|
42
|
+
const parts = [];
|
|
43
|
+
parts.push(`${intent.workUnits.length} work unit${intent.workUnits.length === 1 ? "" : "s"}`);
|
|
44
|
+
if (intent.capabilities.length > 0) {
|
|
45
|
+
parts.push(intent.capabilities.slice(0, 3).join(" + "));
|
|
46
|
+
}
|
|
47
|
+
const maxScope = scopes.includes("architecture")
|
|
48
|
+
? "architecture"
|
|
49
|
+
: scopes.includes("multi-file")
|
|
50
|
+
? "multi-file"
|
|
51
|
+
: scopes.includes("single-file")
|
|
52
|
+
? "single-file"
|
|
53
|
+
: "single-function";
|
|
54
|
+
parts.push(maxScope);
|
|
55
|
+
return parts.join(", ");
|
|
56
|
+
}
|
|
57
|
+
export function scoreComplexity(intent, maxDepth) {
|
|
58
|
+
const scopes = intent.workUnits.map((u) => u.scope);
|
|
59
|
+
const factors = {
|
|
60
|
+
workUnitCount: scoreWorkUnitCount(intent.workUnits.length),
|
|
61
|
+
capabilityCount: scoreCapabilityCount(intent.capabilities.length),
|
|
62
|
+
languageCount: scoreLanguageCount(intent.languages.length),
|
|
63
|
+
dependencyDepth: scoreDependencyDepth(maxDepth),
|
|
64
|
+
scopeIndicators: scoreScopeIndicators(scopes),
|
|
65
|
+
};
|
|
66
|
+
const raw = factors.workUnitCount +
|
|
67
|
+
factors.capabilityCount +
|
|
68
|
+
factors.languageCount +
|
|
69
|
+
factors.dependencyDepth +
|
|
70
|
+
factors.scopeIndicators;
|
|
71
|
+
const score = Math.max(1, Math.min(10, raw));
|
|
72
|
+
return {
|
|
73
|
+
score,
|
|
74
|
+
factors,
|
|
75
|
+
explanation: buildExplanation(intent, scopes),
|
|
76
|
+
recommendedTier: toTier(score),
|
|
77
|
+
};
|
|
78
|
+
}
|
|
79
|
+
//# sourceMappingURL=complexity_scorer.js.map
|