from reportlab.lib.pagesizes import A4
from reportlab.lib import colors
from reportlab.lib.units import mm
from reportlab.platypus import (
    SimpleDocTemplate, Table, TableStyle, Paragraph, Spacer, HRFlowable
)
from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
from reportlab.lib.enums import TA_CENTER, TA_LEFT
from reportlab.platypus import KeepTogether

OUTPUT = "/tmp/workspace/vacutainer-chart/Vacutainer_Quick_Reference.pdf"

# ── Colour palette (RGB 0-1) ────────────────────────────────────────────────
LIGHT_BLUE_CAP  = colors.HexColor("#AED6F1")
RED_CAP         = colors.HexColor("#E74C3C")
GOLD_CAP        = colors.HexColor("#F0B327")
GREEN_CAP       = colors.HexColor("#2ECC71")
LAVENDER_CAP    = colors.HexColor("#9B59B6")
PINK_CAP        = colors.HexColor("#F48FB1")
GRAY_CAP        = colors.HexColor("#95A5A6")
YELLOW_CAP      = colors.HexColor("#F9E231")
ROYAL_BLUE_CAP  = colors.HexColor("#1A5276")
WHITE_CAP       = colors.HexColor("#D5D8DC")
HEADER_BG       = colors.HexColor("#1B2631")
PAGE_BG         = colors.HexColor("#FDFEFE")
STRIPE          = colors.HexColor("#EBF5FB")

doc = SimpleDocTemplate(
    OUTPUT,
    pagesize=A4,
    leftMargin=12*mm, rightMargin=12*mm,
    topMargin=14*mm, bottomMargin=14*mm,
    title="Vacutainer Quick Reference Chart",
    author="Orris Medical Reference"
)

W, H = A4
styles = getSampleStyleSheet()

title_style = ParagraphStyle(
    "title", fontName="Helvetica-Bold", fontSize=17,
    textColor=colors.white, alignment=TA_CENTER, spaceAfter=2
)
sub_style = ParagraphStyle(
    "sub", fontName="Helvetica", fontSize=9,
    textColor=colors.HexColor("#AEB6BF"), alignment=TA_CENTER, spaceAfter=0
)
col_header_style = ParagraphStyle(
    "col_hdr", fontName="Helvetica-Bold", fontSize=8.5,
    textColor=colors.white, alignment=TA_CENTER
)
cell_style = ParagraphStyle(
    "cell", fontName="Helvetica", fontSize=7.5,
    textColor=colors.HexColor("#1B2631"), alignment=TA_LEFT, leading=10
)
cell_bold = ParagraphStyle(
    "cell_bold", fontName="Helvetica-Bold", fontSize=8,
    textColor=colors.HexColor("#1B2631"), alignment=TA_LEFT, leading=11
)
small_style = ParagraphStyle(
    "small", fontName="Helvetica-Oblique", fontSize=6.5,
    textColor=colors.HexColor("#717D7E"), alignment=TA_CENTER
)
note_style = ParagraphStyle(
    "note", fontName="Helvetica", fontSize=7,
    textColor=colors.HexColor("#1B2631"), alignment=TA_LEFT, leading=9
)

# ── Tube data ────────────────────────────────────────────────────────────────
# (cap_color, text_color_on_cap, tube_label, additive, mechanism_short, sample, tests)
TUBES = [
    (
        LIGHT_BLUE_CAP, colors.HexColor("#1A5276"),
        "LIGHT BLUE",
        "Sodium Citrate\n0.109 M (3.2%)",
        "Chelates Ca²⁺ (reversible)\n→ 3–4 inversions\n⚠ Fill EXACTLY to line",
        "Plasma",
        "PT / INR\naPTT\nFibrinogen\nD-Dimer\nThrombin Time\nFactor Assays (II,V,VII–XI)\nLupus Anticoagulant\nTEG / PFA-100",
    ),
    (
        RED_CAP, colors.white,
        "RED (Plain)",
        "None (glass) or\nSilicone clot activator (plastic)",
        "Natural clotting → serum\n0 inversions (glass)\n5 inversions (plastic)",
        "Serum",
        "Drug levels (digoxin, phenytoin,\n  lithium, vancomycin)\nSerology (RPR, rubella, CMV)\nViral antibodies (HIV, HBV, HCV)\nBlood bank (where gel avoided)",
    ),
    (
        GOLD_CAP, colors.HexColor("#1B2631"),
        "GOLD / SST",
        "Clot activator (silica)\n+ Polymer gel barrier",
        "Silica speeds clotting;\ngel forms barrier after spin\n5 inversions, wait 30 min",
        "Serum",
        "CMP (glucose, BUN, Cr, LFTs,\n  electrolytes)\nLipid panel\nTSH, T3, T4\nHormones (FSH, LH, testosterone)\nVitamin D / B12 / folate\nTumor markers (PSA, CEA, AFP,\n  CA-125, CA 19-9)\nTroponin, CK-MB, BNP\nHBsAg, HIV Ag/Ab combo",
    ),
    (
        GREEN_CAP, colors.white,
        "GREEN",
        "Lithium heparin or\nSodium heparin",
        "Activates antithrombin III\n→ inhibits thrombin & Xa\n8 inversions",
        "Plasma",
        "STAT / urgent chemistry\nAmmonia ⚠ (on ice, <15 min)\nABG (heparinised syringe)\nChromosomal / cytogenetics\nHLA typing\nLead level\nCarboxyhemoglobin",
    ),
    (
        LAVENDER_CAP, colors.white,
        "LAVENDER / PURPLE",
        "K₂-EDTA or K₃-EDTA",
        "Chelates Ca²⁺ (irreversible)\nPreserves cell morphology\n8 inversions",
        "Whole Blood",
        "CBC with differential\nHbA1c\nESR\nPeripheral blood smear\nReticulocyte count\nSickle cell screen / malaria smear\nDirect Coombs\nImmunosuppressants (tacrolimus,\n  cyclosporine)\nWhole blood lead\nEBV / CMV PCR",
    ),
    (
        PINK_CAP, colors.HexColor("#1B2631"),
        "PINK",
        "K₂-EDTA\n(blood bank grade)",
        "Same as lavender EDTA\nStrict 2-ID labelling required\n8 inversions",
        "Whole Blood",
        "Type & Screen\nType & Crossmatch\nAntibody identification\nDirect Antiglobulin Test (DAT)\nTransfusion compatibility",
    ),
    (
        GRAY_CAP, colors.white,
        "GRAY",
        "NaF + Potassium oxalate\nor NaF + Na₂-EDTA",
        "NaF inhibits enolase\n→ stops glycolysis\nOxalate/EDTA anticoagulate\n8 inversions",
        "Plasma",
        "Fasting glucose / GTT\nLactate ⚠ (on ice)\nBlood alcohol (ethanol)\nFluoride-sensitive pesticides",
    ),
    (
        YELLOW_CAP, colors.HexColor("#1B2631"),
        "YELLOW",
        "SPS (Sodium Polyanethol\n  Sulfonate)\nor ACD Solution A or B",
        "SPS: anticoagulant +\n  inhibits complement/phagocytosis\nACD: citrate + dextrose\n  preserves RBC metabolism",
        "Whole Blood",
        "SPS → Blood cultures\n  (microbiology)\nACD → HLA antigen typing\nACD → Paternity / DNA testing\nACD → Blood bank studies",
    ),
    (
        ROYAL_BLUE_CAP, colors.white,
        "ROYAL BLUE",
        "K₂-EDTA (plastic) or\nClot activator or\nNo additive",
        "Low trace-metal certified tube\nUse ONLY for metals\n8 inversions (EDTA version)",
        "Serum or Plasma",
        "Trace metals (zinc, copper,\n  selenium, manganese)\nHeavy metal screen\nToxicology panels\nNutritional element testing",
    ),
    (
        WHITE_CAP, colors.HexColor("#1B2631"),
        "WHITE",
        "K₂-EDTA + gel separator",
        "EDTA prevents clotting;\ngel separates plasma from cells\n8 inversions",
        "Plasma",
        "Molecular PCR assays\nViral loads (HIV RNA, HCV RNA,\n  CMV viral load)\nbDNA amplification",
    ),
]

# ── Build table rows ──────────────────────────────────────────────────────────
col_widths = [26*mm, 38*mm, 38*mm, 18*mm, 56*mm]

def cap_cell(cap_color, text_color, label):
    """Rounded coloured cell simulated via a 1-cell mini-table."""
    p = Paragraph(f"<b>{label}</b>", ParagraphStyle(
        "cap", fontName="Helvetica-Bold", fontSize=7.5,
        textColor=text_color, alignment=TA_CENTER, leading=10
    ))
    return p

header_row = [
    Paragraph("TUBE COLOR", col_header_style),
    Paragraph("ADDITIVE", col_header_style),
    Paragraph("MECHANISM / NOTES", col_header_style),
    Paragraph("SAMPLE", col_header_style),
    Paragraph("TESTS / USES", col_header_style),
]

table_data = [header_row]
row_bg_colors = []  # (row_index, color)

for i, (cap_color, text_color, label, additive, mechanism, sample, tests) in enumerate(TUBES):
    row_idx = i + 1
    row = [
        cap_cell(cap_color, text_color, label),
        Paragraph(additive, cell_style),
        Paragraph(mechanism, cell_style),
        Paragraph(sample, cell_style),
        Paragraph(tests, cell_style),
    ]
    table_data.append(row)
    row_bg_colors.append((row_idx, cap_color))

table = Table(table_data, colWidths=col_widths, repeatRows=1)

# ── TableStyle ────────────────────────────────────────────────────────────────
ts = TableStyle([
    # Header
    ("BACKGROUND",  (0,0), (-1,0), HEADER_BG),
    ("TEXTCOLOR",   (0,0), (-1,0), colors.white),
    ("FONTNAME",    (0,0), (-1,0), "Helvetica-Bold"),
    ("FONTSIZE",    (0,0), (-1,0), 8.5),
    ("ALIGN",       (0,0), (-1,0), "CENTER"),
    ("VALIGN",      (0,0), (-1,-1), "MIDDLE"),
    ("TOPPADDING",  (0,0), (-1,0), 5),
    ("BOTTOMPADDING",(0,0),(-1,0), 5),
    # All cells
    ("FONTSIZE",    (0,1), (-1,-1), 7.5),
    ("TOPPADDING",  (0,1), (-1,-1), 4),
    ("BOTTOMPADDING",(0,1),(-1,-1), 4),
    ("LEFTPADDING", (0,0), (-1,-1), 4),
    ("RIGHTPADDING",(0,0), (-1,-1), 4),
    # Grid
    ("GRID",        (0,0), (-1,-1), 0.4, colors.HexColor("#BDC3C7")),
    ("LINEBELOW",   (0,0), (-1,0), 1.2, colors.HexColor("#BDC3C7")),
    # Alternate stripe
    *[("BACKGROUND", (0, r), (-1, r), STRIPE) for r in range(2, len(TUBES)+1, 2)],
])

# Apply per-row cap-column background colours
for row_idx, cap_color in row_bg_colors:
    ts.add("BACKGROUND", (0, row_idx), (0, row_idx), cap_color)

table.setStyle(ts)

# ── Order of draw box ─────────────────────────────────────────────────────────
draw_order_data = [
    [Paragraph("<b>ORDER OF DRAW  (CLSI Standard)</b>", ParagraphStyle(
        "od_title", fontName="Helvetica-Bold", fontSize=8,
        textColor=colors.white, alignment=TA_CENTER))],
    [Paragraph(
        "1. Blood Cultures (Yellow/SPS)  →  2. Sodium Citrate (Light Blue)  →  "
        "3. Serum tubes (Red, then Gold/SST)  →  4. Heparin (Green)  →  "
        "5. EDTA (Lavender / Pink)  →  6. Fluoride/Oxalate (Gray)",
        ParagraphStyle("od_body", fontName="Helvetica", fontSize=7.5,
                       textColor=colors.HexColor("#1B2631"), alignment=TA_CENTER, leading=11)
    )],
]
draw_order_table = Table(draw_order_data, colWidths=[176*mm])
draw_order_table.setStyle(TableStyle([
    ("BACKGROUND",   (0,0), (-1,0), HEADER_BG),
    ("BACKGROUND",   (0,1), (-1,1), colors.HexColor("#D6EAF8")),
    ("TOPPADDING",   (0,0), (-1,-1), 4),
    ("BOTTOMPADDING",(0,0), (-1,-1), 4),
    ("LEFTPADDING",  (0,0), (-1,-1), 6),
    ("RIGHTPADDING", (0,0), (-1,-1), 6),
    ("BOX",          (0,0), (-1,-1), 0.8, colors.HexColor("#85929E")),
]))

# ── Key tips ──────────────────────────────────────────────────────────────────
tips = [
    "• <b>Never shake</b> – invert gently; shaking causes haemolysis.",
    "• <b>Light Blue must be filled exactly</b> to the line (1:9 citrate:blood ratio); under-fill invalidates coagulation results.",
    "• <b>SST (Gold)</b> – allow 30 min clotting before centrifugation.",
    "• <b>Lavender ≠ Pink</b> – both are EDTA, but Pink requires a second positive patient ID for transfusion safety.",
    "• <b>Gray tube glucose</b> is stable for 24 h (NaF halts glycolysis); without NaF, glucose drops ~10 mg/dL per hour at room temp.",
    "• <b>Ammonia (Green)</b> – place on ice immediately; process within 15 min to avoid falsely elevated values.",
    "• <b>Royal Blue</b> – use only low-trace-metal certified tubes for heavy-metal testing; a standard tube will contaminate the sample.",
]
tips_data = [[Paragraph("<b>KEY TIPS FOR PHLEBOTOMISTS</b>", ParagraphStyle(
    "tip_title", fontName="Helvetica-Bold", fontSize=8,
    textColor=colors.white, alignment=TA_CENTER))]]
for t in tips:
    tips_data.append([Paragraph(t, note_style)])

tips_table = Table(tips_data, colWidths=[176*mm])
tips_style = TableStyle([
    ("BACKGROUND",   (0,0), (-1,0), HEADER_BG),
    ("BACKGROUND",   (0,1), (-1,-1), colors.HexColor("#FDFEFE")),
    ("TOPPADDING",   (0,0), (-1,-1), 3),
    ("BOTTOMPADDING",(0,0), (-1,-1), 3),
    ("LEFTPADDING",  (0,0), (-1,-1), 6),
    ("RIGHTPADDING", (0,0), (-1,-1), 6),
    ("BOX",          (0,0), (-1,-1), 0.8, colors.HexColor("#85929E")),
    ("LINEBELOW",    (0,0), (-1,0), 0.8, colors.HexColor("#85929E")),
    *[("BACKGROUND", (0, r), (-1, r), STRIPE) for r in range(2, len(tips)+1, 2)],
])
tips_table.setStyle(tips_style)

# ── Page background colour ───────────────────────────────────────────────────
def add_bg(canvas, doc):
    canvas.saveState()
    canvas.setFillColor(PAGE_BG)
    canvas.rect(0, 0, A4[0], A4[1], fill=1, stroke=0)
    canvas.restoreState()

# ── Assemble document ─────────────────────────────────────────────────────────
title_table_data = [[
    Paragraph("💉 VACUTAINER QUICK-REFERENCE CHART", title_style),
    Paragraph("Orris Medical Reference  •  BD CLSI Standard  •  July 2026", sub_style),
]]
title_block = Table(title_table_data, colWidths=[176*mm])
title_block.setStyle(TableStyle([
    ("BACKGROUND",   (0,0), (-1,-1), HEADER_BG),
    ("TOPPADDING",   (0,0), (-1,-1), 7),
    ("BOTTOMPADDING",(0,0), (-1,-1), 7),
    ("LEFTPADDING",  (0,0), (-1,-1), 8),
    ("RIGHTPADDING", (0,0), (-1,-1), 8),
    ("BOX",          (0,0), (-1,-1), 1, HEADER_BG),
]))

story = [
    title_block,
    Spacer(1, 4*mm),
    table,
    Spacer(1, 4*mm),
    draw_order_table,
    Spacer(1, 3*mm),
    tips_table,
    Spacer(1, 3*mm),
    Paragraph(
        "Reference: BD Vacutainer® Venous Blood Collection Tube Guide (BD-83178) | CLSI GP41-A6 | myOnsite Healthcare",
        small_style
    ),
]

doc.build(story, onFirstPage=add_bg, onLaterPages=add_bg)
print("PDF generated:", OUTPUT)