# test_samples.py
#
# Copyright 2021 James Westman <james@jwestman.net>
#
# This file is free software; you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation; either version 3 of the
# License, or (at your option) any later version.
#
# This file is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# SPDX-License-Identifier: LGPL-3.0-or-later


import difflib # I love Python
from pathlib import Path
import traceback
import unittest

from gtkblueprinttool import tokenizer, parser
from gtkblueprinttool.errors import PrintableError, MultipleErrors, CompileError
from gtkblueprinttool.tokenizer import Token, TokenType, tokenize
from gtkblueprinttool import utils


class TestSamples(unittest.TestCase):
    def assert_sample(self, name):
        try:
            with open((Path(__file__).parent / f"samples/{name}.blp").resolve()) as f:
                blueprint = f.read()
            with open((Path(__file__).parent / f"samples/{name}.ui").resolve()) as f:
                expected = f.read()

            tokens = tokenizer.tokenize(blueprint)
            ast, errors = parser.parse(tokens)

            if errors:
                raise errors
            if len(ast.errors):
                raise MultipleErrors(ast.errors)

            actual = ast.generate()
            if actual.strip() != expected.strip():
                diff = difflib.unified_diff(expected.splitlines(), actual.splitlines())
                print("\n".join(diff))
                raise AssertionError()
        except PrintableError as e:
            e.pretty_print(name + ".blp", blueprint)
            raise AssertionError()


    def assert_sample_error(self, name):
        try:
            with open((Path(__file__).parent / f"sample_errors/{name}.blp").resolve()) as f:
                blueprint = f.read()
            with open((Path(__file__).parent / f"sample_errors/{name}.err").resolve()) as f:
                expected = f.read()

            tokens = tokenizer.tokenize(blueprint)
            ast, errors = parser.parse(tokens)

            if errors:
                raise errors
            if len(ast.errors):
                raise MultipleErrors(ast.errors)
        except PrintableError as e:
            def error_str(error):
                line, col = utils.idx_to_pos(error.start, blueprint)
                len = error.end - error.start
                return ",".join([str(line + 1), str(col + 1), str(len), error.message])

            if isinstance(e, CompileError):
                actual = error_str(e)
            elif isinstance(e, MultipleErrors):
                actual = "\n".join([error_str(error) for error in e.errors])
            else:
                raise AssertionError()

            if actual.strip() != expected.strip():
                diff = difflib.unified_diff(expected.splitlines(), actual.splitlines())
                print("\n".join(diff))
                raise AssertionError()


    def test_samples(self):
        self.assert_sample("binding")
        self.assert_sample("child_type")
        self.assert_sample("flags")
        self.assert_sample("layout")
        self.assert_sample("menu")
        self.assert_sample("property")
        self.assert_sample("signal")
        self.assert_sample("strings")
        self.assert_sample("style")
        self.assert_sample("template")
        self.assert_sample("using")


    def test_sample_errors(self):
        self.assert_sample_error("conflicting_namespaces")
        self.assert_sample_error("duplicate_obj_id")
        self.assert_sample_error("two_templates")
        self.assert_sample_error("using_gtk_3")
        self.assert_sample_error("using_invalid_namespace")