eCourt-Doc-Edit/backend/WordVersioning.Api.IntegrationTests/TestDocx.cs

using System.Text;
using DocumentFormat.OpenXml;
using DocumentFormat.OpenXml.Packaging;
using DocumentFormat.OpenXml.Wordprocessing;

namespace WordVersioning.Api.IntegrationTests;

/// <summary>สร้าง/อ่านไฟล์ .docx สำหรับ test — ไม่ต้องพึ่ง binary file ใน repo
/// สร้างได้ทั้ง template ที่มี content controls (SDT) หลายรูปแบบและไฟล์เปล่าไม่มี SDT</summary>
public static class TestDocx
{
    public const string DocxContentType =
        "application/vnd.openxmlformats-officedocument.wordprocessingml.document";

    /// <summary>สร้าง template มาตรฐาน: SDT แบบ run เดียว / multi-run คนละ formatting / block-level
    /// + tag ซ้ำอีกจุด (detect ต้อง dedup) — ครอบคลุมรูปแบบเดียวกับที่ใช้ตรวจ Phase 3 ตอน dev
    ///
    /// ⚠️ run-level SDT (SdtRun) ต้องอยู่ "ใน" paragraph เท่านั้น (ตาม OOXML schema เหมือนที่ Word author ให้)
    /// — วางเป็นลูกตรงของ Body แล้วตอน reload จะกลายเป็น SdtBlock + OpenXmlUnknownElement ทั้งชุด
    /// ทำให้ generate เข้า branch ผิด (append แทนที่จะ replace)</summary>
    public static byte[] BuildTemplateDocx()
    {
        var buffer = new MemoryStream();
        using (var doc = WordprocessingDocument.Create(buffer, WordprocessingDocumentType.Document))
        {
            var main = doc.AddMainDocumentPart();
            main.Document = new Document(new Body(
                new Paragraph(new Run(new Text("สัญญาจ้างงานฉบับทดสอบ"))),

                // SDT run-level เดี่ยว — รูปแบบที่พบบ่อยสุด
                new Paragraph(SdtRun("employee_name", new Run(
                    new RunProperties(new Bold()),
                    new Text("[ชื่อพนักงาน]")))),

                // SDT multi-run คนละ formatting — generate ต้องเขียนค่าที่ run แรก (รักษา bold) และลบ text ของ run หลัง
                new Paragraph(SdtRun("position",
                    new Run(
                        new RunProperties(new Bold()),
                        new Text("[ตำแหน่ง")),
                    new Run(
                        new RunProperties(new Italic()),
                        new Text("หน้าที่]")))),

                // SDT block-level (ทั้งย่อหน้า) — ตัวนี้เท่านั้นที่เป็นลูกตรงของ Body ได้
                new SdtBlock(
                    new SdtProperties(new Tag { Val = "salary" }),
                    new SdtContentBlock(new Paragraph(new Run(new Text("[เงินเดือน]"))))),

                new Paragraph(SdtRun("start_date", new Run(new Text("[วันเริ่มงาน]")))),

                // tag ซ้ำ (อีกจุดของเอกสาร) — detect ต้องได้ครั้งเดียว
                new Paragraph(SdtRun("employee_name", new Run(new Text("[ชื่อพนักงานอีกครั้ง]")))),

                new Paragraph(new Run(new Text("จบเอกสาร")))));
        }
        return buffer.ToArray();
    }

    /// <summary>template ตัวแปร: มี employee_name + position เท่านั้น — ใช้ตอน replace ไฟล์ (tag เพิ่ม/หาย)</summary>
    public static byte[] BuildReplacementTemplateDocx()
    {
        var buffer = new MemoryStream();
        using (var doc = WordprocessingDocument.Create(buffer, WordprocessingDocumentType.Document))
        {
            var main = doc.AddMainDocumentPart();
            main.Document = new Document(new Body(
                new Paragraph(new Run(new Text("สัญญาฉบับแก้ไข"))),
                new Paragraph(SdtRun("employee_name", new Run(new Text("[ชื่อ]")))),
                new Paragraph(SdtRun("note", new Run(new Text("[หมายเหตุ]"))))));
        }
        return buffer.ToArray();
    }

    /// <summary>ไฟล์ .docx ไม่มี content control เลย — จำลองเอกสารธรรมดาที่ track เองผ่าน add-in</summary>
    public static byte[] BuildPlainDocx(string text = "เอกสารทั่วไปไม่มี content control")
    {
        var buffer = new MemoryStream();
        using (var doc = WordprocessingDocument.Create(buffer, WordprocessingDocumentType.Document))
        {
            var main = doc.AddMainDocumentPart();
            main.Document = new Document(new Body(new Paragraph(new Run(new Text(text)))));
        }
        return buffer.ToArray();
    }

    /// <summary>อ่านข้อความรวมของ SDT ตาม tag (คืน "" เมื่อไม่มี tag นั้นในไฟล์) — ตรวจผลลัพธ์ generate</summary>
    public static string ReadSdtText(byte[] docx, string tag)
    {
        using var doc = WordprocessingDocument.Open(new MemoryStream(docx), false);
        var main = doc.MainDocumentPart!;
        foreach (var root in new OpenXmlPartRootElement?[] { main.Document }
                     .Concat(main.HeaderParts.Select(h => (OpenXmlPartRootElement?)h.Header))
                     .Concat(main.FooterParts.Select(f => (OpenXmlPartRootElement?)f.Footer)))
        {
            if (root is null) continue;
            foreach (var sdtTag in root.Descendants<Tag>())
            {
                if (sdtTag.Val?.Value != tag) continue;
                var sdt = sdtTag.Parent?.Parent;
                var content = sdt?.ChildElements.FirstOrDefault(c => c.LocalName == "sdtContent");
                if (content is null) return "";
                var sb = new StringBuilder();
                foreach (var text in content.Descendants<Text>())
                    sb.Append(text.Text);
                return sb.ToString();
            }
        }
        return "";
    }

    /// <summary>อ่านข้อความ+formatting ของแต่ละ run ใน SDT (ตาม tag) — ใช้ตรวจว่า generate
    /// รวม multi-run เป็น run เดียวแล้วยังรักษา run properties (bold/italic) ของ run แรกไว้</summary>
    public static List<(string Text, bool Bold, bool Italic)> ReadSdtRuns(byte[] docx, string tag)
    {
        using var doc = WordprocessingDocument.Open(new MemoryStream(docx), false);
        var root = doc.MainDocumentPart!.Document!;
        foreach (var sdtTag in root.Descendants<Tag>())
        {
            if (sdtTag.Val?.Value != tag) continue;
            var sdt = sdtTag.Parent?.Parent;
            var content = sdt?.ChildElements.FirstOrDefault(c => c.LocalName == "sdtContent");
            return content?.Descendants<Run>()
                .Select(run => (
                    string.Concat(run.Descendants<Text>().Select(t => t.Text)),
                    run.RunProperties?.Bold is not null,
                    run.RunProperties?.Italic is not null))
                .ToList() ?? [];
        }
        return [];
    }

    /// <summary>อ่าน custom property (VTLPWSTR) — null เมื่อไม่มี property นั้น
    /// ใช้ตรวจ SavedByAddIn / DocumentTrackingId ที่ backend stamp ลงไฟล์ (FR-10)</summary>
    public static string? ReadCustomProperty(byte[] docx, string name)
    {
        using var doc = WordprocessingDocument.Open(new MemoryStream(docx), false);
        var props = doc.CustomFilePropertiesPart?.Properties;
        return props?.Elements<DocumentFormat.OpenXml.CustomProperties.CustomDocumentProperty>()
            .FirstOrDefault(p => p.Name?.Value == name)
            ?.VTLPWSTR?.Text;
    }

    private static SdtRun SdtRun(string tag, params Run[] runs) => new(
        new SdtProperties(new Tag { Val = tag }),
        new SdtContentRun(runs));
}