from __future__ import annotations

import json
import tempfile
import unittest
from io import BytesIO
from pathlib import Path

from reportlab.pdfgen import canvas

from us_cpa.renderers import render_case_forms
from us_cpa.sources import TaxYearCorpus


class RendererTests(unittest.TestCase):
    def test_render_case_forms_writes_overlay_artifacts_and_flags_review(self) -> None:
        with tempfile.TemporaryDirectory() as temp_dir:
            case_dir = Path(temp_dir) / "case"
            (case_dir / "output").mkdir(parents=True)
            corpus = TaxYearCorpus(cache_root=Path(temp_dir) / "cache")
            irs_dir = corpus.paths_for_year(2025).irs_dir
            irs_dir.mkdir(parents=True, exist_ok=True)

            buffer = BytesIO()
            pdf = canvas.Canvas(buffer)
            pdf.drawString(72, 720, "Template")
            pdf.save()
            (irs_dir / "f1040.pdf").write_bytes(buffer.getvalue())

            normalized = {
                "taxYear": 2025,
                "requiredForms": ["f1040"],
                "taxpayer": {"fullName": "Jane Doe"},
                "filingStatus": "single",
                "income": {"wages": 50000.0, "taxableInterest": 100.0, "businessIncome": 0.0},
                "deductions": {"standardDeduction": 15750.0},
                "taxes": {"totalTax": 3883.5},
                "payments": {"federalWithholding": 6000.0},
                "totals": {"adjustedGrossIncome": 50100.0, "taxableIncome": 34350.0, "refund": 2116.5, "balanceDue": 0.0},
            }

            artifacts = render_case_forms(case_dir, corpus, normalized)

            self.assertEqual(artifacts["artifactCount"], 1)
            self.assertEqual(artifacts["artifacts"][0]["renderMethod"], "overlay")
            self.assertTrue(artifacts["artifacts"][0]["reviewRequired"])
            self.assertTrue((case_dir / "output" / "forms" / "f1040.pdf").exists())
            manifest = json.loads((case_dir / "output" / "artifacts.json").read_text())
            self.assertEqual(manifest["artifacts"][0]["formCode"], "f1040")


if __name__ == "__main__":
    unittest.main()