Run Parallel Attempts

import { GitStorage } from '@pierre/storage';

const baseSha = process.env.TIP_SHA;
const sessionId = process.env.SESSION_ID;

const store = new GitStorage({
  key: process.env.PIERRE_PRIVATE_KEY,
  name: 'acme',
});

const repositoryId = 'acme/storefront';
const repo = await store.findOne({ id: repositoryId });

const attemptBranches = Array.from(
  { length: 4 },
  (_, index) => `sessions/${sessionId}/attempt/${index + 1}`,
);

await Promise.all(
  attemptBranches.map((attemptBranch) =>
    repo.createBranch({
      baseRef: baseSha,
      targetBranch: attemptBranch,
      targetIsEphemeral: true,
    }),
  ),
);

const attemptSource = 'export const authEnabled = true;\n';

const attempt = await repo
  .createCommit({
    author: { email: 'agent-1@example.com', name: 'Agent 1' },
    commitMessage: 'Attempt 1',
    ephemeral: true,
    expectedHeadSha: baseSha,
    targetBranch: attemptBranches[0],
  })
  .addFileFromString('src/auth.ts', attemptSource)
  .send();

console.log(attempt.commitSha);

const currentBranches = await repo.listBranches({ ephemeral: true, limit: 100 });

const results = await Promise.all(
  attemptBranches.map(async (attemptBranch) => {
    const current = currentBranches.branches.find((item) => item.name === attemptBranch);
    if (!current) throw new Error(`No branch exists for ${attemptBranch}`);

    return {
      branch: attemptBranch,
      diff: await repo.getCommitDiff({ baseSha, sha: current.headSha }),
    };
  }),
);

for (const result of results) {
  console.log(result.branch, result.diff.stats);
}

const winner = attemptBranches[2];

await repo.promoteEphemeralBranch({
  baseBranch: winner,
  targetBranch: 'agent/fix-auth',
});

await Promise.all(
  attemptBranches
    .filter((attemptBranch) => attemptBranch !== winner)
    .map((attemptBranch) => repo.deleteBranch({ ephemeral: true, name: attemptBranch })),
);