# Copyright  2014-2022 Vincent Texier <vit@free.fr>
#
# DuniterPy is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# DuniterPy 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import unittest

import pypeg2

from duniterpy.grammars.output import CLTV, CSV, SIG, XHX, Condition, Operator

pubkey = "DNann1Lh55eZMEDXeYt59bzHbA3NJR46DeQYCS2qQdLV"


class TestOutputgrammar(unittest.TestCase):
    def test_sig(self):
        condition = "SIG(HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd)"
        result = pypeg2.parse(condition, SIG)
        self.assertEqual(result.pubkey, "HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd")
        self.assertEqual(pypeg2.compose(result, Condition), condition)

    def test_xhx(self):
        condition = "XHX(309BC5E644F797F53E5A2065EAF38A173437F2E6)"
        result = pypeg2.parse(condition, XHX)
        self.assertEqual(result.sha_hash, "309BC5E644F797F53E5A2065EAF38A173437F2E6")

    def test_sig_condition(self):
        condition = "SIG(HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd)"
        result = pypeg2.parse(condition, Condition)
        self.assertEqual(
            result.left.pubkey, "HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd"
        )
        self.assertEqual(pypeg2.compose(result, Condition), condition)

    def test_xhr_condition(self):
        condition = "SIG(HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd)"
        result = pypeg2.parse(condition, Condition)
        self.assertEqual(
            result.left.pubkey, "HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd"
        )
        self.assertEqual(pypeg2.compose(result, Condition), condition)

    def test_simple_condition(self):
        condition = "(SIG(HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd))"
        result = pypeg2.parse(condition, Condition)
        self.assertEqual(
            result.left.left.pubkey, "HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd"
        )
        self.assertEqual(pypeg2.compose(result, Condition), condition)

    def test_simple_and_condition(self):
        condition = (
            "(SIG(HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd) && XHX("
            "309BC5E644F797F53E5A2065EAF38A173437F2E6))"
        )
        result = pypeg2.parse(condition, Condition)
        self.assertEqual(
            result.left.left.pubkey, "HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd"
        )
        self.assertEqual(result.left.op.name, "&&")
        self.assertEqual(
            result.left.right.sha_hash, "309BC5E644F797F53E5A2065EAF38A173437F2E6"
        )
        self.assertEqual(pypeg2.compose(result, Condition), condition)

    def test_simple_or_condition(self):
        condition = (
            "(SIG(HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd) || XHX("
            "309BC5E644F797F53E5A2065EAF38A173437F2E6))"
        )
        result = pypeg2.parse(condition, Condition)
        self.assertEqual(
            result.left.left.pubkey, "HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd"
        )
        self.assertEqual(result.left.op.name, "||")
        self.assertEqual(
            result.left.right.sha_hash, "309BC5E644F797F53E5A2065EAF38A173437F2E6"
        )
        self.assertEqual(pypeg2.compose(result, Condition), condition)

    def test_complex_condition(self):
        condition = (
            "(SIG(HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd) || (SIG("
            "DNann1Lh55eZMEDXeYt59bzHbA3NJR46DeQYCS2qQdLV) && XHX(309BC5E644F797F53E5A2065EAF38A173437F2E6)))"
        )
        result = pypeg2.parse(condition, Condition)
        self.assertEqual(
            result.left.left.pubkey, "HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd"
        )
        self.assertEqual(result.left.op.name, "||")
        self.assertEqual(
            result.left.right.left.pubkey,
            "DNann1Lh55eZMEDXeYt59bzHbA3NJR46DeQYCS2qQdLV",
        )
        self.assertEqual(result.left.right.op.name, "&&")
        self.assertEqual(
            result.left.right.right.sha_hash, "309BC5E644F797F53E5A2065EAF38A173437F2E6"
        )
        self.assertEqual(pypeg2.compose(result, Condition), condition)

    def test_csv_cltv_condition(self):
        condition = "(CSV(1654300) || (SIG(DNann1Lh55eZMEDXeYt59bzHbA3NJR46DeQYCS2qQdLV) && CLTV(2594024)))"
        result = pypeg2.parse(condition, Condition)
        self.assertEqual(result.left.left.time, "1654300")
        self.assertEqual(result.left.op.name, "||")
        self.assertEqual(
            result.left.right.left.pubkey,
            "DNann1Lh55eZMEDXeYt59bzHbA3NJR46DeQYCS2qQdLV",
        )
        self.assertEqual(result.left.right.op.name, "&&")
        self.assertEqual(result.left.right.right.timestamp, "2594024")
        self.assertEqual(pypeg2.compose(result, Condition), condition)

    def test_instanciate_condition(self):
        Condition.token(
            SIG.token("HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd"),
            Operator.token("||"),
            Condition.token(
                SIG.token("DNann1Lh55eZMEDXeYt59bzHbA3NJR46DeQYCS2qQdLV"),
                Operator.token("&&"),
                XHX.token("309BC5E644F797F53E5A2065EAF38A173437F2E6"),
            ),
        )
        condition = (
            "(SIG(HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd) || (SIG("
            "DNann1Lh55eZMEDXeYt59bzHbA3NJR46DeQYCS2qQdLV) && XHX(309BC5E644F797F53E5A2065EAF38A173437F2E6)))"
        )
        inst = pypeg2.parse(condition, Condition)
        self.assertEqual(
            inst.left.left.pubkey, "HgTTJLAQ5sqfknMq7yLPZbehtuLSsKj9CxWN7k8QvYJd"
        )
        self.assertEqual(inst.left.op.name, "||")
        self.assertEqual(
            inst.left.right.left.pubkey, "DNann1Lh55eZMEDXeYt59bzHbA3NJR46DeQYCS2qQdLV"
        )
        self.assertEqual(inst.left.right.op.name, "&&")
        self.assertEqual(
            inst.left.right.right.sha_hash, "309BC5E644F797F53E5A2065EAF38A173437F2E6"
        )
        self.assertEqual(pypeg2.compose(inst, Condition), condition)

    def test_SIG_token_and_compose(self):
        self.assertEqual(SIG.token(pubkey).compose(), "SIG(" + pubkey + ")")

    def test_CSV_token_and_compose(self):
        self.assertEqual(CSV.token(pubkey).compose(), "CSV(" + pubkey + ")")

    def test_CLTV_token_and_compose(self):
        self.assertEqual(CLTV.token(pubkey).compose(), "CLTV(" + pubkey + ")")

    def test_HXH_token_and_compose(self):
        self.assertEqual(XHX.token(pubkey).compose(), "XHX(" + pubkey + ")")