package io.complyance.test;

import io.complyance.sdk.*;
import io.complyance.sdk.Source;
import io.complyance.sdk.SourceType;
import io.complyance.sdk.SDKConfig;
import io.complyance.sdk.Environment;
import io.complyance.sdk.GETSUnifySDK;
import io.complyance.sdk.Country;
import io.complyance.sdk.Operation;
import io.complyance.sdk.Mode;
import io.complyance.sdk.Purpose;
import io.complyance.sdk.UnifyResponse;
import io.complyance.sdk.SDKException;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.List;
import java.util.ArrayList;

/**
 * UAE Tax Invoice Test - Demonstrates UAE-specific TAX_INVOICE functionality
 *
 * This test class demonstrates:
 * - UAE (AE) country-specific compliance
 * - TAX_INVOICE document type mapping
 * - B2B transaction processing with PINT AE
 * - UAE-specific field mappings
 * - Comprehensive payload with UAE invoice data
 */
public class UAETaxInvoiceTest {

    public static void main(String[] args) {
        System.out.println("=== 🇦🇪 UAE Tax Invoice Test ===");
        System.out.println("Testing UAE-specific TAX_INVOICE functionality");
        System.out.println("Demonstrates B2B mapping, AE compliance, and comprehensive payload");

        try {
            // Configure SDK
            configureSDK();

            // Create comprehensive UAE test payload
            Map<String, Object> payload = createUAETestPayload();

            System.out.println("✅ UAE test payload created");

            // Test UAE TAX_INVOICE flow
            testUAETaxInvoiceFlow(payload);

            System.out.println("\n🎉 UAE Tax Invoice test completed successfully!");

        } catch (Exception e) {
            System.err.println("❌ Unexpected error: " + e.getMessage());
            e.printStackTrace();
        }
    }

    /**
     * Configure the SDK for UAE testing
     */
    private static void configureSDK() {
        Source source = new Source("uae-source-system", "1.0", SourceType.FIRST_PARTY);
        SDKConfig config = new SDKConfig("ak_368d073d3de1f19041612fe6cdcf",
                Environment.SANDBOX, Arrays.asList(source));
        GETSUnifySDK.configure(config);
        System.out.println("✅ SDK Configured for UAE Testing");
    }

    /**
     * Creates UAE test payload with PINT AE compliance
     */
    private static Map<String, Object> createUAETestPayload() {
        Map<String, Object> payload = new HashMap<>();

        // Invoice Data - Maps to GETS header fields
        Map<String, Object> invoiceData = new HashMap<>();
        invoiceData.put("document_number", "AE-INV-2026-001");
        invoiceData.put("document_type", "tax_invoice");
        invoiceData.put("invoice_type_code", "380");
        invoiceData.put("invoice_date", "2026-02-18");
        invoiceData.put("invoice_time", "14:30:00+04:00");
        invoiceData.put("currency_code", "AED");
        invoiceData.put("tax_currency_code", "AED");
        invoiceData.put("due_date", "2026-03-20");
        payload.put("invoice_data", invoiceData);

        // Seller Info - UAE company with TRN
        Map<String, Object> sellerInfo = new HashMap<>();
        sellerInfo.put("company_name", "ABC Trading LLC");
        sellerInfo.put("trade_name", "ABC Trading");
        sellerInfo.put("vat_registration", "100000000000003");
        sellerInfo.put("tax_id_type", "TRN");
        sellerInfo.put("tin_number", "1000000000");
        sellerInfo.put("registration_type", "TL");
        sellerInfo.put("registration_number", "CN-1234567");
        sellerInfo.put("peppol_id", "0235:1000000003");
        sellerInfo.put("email", "contact@abctrading.ae");
        sellerInfo.put("phone", "+971-4-1234567");
        sellerInfo.put("street_address", "Sheikh Zayed Road");
        sellerInfo.put("building_number", "123");
        sellerInfo.put("city_name", "Dubai");
        sellerInfo.put("state_province", "DUBAI");
        sellerInfo.put("postal_code", "00000");
        sellerInfo.put("country_code", "AE");
        sellerInfo.put("emirate", "DXB");
        payload.put("seller_info", sellerInfo);

        // Buyer Info
        Map<String, Object> buyerInfo = new HashMap<>();
        buyerInfo.put("company_name", "XYZ Corporation LLC");
        buyerInfo.put("vat_registration", "100000000000003");
        buyerInfo.put("tax_id_type", "TRN");
        buyerInfo.put("buyer_vat", "100000000000003");
        buyerInfo.put("peppol_id", "0235:1000000003");
        buyerInfo.put("email", "purchasing@xyzcorp.ae");
        buyerInfo.put("phone", "+971-2-9876543");
        buyerInfo.put("street_address", "Al Wasl Road");
        buyerInfo.put("building_number", "456");
        buyerInfo.put("city_name", "Dubai");
        buyerInfo.put("state_province", "DUBAI");
        buyerInfo.put("postal_code", "00000");
        buyerInfo.put("country_code", "AE");
        buyerInfo.put("emirate", "DXB");
        payload.put("buyer_info", buyerInfo);

        // Line Items
        List<Map<String, Object>> lineItems = new ArrayList<>();
        Map<String, Object> item1 = new HashMap<>();
        item1.put("item_id", "1");
        item1.put("item_name", "Office Equipment");
        item1.put("description", "Professional office equipment package");
        item1.put("quantity", 10);
        item1.put("unit_code", "EA");
        item1.put("unit_price", 500.00);
        item1.put("tax_rate", 5);
        item1.put("tax_category", "S");
        item1.put("tax_amount", 250.00);
        item1.put("line_extension_amount", 5000.00);
        item1.put("total_amount", 5250.00);
        lineItems.add(item1);
        payload.put("line_items", lineItems);

        // Totals
        Map<String, Object> totals = new HashMap<>();
        totals.put("line_extension_amount", 5000.00);
        totals.put("tax_exclusive_amount", 5000.00);
        totals.put("total_tax_amount", 250.00);
        totals.put("total_amount", 5250.00);
        totals.put("amount_due", 5250.00);
        payload.put("totals", totals);

        // Payment
        Map<String, Object> payment = new HashMap<>();
        payment.put("payment_means_code", "30");
        payment.put("payment_means_text", "Credit Transfer");
        payment.put("payee_account_id", "AE123456789012345678901");
        payment.put("payment_terms", "Net 30 days");
        payload.put("payment", payment);

        return payload;
    }

    /**
     * Test UAE tax invoice flow
     */
    private static void testUAETaxInvoiceFlow(Map<String, Object> payload) {
        GetsDocumentType documentType = GetsDocumentType.builder()
                .base(BASE.TAX_INVOICE)
                .build();

        UnifyResponse response = GETSUnifySDK.pushToUnify(
                "uae-source-system",
                "1.0",
                documentType,
                Country.AE,
                Operation.SINGLE,
                Mode.DOCUMENTS,
                Purpose.INVOICING,
                payload
        );

        if (response != null && "success".equalsIgnoreCase(response.getStatus())) {
            System.out.println("✅ UAE Invoice submitted successfully!");
            if (response.getData() != null && response.getData().getSubmission() != null) {
                System.out.println("🆔 Submission ID: " +
                        response.getData().getSubmission().getSubmissionId());
            }
        } else {
            System.out.println("⚠️ Response: " + response);
        }
    }
}

// Configure SDK
Source source = new Source("your-source", "1.0", SourceType.FIRST_PARTY);
SDKConfig config = new SDKConfig("ak_your_api_key", Environment.PRODUCTION, Arrays.asList(source));
GETSUnifySDK.configure(config);

// Build document type
GetsDocumentType documentType = GetsDocumentType.builder()
        .base(BASE.TAX_INVOICE)
        .build();

// Submit invoice
UnifyResponse response = GETSUnifySDK.pushToUnify(
        "your-source",
        "1.0",
        documentType,
        Country.AE,
        Operation.SINGLE,
        Mode.DOCUMENTS,
        Purpose.INVOICING,
        payload
);