from miners import BaseMiner
from API.bosminer import BOSMinerAPI
import asyncssh
import toml
import time


class BOSminer(BaseMiner):
    def __init__(self, ip: str):
        api = BOSMinerAPI(ip)
        super().__init__(ip, api)
        self.config = None

    def __repr__(self):
        return f"BOSminer: {str(self.ip)}"

    async def get_ssh_connection(self, username: str, password: str) -> asyncssh.connect:
        """Create a new asyncssh connection"""
        conn = await asyncssh.connect(str(self.ip), known_hosts=None, username=username, password=password,
                                      server_host_key_algs=['ssh-rsa'])
        # return created connection
        return conn

    async def send_ssh_command(self, cmd):
        result = None
        conn = await self.get_ssh_connection('root', 'admin')
        for i in range(3):
            try:
                result = await conn.run(cmd)
            except Exception as e:
                print(f"{cmd} error: {e}")
                if i == 3:
                    return
                continue
        # let the user know the result of the command
        if result is not None:
            if result.stdout != "":
                print(result.stdout)
                if result.stderr != "":
                    print("ERROR: " + result.stderr)
            elif result.stderr != "":
                print("ERROR: " + result.stderr)
            else:
                print(cmd)

    async def fault_light_on(self):
        await self.send_ssh_command('miner fault_light on')

    async def fault_light_off(self):
        await self.send_ssh_command('miner fault_light off')

    async def bosminer_restart(self):
        await self.send_ssh_command('/etc/init.d/bosminer restart')

    async def reboot(self):
        await self.send_ssh_command('/sbin/reboot')

    async def get_config(self):
        async with asyncssh.connect(str(self.ip), known_hosts=None, username='root', password='admin',
                                    server_host_key_algs=['ssh-rsa']) as conn:
            async with conn.start_sftp_client() as sftp:
                async with sftp.open('/etc/bosminer.toml') as file:
                    toml_data = toml.loads(await file.read())
        self.config = toml_data

    def update_config(self, config: dict):
        self.config = config

    async def change_config_format(self, update_config: bool = False):
        if not self.config:
            await self.get_config()
        config = self.config
        config['format']['generator'] = 'upstream_data_configurator'
        config['format']['timestamp'] = int(time.time())
        if update_config:
            self.update_config(config)
        return config

    async def change_config_tuning(self, wattage: int, enabled: bool = True, update_config: bool = False):
        if not self.config:
            await self.get_config()
        config = self.config
        config['autotuning']['psu_power_limit'] = wattage
        config['autotuning']['enabled'] = enabled
        if update_config:
            self.update_config(config)
        return config

    async def change_config_temp_ctrl(self, target: float = 80.0, hot: float = 100.0,
                                      dangerous: float = 110.0, update_config: bool = False):
        if not self.config:
            await self.get_config()
        config = self.config
        config['target_temp'] = target
        config['hot_temp'] = hot
        config['dangerous_temp'] = dangerous
        if update_config:
            self.update_config(config)
        return config