from abc import abstractmethod
import logging
from typing import Callable

from siwe_auth.custom_groups.group_manager import GroupManager
from siwe_auth.utils.data_classes import EthereumBaseClass

from web3 import Web3, HTTPProvider


class ERC20Manager(GroupManager):
    contract: str
    abi = [
        {
            "constant": False,
            "inputs": [{"name": "_owner", "type": "address"}],
            "name": "balanceOf",
            "outputs": [{"name": "balance", "type": "uint256"}],
            "payable": False,
            "type": "function",
        }
    ]

    def __init__(self, config: dict):
        if "contract" not in config:
            raise ValueError("ERC20 Manager config is missing contract attribute.")
        self.contract = config["contract"]

    def _is_member(
        self,
        ethereum_address: str,
        provider: HTTPProvider,
        expression: Callable[[str], bool],
    ):
        w3 = Web3(provider)
        contract = w3.eth.contract(
            address=Web3.toChecksumAddress(self.contract.lower()), abi=self.abi
        )
        balance = contract.functions.balanceOf(
            _owner=Web3.toChecksumAddress(ethereum_address.lower())
        ).call()
        return expression(balance)

    @abstractmethod
    def is_member(self, wallet: EthereumBaseClass, provider: HTTPProvider) -> bool:
        pass


class ERC20OwnerManager(ERC20Manager):
    def is_member(self, wallet: EthereumBaseClass, provider: HTTPProvider) -> bool:
        if not self._valid_wallet(wallet=wallet):
            return False
        try:
            return self._is_member(
                ethereum_address=wallet.ethereum_address,
                provider=provider,
                expression=lambda x: x > 0,
            )
        except ValueError:
            logging.error(
                f"Unable to verify membership of invalid address: {wallet.ethereum_address}"
            )
        return False