@treeseed/agent 0.8.5

package/dist/services/workday-manager.d.ts

@@ -0,0 +1,391 @@
+#!/usr/bin/env node
+import { createServiceSdk } from './common.ts';
+import { resolveManagerServiceConfig } from './manager.ts';
+type WorkdayManagerSdk = ReturnType<typeof createServiceSdk>;
+type WorkdayManagerConfig = ReturnType<typeof resolveManagerServiceConfig>;
+export declare function runScheduledWorkdayManager(options?: {
+    sdk?: WorkdayManagerSdk;
+    config?: WorkdayManagerConfig;
+    now?: Date;
+}): Promise<{
+    ok: boolean;
+    skipped: boolean;
+    reason: string;
+    initial?: undefined;
+    latest?: undefined;
+    closed?: undefined;
+} | {
+    ok: boolean;
+    skipped: boolean;
+    reason: string;
+    initial: {
+        ok: boolean;
+        mode: "reconcile";
+        managerId: string;
+        projectId: string;
+        environment: "local" | "staging" | "prod";
+        insideWorkWindow: boolean;
+        workPolicy: import("@treeseed/sdk").WorkdayPolicy;
+        workDay: null;
+        prioritySnapshot: null;
+        seededTasks: never[];
+        queuedCount: number;
+        activeLeases: number;
+        desiredWorkers: number;
+        scaleResult: {
+            applied: boolean;
+            provider: string;
+            desiredWorkers: number;
+            metadata: {
+                reason: string;
+            };
+        };
+        workdaySummary: null;
+    } | {
+        ok: boolean;
+        mode: "reconcile";
+        managerId: string;
+        projectId: string;
+        environment: "local" | "staging" | "prod";
+        insideWorkWindow: boolean;
+        workPolicy: import("@treeseed/sdk").WorkdayPolicy;
+        workDay: Record<string, unknown>;
+        prioritySnapshot: import("@treeseed/sdk").PrioritySnapshot | null;
+        seededTasks: {
+            [x: string]: unknown;
+        }[];
+        queuedCount: number;
+        activeLeases: number;
+        desiredWorkers: number;
+        scaleResult: import("@treeseed/sdk").WorkerPoolScaleResult;
+        workdaySummary: Record<string, unknown> | null;
+    };
+    latest?: undefined;
+    closed?: undefined;
+} | {
+    ok: boolean;
+    initial: {
+        ok: boolean;
+        mode: "reconcile";
+        managerId: string;
+        projectId: string;
+        environment: "local" | "staging" | "prod";
+        insideWorkWindow: boolean;
+        workPolicy: import("@treeseed/sdk").WorkdayPolicy;
+        workDay: null;
+        prioritySnapshot: null;
+        seededTasks: never[];
+        queuedCount: number;
+        activeLeases: number;
+        desiredWorkers: number;
+        scaleResult: {
+            applied: boolean;
+            provider: string;
+            desiredWorkers: number;
+            metadata: {
+                reason: string;
+            };
+        };
+        workdaySummary: null;
+    } | {
+        ok: boolean;
+        mode: "reconcile";
+        managerId: string;
+        projectId: string;
+        environment: "local" | "staging" | "prod";
+        insideWorkWindow: boolean;
+        workPolicy: import("@treeseed/sdk").WorkdayPolicy;
+        workDay: Record<string, unknown>;
+        prioritySnapshot: import("@treeseed/sdk").PrioritySnapshot | null;
+        seededTasks: {
+            [x: string]: unknown;
+        }[];
+        queuedCount: number;
+        activeLeases: number;
+        desiredWorkers: number;
+        scaleResult: import("@treeseed/sdk").WorkerPoolScaleResult;
+        workdaySummary: Record<string, unknown> | null;
+    };
+    latest: {
+        ok: boolean;
+        mode: "reconcile";
+        managerId: string;
+        projectId: string;
+        environment: "local" | "staging" | "prod";
+        insideWorkWindow: boolean;
+        workPolicy: import("@treeseed/sdk").WorkdayPolicy;
+        workDay: null;
+        prioritySnapshot: null;
+        seededTasks: never[];
+        queuedCount: number;
+        activeLeases: number;
+        desiredWorkers: number;
+        scaleResult: {
+            applied: boolean;
+            provider: string;
+            desiredWorkers: number;
+            metadata: {
+                reason: string;
+            };
+        };
+        workdaySummary: null;
+    } | {
+        ok: boolean;
+        mode: "reconcile";
+        managerId: string;
+        projectId: string;
+        environment: "local" | "staging" | "prod";
+        insideWorkWindow: boolean;
+        workPolicy: import("@treeseed/sdk").WorkdayPolicy;
+        workDay: Record<string, unknown>;
+        prioritySnapshot: import("@treeseed/sdk").PrioritySnapshot | null;
+        seededTasks: {
+            [x: string]: unknown;
+        }[];
+        queuedCount: number;
+        activeLeases: number;
+        desiredWorkers: number;
+        scaleResult: import("@treeseed/sdk").WorkerPoolScaleResult;
+        workdaySummary: Record<string, unknown> | null;
+    };
+    closed: {
+        ok: boolean;
+        mode: "reconcile";
+        managerId: string;
+        projectId: string;
+        environment: "local" | "staging" | "prod";
+        insideWorkWindow: boolean;
+        workPolicy: import("@treeseed/sdk").WorkdayPolicy;
+        workDay: null;
+        prioritySnapshot: null;
+        seededTasks: never[];
+        queuedCount: number;
+        activeLeases: number;
+        desiredWorkers: number;
+        scaleResult: {
+            applied: boolean;
+            provider: string;
+            desiredWorkers: number;
+            metadata: {
+                reason: string;
+            };
+        };
+        workdaySummary: null;
+    } | {
+        ok: boolean;
+        mode: "reconcile";
+        managerId: string;
+        projectId: string;
+        environment: "local" | "staging" | "prod";
+        insideWorkWindow: boolean;
+        workPolicy: import("@treeseed/sdk").WorkdayPolicy;
+        workDay: Record<string, unknown>;
+        prioritySnapshot: import("@treeseed/sdk").PrioritySnapshot | null;
+        seededTasks: {
+            [x: string]: unknown;
+        }[];
+        queuedCount: number;
+        activeLeases: number;
+        desiredWorkers: number;
+        scaleResult: import("@treeseed/sdk").WorkerPoolScaleResult;
+        workdaySummary: Record<string, unknown> | null;
+    } | {
+        ok: boolean;
+        created: boolean;
+        workDay: Record<string, unknown>;
+        skipped?: undefined;
+        reason?: undefined;
+        prioritySnapshot?: undefined;
+    } | {
+        ok: boolean;
+        created: boolean;
+        skipped: boolean;
+        reason: string;
+        workDay?: undefined;
+        prioritySnapshot?: undefined;
+    } | {
+        ok: boolean;
+        created: boolean;
+        workDay: import("@treeseed/sdk").SdkWorkDayEntity | null;
+        prioritySnapshot: import("@treeseed/sdk").PrioritySnapshot | null;
+        skipped?: undefined;
+        reason?: undefined;
+    } | {
+        ok: boolean;
+        skipped: boolean;
+        reason: string;
+        workDay?: undefined;
+        summary?: undefined;
+        scale?: undefined;
+    } | {
+        ok: boolean;
+        workDay: import("@treeseed/sdk").SdkWorkDayEntity | null;
+        summary: {
+            contentSnapshot: {
+                relativePath: string;
+                slug: string;
+                reportVersion: string;
+                title: string;
+            };
+            capacity: {
+                providerSplit: {
+                    providerId: string;
+                    laneId: string;
+                    state: import("@treeseed/sdk/types").CapacityReservationState;
+                    reservedCredits: number;
+                    consumedCredits: number;
+                    reservedProviderUnits: number | null;
+                    consumedProviderUnits: number | null;
+                    reservedUsd: number | null;
+                    consumedUsd: number | null;
+                }[];
+                grantedDailyCredits: number;
+                reservedCredits: number;
+                consumedCredits: number;
+                remainingDailyCredits: number | null;
+                providerCount: number;
+                laneCount: number;
+                grantCount: number;
+            } | null;
+            projectId: string;
+            environment: "local" | "staging" | "prod";
+            workDayId: string;
+            state: string;
+            totalTasks: number;
+            completedTasks: number;
+            failedTasks: number;
+            queuedTasks: number;
+            activeTasks: number;
+            dailyTaskCreditBudget: number;
+            usedTaskCredits: number;
+            remainingTaskCredits: number;
+            creditLedgerEntries: number;
+            prioritySnapshotId: string | null;
+            priorityItemCount: number;
+            priorityItems: import("@treeseed/sdk").PrioritySnapshotItem[];
+            taskItems: {
+                id: string;
+                agentId: string | undefined;
+                type: string | undefined;
+                state: string | undefined;
+                priority: number | undefined;
+                idempotencyKey: string | undefined;
+                createdAt: string | null;
+                startedAt: string | null;
+                completedAt: string | null;
+                lastErrorCode: string | null;
+                lastErrorMessage: string | null;
+                lastEventKind: string | null;
+                outputCount: number;
+                changedFiles: string[];
+            }[];
+            changedFiles: string[];
+            releases: {
+                id: string | undefined;
+                deploymentKind: string;
+                status: string;
+                releaseTag: string | null;
+                commitSha: string | null;
+                sourceRef: string | null;
+                startedAt: string | null;
+                finishedAt: string | null;
+                createdAt: string | null;
+            }[];
+            scaleDecision: import("@treeseed/sdk").ScaleDecision;
+            scaleResult: import("@treeseed/sdk").WorkerPoolScaleResult;
+            generatedAt: string;
+        };
+        scale: import("@treeseed/sdk").WorkerPoolScaleResult;
+        skipped?: undefined;
+        reason?: undefined;
+    } | {
+        ok: boolean;
+        skipped: boolean;
+        reason: string;
+        workDayId?: undefined;
+        summary?: undefined;
+    } | {
+        ok: boolean;
+        workDayId: unknown;
+        summary: {
+            contentSnapshot: {
+                relativePath: string;
+                slug: string;
+                reportVersion: string;
+                title: string;
+            };
+            capacity: {
+                providerSplit: {
+                    providerId: string;
+                    laneId: string;
+                    state: import("@treeseed/sdk/types").CapacityReservationState;
+                    reservedCredits: number;
+                    consumedCredits: number;
+                    reservedProviderUnits: number | null;
+                    consumedProviderUnits: number | null;
+                    reservedUsd: number | null;
+                    consumedUsd: number | null;
+                }[];
+                grantedDailyCredits: number;
+                reservedCredits: number;
+                consumedCredits: number;
+                remainingDailyCredits: number | null;
+                providerCount: number;
+                laneCount: number;
+                grantCount: number;
+            } | null;
+            projectId: string;
+            environment: "local" | "staging" | "prod";
+            workDayId: string;
+            state: string;
+            totalTasks: number;
+            completedTasks: number;
+            failedTasks: number;
+            queuedTasks: number;
+            activeTasks: number;
+            dailyTaskCreditBudget: number;
+            usedTaskCredits: number;
+            remainingTaskCredits: number;
+            creditLedgerEntries: number;
+            prioritySnapshotId: string | null;
+            priorityItemCount: number;
+            priorityItems: import("@treeseed/sdk").PrioritySnapshotItem[];
+            taskItems: {
+                id: string;
+                agentId: string | undefined;
+                type: string | undefined;
+                state: string | undefined;
+                priority: number | undefined;
+                idempotencyKey: string | undefined;
+                createdAt: string | null;
+                startedAt: string | null;
+                completedAt: string | null;
+                lastErrorCode: string | null;
+                lastErrorMessage: string | null;
+                lastEventKind: string | null;
+                outputCount: number;
+                changedFiles: string[];
+            }[];
+            changedFiles: string[];
+            releases: {
+                id: string | undefined;
+                deploymentKind: string;
+                status: string;
+                releaseTag: string | null;
+                commitSha: string | null;
+                sourceRef: string | null;
+                startedAt: string | null;
+                finishedAt: string | null;
+                createdAt: string | null;
+            }[];
+            scaleDecision: import("@treeseed/sdk").ScaleDecision;
+            scaleResult: import("@treeseed/sdk").WorkerPoolScaleResult;
+            generatedAt: string;
+        };
+        skipped?: undefined;
+        reason?: undefined;
+    };
+    skipped?: undefined;
+    reason?: undefined;
+}>;
+export {};

package/dist/services/workday-manager.js

@@ -0,0 +1,163 @@
+#!/usr/bin/env node
+import { fileURLToPath } from "node:url";
+import { createServiceSdk } from "./common.js";
+import {
+  resolveManagerServiceConfig,
+  runManagerAction,
+  runManagerCycle
+} from "./manager.js";
+function readDate(value) {
+  if (typeof value !== "string") return null;
+  const parsed = new Date(value);
+  return Number.isFinite(parsed.valueOf()) ? parsed : null;
+}
+function workdayCloseDeadline(workDay, durationMinutes, now) {
+  const startedAt = readDate(workDay?.startedAt) ?? readDate(workDay?.started_at) ?? now;
+  return new Date(startedAt.valueOf() + durationMinutes * 60 * 1e3);
+}
+async function sleep(ms) {
+  await new Promise((resolve) => setTimeout(resolve, ms));
+}
+async function acquireLease(sdk, config, workDayId) {
+  if (typeof sdk.claimWorkdayManagerLease !== "function") {
+    return { payload: { id: `local:${config.managerId}`, managerId: config.managerId } };
+  }
+  return sdk.claimWorkdayManagerLease({
+    projectId: config.projectId,
+    environment: config.environment,
+    workDayId,
+    managerId: config.managerId,
+    ttlSeconds: Math.max(60, Math.ceil(config.pollIntervalMs / 1e3) * 4),
+    staleAfterSeconds: Math.max(120, Math.ceil(config.pollIntervalMs / 1e3) * 8),
+    metadata: {
+      service: "workdayManager"
+    }
+  });
+}
+async function heartbeatLease(sdk, config, leaseId, workDayId) {
+  if (typeof sdk.claimWorkdayManagerLease !== "function") return;
+  await sdk.claimWorkdayManagerLease({
+    id: leaseId,
+    projectId: config.projectId,
+    environment: config.environment,
+    workDayId,
+    managerId: config.managerId,
+    ttlSeconds: Math.max(60, Math.ceil(config.pollIntervalMs / 1e3) * 4),
+    metadata: {
+      service: "workdayManager",
+      heartbeat: true
+    }
+  });
+}
+async function releaseLease(sdk, managerId, leaseId) {
+  if (!leaseId || typeof sdk.releaseWorkdayManagerLease !== "function") return;
+  await sdk.releaseWorkdayManagerLease({ id: leaseId, managerId }).catch(() => null);
+}
+async function recordRunnerCloseDecision(sdk, config, workDayId, action, reason) {
+  if (typeof sdk.recordRunnerScaleDecision !== "function") return;
+  const runners = typeof sdk.listWorkerRunners === "function" ? (await sdk.listWorkerRunners(config.projectId, config.environment).catch(() => ({ payload: [] }))).payload ?? [] : [];
+  if (runners.length === 0) {
+    await sdk.recordRunnerScaleDecision({
+      projectId: config.projectId,
+      environment: config.environment,
+      workDayId,
+      action: "noop",
+      reason: `${reason}:no_runners`,
+      metadata: { service: "workdayManager" }
+    }).catch(() => null);
+    return;
+  }
+  for (const runner of runners) {
+    await sdk.recordRunnerScaleDecision({
+      projectId: config.projectId,
+      environment: config.environment,
+      workDayId,
+      runnerId: typeof runner.runnerId === "string" ? runner.runnerId : null,
+      runnerServiceName: typeof runner.runnerServiceName === "string" ? runner.runnerServiceName : null,
+      action,
+      reason,
+      metadata: { service: "workdayManager" }
+    }).catch(() => null);
+  }
+}
+async function runScheduledWorkdayManager(options = {}) {
+  const sdk = options.sdk ?? createServiceSdk();
+  const config = options.config ?? {
+    ...resolveManagerServiceConfig(),
+    mode: "reconcile"
+  };
+  let now = options.now ?? /* @__PURE__ */ new Date();
+  const policyEnvelope = await sdk.getWorkPolicy(config.projectId, config.environment);
+  const policy = policyEnvelope.payload ?? await sdk.upsertWorkPolicy({
+    projectId: config.projectId,
+    environment: config.environment,
+    schedule: config.defaultSchedule,
+    enabled: true,
+    startCron: process.env.TREESEED_WORKDAY_START_CRON?.trim() || "0 9 * * 1-5",
+    durationMinutes: Number(process.env.TREESEED_WORKDAY_DURATION_MINUTES ?? 480),
+    maxRunners: config.autoscale.maxWorkers,
+    maxWorkersPerRunner: Number(process.env.TREESEED_RUNNER_MAX_LOCAL_WORKERS ?? 4),
+    dailyCreditBudget: config.dailyTaskCreditBudget,
+    closeoutGraceMinutes: Number(process.env.TREESEED_WORKDAY_CLOSEOUT_GRACE_MINUTES ?? 15),
+    dailyTaskCreditBudget: config.dailyTaskCreditBudget,
+    maxQueuedTasks: config.maxQueuedTasks,
+    maxQueuedCredits: config.maxQueuedCredits,
+    autoscale: config.autoscale,
+    creditWeights: config.creditWeights,
+    metadata: { managedBy: "workdayManager" }
+  }).then((created) => created.payload);
+  if (!policy?.enabled) {
+    return { ok: true, skipped: true, reason: "workday_policy_disabled" };
+  }
+  const requests = typeof sdk.listWorkdayRequests === "function" ? (await sdk.listWorkdayRequests(config.projectId, config.environment, "pending").catch(() => ({ payload: [] }))).payload ?? [] : [];
+  const oneOffRunRequested = requests.some((entry) => entry.type === "one_off_run");
+  const initial = await runManagerCycle({ sdk, config, now });
+  const workDay = initial.workDay;
+  if (!workDay && !oneOffRunRequested) {
+    return { ok: true, skipped: true, reason: "no_workday_started", initial };
+  }
+  const lease = await acquireLease(sdk, config, workDay ? String(workDay.id ?? "") : null);
+  if (!lease.payload) {
+    return { ok: true, skipped: true, reason: "healthy_manager_lease_exists" };
+  }
+  const leaseId = String(lease.payload.id ?? "");
+  let latest = initial;
+  try {
+    let activeWorkDay = workDay;
+    let closeDeadline = workdayCloseDeadline(activeWorkDay, Number(policy.durationMinutes ?? 480), now);
+    while (Date.now() < closeDeadline.valueOf()) {
+      await heartbeatLease(sdk, config, leaseId, activeWorkDay ? String(activeWorkDay.id ?? "") : null);
+      await sleep(config.pollIntervalMs);
+      now = /* @__PURE__ */ new Date();
+      latest = await runManagerCycle({ sdk, config, now });
+      activeWorkDay = latest.workDay;
+      closeDeadline = workdayCloseDeadline(activeWorkDay, Number(policy.durationMinutes ?? 480), now);
+    }
+    const workDayId = activeWorkDay ? String(activeWorkDay.id ?? "") : null;
+    await recordRunnerCloseDecision(sdk, config, workDayId, "drain", "workday_closeout");
+    const graceMs = Number(policy.closeoutGraceMinutes ?? 15) * 60 * 1e3;
+    const graceEnd = Date.now() + graceMs;
+    while (Date.now() < graceEnd) {
+      const result = await runManagerCycle({ sdk, config, now: /* @__PURE__ */ new Date() });
+      latest = result;
+      if (Number(result.queuedCount ?? 0) === 0 && Number(result.activeLeases ?? 0) === 0) {
+        break;
+      }
+      await sleep(config.pollIntervalMs);
+    }
+    const closed = await runManagerAction({ sdk, config, mode: "close-workday" });
+    await recordRunnerCloseDecision(sdk, config, workDayId, "sleep", "workday_closed");
+    return { ok: true, initial, latest, closed };
+  } finally {
+    await releaseLease(sdk, config.managerId, leaseId);
+  }
+}
+const currentFile = fileURLToPath(import.meta.url);
+const entryFile = process.argv[1] ?? "";
+if (entryFile === currentFile) {
+  process.stdout.write(`${JSON.stringify(await runScheduledWorkdayManager(), null, 2)}
+`);
+}
+export {
+  runScheduledWorkdayManager
+};