@treeseed/sdk 0.1.1

package/dist/d1-store.js

@@ -0,0 +1,549 @@
+import crypto from "node:crypto";
+import { applyFilters, applySort } from "./sdk-filters.js";
+import { CursorStore } from "./stores/cursor-store.js";
+import { LeaseStore } from "./stores/lease-store.js";
+import { MessageStore } from "./stores/message-store.js";
+import { RunStore } from "./stores/run-store.js";
+import { SubscriptionStore } from "./stores/subscription-store.js";
+function nowIso() {
+  return (/* @__PURE__ */ new Date()).toISOString();
+}
+function nextLeaseToken() {
+  return crypto.randomUUID();
+}
+function filterSinceField(model) {
+  switch (model) {
+    case "message":
+    case "subscription":
+      return "updated_at";
+    case "agent_run":
+      return "startedAt";
+    case "agent_cursor":
+      return "updatedAt";
+    case "content_lease":
+      return "leaseExpiresAt";
+    default:
+      return "updatedAt";
+  }
+}
+class MemoryAgentDatabase {
+  subscriptions = /* @__PURE__ */ new Map();
+  messages = /* @__PURE__ */ new Map();
+  runs = /* @__PURE__ */ new Map();
+  contentLeases = /* @__PURE__ */ new Map();
+  cursors = /* @__PURE__ */ new Map();
+  messageId = 0;
+  constructor(seed) {
+    for (const item of seed?.subscriptions ?? []) {
+      this.subscriptions.set(String(item.id ?? item.email), item);
+    }
+    for (const message of seed?.messages ?? []) {
+      this.messages.set(message.id, message);
+      this.messageId = Math.max(this.messageId, message.id);
+    }
+    for (const run of seed?.runs ?? []) {
+      this.runs.set(run.runId, run);
+    }
+    for (const cursor of seed?.cursors ?? []) {
+      this.cursors.set(`${cursor.agentSlug}:${cursor.cursorKey}`, cursor.cursorValue);
+    }
+    for (const lease of seed?.leases ?? []) {
+      this.contentLeases.set(`${lease.model}:${lease.itemKey}`, lease);
+    }
+  }
+  rowsForModel(model) {
+    if (model === "subscription") {
+      return [...this.subscriptions.values()];
+    }
+    if (model === "message") {
+      return [...this.messages.values()];
+    }
+    if (model === "agent_run") {
+      return [...this.runs.values()];
+    }
+    if (model === "agent_cursor") {
+      return [...this.cursors.entries()].map(([key, value]) => {
+        const [agentSlug, cursorKey] = key.split(":", 2);
+        return {
+          agentSlug,
+          cursorKey,
+          cursorValue: value,
+          updatedAt: null
+        };
+      });
+    }
+    if (model === "content_lease") {
+      return [...this.contentLeases.values()].map((lease) => ({
+        model: lease.model,
+        itemKey: lease.itemKey,
+        claimedBy: lease.claimedBy,
+        claimedAt: lease.claimedAt,
+        leaseExpiresAt: lease.leaseExpiresAt,
+        token: lease.token
+      }));
+    }
+    throw new Error(`Unsupported D1 model "${model}".`);
+  }
+  async get(request) {
+    const key = String(request.id ?? request.slug ?? request.key ?? "");
+    if (request.model === "agent_cursor") {
+      if (!key) {
+        return null;
+      }
+      const [agentSlug, cursorKey] = key.split(":", 2);
+      const value = this.cursors.get(`${agentSlug}:${cursorKey}`);
+      return value ? {
+        agentSlug,
+        cursorKey,
+        cursorValue: value,
+        updatedAt: null
+      } : null;
+    }
+    if (request.model === "content_lease") {
+      const lease = this.contentLeases.get(key);
+      return lease ? {
+        model: lease.model,
+        itemKey: lease.itemKey,
+        claimedBy: lease.claimedBy,
+        claimedAt: lease.claimedAt,
+        leaseExpiresAt: lease.leaseExpiresAt,
+        token: lease.token
+      } : null;
+    }
+    return this.rowsForModel(request.model).find(
+      (row) => [row.id, row.email, row.runId].map((value) => String(value ?? "")).includes(key)
+    ) ?? null;
+  }
+  async search(request) {
+    const filtered = applyFilters(this.rowsForModel(request.model), request.filters);
+    const sorted = applySort(filtered, request.sort);
+    return sorted.slice(0, request.limit ?? sorted.length);
+  }
+  async follow(request) {
+    const filters = [
+      ...request.filters ?? [],
+      {
+        field: filterSinceField(request.model),
+        op: "updated_since",
+        value: request.since
+      }
+    ];
+    return {
+      items: await this.search({
+        model: request.model,
+        filters
+      }),
+      since: request.since
+    };
+  }
+  async pick(request) {
+    if (request.model === "message") {
+      const item = await this.claimMessage({
+        workerId: request.workerId,
+        messageTypes: request.filters?.filter((filter) => filter.field === "type" && filter.op === "in").flatMap((filter) => Array.isArray(filter.value) ? filter.value.map(String) : []),
+        leaseSeconds: request.leaseSeconds
+      });
+      return {
+        item,
+        leaseToken: item ? nextLeaseToken() : null
+      };
+    }
+    if (request.model === "content_lease") {
+      const item = (await this.search({
+        model: request.model,
+        filters: request.filters,
+        sort: [{ field: "leaseExpiresAt", direction: "desc" }],
+        limit: 1
+      }))[0];
+      return {
+        item: item ?? null,
+        leaseToken: item ? String(item.token) : null
+      };
+    }
+    const items = await this.search({
+      model: request.model,
+      filters: request.filters,
+      sort: [{ field: filterSinceField(request.model), direction: "desc" }]
+    });
+    return {
+      item: items[0] ?? null,
+      leaseToken: null
+    };
+  }
+  async create(request) {
+    switch (request.model) {
+      case "message":
+        return await this.createMessage({
+          type: String(request.data.type ?? "message.created"),
+          payload: request.data.payload ?? request.data,
+          relatedModel: typeof request.data.relatedModel === "string" ? request.data.relatedModel : null,
+          relatedId: typeof request.data.relatedId === "string" ? request.data.relatedId : null,
+          priority: Number(request.data.priority ?? 0),
+          maxAttempts: Number(request.data.maxAttempts ?? 3),
+          actor: request.actor
+        });
+      case "subscription": {
+        const record = {
+          id: this.subscriptions.size + 1,
+          email: String(request.data.email ?? ""),
+          name: request.data.name ? String(request.data.name) : null,
+          status: String(request.data.status ?? "active"),
+          source: String(request.data.source ?? "sdk"),
+          consent_at: String(request.data.consent_at ?? nowIso()),
+          created_at: String(request.data.created_at ?? nowIso()),
+          updated_at: String(request.data.updated_at ?? nowIso()),
+          ip_hash: String(request.data.ip_hash ?? "")
+        };
+        this.subscriptions.set(String(record.id), record);
+        return record;
+      }
+      case "agent_run":
+        return this.recordRun({ run: request.data });
+      case "agent_cursor": {
+        const agentSlug = String(request.data.agentSlug ?? "");
+        const cursorKey = String(request.data.cursorKey ?? "");
+        const cursorValue = String(request.data.cursorValue ?? "");
+        this.cursors.set(`${agentSlug}:${cursorKey}`, cursorValue);
+        return {
+          agentSlug,
+          cursorKey,
+          cursorValue,
+          updatedAt: nowIso()
+        };
+      }
+      case "content_lease": {
+        const token = await this.tryClaimContentLease({
+          model: String(request.data.model ?? ""),
+          itemKey: String(request.data.itemKey ?? ""),
+          claimedBy: String(request.data.claimedBy ?? request.actor),
+          leaseSeconds: Number(request.data.leaseSeconds ?? 300)
+        });
+        const lease = this.contentLeases.get(`${request.data.model}:${request.data.itemKey}`);
+        return {
+          model: String(request.data.model ?? ""),
+          itemKey: String(request.data.itemKey ?? ""),
+          claimedBy: String(request.data.claimedBy ?? request.actor),
+          claimedAt: String(lease?.claimedAt ?? nowIso()),
+          leaseExpiresAt: String(lease?.leaseExpiresAt ?? nowIso()),
+          token: String(token ?? lease?.token ?? "")
+        };
+      }
+      default:
+        throw new Error(`Unsupported D1 create model "${request.model}".`);
+    }
+  }
+  async update(request) {
+    switch (request.model) {
+      case "message": {
+        const current = this.messages.get(Number(request.id ?? request.key ?? request.data.id ?? 0));
+        if (!current) {
+          return null;
+        }
+        const next = {
+          ...current,
+          ...request.data,
+          updatedAt: nowIso()
+        };
+        this.messages.set(next.id, next);
+        return next;
+      }
+      case "subscription": {
+        const key = String(request.id ?? request.key ?? request.data.email ?? "");
+        const current = await this.get({ model: "subscription", key });
+        if (!current) {
+          return null;
+        }
+        const next = {
+          ...current,
+          ...request.data,
+          updated_at: nowIso()
+        };
+        this.subscriptions.set(String(next.id ?? next.email), next);
+        return next;
+      }
+      case "agent_run":
+        return this.recordRun({ run: { ...request.data, runId: request.id ?? request.key ?? request.data.runId } });
+      case "agent_cursor":
+        return this.create({
+          model: "agent_cursor",
+          data: request.data,
+          actor: request.actor
+        });
+      case "content_lease":
+        return this.create({
+          model: "content_lease",
+          data: request.data,
+          actor: request.actor
+        });
+      default:
+        throw new Error(`Unsupported D1 update model "${request.model}".`);
+    }
+  }
+  async claimMessage(request) {
+    const pending = [...this.messages.values()].filter(
+      (message) => (message.status === "pending" || message.status === "failed") && new Date(message.availableAt).valueOf() <= Date.now() && (!request.messageTypes?.length || request.messageTypes.includes(message.type))
+    ).sort((left, right) => right.priority - left.priority || left.availableAt.localeCompare(right.availableAt))[0];
+    if (!pending) {
+      return null;
+    }
+    const claimedAt = nowIso();
+    const next = {
+      ...pending,
+      status: "claimed",
+      claimedBy: request.workerId,
+      claimedAt,
+      leaseExpiresAt: new Date(Date.now() + request.leaseSeconds * 1e3).toISOString(),
+      attempts: pending.attempts + 1,
+      updatedAt: claimedAt
+    };
+    this.messages.set(next.id, next);
+    return next;
+  }
+  async ackMessage(request) {
+    const current = this.messages.get(request.id);
+    if (!current) {
+      return;
+    }
+    this.messages.set(request.id, {
+      ...current,
+      status: request.status,
+      updatedAt: nowIso()
+    });
+  }
+  async createMessage(request) {
+    this.messageId += 1;
+    const record = {
+      id: this.messageId,
+      type: request.type,
+      status: "pending",
+      payloadJson: JSON.stringify(request.payload),
+      relatedModel: request.relatedModel ?? null,
+      relatedId: request.relatedId ?? null,
+      priority: request.priority ?? 0,
+      availableAt: nowIso(),
+      claimedBy: null,
+      claimedAt: null,
+      leaseExpiresAt: null,
+      attempts: 0,
+      maxAttempts: request.maxAttempts ?? 3,
+      createdAt: nowIso(),
+      updatedAt: nowIso()
+    };
+    this.messages.set(record.id, record);
+    return record;
+  }
+  async recordRun(request) {
+    const run = request.run;
+    this.runs.set(String(run.runId), run);
+    return run;
+  }
+  async getCursor(request) {
+    return this.cursors.get(`${request.agentSlug}:${request.cursorKey}`) ?? null;
+  }
+  async upsertCursor(request) {
+    this.cursors.set(`${request.agentSlug}:${request.cursorKey}`, request.cursorValue);
+  }
+  async releaseLease(request) {
+    this.contentLeases.delete(`${request.model}:${request.itemKey}`);
+  }
+  async tryClaimContentLease(input) {
+    const key = `${input.model}:${input.itemKey}`;
+    const existing = this.contentLeases.get(key);
+    if (existing && new Date(existing.leaseExpiresAt).valueOf() > Date.now()) {
+      return null;
+    }
+    const token = nextLeaseToken();
+    this.contentLeases.set(key, {
+      model: input.model,
+      itemKey: input.itemKey,
+      claimedBy: input.claimedBy,
+      claimedAt: nowIso(),
+      leaseExpiresAt: new Date(Date.now() + input.leaseSeconds * 1e3).toISOString(),
+      token
+    });
+    return token;
+  }
+  async releaseAllLeases() {
+    const count = this.contentLeases.size;
+    this.contentLeases.clear();
+    return count;
+  }
+  inspectRuns() {
+    return [...this.runs.values()];
+  }
+  inspectLeases() {
+    return [...this.contentLeases.values()];
+  }
+}
+class CloudflareD1AgentDatabase {
+  constructor(db) {
+    this.db = db;
+    this.subscriptions = new SubscriptionStore(db);
+    this.messages = new MessageStore(db);
+    this.runs = new RunStore(db);
+    this.cursors = new CursorStore(db);
+    this.leases = new LeaseStore(db);
+  }
+  db;
+  subscriptions;
+  messages;
+  runs;
+  cursors;
+  leases;
+  async get(request) {
+    if (request.model === "subscription") {
+      return this.subscriptions.getByKey(String(request.id ?? request.slug ?? request.key ?? ""));
+    }
+    if (request.model === "message") {
+      return this.messages.getById(Number(request.id ?? request.slug ?? request.key ?? 0));
+    }
+    if (request.model === "agent_run") {
+      return this.runs.getByKey(String(request.id ?? request.key ?? request.slug ?? ""));
+    }
+    if (request.model === "agent_cursor") {
+      return this.cursors.getByKey(String(request.id ?? request.key ?? request.slug ?? ""));
+    }
+    if (request.model === "content_lease") {
+      return this.leases.getByKey(String(request.id ?? request.key ?? request.slug ?? ""));
+    }
+    throw new Error(`Unsupported D1 get model "${request.model}".`);
+  }
+  async search(request) {
+    if (request.model === "subscription") {
+      return this.subscriptions.search(request);
+    }
+    if (request.model === "message") {
+      return this.messages.search(request);
+    }
+    if (request.model === "agent_run") {
+      return this.runs.search(request);
+    }
+    if (request.model === "agent_cursor") {
+      return this.cursors.search(request);
+    }
+    if (request.model === "content_lease") {
+      return this.leases.search(request);
+    }
+    throw new Error(`Unsupported D1 search model "${request.model}".`);
+  }
+  async follow(request) {
+    const field = request.model === "subscription" || request.model === "message" ? "updated_at" : request.model === "agent_run" ? "started_at" : request.model === "agent_cursor" ? "updated_at" : "lease_expires_at";
+    return this.search({
+      model: request.model,
+      filters: [
+        ...request.filters ?? [],
+        { field, op: "updated_since", value: request.since }
+      ]
+    }).then((items) => ({ items, since: request.since }));
+  }
+  async pick(request) {
+    if (request.model === "message") {
+      const claimed = await this.claimMessage({
+        workerId: request.workerId,
+        messageTypes: request.filters?.filter((filter) => filter.field === "type" && filter.op === "in").flatMap((filter) => Array.isArray(filter.value) ? filter.value.map(String) : []),
+        leaseSeconds: request.leaseSeconds
+      });
+      return {
+        item: claimed,
+        leaseToken: claimed ? nextLeaseToken() : null
+      };
+    }
+    if (request.model === "content_lease") {
+      const items = await this.leases.search({
+        model: "content_lease",
+        filters: request.filters,
+        sort: [{ field: "lease_expires_at", direction: "desc" }],
+        limit: 1
+      });
+      return {
+        item: items[0] ?? null,
+        leaseToken: items[0]?.token ?? null
+      };
+    }
+    return {
+      item: null,
+      leaseToken: null
+    };
+  }
+  async create(request) {
+    if (request.model === "message") {
+      return await this.createMessage({
+        type: String(request.data.type ?? "message.created"),
+        payload: request.data.payload ?? request.data,
+        relatedModel: typeof request.data.relatedModel === "string" ? request.data.relatedModel : null,
+        relatedId: typeof request.data.relatedId === "string" ? request.data.relatedId : null,
+        priority: Number(request.data.priority ?? 0),
+        maxAttempts: Number(request.data.maxAttempts ?? 3),
+        actor: request.actor
+      });
+    }
+    if (request.model === "subscription") {
+      return await this.subscriptions.create(request);
+    }
+    if (request.model === "agent_run") {
+      return await this.runs.record({ run: request.data });
+    }
+    if (request.model === "agent_cursor") {
+      return await this.cursors.update({
+        ...request,
+        model: "agent_cursor"
+      });
+    }
+    if (request.model === "content_lease") {
+      return await this.leases.update({
+        ...request,
+        model: "content_lease"
+      });
+    }
+    throw new Error(`Unsupported D1 create model "${request.model}".`);
+  }
+  async update(request) {
+    if (request.model === "message") {
+      return this.messages.update(request);
+    }
+    if (request.model === "subscription") {
+      return this.subscriptions.update(request);
+    }
+    if (request.model === "agent_run") {
+      return this.runs.update(request);
+    }
+    if (request.model === "agent_cursor") {
+      return this.cursors.update(request);
+    }
+    if (request.model === "content_lease") {
+      return this.leases.update(request);
+    }
+    throw new Error(`Unsupported D1 update model "${request.model}".`);
+  }
+  claimMessage(request) {
+    return this.messages.claim(request);
+  }
+  ackMessage(request) {
+    return this.messages.ack(request);
+  }
+  createMessage(request) {
+    return this.messages.create(request);
+  }
+  recordRun(request) {
+    return this.runs.record(request);
+  }
+  getCursor(request) {
+    return this.cursors.get(request);
+  }
+  upsertCursor(request) {
+    return this.cursors.upsert(request);
+  }
+  releaseLease(request) {
+    return this.leases.release(request);
+  }
+  tryClaimContentLease(input) {
+    return this.leases.tryClaim(input);
+  }
+  releaseAllLeases() {
+    return this.leases.releaseAll();
+  }
+}
+export {
+  CloudflareD1AgentDatabase,
+  MemoryAgentDatabase
+};

package/dist/frontmatter.js

@@ -0,0 +1,33 @@
+import { parse as parseYaml, stringify as stringifyYaml } from "yaml";
+function parseFrontmatterDocument(source) {
+  if (!source.startsWith("---\n")) {
+    return {
+      frontmatter: {},
+      body: source
+    };
+  }
+  const delimiterIndex = source.indexOf("\n---\n", 4);
+  if (delimiterIndex < 0) {
+    return {
+      frontmatter: {},
+      body: source
+    };
+  }
+  const yamlSource = source.slice(4, delimiterIndex);
+  const body = source.slice(delimiterIndex + 5);
+  const parsed = parseYaml(yamlSource);
+  return {
+    frontmatter: parsed && typeof parsed === "object" && !Array.isArray(parsed) ? parsed : {},
+    body
+  };
+}
+function serializeFrontmatterDocument(frontmatter, body) {
+  return `---
+${stringifyYaml(frontmatter).trimEnd()}
+---
+${body.replace(/^\n*/, "")}`;
+}
+export {
+  parseFrontmatterDocument,
+  serializeFrontmatterDocument
+};

package/dist/git-runtime.js

@@ -0,0 +1,67 @@
+import { execFile } from "node:child_process";
+import path from "node:path";
+import { promisify } from "node:util";
+const execFileAsync = promisify(execFile);
+class GitRuntime {
+  constructor(repoRoot, disabled = false) {
+    this.repoRoot = repoRoot;
+    this.disabled = disabled;
+  }
+  repoRoot;
+  disabled;
+  async currentBranch() {
+    const { stdout } = await execFileAsync("git", ["rev-parse", "--abbrev-ref", "HEAD"], {
+      cwd: this.repoRoot
+    });
+    return stdout.trim();
+  }
+  async ensureWorktree(branchName) {
+    const worktreePath = path.join(this.repoRoot, ".agent-worktrees", branchName);
+    if (this.disabled) {
+      return worktreePath;
+    }
+    try {
+      await execFileAsync("git", ["worktree", "list", "--porcelain"], {
+        cwd: this.repoRoot
+      }).then(async ({ stdout }) => {
+        if (stdout.includes(`worktree ${worktreePath}`)) {
+          await execFileAsync("git", ["switch", branchName], { cwd: worktreePath });
+          return;
+        }
+        await execFileAsync("git", ["worktree", "add", "-B", branchName, worktreePath, "HEAD"], {
+          cwd: this.repoRoot
+        });
+      });
+    } catch {
+      await execFileAsync("git", ["worktree", "add", "-B", branchName, worktreePath, "HEAD"], {
+        cwd: this.repoRoot
+      });
+    }
+    return worktreePath;
+  }
+  async commitFileChange(filePath, branchName, commitMessage) {
+    return this.commitFileChanges([filePath], branchName, commitMessage);
+  }
+  async commitFileChanges(filePaths, branchName, commitMessage) {
+    const worktreePath = await this.ensureWorktree(branchName);
+    if (this.disabled) {
+      return { branchName, commitMessage, worktreePath, commitSha: null, changedPaths: filePaths };
+    }
+    const relativeFilePaths = filePaths.map((filePath) => path.relative(worktreePath, filePath));
+    await execFileAsync("git", ["add", ...relativeFilePaths], { cwd: worktreePath });
+    try {
+      await execFileAsync("git", ["commit", "-m", commitMessage], { cwd: worktreePath });
+    } catch (error) {
+      const message = error && typeof error === "object" && "stderr" in error ? String(error.stderr ?? "") : "";
+      if (!message.includes("nothing to commit")) {
+        throw error;
+      }
+    }
+    const { stdout } = await execFileAsync("git", ["rev-parse", "HEAD"], { cwd: worktreePath });
+    const commitSha = stdout.trim();
+    return { branchName, commitMessage, worktreePath, commitSha, changedPaths: filePaths };
+  }
+}
+export {
+  GitRuntime
+};

package/dist/index.js

@@ -0,0 +1,12 @@
+import { AgentSdk, ScopedAgentSdk } from "./sdk.js";
+import { MODEL_REGISTRY, buildModelRegistry, resolveModelDefinition } from "./model-registry.js";
+import { normalizeAgentCliOptions, buildCopilotAllowToolArgs } from "./cli-tools.js";
+export {
+  AgentSdk,
+  MODEL_REGISTRY,
+  ScopedAgentSdk,
+  buildCopilotAllowToolArgs,
+  buildModelRegistry,
+  normalizeAgentCliOptions,
+  resolveModelDefinition
+};