import os
import tempfile
import unittest

from renamer import CliAction, _is_media_file, process_path


class TestIsMediaFile(unittest.TestCase):
    def test_extension_is_case_insensitive(self):
        self.assertTrue(_is_media_file("Show.S04E06.MKV"))
        self.assertTrue(_is_media_file("Show.S04E06.mkv"))
        self.assertTrue(_is_media_file("Show.S04E06.AVI"))

    def test_unsupported_extension(self):
        self.assertFalse(_is_media_file("Show.S04E06.txt"))


class TestRenameCaseInsensitiveExtension(unittest.TestCase):
    def test_uppercase_extension_is_processed(self):
        with tempfile.TemporaryDirectory() as root:
            original = "Show S04E06.MKV"
            open(os.path.join(root, original), "w").close()

            process_path(CliAction.rename, os.path.join(root, original))

            self.assertEqual(os.listdir(root), ["Show.S04E06.MKV"])


class TestSeasonNumbering(unittest.TestCase):
    def test_scoped_to_directory_not_whole_tree(self):
        with tempfile.TemporaryDirectory() as root:
            season_dir = os.path.join(root, "Season 1")
            extras_dir = os.path.join(root, "Extras")
            os.mkdir(season_dir)
            os.mkdir(extras_dir)
            for name in (
                "Show.(01.serija.iz.10).mkv",
                "Show.(02.serija.iz.10).mkv",
            ):
                open(os.path.join(season_dir, name), "w").close()
            open(os.path.join(extras_dir, "Blooper.reel.mkv"), "w").close()

            process_path(
                CliAction.rename, root, overrides={"name": "Show"}, season=1,
            )

            self.assertEqual(
                sorted(os.listdir(season_dir)),
                ["Show.S01E01.mkv", "Show.S01E02.mkv"],
            )
            # Extras is a separate directory: it gets its own numbering,
            # starting fresh at E01, instead of continuing the season's count.
            self.assertEqual(os.listdir(extras_dir), ["Show.S01E01.mkv"])

    def test_single_file_target_defaults_to_episode_one(self):
        with tempfile.TemporaryDirectory() as root:
            original = "Show.(01.serija.iz.10).mkv"
            open(os.path.join(root, original), "w").close()

            process_path(
                CliAction.rename, os.path.join(root, original),
                overrides={"name": "Show"}, season=1,
            )

            self.assertEqual(os.listdir(root), ["Show.S01E01.mkv"])


class TestSetOverride(unittest.TestCase):
    def test_name_override_drops_unrecognized_leftovers(self):
        with tempfile.TemporaryDirectory() as root:
            # nothing else in the title is recognized (no year/quality/etc),
            # so it would otherwise land entirely under "unknown"
            original = "Show.(01.serija.iz.10).mkv"
            open(os.path.join(root, original), "w").close()

            process_path(
                CliAction.rename, os.path.join(root, original),
                overrides={"name": "Show"}, season=1,
            )

            self.assertEqual(os.listdir(root), ["Show.S01E01.mkv"])

    def test_name_override_preserves_unknown_when_name_was_recognized(self):
        with tempfile.TemporaryDirectory() as root:
            original = "The.Walking.Dead.S04E06.1080p.WEB-DL.Rus.Eng.HDCLUB.mkv"
            open(os.path.join(root, original), "w").close()

            process_path(
                CliAction.rename, os.path.join(root, original),
                overrides={"name": "TWD"},
            )

            self.assertEqual(
                os.listdir(root),
                ["TWD.S04E06.1080p.WEB-DL.Rus.Eng.HDCLUB.mkv"],
            )


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