Advance Playwright framework + AI

import { AIClient } from './AIClient';
import { summariseTrace } from './trace-summary';

export async function explainFailure(
  traceZipPath: string,
  opts: { model?: string; maxTokens?: number } = {},
) {
  // last 50 events + DOM at failure + last network errors
  const trace = await summariseTrace(traceZipPath);
  const prompt =
    'A Playwright test failed. Trace summary follows.\n\n' +
    trace +
    '\n\nReturn: 1) a 2-sentence root cause, 2) the single most likely fix.';

  return AIClient.complete({
    model: opts.model ?? process.env.TTA_AI_MODEL ?? 'deepseek-chat',
    prompt,
    maxTokens: opts.maxTokens ?? 300,
  });
}

async click(target: Flex, opts: { healOnFail?: boolean } = {}) {
  const heal = opts.healOnFail ?? true;
  const loc = typeof target === 'string' ? this.page.locator(target) : target;

  try {
    await loc.click({ timeout: 5_000 });
  } catch (err) {
    if (!heal) throw err;
    const dom = await this.snapshotDom();              // redacted
    const guesses = await AIClient.locators({
      intent: 'click',
      original: target.toString(),
      dom,
      n: 3,
    });
    for (const g of guesses) {
      try {
        await this.page.locator(g.selector).click({ timeout: 3_000 });
        await this.recordHeal({ from: target, to: g, confidence: g.score });
        return;
      } catch { /* try the next fallback */ }
    }
    throw err;
  }
}

import { AIClient } from './AIClient';

const SYSTEM = `You write Playwright TypeScript specs.
Target pages live at /playwright/ttacart/*.html.
Use the existing Page Objects in src/pages/ttacart/:
  LoginPage, InventoryPage, ItemDetailPage, CartPage,
  CheckoutStepOnePage, CheckoutStepTwoPage, CheckoutCompletePage.
Use the test-base fixture (import { test, expect } from '../fixtures/test-base').
No raw page.click outside POMs. No waitForTimeout. Use expect().toHaveX.`;

export async function generateSpec(prompt: string) {
  const out = await AIClient.complete({
    model: process.env.TTA_AI_MODEL ?? 'deepseek-chat',
    system: SYSTEM,
    prompt,
    maxTokens: 1200,
  });
  return out.trim();
}

// CLI:
//   npx tsx scripts/ai-gen-spec.ts \
//     "log in as standard_user and complete checkout with 1 backpack"

import { z } from 'zod';
import { AIClient } from '../ai/AIClient';
import { cacheGet, cachePut } from '../ai/cache';

const EdgeCaseList = z.array(z.string()).max(20);

export const DataFactory = {
  // V1 helpers kept verbatim ...
  generateUser, generateCard, randomZip, pickProduct,

  // V2:
  async smart(prompt: string, schema = EdgeCaseList) {
    const key = await sha256(prompt + (process.env.TTA_AI_MODEL ?? 'deepseek-chat'));
    const hit = await cacheGet(key);
    if (hit) return schema.parse(hit);

    const raw = await AIClient.completeJSON({
      model: process.env.TTA_AI_MODEL ?? 'deepseek-chat',
      prompt: 'Return JSON array of strings. ' + prompt,
      maxTokens: 600,
    });
    const out = schema.parse(raw);
    await cachePut(key, out);
    return out;
  },
};

import { execSync } from 'child_process';
import { AIClient } from '../src/ai/AIClient';

const diff = execSync('git diff --unified=3 origin/main -- "**/*.spec.ts"').toString();
if (!diff) process.exit(0);

const SYSTEM = `You review Playwright TypeScript spec diffs.
Flag ONLY:
  - waitForTimeout
  - absolute XPath
  - missing expect()
  - raw page.click outside a Page Object
Return JSON: { comments: Array<{ file, line, body }> }`;

const review = await AIClient.completeJSON({
  model: process.env.TTA_AI_MODEL ?? 'deepseek-chat',
  system: SYSTEM,
  prompt: diff,
  maxTokens: 800,
});

for (const c of review.comments) {
  execSync(
    `gh pr review --comment -b ${JSON.stringify(c.body)} \
     -F ${c.file}:${c.line}`,
  );
}