const {
  Document, Packer, Paragraph, TextRun, Table, TableRow, TableCell,
  HeadingLevel, AlignmentType, BorderStyle, WidthType, ShadingType,
  VerticalAlign, PageNumber, Header, Footer, UnderlineType
} = require('docx');
const fs = require('fs');

const redShading = { type: ShadingType.CLEAR, fill: "FFE0E0" };
const yellowShading = { type: ShadingType.CLEAR, fill: "FFFBE6" };
const greenShading = { type: ShadingType.CLEAR, fill: "E6F9EE" };
const headerShading = { type: ShadingType.CLEAR, fill: "1A3C5E" };
const lightBlueShading = { type: ShadingType.CLEAR, fill: "E8F0FB" };

const cellBorders = {
  top: { style: BorderStyle.SINGLE, size: 4, color: "CCCCCC" },
  bottom: { style: BorderStyle.SINGLE, size: 4, color: "CCCCCC" },
  left: { style: BorderStyle.SINGLE, size: 4, color: "CCCCCC" },
  right: { style: BorderStyle.SINGLE, size: 4, color: "CCCCCC" },
};

function makeHeaderCell(text, shading = headerShading) {
  return new TableCell({
    shading,
    borders: cellBorders,
    verticalAlign: VerticalAlign.CENTER,
    children: [new Paragraph({
      alignment: AlignmentType.CENTER,
      children: [new TextRun({ text, bold: true, color: "FFFFFF", size: 20 })]
    })]
  });
}

function makeCell(text, shading = {}, bold = false, color = "000000") {
  return new TableCell({
    shading: shading.fill ? shading : undefined,
    borders: cellBorders,
    verticalAlign: VerticalAlign.CENTER,
    children: [new Paragraph({
      alignment: AlignmentType.CENTER,
      children: [new TextRun({ text, bold, size: 20, color })]
    })]
  });
}

function makeLeftCell(text, shading, bold = false) {
  return new TableCell({
    shading: shading && shading.fill ? shading : undefined,
    borders: cellBorders,
    children: [new Paragraph({
      children: [new TextRun({ text, bold, size: 20 })]
    })]
  });
}

const doc = new Document({
  sections: [{
    headers: {
      default: new Header({
        children: [new Paragraph({
          alignment: AlignmentType.CENTER,
          children: [
            new TextRun({ text: "ECG REPORT SUMMARY", bold: true, size: 28, color: "1A3C5E" }),
            new TextRun({ text: "  |  For Doctor Consultation", size: 22, color: "666666" })
          ]
        })]
      })
    },
    footers: {
      default: new Footer({
        children: [new Paragraph({
          alignment: AlignmentType.CENTER,
          children: [
            new TextRun({ text: "⚠ This document is for informational purposes only and does not replace professional medical advice.", size: 16, color: "888888", italics: true })
          ]
        })]
      })
    },
    children: [

      // Title
      new Paragraph({
        alignment: AlignmentType.CENTER,
        spacing: { before: 200, after: 100 },
        children: [new TextRun({ text: "ECG REPORT SUMMARY", bold: true, size: 40, color: "1A3C5E" })]
      }),
      new Paragraph({
        alignment: AlignmentType.CENTER,
        spacing: { after: 400 },
        children: [new TextRun({ text: "Prepared for Doctor Consultation  |  Date of ECG: 19 May 2026", size: 20, color: "555555", italics: true })]
      }),

      // --- PATIENT INFORMATION ---
      new Paragraph({
        heading: HeadingLevel.HEADING_1,
        spacing: { before: 200, after: 100 },
        children: [new TextRun({ text: "1. PATIENT INFORMATION", bold: true, size: 26, color: "1A3C5E" })]
      }),
      new Table({
        width: { size: 100, type: WidthType.PERCENTAGE },
        rows: [
          new TableRow({ children: [makeHeaderCell("Field"), makeHeaderCell("Details")] }),
          new TableRow({ children: [makeLeftCell("Full Name", lightBlueShading, true), makeLeftCell("HARJEET", lightBlueShading)] }),
          new TableRow({ children: [makeLeftCell("Age", {}, true), makeLeftCell("19 Years")] }),
          new TableRow({ children: [makeLeftCell("Gender", lightBlueShading, true), makeLeftCell("Male", lightBlueShading)] }),
          new TableRow({ children: [makeLeftCell("ECG Date & Time", {}, true), makeLeftCell("19 May 2026, 13:09:08")] }),
          new TableRow({ children: [makeLeftCell("Lab / Facility", lightBlueShading, true), makeLeftCell("H S LABS PHILLAUR - 01826220014", lightBlueShading)] }),
        ]
      }),

      // --- ECG MEASUREMENTS ---
      new Paragraph({
        heading: HeadingLevel.HEADING_1,
        spacing: { before: 400, after: 100 },
        children: [new TextRun({ text: "2. ECG MEASUREMENTS", bold: true, size: 26, color: "1A3C5E" })]
      }),
      new Table({
        width: { size: 100, type: WidthType.PERCENTAGE },
        rows: [
          new TableRow({ children: [makeHeaderCell("Parameter"), makeHeaderCell("Recorded Value"), makeHeaderCell("Normal Range"), makeHeaderCell("Status")] }),
          new TableRow({ children: [makeCell("Heart Rate (HR)", lightBlueShading), makeCell("71 bpm", lightBlueShading), makeCell("60-100 bpm", lightBlueShading), makeCell("✔ Normal", greenShading, true, "1A6B35")] }),
          new TableRow({ children: [makeCell("P-R Interval"), makeCell("116 ms"), makeCell("120-200 ms"), makeCell("⚠ Slightly Short", yellowShading, true, "7A5F00")] }),
          new TableRow({ children: [makeCell("QRS Duration", lightBlueShading), makeCell("120 ms", lightBlueShading), makeCell("< 120 ms", lightBlueShading), makeCell("⚠ Borderline Wide", yellowShading, true, "7A5F00")] }),
          new TableRow({ children: [makeCell("QT / QTc"), makeCell("383 / 420 ms"), makeCell("QTc < 440 ms"), makeCell("✔ Normal", greenShading, true, "1A6B35")] }),
          new TableRow({ children: [makeCell("P / QRS / T Axes", lightBlueShading), makeCell("72 / 185 / 72°", lightBlueShading), makeCell("QRS 0–90°", lightBlueShading), makeCell("⚠ Abnormal QRS Axis", yellowShading, true, "7A5F00")] }),
          new TableRow({ children: [makeCell("RV5 / SV1"), makeCell("2.98 / 1.53 mV"), makeCell("-"), makeCell("-")] }),
          new TableRow({ children: [makeCell("RV5 + SV1 (LVH index)", lightBlueShading), makeCell("4.50 mV", lightBlueShading), makeCell("< 3.5 mV", lightBlueShading), makeCell("⚠ Elevated (LVH criteria)", redShading, true, "8B0000")] }),
        ]
      }),

      // --- COMPUTER INTERPRETATION ---
      new Paragraph({
        heading: HeadingLevel.HEADING_1,
        spacing: { before: 400, after: 100 },
        children: [new TextRun({ text: "3. MACHINE INTERPRETATION (Auto-Analysis)", bold: true, size: 26, color: "1A3C5E" })]
      }),
      new Table({
        width: { size: 100, type: WidthType.PERCENTAGE },
        rows: [
          new TableRow({ children: [makeHeaderCell("Finding"), makeHeaderCell("Significance")] }),
          new TableRow({ children: [makeLeftCell("Sinus Rhythm", lightBlueShading, true), makeLeftCell("Normal heart rhythm - heart is beating regularly from the sinus node", lightBlueShading)] }),
          new TableRow({ children: [makeLeftCell("ST Elevation", { type: ShadingType.CLEAR, fill: "FFD0D0" }, true), makeLeftCell("URGENT: Must be evaluated clinically. May indicate early repolarisation, pericarditis, or rarely ischaemia")] }),
          new TableRow({ children: [makeLeftCell("LVH - Probably", yellowShading, true), makeLeftCell("Left Ventricular Hypertrophy suspected. May be due to athletic heart, hypertension, or hypertrophic cardiomyopathy", yellowShading)] }),
          new TableRow({ children: [makeLeftCell(">> SUSPECTED ABNORMAL ECG <<", redShading, true), makeLeftCell("Machine has flagged this ECG as abnormal. Requires doctor confirmation", redShading)] }),
        ]
      }),

      // --- QUESTIONS TO ASK DOCTOR ---
      new Paragraph({
        heading: HeadingLevel.HEADING_1,
        spacing: { before: 400, after: 100 },
        children: [new TextRun({ text: "4. SUGGESTED QUESTIONS FOR YOUR DOCTOR", bold: true, size: 26, color: "1A3C5E" })]
      }),
      new Paragraph({ spacing: { after: 80 }, children: [new TextRun({ text: "1.  What is causing the ST elevation on my ECG? Is it dangerous?", size: 20 })] }),
      new Paragraph({ spacing: { after: 80 }, children: [new TextRun({ text: "2.  Do I have Left Ventricular Hypertrophy (LVH)? What could be causing it?", size: 20 })] }),
      new Paragraph({ spacing: { after: 80 }, children: [new TextRun({ text: "3.  Should I have an Echocardiogram (heart ultrasound) to assess my heart structure?", size: 20 })] }),
      new Paragraph({ spacing: { after: 80 }, children: [new TextRun({ text: "4.  Is the abnormal QRS axis clinically significant for me?", size: 20 })] }),
      new Paragraph({ spacing: { after: 80 }, children: [new TextRun({ text: "5.  Do I need any blood tests (e.g. troponin, BNP) or further cardiac investigations?", size: 20 })] }),
      new Paragraph({ spacing: { after: 80 }, children: [new TextRun({ text: "6.  Are there any restrictions on physical activity or exercise until this is clarified?", size: 20 })] }),
      new Paragraph({ spacing: { after: 80 }, children: [new TextRun({ text: "7.  Is a follow-up ECG or Holter monitor needed?", size: 20 })] }),

      // --- URGENT NOTICE ---
      new Paragraph({
        heading: HeadingLevel.HEADING_1,
        spacing: { before: 400, after: 100 },
        children: [new TextRun({ text: "5. IMPORTANT NOTICE", bold: true, size: 26, color: "CC0000" })]
      }),
      new Table({
        width: { size: 100, type: WidthType.PERCENTAGE },
        rows: [
          new TableRow({
            children: [new TableCell({
              shading: redShading,
              borders: cellBorders,
              children: [
                new Paragraph({ spacing: { before: 100, after: 60 }, children: [new TextRun({ text: "⚠ This ECG has been flagged as SUSPECTED ABNORMAL by the machine.", bold: true, size: 22, color: "8B0000" })] }),
                new Paragraph({ spacing: { after: 60 }, children: [new TextRun({ text: "The ST elevation and possible LVH findings need to be urgently reviewed by a qualified doctor or cardiologist.", size: 20 })] }),
                new Paragraph({ spacing: { after: 60 }, children: [new TextRun({ text: "The ECG report itself states: \"The result must be confirmed by doctor\"", size: 20, italics: true })] }),
                new Paragraph({ spacing: { after: 60 }, children: [new TextRun({ text: "If you are experiencing chest pain, shortness of breath, or palpitations, seek emergency medical care immediately.", bold: true, size: 20, color: "8B0000" })] }),
              ]
            })]
          })
        ]
      }),

      // Footer note
      new Paragraph({
        spacing: { before: 400 },
        alignment: AlignmentType.CENTER,
        children: [new TextRun({ text: "Document prepared by Orris AI | For use in doctor consultation only | Not a substitute for medical diagnosis", size: 16, color: "999999", italics: true })]
      }),
    ]
  }]
});

Packer.toBuffer(doc).then(buffer => {
  fs.writeFileSync("/home/daytona/workspace/ecg-report/ECG_Summary_HARJEET.docx", buffer);
  console.log("Document created successfully.");
});