import importlib.util
import pathlib
import unittest


ROOT = pathlib.Path(__file__).resolve().parents[1]
MODULE_PATH = ROOT / "app" / "utils" / "screenshot_marker.py"
spec = importlib.util.spec_from_file_location("screenshot_marker", MODULE_PATH)
if spec is None or spec.loader is None:
    raise ImportError("screenshot_marker module spec not found")
screenshot_marker = importlib.util.module_from_spec(spec)
spec.loader.exec_module(screenshot_marker)
extract_screenshot_timestamps = screenshot_marker.extract_screenshot_timestamps
extract_content_timestamps = screenshot_marker.extract_content_timestamps
ensure_screenshot_markers = screenshot_marker.ensure_screenshot_markers


class TestScreenshotMarker(unittest.TestCase):
    def test_extract_accepts_star_bracket_format(self):
        markdown = "A\n*Screenshot-[01:02]\nB"
        matches = extract_screenshot_timestamps(markdown)
        self.assertEqual(matches, [("*Screenshot-[01:02]", 62)])

    def test_extract_accepts_legacy_formats(self):
        markdown = "*Screenshot-03:04 and Screenshot-[05:06]"
        matches = extract_screenshot_timestamps(markdown)
        self.assertEqual(
            matches,
            [
                ("*Screenshot-03:04", 184),
                ("Screenshot-[05:06]", 306),
            ],
        )

    def test_extract_content_timestamps_for_fallback(self):
        markdown = "## A *Content-[00:12]\n## B *Content-[01:03]\n## C *Content-[01:03]"
        matches = extract_content_timestamps(markdown)
        self.assertEqual(matches, [12, 63])

    def test_ensure_screenshot_markers_adds_duration_fallback(self):
        markdown = "## A\ncontent"
        with_markers = ensure_screenshot_markers(markdown, 120)
        self.assertIn("*Screenshot-[00:30]", with_markers)
        self.assertIn("*Screenshot-[01:00]", with_markers)
        self.assertIn("*Screenshot-[01:30]", with_markers)


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