import { generateObject } from 'ai';
import { openai } from '@ai-sdk/openai';
import { z } from 'zod';

export interface AuditResult {
  clause: string;
  riskLevel: 'LOW' | 'MEDIUM' | 'HIGH';
  statutoryBasis: string;
  recommendation: string;
  verifiedCitations: Array<{
    marker: number;
    canonicalId: string;
    citationString: string;
  }>;
}

export async function auditContractClause(clauseText: string): Promise<AuditResult> {
  // 1. Evaluate legal enforceability against Egyptian Civil Code via Dike Reason
  const reasonResponse = await fetch('https://api.dike.it.com/v1/reason', {
    method: 'POST',
    headers: {
      'Authorization': `Bearer ${process.env.DIKE_API_KEY}`,
      'Content-Type': 'application/json',
    },
    body: JSON.stringify({
      question: `ما مدى صحة ونفاذ هذا البند وفقاً لأحكام القانون المدني المصري وأحكام محكمة النقض: "${clauseText}"؟`,
      filters: { jurisdiction: 'EG', status: ['in_force'] },
      language: 'ar',
    }),
  });

  const dikeResult = await reasonResponse.json();

  // 2. Format into structured enterprise risk payload
  const { object } = await generateObject({
    model: openai('gpt-4o'),
    schema: z.object({
      clause: z.string(),
      riskLevel: z.enum(['LOW', 'MEDIUM', 'HIGH']),
      statutoryBasis: z.string().describe('Mandatory law provisions from Dike response'),
      recommendation: z.string().describe('Actionable redlining guidance for procurement'),
      verifiedCitations: z.array(
        z.object({
          marker: z.number(),
          canonicalId: z.string(),
          citationString: z.string(),
        })
      ),
    }),
    prompt: `
Dike Verified Legal Analysis:
${dikeResult.answer}

Dike Verified Citations:
${JSON.stringify(dikeResult.citations)}

Contract Clause to Evaluate:
"${clauseText}"
    `,
  });

  return object;
}