Function updateRun

packages/core/src/progress/store.ts:151–206

(
  id: string,
  owner: string,
  input: UpdateProgressInput,
)

}

export async function updateRun(
  id: string,
  owner: string,
  input: UpdateProgressInput,
): Promise<AgentRun | null> {
  await ensureTable();
  const client = getDbExec();
  // Read current row first so we can return a consistent snapshot of this
  // caller's update (avoids the UPDATE→SELECT race where a concurrent writer
  // could have their change reflected in the returned value).
  const current = await getRun(id, owner);
  if (!current) return null;

  const now = Date.now();
  const sets: string[] = ["updated_at = ?"];
  const args: Array<string | number | null> = [now];
  const next: AgentRun = {
    ...current,
    updatedAt: new Date(now).toISOString(),
  };

  if (Object.prototype.hasOwnProperty.call(input, "percent")) {
    const percent = input.percent == null ? null : clampPercent(input.percent);
    sets.push("percent = ?");
    args.push(percent);
    next.percent = percent;
  }
  if (input.step !== undefined) {
    sets.push("step = ?");
    args.push(input.step);
    next.step = input.step;
  }
  if (input.metadata !== undefined) {
    sets.push("metadata = ?");
    args.push(JSON.stringify(input.metadata));
    next.metadata = input.metadata;
  }
  if (input.status !== undefined) {
    sets.push("status = ?");
    args.push(input.status);
    next.status = input.status;
    if (input.status !== "running") {
      sets.push("completed_at = ?");
      args.push(now);
      next.completedAt = new Date(now).toISOString();
    }
  }
  args.push(id, owner);

  await client.execute({
    sql: `UPDATE progress_runs SET ${sets.join(", ")} WHERE id = ? AND owner = ?`,
    args,
  });
  bumpPoll(owner);
  return next;
}

function clampPercent(n: number): number {