#!/usr/bin/env node

/**
 * SAP AI Provider - Data Masking Example (DPI)
 *
 * This example demonstrates data masking/anonymization using
 * SAP Data Privacy Integration (DPI) through the Orchestration API.
 *
 * Authentication:
 * - On SAP BTP: Automatically uses service binding (VCAP_SERVICES)
 * - Locally: Set AICORE_SERVICE_KEY environment variable with your service key JSON
 */

// Load environment variables
import "dotenv/config";
import { APICallError, LoadAPIKeyError, NoSuchModelError } from "@ai-sdk/provider";
import { generateText } from "ai";

// This example uses relative imports for local development within this repo.
// In YOUR production project, use the published package instead:
// import { createSAPAIProvider, buildDpiMaskingProvider } from "@jerome-benoit/sap-ai-provider";
import { buildDpiMaskingProvider, createSAPAIProvider } from "../src/index";

/**
 *
 */
async function dataMaskingExample() {
  console.log("🔒 SAP AI Data Masking Example (DPI)\n");

  // Verify AICORE_SERVICE_KEY is set for local development
  if (!process.env.AICORE_SERVICE_KEY && !process.env.VCAP_SERVICES) {
    console.warn("⚠️  Warning: AICORE_SERVICE_KEY environment variable not set.");
    console.warn("   Set it in your .env file or environment for local development.\n");
  }

  try {
    // Build DPI masking configuration using the SDK helper
    const dpiMaskingConfig = buildDpiMaskingProvider({
      entities: [
        // Standard entities
        "profile-email",
        "profile-person",
        // Custom entity with replacement strategy
        {
          replacement_strategy: {
            method: "constant",
            value: "PHONE_REDACTED",
          },
          type: "profile-phone",
        },
      ],
      method: "anonymization",
    });

    // Provider with masking enabled by default
    const provider = createSAPAIProvider({
      defaultSettings: {
        masking: {
          masking_providers: [dpiMaskingConfig],
        },
      },
    });

    const model = provider("gpt-4.1");

    console.log("📝 Testing with data masking enabled...\n");

    const { text } = await generateText({
      messages: [
        {
          content:
            "Please email Jane Doe (jane.doe@example.com) at +1-555-123-4567 about the meeting.",
          role: "user",
        },
      ],
      model,
    });

    console.log("🤖 Response:", text);
    console.log("\n📌 Note: Personal data like names, emails, and phone numbers should be");
    console.log("   masked by DPI before reaching the model.");

    // Test without masking for comparison
    console.log("\n================================");
    console.log("🧪 Same prompt WITHOUT data masking (for comparison)");
    console.log("================================\n");

    const providerNoMask = createSAPAIProvider();
    const modelNoMask = providerNoMask("gpt-4.1");

    const { text: textNoMask } = await generateText({
      messages: [
        {
          content:
            "Please email Jane Doe (jane.doe@example.com) at +1-555-123-4567 about the meeting.",
          role: "user",
        },
      ],
      model: modelNoMask,
    });

    console.log("🤖 Response (no masking):", textNoMask);

    // Verbatim echo test
    console.log("\n================================");
    console.log("📎 Verbatim echo test (shows what model receives)");
    console.log("================================\n");

    const original = "My name is John Smith, email: john.smith@company.com, phone: 555-987-6543";

    const { text: echoMasked } = await generateText({
      messages: [
        {
          content: `Repeat this exactly: ${original}`,
          role: "user",
        },
      ],
      model,
    });
    console.log("🔒 Echo with masking:", echoMasked);

    const { text: echoNoMask } = await generateText({
      messages: [
        {
          content: `Repeat this exactly: ${original}`,
          role: "user",
        },
      ],
      model: modelNoMask,
    });
    console.log("🔓 Echo without masking:", echoNoMask);

    console.log("\n✅ Data masking example completed!");
  } catch (error: unknown) {
    if (error instanceof LoadAPIKeyError) {
      console.error("❌ Authentication Error:", error.message);
    } else if (error instanceof NoSuchModelError) {
      console.error("❌ Model Not Found:", error.modelId);
    } else if (error instanceof APICallError) {
      console.error("❌ API Call Error:", error.statusCode, error.message);

      // Parse SAP-specific metadata
      const sapError = JSON.parse(error.responseBody ?? "{}") as {
        error?: { code?: string; request_id?: string };
      };
      if (sapError.error?.request_id) {
        console.error("   SAP Request ID:", sapError.error.request_id);
        console.error("   SAP Error Code:", sapError.error.code);
      }
    } else {
      const errorMessage = error instanceof Error ? error.message : String(error);
      console.error("❌ Example failed:", errorMessage);
    }

    console.error("\n💡 Troubleshooting tips:");
    console.error("   - Ensure AICORE_SERVICE_KEY is set with valid credentials");
    console.error("   - Check that your SAP AI Core instance is accessible");
    console.error("   - Verify the model is available in your deployment");
  }
}

dataMaskingExample().catch(console.error);

export { dataMaskingExample };