import json
from base import BaseCLi
from constants import (
    DEFAULT_PORTAINER_JWT_CONFIG_FILE,
    FUNCTIONAL_USER_USERNAME_ENV,
    FUNCTIONAL_USER_PASSWORD_ENV,
)
import os
import requests

class MissingEnvironmentVariable(Exception):
    pass

class EmptyConfigException(Exception):
    pass

class PortainerHost(BaseCLi):

    def __init__(self, host_url):
        super().__init__()
        self.host_url = host_url
        self.api_url = f'{host_url}/api'
        self.jwt_config = {}
        self.jwt_token = ''
        home_directory = os.path.expanduser("~")
        config_file_path = os.path.join(home_directory, DEFAULT_PORTAINER_JWT_CONFIG_FILE)
        try:
            self.__read_current_config(config_file_path)
        except EmptyConfigException:
            self.__create_config(config_file_path)
            
    def __read_current_config(self, file_path):
        try:
            with open(file_path, 'r') as file:
                self.jwt_config = json.load(file)

        except FileNotFoundError:
            print(f"Configuration does not exist. Creating...")
            raise EmptyConfigException()
        try:
            self.jwt_token = self.jwt_config[self.host_url]
        except KeyError:
            print(f"No token present in configuration file for host {self.host_url}. Creating...")
            raise EmptyConfigException()
            
    def __create_config(self, file_path):
        self.jwt_token = self.__get_jwt_token()
        self.jwt_config[self.host_url] = {
            "jwt": self.jwt_token,
        }
        with open(file_path, 'w') as file:
            json.dump(self.jwt_config, file, indent=4)

    def __get_jwt_token(self):
        def get_env(env_key):
            try:
                return os.environ[env_key]
            except KeyError:
                raise MissingEnvironmentVariable(f"{env_key} env does not exist")

        payload = {
            "Username": get_env(FUNCTIONAL_USER_USERNAME_ENV),
            "Password": get_env(FUNCTIONAL_USER_PASSWORD_ENV)
        }
        auth_endpoint = f'{self.api_url}/auth'
        try:
            response = requests.post(
                auth_endpoint,
                json=payload
            )
            if response.status_code != 200:
                self.error(f'Return code {response.status_code} from {auth_endpoint} with message {response.text}')
        except requests.exceptions.RequestException as e:
            self.error("An error occurred:", e)
        data = response.json()
        return data['jwt']