# This file is part of the Trezor project.
#
# Copyright (C) 2012-2019 SatoshiLabs and contributors
#
# This library is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License version 3
# as published by the Free Software Foundation.
#
# This library 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 License along with this library.
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.

import pytest

from trezorlib import btc, ckd_public as bip32, messages as proto
from trezorlib.tools import H_

from ..common import MNEMONIC12


class TestMsgGetpublickey:
    @pytest.mark.setup_client(mnemonic=MNEMONIC12)
    def test_btc(self, client):
        assert (
            bip32.serialize(btc.get_public_node(client, []).node, 0x0488B21E)
            == "xpub661MyMwAqRbcF1zGijBb2K6x9YiJPh58xpcCeLvTxMX6spkY3PcpJ4ABcCyWfskq5DDxM3e6Ez5ePCqG5bnPUXR4wL8TZWyoDaUdiWW7bKy"
        )
        assert (
            btc.get_public_node(client, [], coin_name="Bitcoin").xpub
            == "xpub661MyMwAqRbcF1zGijBb2K6x9YiJPh58xpcCeLvTxMX6spkY3PcpJ4ABcCyWfskq5DDxM3e6Ez5ePCqG5bnPUXR4wL8TZWyoDaUdiWW7bKy"
        )
        assert (
            bip32.serialize(btc.get_public_node(client, [1]).node, 0x0488B21E)
            == "xpub68zNxjsTrV8y9AadThLW7dTAqEpZ7xBLFSyJ3X9pjTv6Njg6kxgjXJkzxq8u3ttnjBw1jupQHMP3gpGZzZqd1eh5S4GjkaMhPR18vMyUi8N"
        )
        assert (
            btc.get_public_node(client, [1], coin_name="Bitcoin").xpub
            == "xpub68zNxjsTrV8y9AadThLW7dTAqEpZ7xBLFSyJ3X9pjTv6Njg6kxgjXJkzxq8u3ttnjBw1jupQHMP3gpGZzZqd1eh5S4GjkaMhPR18vMyUi8N"
        )
        assert (
            bip32.serialize(btc.get_public_node(client, [0, H_(1)]).node, 0x0488B21E)
            == "xpub6A3FoZqYXj1AbW4thRwBh26YwZWbmoyjTaZwwxJjY1oKUpefLepL3RFS9DHKQrjAfxDrzDepYMDZPqXN6upQm3bHQ9xaXD5a3mqni3goF4v"
        )
        assert (
            btc.get_public_node(client, [0, H_(1)], coin_name="Bitcoin").xpub
            == "xpub6A3FoZqYXj1AbW4thRwBh26YwZWbmoyjTaZwwxJjY1oKUpefLepL3RFS9DHKQrjAfxDrzDepYMDZPqXN6upQm3bHQ9xaXD5a3mqni3goF4v"
        )
        assert (
            bip32.serialize(btc.get_public_node(client, [H_(9), 0]).node, 0x0488B21E)
            == "xpub6A2h5mzLDfYginoD7q7wCWbq18wTbN9gducRr2w5NRTwdLeoT3cJSwefFqW7uXTpVFGtpUyDMBNYs3DNvvXx6NPjF9YEbUQrtxFSWnPtVrv"
        )
        assert (
            btc.get_public_node(client, [H_(9), 0], coin_name="Bitcoin").xpub
            == "xpub6A2h5mzLDfYginoD7q7wCWbq18wTbN9gducRr2w5NRTwdLeoT3cJSwefFqW7uXTpVFGtpUyDMBNYs3DNvvXx6NPjF9YEbUQrtxFSWnPtVrv"
        )
        assert (
            bip32.serialize(btc.get_public_node(client, [0, 9999999]).node, 0x0488B21E)
            == "xpub6A3FoZqQEK6iwLZ4HFkqSo5fb35BH4bpjC4SPZ63prfLdGYPwYxEuC6o91bUvFFdMzKWe5rs3axHRUjxJaSvBnKKFtnfLwDACRxPxabsv2r"
        )
        assert (
            btc.get_public_node(client, [0, 9999999], coin_name="Bitcoin").xpub
            == "xpub6A3FoZqQEK6iwLZ4HFkqSo5fb35BH4bpjC4SPZ63prfLdGYPwYxEuC6o91bUvFFdMzKWe5rs3axHRUjxJaSvBnKKFtnfLwDACRxPxabsv2r"
        )

    @pytest.mark.altcoin
    @pytest.mark.setup_client(mnemonic=MNEMONIC12)
    def test_ltc(self, client):
        assert (
            bip32.serialize(btc.get_public_node(client, []).node, 0x019DA462)
            == "Ltub2SSUS19CirucVPGDKDBatBDBEM2s9UbH66pBURfaKrMocCPLhQ7Z7hecy5VYLHA5fRdXwB2e61j2VJCNzVsqKTCVEU1vECjqi5EyczFX9xp"
        )
        assert (
            btc.get_public_node(client, [], coin_name="Litecoin").xpub
            == "Ltub2SSUS19CirucVPGDKDBatBDBEM2s9UbH66pBURfaKrMocCPLhQ7Z7hecy5VYLHA5fRdXwB2e61j2VJCNzVsqKTCVEU1vECjqi5EyczFX9xp"
        )
        assert (
            bip32.serialize(btc.get_public_node(client, [1]).node, 0x019DA462)
            == "Ltub2VRVRP5VjvSyPXra4BLVyVZPv397sjhUNjBGsbtw6xko77JuQyBULxFSKheviJJ3KQLbL3Cx8P2RnudguTw4raUVjCACRG7jsumUptYx55C"
        )
        assert (
            btc.get_public_node(client, [1], coin_name="Litecoin").xpub
            == "Ltub2VRVRP5VjvSyPXra4BLVyVZPv397sjhUNjBGsbtw6xko77JuQyBULxFSKheviJJ3KQLbL3Cx8P2RnudguTw4raUVjCACRG7jsumUptYx55C"
        )
        assert (
            bip32.serialize(btc.get_public_node(client, [0, H_(1)]).node, 0x019DA462)
            == "Ltub2WUNGD3aRAKAqsLqHuwBYtCn2MqAXbVsarmvn33quWe2DCHTzfK4s4jsW5oM5G8RGAdSaM3NPNrwVvtV1ourbyNhhHr3BtqcYGc8caf5GoT"
        )
        assert (
            btc.get_public_node(client, [0, H_(1)], coin_name="Litecoin").xpub
            == "Ltub2WUNGD3aRAKAqsLqHuwBYtCn2MqAXbVsarmvn33quWe2DCHTzfK4s4jsW5oM5G8RGAdSaM3NPNrwVvtV1ourbyNhhHr3BtqcYGc8caf5GoT"
        )
        assert (
            bip32.serialize(btc.get_public_node(client, [H_(9), 0]).node, 0x019DA462)
            == "Ltub2WToYRCN76rgyA59iK7w4Ni45wG2M9fpmBpQg7gBjvJeMiHc7473Gb96ci29Zvs55TgUQcMmCD1vy8aVqpdPwJB9YHRhGAAuPT1nRLLXmFu"
        )
        assert (
            btc.get_public_node(client, [H_(9), 0], coin_name="Litecoin").xpub
            == "Ltub2WToYRCN76rgyA59iK7w4Ni45wG2M9fpmBpQg7gBjvJeMiHc7473Gb96ci29Zvs55TgUQcMmCD1vy8aVqpdPwJB9YHRhGAAuPT1nRLLXmFu"
        )
        assert (
            bip32.serialize(btc.get_public_node(client, [0, 9999999]).node, 0x019DA462)
            == "Ltub2WUNGD3S7kQjBhpzsjkqJfBtfqPk2r7xrUGRDdqACMW3MeBCbZSyiqbEVt7WaeesxCj6EDFQtcbfXa75DUYN2i6jZ2g81cyCgvijs9J2u2n"
        )
        assert (
            btc.get_public_node(client, [0, 9999999], coin_name="Litecoin").xpub
            == "Ltub2WUNGD3S7kQjBhpzsjkqJfBtfqPk2r7xrUGRDdqACMW3MeBCbZSyiqbEVt7WaeesxCj6EDFQtcbfXa75DUYN2i6jZ2g81cyCgvijs9J2u2n"
        )

    @pytest.mark.setup_client(mnemonic=MNEMONIC12)
    def test_tbtc(self, client):
        assert (
            bip32.serialize(btc.get_public_node(client, [111, 42]).node, 0x043587CF)
            == "tpubDAgixSyai5PWbc8N1mBkHDR5nLgAnHFtY7r4y5EzxqAxrt9YUDpZL3kaRoHVvCfrcwNo31c2isBP2uTHcZxEosuKbyJhCAbrvGoPuLUZ7Mz"
        )
        assert (
            btc.get_public_node(client, [111, 42], coin_name="Testnet").xpub
            == "tpubDAgixSyai5PWbc8N1mBkHDR5nLgAnHFtY7r4y5EzxqAxrt9YUDpZL3kaRoHVvCfrcwNo31c2isBP2uTHcZxEosuKbyJhCAbrvGoPuLUZ7Mz"
        )

    @pytest.mark.setup_client(mnemonic=MNEMONIC12)
    def test_script_type(self, client):
        assert (
            btc.get_public_node(client, [], coin_name="Bitcoin").xpub
            == "xpub661MyMwAqRbcF1zGijBb2K6x9YiJPh58xpcCeLvTxMX6spkY3PcpJ4ABcCyWfskq5DDxM3e6Ez5ePCqG5bnPUXR4wL8TZWyoDaUdiWW7bKy"
        )
        assert (
            btc.get_public_node(
                client,
                [],
                coin_name="Bitcoin",
                script_type=proto.InputScriptType.SPENDADDRESS,
            ).xpub
            == "xpub661MyMwAqRbcF1zGijBb2K6x9YiJPh58xpcCeLvTxMX6spkY3PcpJ4ABcCyWfskq5DDxM3e6Ez5ePCqG5bnPUXR4wL8TZWyoDaUdiWW7bKy"
        )
        assert (
            btc.get_public_node(
                client,
                [],
                coin_name="Bitcoin",
                script_type=proto.InputScriptType.SPENDP2SHWITNESS,
            ).xpub
            == "ypub6QqdH2c5z7966KBPZ5yDEQCTKWrkLK4dsw8RRjpMLMtyvvZmJ3nNv7pKdQw6fnQkUrLm6XEeheSCGVSpoJCQGm6fofpt9RoHVJYH72ecmVm"
        )
        assert (
            btc.get_public_node(
                client,
                [],
                coin_name="Bitcoin",
                script_type=proto.InputScriptType.SPENDWITNESS,
            ).xpub
            == "zpub6jftahH18ngZwcNWPSkqSVHxVV1CGw48o3eeD8iEiNGrz2NzYhwwYBUTectgfh4ftVTZqzqDAJnk9n4PWzcR4znGg1XJjLcmm2bvVc3Honv"
        )
