from datetime import datetime
from pathlib import Path
import re
import unittest


ROOT = Path(__file__).resolve().parents[2]
ART = ROOT / "_art"
JOB = "7c832afa1b73023d8f2391ff40c221be9ffd468a1a96a6bf4900cbf95595d31d"
ARMENIA = [f"/img/arts/uploads/{JOB}/{index:02}.jpg" for index in (9, 8, 7, 6, 5, 4)] + [
    "/img/arts/uploads/474ab3283ce8801ac6e59b67b65b4169167053873a5a8db4a15b840a3448266c/00.jpg"]
ROMANIA = [f"/img/arts/uploads/{JOB}/{index:02}.jpg" for index in (0, 1)] + [
    f"/img/arts/uploads/c53f20236f166632bff163339c8e789cc09b0d7e63c5b531b1fc20705460975b/{index:02}.jpg"
    for index in range(5)] + [
    "/img/arts/uploads/260cc99527c13f256d4ea39d0959f5daa942c888ea882852e73fd94f6c30b7db/00.jpg",
    "/img/arts/uploads/b18a2f1c7127077c801160782012cb164759f0b9dd41762205b45e4c9a07d673/00.jpg"]


class PhotoHighlightTests(unittest.TestCase):
    def test_uploaded_photos_are_individual_and_grouped(self):
        records = []
        for path in ART.glob("*photo-*.md"):
            content = path.read_text()
            images = re.findall(r'<img src="([^"]+/uploads/[^"]+)"', content)
            self.assertEqual(len(images), 1, path)
            date = re.search(r'^date: "([^"]+)"', content, re.MULTILINE)
            anchor = re.search(r'^anchor: "([^"]+)"', content, re.MULTILINE)
            records.append((datetime.fromisoformat(date.group(1)), images[0],
                            anchor.group(1) if anchor else None, content))
        records.sort(reverse=True)
        grouped = [record for record in records if record[1] in ARMENIA + ROMANIA]
        self.assertEqual([record[1] for record in grouped], ARMENIA + ROMANIA)
        self.assertEqual(grouped[0][2], "Armenia")
        self.assertEqual(grouped[7][2], "Romania")
        self.assertIn("Transmission of Flesh by Erfan Ashourioun", grouped[2][3])

    def test_existing_collection_highlight_starts(self):
        iran = (ART / "2024-10-25-Taq-e-Bostan_000521270020.md").read_text()
        kazakhstan = (ART / "2024-10-09-x_DSC-0999.md").read_text()
        self.assertIn('anchor: "Iran - Hamadan, Kermanshah, Tabriz"', iran)
        self.assertIn('anchor: "Kazakhstan - World Nomad Games"', kazakhstan)


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