using System.Globalization;
using System.Text.Json;
using CC.Art2link.Pipelines.Domain.Models.PipelineComponents;

public sealed class EdiConfig
{
    public string SegmentTerminator { get; set; } = "~";
    public string ElementSeparator  { get; set; } = "*";
}

public sealed class Edi850ApiDisassembler : PipelineComponentBase<EdiConfig>
{
    public override string Name => "Edi850ApiDisassembler";

    protected override Task<PipelineComponentOutput> ExecuteAsync(
        PipelineComponentInput input, EdiConfig config, CancellationToken ct)
    {
        var seg = input.Body
            .Split(config.SegmentTerminator, StringSplitOptions.RemoveEmptyEntries)
            .Select(s => s.Trim().Split(config.ElementSeparator))
            .ToList();

        var isa = seg.FirstOrDefault(s => s[0] == "ISA");
        var gs  = seg.FirstOrDefault(s => s[0] == "GS");
        var ge  = seg.FirstOrDefault(s => s[0] == "GE");
        if (isa is null || gs is null)                 // no readable envelope: cannot acknowledge
            return Task.FromResult(new PipelineComponentOutput
                { Success = false, ErrorMessage = "No ISA/GS envelope" });

        var sender = isa[6].Trim();          // ISA06, the partner (the 997 receiver)
        var self   = isa[8].Trim();          // ISA08, our own id (the 997 sender)
        var icn    = isa[13].Trim();          // ISA13 interchange control number
        var funcId = gs[1].Trim();            // GS01, "PO" for an 850
        var gcn    = gs[6].Trim();            // GS06 group control number
        var correlation = $"{sender}:{icn}:{gcn}";

        // ---- Group-level checks -> AK9 / AK905..909 ----
        var groupErrors = new List<object>();
        if (ge is not null && ge.Length > 2 && ge[2].Trim() != gcn)
            groupErrors.Add(new { code = "4", meaning = "Group control numbers in GS and GE disagree" });
        var stCount = seg.Count(s => s[0] == "ST");
        if (ge is not null && ge.Length > 1 && int.TryParse(ge[1], out var declaredSets) && declaredSets != stCount)
            groupErrors.Add(new { code = "5", meaning = "Number of included transaction sets does not match count" });

        // ---- Transaction-set checks -> AK5 (502..506) and AK3/AK4 ----
        var st  = seg.First(s => s[0] == "ST");
        var se  = seg.FirstOrDefault(s => s[0] == "SE");
        var beg = seg.FirstOrDefault(s => s[0] == "BEG");
        var segErrors  = new List<object>();   // AK3/AK4
        var setReasons = new List<object>();   // AK5 502..506

        if (se is not null && se.Length > 2 && se[2].Trim() != st[2].Trim())
            setReasons.Add(new { code = "3", meaning = "Control numbers in ST and SE do not match" });
        if (se is not null && se.Length > 1 && int.TryParse(se[1], out var declaredSeg) && declaredSeg != CountSetSegments(seg))
            setReasons.Add(new { code = "4", meaning = "Number of included segments does not match count" });
        if (beg is null)
            segErrors.Add(SegErr("BEG", 2, "3", "Mandatory segment missing", null));
        else if (beg.Length > 5 && !IsDate(beg[5]))
            segErrors.Add(SegErr("BEG", 2, "8", "Segment has data element errors",
                new { position = 5, code = "8", meaning = "Invalid date", badValue = beg[5] }));
        // ...add more segment/element rules here as the trading relationship needs them...
        if (segErrors.Count > 0)
            setReasons.Add(new { code = "5", meaning = "One or more segments in error" });

        var setAccepted = setReasons.Count == 0;
        var accepted    = setAccepted && groupErrors.Count == 0;
        var ack = new
        {
            correlation, partner = sender, self,
            ak1 = new { functionalIdCode = funcId, groupControlNumber = gcn },
            transactionSets = new[] { new {
                setId = st[1], controlNumber = st[2],
                ak5 = setAccepted ? "A" : "R",
                ak5Reasons = setReasons,
                segmentErrors = segErrors
            } },
            ak9 = new { code = accepted ? "A" : "R", setsIncluded = stCount, setsReceived = stCount,
                          setsAccepted = setAccepted ? 1 : 0, errors = groupErrors },
            result = accepted ? "ACCEPTED" : "REJECTED"
        };

        var messages = new List<PipelineMessage>
        {
            new() { Body = JsonSerializer.Serialize(ack), MessageType = "Edi997Ack" }
        };
        if (accepted)
            messages.Add(new() {
                Body = JsonSerializer.Serialize(BuildOrder(seg, sender, isa, gs, st, beg, correlation)),
                MessageType = "Edi850Order" });

        return Task.FromResult(new PipelineComponentOutput
        {
            Success   = true,
            Messages  = messages,
            Variables = { ["AckCorrelation"] = correlation }   // the response subscription matches this
        });
    }

    private static int CountSetSegments(List<string[]> seg)
    {
        int st = seg.FindIndex(s => s[0] == "ST"), se = seg.FindIndex(s => s[0] == "SE");
        return (st >= 0 && se >= st) ? se - st + 1 : 0;   // ST..SE inclusive
    }

    private static bool IsDate(string s) =>
        DateTime.TryParseExact(s, "yyyyMMdd", CultureInfo.InvariantCulture,
            DateTimeStyles.None, out _);

    private static object SegErr(string id, int pos, string code, string meaning, object? el) =>
        new { segmentId = id, position = pos, code, meaning,
              elementErrors = el is null ? Array.Empty<object>() : new[] { el } };

    private static object BuildOrder(List<string[]> seg, string sender,
        string[] isa, string[] gs, string[] st, string[] beg, string correlation)
    {
        var n1  = seg.First(s => s[0] == "N1" && s[1] == "BY");
        var dtm = seg.FirstOrDefault(s => s[0] == "DTM" && s[1] == "002");

        var lines = new List<object>();
        for (var i = 0; i < seg.Count; i++)
        {
            if (seg[i][0] != "PO1") continue;
            var p = seg[i];
            var pid = (i + 1 < seg.Count && seg[i + 1][0] == "PID") ? seg[i + 1] : null;
            lines.Add(new {
                lineNumber  = int.Parse(p[1]),
                quantity    = int.Parse(p[2]),
                uom         = p[3],
                unitPrice   = decimal.Parse(p[4], CultureInfo.InvariantCulture),
                vendorPart  = p[7],
                description = pid is not null && pid.Length > 5 ? pid[5] : null
            });
        }

        return new
        {
            interchange = new { senderId = sender, receiverId = isa[8].Trim(), controlNumber = isa[13].Trim() },
            group       = new { functionalId = gs[1].Trim(), controlNumber = gs[6].Trim(), version = gs[8].Trim() },
            transaction = new { setId = st[1], controlNumber = st[2] },
            correlation,
            purchaseOrder = new
            {
                purpose = beg[1], type = beg[2], poNumber = beg[3],
                poDate = Iso(beg[5]), requestedDelivery = dtm is not null ? Iso(dtm[2]) : null,
                buyer = new { name = n1[2], idQualifier = n1[3], id = n1[4] },
                lines, totalLineItems = lines.Count
            }
        };
    }

    private static string Iso(string d) =>
        d.Length == 8 ? $"{d[..4]}-{d.Substring(4, 2)}-{d.Substring(6, 2)}" : d;
}

CREATE OR ALTER PROCEDURE dbo.usp_BookPurchaseOrder
    @doc XML
AS
BEGIN
    SET NOCOUNT ON;

    DECLARE @PartnerCode VARCHAR(30) = @doc.value('(/BookOrder/Client/PartnerCode)[1]', 'VARCHAR(30)');
    DECLARE @AccountRef  VARCHAR(40) = @doc.value('(/BookOrder/Client/AccountRef)[1]', 'VARCHAR(40)');
    DECLARE @DisplayName NVARCHAR(120) = @doc.value('(/BookOrder/Client/DisplayName)[1]', 'NVARCHAR(120)');
    DECLARE @CustRef VARCHAR(40) = @doc.value('(/BookOrder/Order/CustomerReference)[1]', 'VARCHAR(40)');

    -- Keep the client current, but never touch NextControlNumber here.
    UPDATE dbo.Clients SET AccountRef = @AccountRef, DisplayName = @DisplayName
    WHERE PartnerCode = @PartnerCode;
    IF @@ROWCOUNT = 0
        INSERT dbo.Clients (PartnerCode, AccountRef, DisplayName)
        VALUES (@PartnerCode, @AccountRef, @DisplayName);

    DECLARE @ClientKey INT = (SELECT ClientKey FROM dbo.Clients WHERE PartnerCode = @PartnerCode);

    -- Idempotent on (client, PO number): a re-sent order updates, never duplicates.
    DECLARE @OrderKey INT =
        (SELECT OrderKey FROM dbo.Orders WHERE ClientKey = @ClientKey AND CustomerReference = @CustRef);
    IF @OrderKey IS NULL
    BEGIN
        INSERT dbo.Orders (ClientKey, CustomerReference, OrderKind, PlacedOn, NeededBy, SourceGroupControl)
        VALUES (@ClientKey, @CustRef,
            @doc.value('(/BookOrder/Order/OrderKind)[1]', 'VARCHAR(4)'),
            @doc.value('(/BookOrder/Order/PlacedOn)[1]', 'DATE'),
            @doc.value('(/BookOrder/Order/NeededBy)[1]', 'DATE'),
            @doc.value('(/BookOrder/Order/SourceGroupControl)[1]', 'VARCHAR(20)'));
        SET @OrderKey = SCOPE_IDENTITY();
    END
    ELSE
        UPDATE dbo.Orders SET NeededBy = @doc.value('(/BookOrder/Order/NeededBy)[1]', 'DATE')
        WHERE OrderKey = @OrderKey;

    DELETE dbo.OrderLines WHERE OrderKey = @OrderKey;
    INSERT dbo.OrderLines (OrderKey, LinePosition, CatalogCode, LineText, Cases, UnitOfMeasure, PriceEach)
    SELECT @OrderKey,
        L.n.value('(LinePosition)[1]', 'INT'),
        L.n.value('(CatalogCode)[1]', 'VARCHAR(40)'),
        L.n.value('(LineText)[1]', 'NVARCHAR(200)'),
        L.n.value('(Cases)[1]', 'INT'),
        L.n.value('(UnitOfMeasure)[1]', 'VARCHAR(4)'),
        L.n.value('(PriceEach)[1]', 'DECIMAL(18,2)')
    FROM @doc.nodes('/BookOrder/Order/Lines/Line') AS L(n);
END;

using System.Text;
using System.Text.Json;
using Microsoft.Data.SqlClient;
using CC.Art2link.Pipelines.Domain.Models.PipelineComponents;

public sealed class AssemblerConfig
{
    public string ConnectionString { get; set; } = "";   // bound to {{Constant.EdiDbConn}}
}

public sealed class Edi997Assembler : PipelineComponentBase<AssemblerConfig>
{
    public override string Name => "Edi997Assembler";

    protected override async Task<PipelineComponentOutput> ExecuteAsync(
        PipelineComponentInput input, AssemblerConfig config, CancellationToken ct)
    {
        using var doc = JsonDocument.Parse(input.Body);
        var a = doc.RootElement;
        var partner = a.GetProperty("partner").GetString()!;   // the 997 receiver
        var self    = a.GetProperty("self").GetString()!;      // our id, taken from the inbound envelope
        var ak1 = a.GetProperty("ak1");
        var ts  = a.GetProperty("transactionSets")[0];
        var ak9 = a.GetProperty("ak9");

        // One deliberate, transactional draw per outbound 997. Gaps are normal in EDI.
        long ctl = await NextControlNumberAsync(config.ConnectionString, partner, ct);
        var ctl9 = ctl.ToString("000000000");
        var now  = DateTime.UtcNow;
        var pad  = (string s) => (s + new string(' ', 15))[..15];

        var seg = new List<string>
        {
            $"ISA*00*          *00*          *ZZ*{pad(self)}*ZZ*{pad(partner)}*{now:yyMMdd}*{now:HHmm}*U*00501*{ctl9}*0*P*>",
            $"GS*FA*{self}*{partner}*{now:yyyyMMdd}*{now:HHmm}*{ctl}*X*005010",
            "ST*997*0001",
            $"AK1*{ak1.GetProperty("functionalIdCode").GetString()}*{ak1.GetProperty("groupControlNumber").GetString()}",
            $"AK2*{ts.GetProperty("setId").GetString()}*{ts.GetProperty("controlNumber").GetString()}"
        };

        // AK3/AK4 only when the set was rejected for segment/element faults.
        if (ts.GetProperty("ak5").GetString() == "R")
            foreach (var e in ts.GetProperty("segmentErrors").EnumerateArray())
            {
                seg.Add($"AK3*{e.GetProperty("segmentId").GetString()}*{e.GetProperty("position").GetInt32()}**{e.GetProperty("code").GetString()}");
                foreach (var el in e.GetProperty("elementErrors").EnumerateArray())
                    seg.Add($"AK4*{el.GetProperty("position").GetInt32()}**{el.GetProperty("code").GetString()}");
            }

        var ak5line = ts.GetProperty("ak5").GetString() == "A"
            ? "AK5*A" : "AK5*R*5";
        seg.Add(ak5line);
        seg.Add($"AK9*{ak9.GetProperty("code").GetString()}*{ak9.GetProperty("setsIncluded").GetInt32()}*{ak9.GetProperty("setsReceived").GetInt32()}*{ak9.GetProperty("setsAccepted").GetInt32()}");
        seg.Add($"SE*{seg.Count - 1}*0001");   // count ST..SE inclusive; two envelope lines precede ST
        seg.Add($"GE*1*{ctl}");
        seg.Add($"IEA*1*{ctl9}");

        var x12 = string.Join("", seg.Select(s => s + "~\n"));
        return new PipelineComponentOutput
        {
            Success  = true,
            Messages = [new PipelineMessage { Body = x12, MessageType = "Edi997Interchange" }]
        };
    }

    private static async Task<long> NextControlNumberAsync(string conn, string partner, CancellationToken ct)
    {
        await using var c = new SqlConnection(conn);
        await c.OpenAsync(ct);
        await using var cmd = new SqlCommand("dbo.usp_NextOutboundControlNumber", c)
            { CommandType = System.Data.CommandType.StoredProcedure };
        cmd.Parameters.AddWithValue("@PartnerCode", partner);
        return Convert.ToInt64(await cmd.ExecuteScalarAsync(ct));
    }
}

{
  "$schema": "https://json-schema.org/draft/2020-12/schema",
  "title": "Edi850Order",
  "type": "object",
  "required": ["interchange", "purchaseOrder"],
  "properties": {
    "purchaseOrder": {
      "type": "object",
      "required": ["poNumber", "buyer", "lines"],
      "properties": {
        "poNumber": { "type": "string" },
        "buyer": {
          "type": "object",
          "required": ["id"],
          "properties": { "id": { "type": "string" } }
        },
        "lines": {
          "type": "array",
          "minItems": 1,
          "items": {
            "type": "object",
            "required": ["vendorPart", "quantity"],
            "properties": {
              "vendorPart": { "type": "string" },
              "quantity":   { "type": "integer", "minimum": 1 },
              "unitPrice":  { "type": "number" }
            }
          }
        }
      }
    }
  }
}

<xsl:stylesheet version="3.0"
    xmlns:xsl="http://www.w3.org/1999/XSL/Transform">
  <xsl:output method="xml" indent="yes" omit-xml-declaration="yes"/>
  <xsl:template match="/">
    <xsl:variable name="o" select="parse-json(.)"/>
    <xsl:variable name="po" select="$o?purchaseOrder"/>
    <BookOrder>
      <Client>
        <PartnerCode><xsl:value-of select="$o?interchange?senderId"/></PartnerCode>
        <AccountRef><xsl:value-of select="$po?buyer?id"/></AccountRef>
        <DisplayName><xsl:value-of select="$po?buyer?name"/></DisplayName>
      </Client>
      <Order>
        <CustomerReference><xsl:value-of select="$po?poNumber"/></CustomerReference>
        <OrderKind><xsl:value-of select="$po?type"/></OrderKind>
        <PlacedOn><xsl:value-of select="$po?poDate"/></PlacedOn>
        <NeededBy><xsl:value-of select="$po?requestedDelivery"/></NeededBy>
        <SourceGroupControl><xsl:value-of select="$o?group?controlNumber"/></SourceGroupControl>
        <Lines>
          <xsl:for-each select="$po?lines?*">
            <Line>
              <LinePosition><xsl:value-of select="?lineNumber"/></LinePosition>
              <CatalogCode><xsl:value-of select="?vendorPart"/></CatalogCode>
              <LineText><xsl:value-of select="?description"/></LineText>
              <Cases><xsl:value-of select="?quantity"/></Cases>
              <UnitOfMeasure><xsl:value-of select="?uom"/></UnitOfMeasure>
              <PriceEach><xsl:value-of select="?unitPrice"/></PriceEach>
            </Line>
          </xsl:for-each>
        </Lines>
      </Order>
    </BookOrder>
  </xsl:template>
</xsl:stylesheet>