PaperClipAI/server/src/services/native-runtime/completion-contracts.ts

import { createHash } from "node:crypto";
import { and, desc, eq, sql } from "drizzle-orm";

import type { Db } from "@paperclipai/db";
import { completionContracts } from "@paperclipai/db";
import type { NativeCompletionSource, StrictCompletionContractInput } from "../../vendor/paperclip-runner/index.js";

import { nativeSha256 } from "./canonical.js";

export const NATIVE_COMPLETION_CONTRACT_SCHEMA = "paperclip.completion-contract.v1";
export const NATIVE_COMPLETION_POLICY_VERSION = "phase6-v4";

/** Canonical persisted receipt includes the policy that governs the contract. */
export function nativeCompletionContractSha256(row: Pick<typeof completionContracts.$inferSelect,
  "schemaVersion" | "policyVersion" | "risk" | "completionAuthority" | "contractJson">): string {
  return nativeSha256({
    schemaVersion: row.schemaVersion,
    policyVersion: row.policyVersion,
    risk: row.risk,
    completionAuthority: row.completionAuthority,
    contract: row.contractJson,
  });
}

type CompletionComment = {
  id?: string;
  body: string;
  attachments?: readonly unknown[];
};

export function nativeCompletionSource(kind: NativeCompletionSource["kind"], id: string, content: string): NativeCompletionSource {
  return { kind, id, revision: createHash("sha256").update(content.trim()).digest("hex") };
}

/** Select only the server-owned description source for the ordinary objective path. */
export function nativeImmediateObjectiveSource(input: {
  issueId: string;
  objectiveSource?: {
    kind: "comment" | "description" | "title";
    id: string;
    revision: string | null;
  } | null;
  excluded: boolean;
}): NativeCompletionSource | null {
  const source = input.objectiveSource;
  return !input.excluded && source?.kind === "description" && source.id === input.issueId && source.revision
    ? { kind: "description", id: source.id, revision: source.revision }
    : null;
}

/** The same source operation produces both the requirement and its provenance. */
export function nativeCompletionRequestsWithSources(
  comments: readonly CompletionComment[],
  options: { requiredFullWakeCommentCount?: number } = {},
): { requests: string[]; sources: Array<NativeCompletionSource | null> } {
  if (Number.isSafeInteger(options.requiredFullWakeCommentCount) && (options.requiredFullWakeCommentCount ?? 0) > 0) {
    return { requests: [
      `Read every server-bound pending external-chat comment with read_current_wake_comments until complete=true, then answer all ${options.requiredFullWakeCommentCount} accepted comments in order without omitting a request. Report any unavailable attachment honestly; metadata alone is not its content.`,
    ], sources: [null] };
  }
  const entries = comments.flatMap((comment, index) => {
    const body = comment.body.trim();
    if (body) return [{ request: body, source: comment.id ? nativeCompletionSource("comment", comment.id, body) : null }];
    // A file-only message has an independent inspection requirement.
    return comment.attachments?.length
      ? [{ request: `Inspect and respond to the attached file(s) on pending comment ${index + 1}.`, source: null }]
      : [];
  });
  return { requests: entries.map((entry) => entry.request), sources: entries.map((entry) => entry.source) };
}

export function nativeCompletionRequestsForComments(
  comments: readonly CompletionComment[],
  options: { requiredFullWakeCommentCount?: number } = {},
): string[] {
  return nativeCompletionRequestsWithSources(comments, options).requests;
}

export function resolveNativeCompletionPolicy(issue: {
  reviewPolicy?: string | null;
}) {
  const externalReviewRequired =
    issue.reviewPolicy === "human_only" || issue.reviewPolicy === "not_creator";
  return externalReviewRequired
    ? { risk: "standard", completionAuthority: "server_arbiter" } as const
    : { risk: "low", completionAuthority: "agent_claim_policy" } as const;
}

export function buildNativeCompletionContract(
  issue: { title: string; description: string | null },
  options: {
    readonly revision?: number;
    readonly immediateRequest?: string | null;
    readonly immediateRequests?: readonly string[] | null;
    /** ID of this wake's server-verified humanResponses entry; its content is already in task context. */
    readonly humanResponseId?: string | null;
  } = {},
): StrictCompletionContractInput {
  const immediateRequests = (
    options.immediateRequests ??
    (options.immediateRequest == null ? [] : [options.immediateRequest])
  )
    .map((request) => request.trim())
    .filter((request) => request.length > 0);
  const humanResponseId = options.humanResponseId?.trim();
  const hasFollowUp = immediateRequests.length > 0 || Boolean(humanResponseId);
  // Reference existing context rather than copying the brief/history into every
  // follow-up contract. Keep this guidance stable across comments and resumes.
  const followUpObjective = "Complete the current authorized stage using the task brief and current user direction in the supplied context. Later human direction replaces conflicting scope; preserve other requirements, assigned-skill instructions, and approval gates. Apply authenticated humanResponses only to their question or decision. Clarification is not approval. If acceptance is required, propose or save the requested plan and wait before executing.";
  return {
    revision: String(options.revision ?? 1),
    objective: hasFollowUp
      ? followUpObjective
      : issue.title,
    criteria: hasFollowUp
      ? [
          ...immediateRequests.map((request, index) => ({
            id: immediateRequests.length === 1 ? "objective" : `pending_comment_${index + 1}`,
            requirement: request,
          })),
          ...(humanResponseId ? [{
            id: "human_response",
            requirement: `Apply the server-verified humanResponses entry with id ${JSON.stringify(humanResponseId)} in the supplied current request context, within its question or decision scope and subject to later user direction.`,
          }] : []),
        ]
      : [
          {
            id: "objective",
            requirement:
              issue.description?.trim() || `Complete: ${issue.title}`,
          },
        ],
  };
}

function nativeCompletionContractLockKey(companyId: string, issueId: string): string {
  return `paperclip:native-completion-contract:${companyId}:${issueId}`;
}

export async function lockCurrentNativeCompletionContract(tx: Db, input: {
  companyId: string; issueId: string; contractId: string;
}): Promise<boolean> {
  const [lock] = await tx.execute(sql`select pg_try_advisory_xact_lock(hashtextextended(
    ${nativeCompletionContractLockKey(input.companyId, input.issueId)}, 0)) as acquired`) as unknown as Array<{ acquired: boolean }>;
  if (!lock.acquired) return false;
  const [latest] = await tx.select({ id: completionContracts.id }).from(completionContracts)
    .where(and(eq(completionContracts.companyId, input.companyId), eq(completionContracts.issueId, input.issueId)))
    .orderBy(desc(completionContracts.revision)).limit(1);
  return latest?.id === input.contractId;
}

export async function ensureNativeCompletionContract(input: {
  db: Db;
  companyId: string;
  issue: {
    id: string;
    title: string;
    description: string | null;
    reviewPolicy?: string | null;
  };
  actorId: string;
  immediateRequest?: string | null;
  /** Provenance for a singular server-selected request, when explicitly known. */
  immediateRequestSource?: NativeCompletionSource | null;
  immediateRequests?: readonly string[] | null;
  immediateRequestSources?: readonly (NativeCompletionSource | null)[];
  humanResponseId?: string | null;
}) {
  return input.db.transaction(async (tx) => {
    await tx.execute(sql`select pg_advisory_xact_lock(hashtextextended(
      ${nativeCompletionContractLockKey(input.companyId, input.issue.id)}, 0))`);
    const policy = resolveNativeCompletionPolicy(input.issue);
    const latest = await tx
      .select()
      .from(completionContracts)
      .where(and(
        eq(completionContracts.companyId, input.companyId),
        eq(completionContracts.issueId, input.issue.id),
      ))
      .orderBy(desc(completionContracts.revision))
      .limit(1)
      .then((rows) => rows[0] ?? null);
    const latestRevision = latest?.revision ?? 1;
    const latestCandidate = buildNativeCompletionContract(input.issue, {
      revision: latestRevision,
      immediateRequest: input.immediateRequest,
      immediateRequests: input.immediateRequests,
      humanResponseId: input.humanResponseId,
    });
    const latestCandidateSha256 = nativeCompletionContractSha256({
      schemaVersion: NATIVE_COMPLETION_CONTRACT_SCHEMA,
      policyVersion: NATIVE_COMPLETION_POLICY_VERSION,
      ...policy,
      contractJson: latestCandidate as unknown as Record<string, unknown>,
    });
    if (latest?.canonicalSha256 === latestCandidateSha256) {
      return { row: latest, contract: latestCandidate, sources: buildNativeCompletionContractSources(input) };
    }

    const nextRevision = latest ? latest.revision + 1 : 1;
    const contract = buildNativeCompletionContract(input.issue, {
      revision: nextRevision,
      immediateRequest: input.immediateRequest,
      immediateRequests: input.immediateRequests,
      humanResponseId: input.humanResponseId,
    });
    const canonicalSha256 = nativeCompletionContractSha256({
      schemaVersion: NATIVE_COMPLETION_CONTRACT_SCHEMA,
      policyVersion: NATIVE_COMPLETION_POLICY_VERSION,
      ...policy,
      contractJson: contract as unknown as Record<string, unknown>,
    });
    const [row] = await tx.insert(completionContracts).values({
      companyId: input.companyId,
      issueId: input.issue.id,
      revision: nextRevision,
      schemaVersion: NATIVE_COMPLETION_CONTRACT_SCHEMA,
      policyVersion: NATIVE_COMPLETION_POLICY_VERSION,
      ...policy,
      incompleteCriteriaPolicy: "preserve_non_terminal",
      contractJson: contract as unknown as Record<string, unknown>,
      canonicalSha256,
      createdByActorType: "system",
      createdByActorId: input.actorId,
      supersedesContractId: latest?.id ?? null,
    }).returning();
    if (!row) throw new Error("native_completion_contract_not_persisted");
    return { row, contract, sources: buildNativeCompletionContractSources(input) };
  });
}


export function buildNativeCompletionContractSources(input: {
  issue: { id: string; description: string | null };
  immediateRequest?: string | null;
  immediateRequestSource?: NativeCompletionSource | null;
  immediateRequests?: readonly string[] | null;
  immediateRequestSources?: readonly (NativeCompletionSource | null)[];
  humanResponseId?: string | null;
}): Array<{ id: string; source: NativeCompletionSource }> {
  const requests = (input.immediateRequests ?? (input.immediateRequest == null ? [] : [input.immediateRequest]))
    .map((body, index) => ({
      body: body.trim(),
      source: input.immediateRequests
        ? input.immediateRequestSources?.[index]
        : input.immediateRequestSource,
    }))
    .filter((entry) => entry.body.length > 0);
  if (requests.length > 0) {
    return requests.flatMap((entry, index) => entry.source ? [{
      id: requests.length === 1 ? "objective" : `pending_comment_${index + 1}`,
      source: entry.source,
    }] : []);
  }
  return !input.humanResponseId?.trim() && input.issue.description?.trim()
    ? [{ id: "objective", source: nativeCompletionSource("description", input.issue.id, input.issue.description) }]
    : [];
}