@treeseed/sdk 0.12.62 → 0.13.0-rc.10

package/dist/platform-operation-store.js

@@ -1,531 +0,0 @@
-import { randomUUID } from "node:crypto";
-import { NodeSqliteD1Database } from "./db/node-sqlite.js";
-const PLATFORM_OPERATION_SCHEMA_SQL = `
-CREATE TABLE IF NOT EXISTS platform_operations (
-	id TEXT PRIMARY KEY,
-	namespace TEXT NOT NULL,
-	operation TEXT NOT NULL,
-	status TEXT NOT NULL,
-	target TEXT NOT NULL,
-	idempotency_key TEXT,
-	input_json TEXT NOT NULL DEFAULT '{}',
-	output_json TEXT,
-	error_json TEXT,
-	requested_by_type TEXT NOT NULL,
-	requested_by_id TEXT,
-	assigned_runner_id TEXT,
-	lease_expires_at TEXT,
-	created_at TEXT NOT NULL,
-	updated_at TEXT NOT NULL,
-	started_at TEXT,
-	finished_at TEXT,
-	cancelled_at TEXT
-);
-
-CREATE UNIQUE INDEX IF NOT EXISTS idx_platform_operations_idempotency
-	ON platform_operations(namespace, operation, idempotency_key)
-	WHERE idempotency_key IS NOT NULL;
-
-CREATE INDEX IF NOT EXISTS idx_platform_operations_runnable
-	ON platform_operations(status, created_at ASC);
-
-CREATE TABLE IF NOT EXISTS platform_operation_events (
-	id TEXT PRIMARY KEY,
-	operation_id TEXT NOT NULL,
-	seq INTEGER NOT NULL,
-	kind TEXT NOT NULL,
-	data_json TEXT NOT NULL DEFAULT '{}',
-	created_at TEXT NOT NULL,
-	FOREIGN KEY (operation_id) REFERENCES platform_operations(id) ON DELETE CASCADE
-);
-
-CREATE UNIQUE INDEX IF NOT EXISTS idx_platform_operation_events_seq
-	ON platform_operation_events(operation_id, seq);
-
-CREATE TABLE IF NOT EXISTS market_operation_runners (
-	id TEXT PRIMARY KEY,
-	runner_key TEXT NOT NULL UNIQUE,
-	name TEXT NOT NULL,
-	environment TEXT NOT NULL,
-	status TEXT NOT NULL DEFAULT 'online',
-	version TEXT,
-	capabilities_json TEXT NOT NULL DEFAULT '[]',
-	active_job_count INTEGER NOT NULL DEFAULT 0,
-	max_concurrent_jobs INTEGER NOT NULL DEFAULT 1,
-	heartbeat_at TEXT,
-	metadata_json TEXT NOT NULL DEFAULT '{}',
-	created_at TEXT NOT NULL,
-	updated_at TEXT NOT NULL
-);
-
-CREATE TABLE IF NOT EXISTS platform_repository_claims (
-	id TEXT PRIMARY KEY,
-	repository_key TEXT NOT NULL,
-	runner_id TEXT NOT NULL,
-	workspace_path TEXT NOT NULL,
-	branch TEXT,
-	commit_sha TEXT,
-	claim_state TEXT NOT NULL DEFAULT 'active',
-	lease_expires_at TEXT,
-	metadata_json TEXT NOT NULL DEFAULT '{}',
-	created_at TEXT NOT NULL,
-	updated_at TEXT NOT NULL
-);
-
-CREATE UNIQUE INDEX IF NOT EXISTS idx_platform_repository_claims_active
-	ON platform_repository_claims(repository_key, runner_id)
-	WHERE claim_state = 'active';
-
-CREATE INDEX IF NOT EXISTS idx_platform_repository_claims_runner
-	ON platform_repository_claims(runner_id, claim_state);
-`;
-function isoNow(now) {
-  return now().toISOString();
-}
-function parseJson(value, fallback) {
-  if (typeof value !== "string" || !value) return fallback;
-  try {
-    return JSON.parse(value);
-  } catch {
-    return fallback;
-  }
-}
-function rowOperation(row) {
-  if (!row) return null;
-  return {
-    id: String(row.id),
-    namespace: String(row.namespace),
-    operation: String(row.operation),
-    status: String(row.status),
-    target: String(row.target),
-    idempotencyKey: row.idempotency_key == null ? null : String(row.idempotency_key),
-    input: parseJson(row.input_json, {}),
-    output: parseJson(row.output_json, null),
-    error: parseJson(row.error_json, null),
-    requestedByType: String(row.requested_by_type),
-    requestedById: row.requested_by_id == null ? null : String(row.requested_by_id),
-    assignedRunnerId: row.assigned_runner_id == null ? null : String(row.assigned_runner_id),
-    leaseExpiresAt: row.lease_expires_at == null ? null : String(row.lease_expires_at),
-    createdAt: String(row.created_at),
-    updatedAt: String(row.updated_at),
-    startedAt: row.started_at == null ? null : String(row.started_at),
-    finishedAt: row.finished_at == null ? null : String(row.finished_at),
-    cancelledAt: row.cancelled_at == null ? null : String(row.cancelled_at)
-  };
-}
-function rowEvent(row) {
-  if (!row) return null;
-  return {
-    id: String(row.id),
-    operationId: String(row.operation_id),
-    seq: Number(row.seq),
-    kind: String(row.kind),
-    data: parseJson(row.data_json, {}),
-    createdAt: String(row.created_at)
-  };
-}
-function repositoryKey(repository = {}) {
-  return [repository.provider ?? "git", repository.owner ?? "local", repository.name ?? "repository"].join("-").toLowerCase().replace(/[^a-z0-9.-]+/gu, "-").replace(/^-+|-+$/gu, "") || "repository";
-}
-function repositoryWorkspacePath(workspaceRoot, repository = {}) {
-  const root = String(workspaceRoot ?? "/data").replace(/\/+$/u, "") || "/data";
-  return `${root}/repositories/${repositoryKey(repository)}/repo`;
-}
-function normalizeOperationCapabilities(capabilities) {
-  return Array.isArray(capabilities) ? capabilities.map((entry) => String(entry ?? "").trim()).filter(Boolean) : [];
-}
-function convertQuestionPlaceholders(query) {
-  let index = 0;
-  return query.replace(/\?/gu, () => `$${++index}`);
-}
-function createD1RelationalAdapter(db) {
-  return {
-    provider: "d1",
-    async run(query, params = []) {
-      await db.prepare(query).bind(...params).run();
-    },
-    async first(query, params = []) {
-      return db.prepare(query).bind(...params).first();
-    },
-    async all(query, params = []) {
-      const result = await db.prepare(query).bind(...params).all();
-      return Array.isArray(result) ? result : result.results ?? [];
-    },
-    async exec(query) {
-      if (!db.exec) {
-        for (const statement of query.split(/;\s*/u).map((entry) => entry.trim()).filter(Boolean)) {
-          await this.run(statement);
-        }
-        return;
-      }
-      await db.exec(query);
-    }
-  };
-}
-function createSqliteRelationalAdapter(path) {
-  const database = new NodeSqliteD1Database(path);
-  return {
-    ...createD1RelationalAdapter(database),
-    provider: "sqlite",
-    close: () => database.close()
-  };
-}
-async function createPostgresRelationalAdapter(databaseUrl) {
-  const importer = new Function("specifier", "return import(specifier)");
-  const { Pool } = await importer("pg");
-  const pool = new Pool({ connectionString: databaseUrl });
-  return {
-    provider: "postgres",
-    async run(query, params = []) {
-      await pool.query(convertQuestionPlaceholders(query), params);
-    },
-    async first(query, params = []) {
-      const result = await pool.query(convertQuestionPlaceholders(query), params);
-      return result.rows[0] ?? null;
-    },
-    async all(query, params = []) {
-      const result = await pool.query(convertQuestionPlaceholders(query), params);
-      return result.rows;
-    },
-    async exec(query) {
-      await pool.query(query);
-    },
-    async transaction(callback) {
-      await pool.query("BEGIN");
-      try {
-        const result = await callback();
-        await pool.query("COMMIT");
-        return result;
-      } catch (error) {
-        await pool.query("ROLLBACK");
-        throw error;
-      }
-    },
-    close: () => pool.end()
-  };
-}
-async function createRelationalAdapterFromUrl(databaseUrl) {
-  const value = databaseUrl.trim();
-  if (/^postgres(ql)?:\/\//iu.test(value)) return createPostgresRelationalAdapter(value);
-  if (/^sqlite:\/\//iu.test(value)) return createSqliteRelationalAdapter(value.replace(/^sqlite:\/\//iu, ""));
-  return createSqliteRelationalAdapter(value);
-}
-async function createPlatformOperationStoreFromEnv(options = {}) {
-  const databaseUrl = options.databaseUrl ?? globalThis.process?.env?.TREESEED_DATABASE_URL ?? null;
-  if (!databaseUrl?.trim()) throw new Error("TREESEED_DATABASE_URL is required for direct database platform operations.");
-  const database = await createRelationalAdapterFromUrl(databaseUrl);
-  return new PlatformOperationStore({ database, initializeSchema: options.initializeSchema ?? true });
-}
-class PlatformOperationStore {
-  initialized = false;
-  database;
-  now;
-  initializeSchema;
-  constructor(options) {
-    this.database = options.database;
-    this.now = options.now ?? (() => /* @__PURE__ */ new Date());
-    this.initializeSchema = options.initializeSchema ?? true;
-  }
-  async close() {
-    await this.database.close?.();
-  }
-  async ensureInitialized() {
-    if (this.initialized) return;
-    if (this.initializeSchema) {
-      if (this.database.exec) await this.database.exec(PLATFORM_OPERATION_SCHEMA_SQL);
-      else {
-        for (const statement of PLATFORM_OPERATION_SCHEMA_SQL.split(/;\s*/u).map((entry) => entry.trim()).filter(Boolean)) {
-          await this.database.run(statement);
-        }
-      }
-    }
-    this.initialized = true;
-  }
-  async appendPlatformOperationEvent(operationId, kind, data = {}) {
-    await this.ensureInitialized();
-    const row = await this.database.first(
-      `SELECT COALESCE(MAX(seq), 0) + 1 AS next_seq FROM platform_operation_events WHERE operation_id = ?`,
-      [operationId]
-    );
-    const seq = Number(row?.next_seq ?? 1);
-    const timestamp = isoNow(this.now);
-    const id = randomUUID();
-    await this.database.run(
-      `INSERT INTO platform_operation_events (id, operation_id, seq, kind, data_json, created_at)
-			 VALUES (?, ?, ?, ?, ?, ?)`,
-      [id, operationId, seq, kind, JSON.stringify(data ?? {}), timestamp]
-    );
-    return rowEvent(await this.database.first(`SELECT * FROM platform_operation_events WHERE id = ?`, [id]));
-  }
-  async register(request) {
-    return { ok: true, runner: await this.upsertRunner(request) };
-  }
-  async heartbeat(request) {
-    return { ok: true, runner: await this.upsertRunner(request) };
-  }
-  async upsertRunner(input) {
-    await this.ensureInitialized();
-    const timestamp = isoNow(this.now);
-    const id = input.runnerId;
-    await this.database.run(
-      `INSERT INTO market_operation_runners (
-				id, runner_key, name, environment, status, version, capabilities_json,
-				active_job_count, max_concurrent_jobs, heartbeat_at, metadata_json, created_at, updated_at
-			) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-			ON CONFLICT(id) DO UPDATE SET
-				runner_key = excluded.runner_key,
-				name = excluded.name,
-				environment = excluded.environment,
-				status = excluded.status,
-				version = excluded.version,
-				capabilities_json = excluded.capabilities_json,
-				active_job_count = excluded.active_job_count,
-				max_concurrent_jobs = excluded.max_concurrent_jobs,
-				heartbeat_at = excluded.heartbeat_at,
-				metadata_json = excluded.metadata_json,
-				updated_at = excluded.updated_at`,
-      [
-        id,
-        ("runnerKey" in input ? input.runnerKey : void 0) ?? id,
-        ("name" in input ? input.name : void 0) ?? id,
-        input.environment ?? "unknown",
-        ("status" in input ? input.status : void 0) ?? "online",
-        input.version ?? null,
-        JSON.stringify(Array.isArray(input.capabilities) ? input.capabilities : []),
-        Math.max(0, Number(("activeJobCount" in input ? input.activeJobCount : void 0) ?? 0) || 0),
-        Math.max(1, Number(input.maxConcurrentJobs ?? 1) || 1),
-        timestamp,
-        JSON.stringify(input.metadata ?? {}),
-        timestamp,
-        timestamp
-      ]
-    );
-    return this.database.first(`SELECT * FROM market_operation_runners WHERE id = ?`, [id]);
-  }
-  async getOperation(operationId) {
-    await this.ensureInitialized();
-    const operation = rowOperation(await this.database.first(`SELECT * FROM platform_operations WHERE id = ?`, [operationId]));
-    if (!operation) throw new Error(`Unknown platform operation "${operationId}".`);
-    return { ok: true, operation };
-  }
-  async claimJob(input) {
-    await this.ensureInitialized();
-    const runnerId = input.runnerId;
-    const leaseSeconds = Math.max(30, Math.min(Number(input.leaseSeconds ?? 300), 3600));
-    const now = isoNow(this.now);
-    const leaseExpiresAt = new Date(this.now().getTime() + leaseSeconds * 1e3).toISOString();
-    const capabilities = normalizeOperationCapabilities(input.capabilities);
-    const capabilityWhere = capabilities.length > 0 ? ` AND (${capabilities.map(() => `(namespace || ':' || operation) = ?`).join(" OR ")})` : "";
-    const capabilityParams = capabilities;
-    const rows = input.operationId ? await this.database.all(
-      `SELECT * FROM platform_operations
-				 WHERE id = ? AND (
-				    status = 'queued'
-				    OR (status = 'leased' AND lease_expires_at IS NOT NULL AND lease_expires_at < ?)
-				 )
-				 ${capabilityWhere}
-				 ORDER BY created_at ASC LIMIT 1`,
-      [input.operationId, now, ...capabilityParams]
-    ) : await this.database.all(
-      `SELECT * FROM platform_operations
-				 WHERE (
-				    status = 'queued'
-				    OR (status = 'leased' AND lease_expires_at IS NOT NULL AND lease_expires_at < ?)
-				 )
-				 ${capabilityWhere}
-				 ORDER BY created_at ASC LIMIT 1`,
-      [now, ...capabilityParams]
-    );
-    const row = rows[0];
-    if (!row) return { ok: true, operation: null };
-    await this.database.run(
-      `UPDATE platform_operations
-			 SET status = 'leased',
-			     assigned_runner_id = ?,
-			     lease_expires_at = ?,
-			     started_at = COALESCE(started_at, ?),
-			     updated_at = ?
-			 WHERE id = ?`,
-      [runnerId, leaseExpiresAt, now, now, row.id]
-    );
-    await this.appendPlatformOperationEvent(String(row.id), "claimed", { runnerId, leaseExpiresAt });
-    const operation = rowOperation(await this.database.first(`SELECT * FROM platform_operations WHERE id = ?`, [row.id]));
-    if (operation?.input?.repository && typeof operation.input.repository === "object" && !Array.isArray(operation.input.repository)) {
-      const runner = await this.database.first(`SELECT * FROM market_operation_runners WHERE id = ?`, [runnerId]);
-      const metadata = parseJson(runner?.metadata_json, {});
-      const workspaceRoot = metadata.dataDir ?? "/data";
-      const repository = operation.input.repository;
-      await this.upsertRepositoryClaim({
-        runnerId,
-        repository,
-        workspaceRoot,
-        branch: String(repository.defaultBranch ?? ""),
-        leaseSeconds,
-        metadata: { operationId: operation.id, namespace: operation.namespace, operation: operation.operation }
-      });
-    }
-    return { ok: true, operation };
-  }
-  async assertRunnerUpdate(operationId, runnerId) {
-    const operation = (await this.getOperation(operationId)).operation;
-    if (!runnerId) throw new Error("runnerId is required.");
-    if (operation.assignedRunnerId !== runnerId) throw new Error("Platform operation is assigned to a different runner.");
-    if (["succeeded", "failed", "cancelled"].includes(operation.status)) throw new Error(`Platform operation is already ${operation.status}.`);
-    return operation;
-  }
-  async appendEvent(operationId, request) {
-    await this.assertRunnerUpdate(operationId, request.runnerId);
-    return { ok: true, event: await this.appendPlatformOperationEvent(operationId, request.event?.kind ?? "event", request.event?.data ?? {}) };
-  }
-  async renewLease(operationId, request) {
-    await this.assertRunnerUpdate(operationId, request.runnerId);
-    const leaseSeconds = Math.max(30, Math.min(Number(request.leaseSeconds ?? 300), 3600));
-    const timestamp = isoNow(this.now);
-    const leaseExpiresAt = new Date(this.now().getTime() + leaseSeconds * 1e3).toISOString();
-    await this.database.run(
-      `UPDATE platform_operations SET lease_expires_at = ?, updated_at = ? WHERE id = ?`,
-      [leaseExpiresAt, timestamp, operationId]
-    );
-    await this.appendPlatformOperationEvent(operationId, request.event?.kind ?? "runner.lease_renewed", request.event?.data ?? { runnerId: request.runnerId, leaseExpiresAt });
-    await this.renewRepositoryClaimsForRunner(request.runnerId, leaseSeconds);
-    return this.getOperation(operationId);
-  }
-  async checkpoint(operationId, request) {
-    await this.assertRunnerUpdate(operationId, request.runnerId);
-    const timestamp = isoNow(this.now);
-    await this.database.run(
-      `UPDATE platform_operations SET status = 'running', output_json = ?, updated_at = ? WHERE id = ?`,
-      [JSON.stringify(request.output ?? null), timestamp, operationId]
-    );
-    await this.appendPlatformOperationEvent(operationId, request.event?.kind ?? "checkpoint", request.event?.data ?? { runnerId: request.runnerId ?? null });
-    return this.getOperation(operationId);
-  }
-  async complete(operationId, request) {
-    await this.assertRunnerUpdate(operationId, request.runnerId);
-    const timestamp = isoNow(this.now);
-    await this.database.run(
-      `UPDATE platform_operations
-			 SET status = 'succeeded', output_json = ?, error_json = NULL, lease_expires_at = NULL, updated_at = ?, finished_at = ?
-			 WHERE id = ?`,
-      [JSON.stringify(request.output ?? null), timestamp, timestamp, operationId]
-    );
-    await this.appendPlatformOperationEvent(operationId, request.event?.kind ?? "completed", request.event?.data ?? {});
-    const output = request.output && typeof request.output === "object" ? request.output : {};
-    await this.releaseRepositoryClaimsForRunner(request.runnerId, {
-      branch: output.operationBranch ?? output.branch ?? null,
-      commitSha: output.commitSha ?? null,
-      metadata: { operationId, status: "succeeded" }
-    });
-    return this.getOperation(operationId);
-  }
-  async fail(operationId, request) {
-    await this.assertRunnerUpdate(operationId, request.runnerId);
-    const timestamp = isoNow(this.now);
-    await this.database.run(
-      `UPDATE platform_operations
-			 SET status = 'failed', error_json = ?, lease_expires_at = NULL, updated_at = ?, finished_at = ?
-			 WHERE id = ?`,
-      [JSON.stringify(request.error ?? { message: "Platform operation failed." }), timestamp, timestamp, operationId]
-    );
-    await this.appendPlatformOperationEvent(operationId, request.event?.kind ?? "failed", request.event?.data ?? {});
-    await this.releaseRepositoryClaimsForRunner(request.runnerId, {
-      claimState: "released",
-      metadata: { operationId, status: "failed" }
-    });
-    return this.getOperation(operationId);
-  }
-  async cancel(operationId, request) {
-    await this.assertRunnerUpdate(operationId, request.runnerId);
-    const timestamp = isoNow(this.now);
-    await this.database.run(
-      `UPDATE platform_operations
-			 SET status = 'cancelled', error_json = ?, lease_expires_at = NULL, cancelled_at = COALESCE(cancelled_at, ?), updated_at = ?, finished_at = COALESCE(finished_at, ?)
-			 WHERE id = ?`,
-      [JSON.stringify(request.error ?? { message: "Platform operation was cancelled." }), timestamp, timestamp, timestamp, operationId]
-    );
-    await this.appendPlatformOperationEvent(operationId, request.event?.kind ?? "runner.cancelled", request.event?.data ?? {});
-    await this.releaseRepositoryClaimsForRunner(request.runnerId, {
-      claimState: "released",
-      metadata: { operationId, status: "cancelled" }
-    });
-    return this.getOperation(operationId);
-  }
-  async upsertRepositoryClaim(input) {
-    const repositoryKeyValue = repositoryKey(input.repository);
-    const timestamp = isoNow(this.now);
-    const leaseExpiresAt = new Date(this.now().getTime() + input.leaseSeconds * 1e3).toISOString();
-    const existing = await this.database.first(
-      `SELECT * FROM platform_repository_claims WHERE repository_key = ? AND runner_id = ? AND claim_state = 'active' LIMIT 1`,
-      [repositoryKeyValue, input.runnerId]
-    );
-    if (existing) {
-      await this.database.run(
-        `UPDATE platform_repository_claims SET lease_expires_at = ?, metadata_json = ?, updated_at = ? WHERE id = ?`,
-        [leaseExpiresAt, JSON.stringify(input.metadata ?? parseJson(existing.metadata_json, {})), timestamp, existing.id]
-      );
-      return;
-    }
-    await this.database.run(
-      `INSERT INTO platform_repository_claims (
-				id, repository_key, runner_id, workspace_path, branch, commit_sha, claim_state, lease_expires_at, metadata_json, created_at, updated_at
-			) VALUES (?, ?, ?, ?, ?, NULL, 'active', ?, ?, ?, ?)`,
-      [
-        randomUUID(),
-        repositoryKeyValue,
-        input.runnerId,
-        repositoryWorkspacePath(input.workspaceRoot, input.repository),
-        input.branch ?? null,
-        leaseExpiresAt,
-        JSON.stringify(input.metadata ?? {}),
-        timestamp,
-        timestamp
-      ]
-    );
-  }
-  async renewRepositoryClaimsForRunner(runnerId, leaseSeconds = 300) {
-    if (!runnerId) return;
-    const timestamp = isoNow(this.now);
-    const leaseExpiresAt = new Date(this.now().getTime() + leaseSeconds * 1e3).toISOString();
-    await this.database.run(
-      `UPDATE platform_repository_claims SET lease_expires_at = ?, updated_at = ? WHERE runner_id = ? AND claim_state = 'active'`,
-      [leaseExpiresAt, timestamp, runnerId]
-    );
-  }
-  async releaseRepositoryClaimsForRunner(runnerId, input = {}) {
-    if (!runnerId) return;
-    const rows = await this.database.all(
-      `SELECT * FROM platform_repository_claims WHERE runner_id = ? AND claim_state = 'active'`,
-      [runnerId]
-    );
-    const timestamp = isoNow(this.now);
-    for (const row of rows) {
-      await this.database.run(
-        `UPDATE platform_repository_claims
-				 SET claim_state = ?,
-				     branch = COALESCE(?, branch),
-				     commit_sha = COALESCE(?, commit_sha),
-				     lease_expires_at = NULL,
-				     metadata_json = ?,
-				     updated_at = ?
-				 WHERE id = ?`,
-        [
-          input.claimState ?? "released",
-          input.branch ?? null,
-          input.commitSha ?? null,
-          JSON.stringify({ ...parseJson(row.metadata_json, {}), ...input.metadata ?? {} }),
-          timestamp,
-          row.id
-        ]
-      );
-    }
-  }
-}
-export {
-  PLATFORM_OPERATION_SCHEMA_SQL,
-  PlatformOperationStore,
-  createD1RelationalAdapter,
-  createPlatformOperationStoreFromEnv,
-  createPostgresRelationalAdapter,
-  createRelationalAdapterFromUrl,
-  createSqliteRelationalAdapter
-};