from pydantic import BaseModel
import requests as r
import json
from time import sleep
from typing import List

URL = "https://hackathon.lsp.team/hk"

base_headers = {"Content-Type": "application/json", "Accept": "application/json"}


class Wallet(BaseModel):
    publicKey: str
    privateKey: str


class WalletBalance(BaseModel):
    matic: float
    coins: float


class TransHash(BaseModel):
    transaction_hash: str


class NftInfo(BaseModel):
    uri: str
    tokens: List[int]


class TransactionHistory(BaseModel):
    hash: str
    timestamp: int
    token_name: str
    from_wallet: str
    to_wallet: str


def create_wallet() -> Wallet:
    response = r.post(URL + "/v1/wallets/new")
    data = response.json()
    return Wallet(publicKey=data["publicKey"], privateKey=data["privateKey"])


def check_transaction(trans_hash: str) -> str:
    res = r.get(
        URL + "/v1/transfers/status/" + trans_hash,
        data=json.dumps({"transactionHash": trans_hash}),
        headers=base_headers,
    )
    return res.json()["status"]


def transfer_rubbles(
    my_wallet_private_key, transfer_publick_key: str, amount: float
) -> TransHash:
    response = r.post(
        URL + "/v1/transfers/ruble",
        data=json.dumps(
            {
                "fromPrivateKey": my_wallet_private_key,
                "toPublicKey": transfer_publick_key,
                "amount": amount,
            }
        ),
        headers={"Content-Type": "application/json", "Accept": "application/json"},
    )
    return TransHash(transaction_hash=response.json()["transaction"])


def create_nft(wallet_public_key: str, string_url: str) -> TransHash:
    response = r.post(
        URL + "/v1/nft/generate",
        data=json.dumps(
            {"toPublicKey": wallet_public_key, "uri": string_url, "nftCount": 1}
        ),
        headers=base_headers,
    )
    return TransHash(transaction_hash=response.json()["transaction_hash"])


def get_nfts(wallet_public_key: str) -> List[NftInfo]:
    res = r.get(URL + f"/v1/wallets/{wallet_public_key}/nft/balance")
    return list(
        map(
            lambda res: NftInfo(uri=res["uri"], tokens=res["tokens"]),
            res.json()["balance"],
        ),
    )


def get_history(wallet_public_key: str) -> List[TransactionHistory]:
    res = r.post(
        URL + f"/v1/wallets/{wallet_public_key}/history",
        data=json.dumps({"page": 0, "offset": 100, "sort": "asc"}),
        headers=base_headers,
    )
    return list(
        map(
            lambda res: TransactionHistory(
                hash=res["hash"],
                timestamp=res["timeStamp"],
                token_name=res["tokenName"],
                from_wallet=res["from"],
                to_wallet=res["to"],
            ),
            res.json()["history"],
        ),
    )


def transfer_nft(
    my_private_wallet_key: str, transfer_publick_key: str, token_id: int
) -> TransHash:
    res = r.post(
        URL + "/v1/transfers/nft",
        data=json.dumps(
            {
                "fromPrivateKey": my_private_wallet_key,
                "toPublicKey": transfer_publick_key,
                "tokenId": token_id,
            }
        ),
        headers=base_headers,
    )
    return TransHash(transaction_hash=res.json()["transaction_hash"])


def get_balance(my_public_wallet_key: str) -> WalletBalance:
    res = r.get(URL + f"/v1/wallets/{my_public_wallet_key}/balance")
    return WalletBalance(
        matic=res.json()["maticAmount"], coins=res.json()["coinsAmount"]
    )


def send_matic(
    my_private_wallet_key: str, transfer_publick_key: str, amount: float
) -> TransHash:
    res = r.post(
        URL + "/v1/transfers/matic",
        data=json.dumps(
            {
                "fromPrivateKey": my_private_wallet_key,
                "toPublicKey": transfer_publick_key,
                "amount": amount,
            }
        ),
        headers=base_headers,
    )
    return TransHash(transaction_hash=res.json()["transaction"])