import ast

import pytest

from core.builtin_concepts import ParserResultConcept, BuiltinConcepts, ReturnValueConcept
from core.concept import Concept
from core.rule import Rule
from core.sheerka.services.SheerkaExecute import ParserInput
from core.tokenizer import Token, TokenKind, Tokenizer
from parsers.BaseNodeParser import ConceptNode, UnrecognizedTokensNode, RuleNode, SourceCodeNode
from parsers.PythonParser import PythonNode
from parsers.PythonWithConceptsParser import PythonWithConceptsParser
from parsers.UnrecognizedNodeParser import UnrecognizedNodeParser
from tests.TestUsingMemoryBasedSheerka import TestUsingMemoryBasedSheerka
from tests.parsers.parsers_utils import get_source_code_node

unrecognized_nodes_parser = UnrecognizedNodeParser()


def ret_val(*args):
    result = []
    index = 0
    for item in args:
        if isinstance(item, Concept):
            tokens = [Token(TokenKind.IDENTIFIER, item.name, 0, 0, 0)]
            result.append(ConceptNode(item, index, index, tokens, item.name))
            index += 1
        elif isinstance(item, Rule):
            tokens = [Token(TokenKind.RULE, (None, item.id), 0, 0, 0)]
            result.append(RuleNode(item, index, index, tokens, f"r:|{item.id}:"))
            index += 1
        else:
            tokens = list(Tokenizer(item))
            result.append(UnrecognizedTokensNode(index, index + len(tokens) - 1, tokens))
            index += len(tokens)

    return ReturnValueConcept("who", False, ParserResultConcept(parser=unrecognized_nodes_parser, value=result))


def to_str_ast(expression):
    return PythonNode.get_dump(ast.parse(expression, mode="eval"))


class TestPythonWithConceptsParser(TestUsingMemoryBasedSheerka):

    @pytest.mark.parametrize("text, interested", [
        ("not parser result", False),
        (ParserInput("not parser result"), False),
        (ParserResultConcept(parser="not multiple_concepts_parser"), False),
        (ParserResultConcept(parser=unrecognized_nodes_parser, value=[UnrecognizedTokensNode(0, 0, [])]), True),
    ])
    def test_not_interested(self, text, interested):
        context = self.get_context()

        res = PythonWithConceptsParser().parse(context, text)
        if interested:
            assert res is not None
        else:
            assert res is None

    def test_i_can_parse_concepts_and_python(self):
        context = self.get_context()
        foo = Concept("foo")
        input_return_value = ret_val(foo, " + 1")

        parser = PythonWithConceptsParser()
        result = parser.parse(context, input_return_value.body)
        wrapper = result.value
        return_value = result.value.value

        assert result.status
        assert result.who == parser.name
        assert context.sheerka.isinstance(wrapper, BuiltinConcepts.PARSER_RESULT)
        assert wrapper.source == "foo + 1"
        assert isinstance(return_value, PythonNode)
        assert return_value.source == "__C__foo__C__ + 1"
        assert return_value.original_source == "foo + 1"
        assert return_value.get_dump(return_value.ast_) == to_str_ast("__C__foo__C__ + 1")
        assert return_value.objects["__C__foo__C__"] == foo

    def test_i_can_parse_concepts_and_python_when_concept_is_known(self):
        context = self.get_context()
        foo = Concept("foo")
        foo = context.sheerka.create_new_concept(context, foo).body.body
        input_return_value = ret_val(foo, " + 1")

        parser = PythonWithConceptsParser()
        result = parser.parse(context, input_return_value.body)
        wrapper = result.value
        return_value = result.value.value

        assert result.status
        assert result.who == parser.name
        assert context.sheerka.isinstance(wrapper, BuiltinConcepts.PARSER_RESULT)
        assert wrapper.source == "foo + 1"
        assert isinstance(return_value, PythonNode)
        assert return_value.source == "__C__foo__1001__C__ + 1"
        assert return_value.original_source == "foo + 1"
        assert return_value.get_dump(return_value.ast_) == to_str_ast("__C__foo__1001__C__ + 1")
        assert return_value.objects["__C__foo__1001__C__"] == foo

    def test_i_can_parse_when_concept_name_has_invalid_characters(self):
        context = self.get_context()
        foo = Concept("foo et > (,")
        foo = context.sheerka.create_new_concept(context, foo).body.body
        input_return_value = ret_val(foo, " + 1")

        parser = PythonWithConceptsParser()
        result = parser.parse(context, input_return_value.body)
        return_value = result.value.value

        assert result.status
        assert return_value.objects["__C__foo0et000000__1001__C__"] == foo

    def test_i_can_parse_when_multiple_concepts(self):
        sheerka, context, foo, bar = self.init_concepts("foo", "bar")
        input_return_value = ret_val("func(", foo, ", ", bar, ")")

        parser = PythonWithConceptsParser()
        result = parser.parse(context, input_return_value.body)
        parser_result = result.value
        return_value = result.value.value

        assert result.status
        assert result.who == parser.name
        assert context.sheerka.isinstance(parser_result, BuiltinConcepts.PARSER_RESULT)
        assert parser_result.source == "func(foo, bar)"
        assert isinstance(return_value, PythonNode)
        assert return_value.source == "func(__C__foo__1001__C__, __C__bar__1002__C__)"
        assert return_value.original_source == "func(foo, bar)"
        assert return_value.get_dump(return_value.ast_) == to_str_ast("func(__C__foo__1001__C__, __C__bar__1002__C__)")
        assert return_value.objects["__C__foo__1001__C__"] == foo
        assert return_value.objects["__C__bar__1002__C__"] == bar

    def test_i_can_parse_when_python_and_rule(self):
        context = self.get_context()
        rule = Rule().set_id("rule_id")
        input_return_value = ret_val(rule, " + 1")

        parser = PythonWithConceptsParser()
        result = parser.parse(context, input_return_value.body)
        wrapper = result.value
        return_value = result.value.value

        assert result.status
        assert result.who == parser.name
        assert context.sheerka.isinstance(wrapper, BuiltinConcepts.PARSER_RESULT)
        assert wrapper.source == "r:|rule_id: + 1"
        assert isinstance(return_value, PythonNode)
        assert return_value.source == "__R____rule_id__R__ + 1"
        assert return_value.original_source == "r:|rule_id: + 1"
        assert return_value.get_dump(return_value.ast_) == to_str_ast("__R____rule_id__R__ + 1")
        assert return_value.objects["__R____rule_id__R__"] == rule

    def test_python_ids_mappings_are_correct_when_rules_with_the_same_id(self):
        context = self.get_context()
        rule1 = Rule().set_id("rule_id")
        rule2 = Rule().set_id("rule_id")

        input_return_value = ret_val(rule1, "+", rule2)

        parser = PythonWithConceptsParser()
        result = parser.parse(context, input_return_value.body)
        return_value = result.value.value

        assert result.status
        assert return_value.objects["__R____rule_id__R__"] == rule1
        assert return_value.objects["__R____rule_id_1__R__"] == rule2

    def test_python_ids_mappings_are_correct_when_concepts_with_the_same_name(self):
        context = self.get_context()
        foo1 = Concept("foo")
        foo2 = Concept("foo")
        foo3 = context.sheerka.create_new_concept(context, Concept("foo", body="foo3")).body.body
        foo4 = context.sheerka.create_new_concept(context, Concept("foo", body="foo4")).body.body

        input_return_value = ret_val(foo1, "+", foo2, "+", foo3, "+", foo4)

        parser = PythonWithConceptsParser()
        result = parser.parse(context, input_return_value.body)
        return_value = result.value.value

        assert result.status
        assert return_value.objects["__C__foo__C__"] == foo1
        assert return_value.objects["__C__foo_1__C__"] == foo2
        assert return_value.objects["__C__foo__1001__C__"] == foo3
        assert return_value.objects["__C__foo__1002__C__"] == foo4

    def test_i_cannot_parse_if_syntax_error(self):
        context = self.get_context()
        foo = Concept("foo")
        foo = context.sheerka.create_new_concept(context, foo).body.body
        input_return_value = ret_val(foo, " + ")

        parser = PythonWithConceptsParser()
        result = parser.parse(context, input_return_value.body)

        assert not result.status
        assert context.sheerka.isinstance(result.value, BuiltinConcepts.NOT_FOR_ME)

    def test_i_can_parse_nodes_when_source_code_node(self):
        context = self.get_context()
        foo = Concept("foo")
        bar = Concept("bar")

        nodes = [
            UnrecognizedTokensNode(0, 1, list(Tokenizer("not ", yield_eof=False))),
            get_source_code_node(2, "foo == 1", {"foo": foo}),
            UnrecognizedTokensNode(7, 9, list(Tokenizer(" and ", yield_eof=False))),
            get_source_code_node(10, "bar < 1", {"bar": bar}),
        ]
        expected_ast = ast.parse('not __C__foo__C__ == 1 and __C__bar__C__ < 1', "<source>", 'eval')

        parser = PythonWithConceptsParser()
        result = parser.parse_nodes(context, nodes)

        result_python_node = result.value.value
        assert isinstance(result_python_node, PythonNode)
        assert result_python_node.source == 'not __C__foo__C__ == 1 and __C__bar__C__ < 1'
        assert result_python_node.ast_str == PythonNode.get_dump(expected_ast)
        assert result_python_node.original_source == "not foo == 1 and bar < 1"
        assert result_python_node.objects == {"__C__foo__C__": foo, "__C__bar__C__": bar}