import json
import os
import unittest

from main import TodoParser


def count_issues_for_file_type(raw_issues, file_type):
    num_issues = 0
    for issue in raw_issues:
        if issue.markdown_language == file_type:
            num_issues += 1
    return num_issues


class NewIssueTest(unittest.TestCase):
    # Check for newly added TODOs across the files specified.
    def setUp(self):
        diff_file = open('tests/test_new.diff', 'r')
        parser = TodoParser()
        with open('syntax.json', 'r') as syntax_json:
            parser.syntax_dict = json.load(syntax_json)
        self.raw_issues = parser.parse(diff_file)

    def test_python_issues(self):
        # Includes 4 tests for Starlark.
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'python'), 8)

    def test_yaml_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'yaml'), 2)

    def test_toml_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'toml'), 2)

    def test_php_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'php'), 4)

    def test_java_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'java'), 2)

    def test_javascript_issues(self):
        # Includes 1 test for JSON with Comments, 1 test for JSON5, 3 tests for TSX.
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'javascript'), 5)

    def test_ruby_issues(self):
        # Includes 2 tests for Crystal.
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'ruby'), 5)

    def test_abap_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'abap'), 2)

    def test_sql_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'sql'), 1)

    def test_tex_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'tex'), 2)

    def test_julia_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'julia'), 2)

    def test_starlark_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'python'), 8)

    def test_autohotkey_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'autohotkey'), 1)

    def test_handlebars_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'handlebars'), 2)

    def test_text_issues(self):
        # Includes 2 tests for Org, 2 tests for GAP, 2 tests for Visual Basic, 2 tests for Agda, 4 tests for Sol.
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'text'), 12)

    def test_scss_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'scss'), 2)

    def test_twig_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'twig'), 2)

    def test_makefile_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'makefile'), 3)

    def test_md_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'markdown'), 8)

    def test_r_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'r'), 2)

    def test_haskell_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'haskell'), 4)

    def test_clojure_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'clojure'), 2)

    def test_nix_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'nix'), 2)

    def test_xaml_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'xml'), 2)

    def test_c_cpp_like_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'c_cpp'), 2)

    def test_liquid_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'liquid'), 3)

    def test_lua_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'lua'), 2)

class ClosedIssueTest(unittest.TestCase):
    # Check for removed TODOs across the files specified.
    def setUp(self):
        diff_file = open('tests/test_closed.diff', 'r')
        parser = TodoParser()
        with open('syntax.json', 'r') as syntax_json:
            parser.syntax_dict = json.load(syntax_json)
        self.raw_issues = parser.parse(diff_file)

    def test_python_issues(self):
        # Includes 1 test for Starlark.
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'python'), 5)

    def test_yaml_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'yaml'), 2)

    def test_toml_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'toml'), 2)

    def test_php_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'php'), 4)

    def test_java_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'java'), 2)

    def test_ruby_issues(self):
        # Includes 2 tests for Crystal.
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'ruby'), 5)

    def test_abap_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'abap'), 2)

    def test_sql_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'sql'), 1)

    def test_tex_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'tex'), 2)

    def test_julia_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'julia'), 4)

    def test_starlark_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'python'), 5)

    def test_javascript_issues(self):
        # Includes 1 test for JSON with Comments, 1 test for JSON5, 3 tests for TSX.
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'javascript'), 5)

    def test_autohotkey_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'autohotkey'), 1)

    def test_handlebars_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'handlebars'), 2)

    def test_text_issues(self):
        # Includes 2 tests for Org, 2 tests for GAP, 2 tests for Visual Basic, 2 tests for Agda, 4 tests for Sol.
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'text'), 12)

    def test_scss_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'scss'), 2)

    def test_twig_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'twig'), 2)

    def test_makefile_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'makefile'), 3)

    def test_md_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'markdown'), 8)

    def test_r_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'r'), 2)

    def test_haskell_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'haskell'), 4)

    def test_clojure_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'clojure'), 2)

    def test_nix_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'nix'), 2)

    def test_xaml_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'xml'), 2)

    def test_c_cpp_like_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'c_cpp'), 2)

    def test_liquid_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'liquid'), 3)

    def test_lua_issues(self):
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'lua'), 2)

class IgnorePatternTest(unittest.TestCase):
    def test_single_ignore(self):
        os.environ['INPUT_IGNORE'] = '.*\\.java'
        parser = TodoParser()
        with open('syntax.json', 'r') as syntax_json:
            parser.syntax_dict = json.load(syntax_json)
        diff_file = open('tests/test_closed.diff', 'r')
        self.raw_issues = parser.parse(diff_file)
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'python'), 5)
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'yaml'), 2)
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'php'), 4)
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'java'), 0)
        # Includes 2 tests for Crystal.
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'ruby'), 5)
        os.environ['INPUT_IGNORE'] = ''

    def test_multiple_ignores(self):
        os.environ['INPUT_IGNORE'] = '.*\\.java, tests/example-file\\.php'
        parser = TodoParser()
        with open('syntax.json', 'r') as syntax_json:
            parser.syntax_dict = json.load(syntax_json)
        diff_file = open('tests/test_closed.diff', 'r')
        self.raw_issues = parser.parse(diff_file)
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'python'), 5)
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'yaml'), 2)
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'php'), 0)
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'java'), 0)
        # Includes 2 tests for Crystal.
        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'ruby'), 5)
        os.environ['INPUT_IGNORE'] = ''


class EscapeMarkdownTest(unittest.TestCase):
    def test_simple_escape(self):
        os.environ['INPUT_ESCAPE'] = 'true'
        parser = TodoParser()
        with open('syntax.json', 'r') as syntax_json:
            parser.syntax_dict = json.load(syntax_json)
        diff_file = open('tests/test_escape.diff', 'r')

        # I had no other idea to make these checks dynamic.
        self.raw_issues = parser.parse(diff_file)
        self.assertEqual(len(self.raw_issues), 2)

        issue = self.raw_issues[0]
        self.assertEqual(len(issue.body), 2)
        self.assertEqual(issue.body[0], '\\# Some title')
        self.assertEqual(issue.body[1], '\\<SomeTag\\>')

        issue = self.raw_issues[1]
        self.assertEqual(len(issue.body), 2)
        self.assertEqual(issue.body[0], '\\# Another title')
        self.assertEqual(issue.body[1], '\\<AnotherTag\\>')


class BaseCustomLanguageTests:
    class BaseTest(unittest.TestCase):
        @staticmethod
        def count_syntax(parser: TodoParser, name: str):
            counter = 0

            for syntax in parser.syntax_dict:
                if syntax['language'] == name:
                    counter = counter + 1

            return counter


class CustomLanguageFileTest(BaseCustomLanguageTests.BaseTest):
    def setUp(self):
        os.environ['INPUT_LANGUAGES'] = 'tests/custom_languages.json'
        self.parser = TodoParser()

    def test_custom_lang_load(self):
        # Test if the custom language ILS is actually loaded into the system
        self.assertIsNotNone(self.parser.languages_dict['ILS'])
        self.assertEqual(self.count_syntax(self.parser, 'ILS'), 1)

    def test_custom_lang_not_dupplicate(self):

        # Test if a custom language can overwrite the rules of an existing one
        self.assertEqual(self.count_syntax(self.parser, 'Java'), 1)
        for syntax in self.parser.syntax_dict:
            if syntax['language'] == 'Java':
                self.assertEqual(len(syntax['markers']), 2)
                self.assertEqual(syntax['markers'][0]['pattern'], "////")
                self.assertEqual(syntax['markers'][1]['pattern']['start'], '+=')
                self.assertEqual(syntax['markers'][1]['pattern']['end'], '=+')
                break

        self.assertIsNotNone(self.parser.languages_dict['Java'])
        self.assertEqual(len(self.parser.languages_dict['Java']['extensions']), 1)
        self.assertEqual(self.parser.languages_dict['Java']['extensions'][0], ".java2")

    def tearDown(self):
        del os.environ['INPUT_LANGUAGES']


class CustomLanguageUrlTest(BaseCustomLanguageTests.BaseTest):
    def setUp(self):
        os.environ['INPUT_LANGUAGES'] = 'https://raw.githubusercontent.com/alstr/todo-to-issue-action/master/tests/custom_languages.json'
        os.environ['INPUT_NO_STANDARD'] = 'true'
        self.parser = TodoParser()

    def test_url_load(self):
        self.assertEqual(len(self.parser.languages_dict), 2)
        self.assertEqual(len(self.parser.syntax_dict), 2)

    def tearDown(self):
        del os.environ['INPUT_LANGUAGES']
        del os.environ['INPUT_NO_STANDARD']