import base64
from dataclasses import dataclass
from datetime import datetime
import json
import logging
from pydantic import BaseModel, Field, TypeAdapter, field_validator
from requests.sessions import Session


@dataclass
class Configuration:
    url_base: str = None
    url_auth: str = None
    client_id: str = None
    refresh_token: str = None
    token: str = None
    unit_id: str = None


@dataclass
class Response:
    status_code: int
    content: list | dict | str | None = None


class UiPath(object):
    def __init__(self, credentials: dict) -> None:
        """
        Initializes variables and performs authentication.

        Args:
            credentials (dict): Credentials from Fortanix.
        """
        logging.basicConfig(level=logging.INFO)
        self.session: Session = Session()
        self.__configuration = self.auth(credentials=credentials)

    def __del__(self) -> None:
        """
        Clean the house at the exit.
        """
        logging.info(msg="Closes session")
        self.session.close()

    def is_auth(self) -> bool:
        """
        Checks whether authentication was successful.

        Returns:
            bool: If true, then authentication was successful.
        """
        logging.info(msg="Gets authentication status")
        return False if self.__configuration.token is None else True

    def auth(self, credentials: dict) -> Configuration:
        """
        Authenticate (On-Premise).
        Performs authentication process to obtain the token.

        Args:
            credentials (dict): Credentials from Fortanix.

        Returns:
            Configuration: Access configuration.
        """
        logging.info(msg="Opens session")
        configuration = Configuration()
        configuration.client_id = credentials["custom_metadata"]["client_id"]
        configuration.unit_id = credentials["custom_metadata"]["unit_id"]
        configuration.url_base = credentials["custom_metadata"]["url_base"]
        configuration.url_auth = credentials["custom_metadata"]["url_auth"]
        configuration.refresh_token = base64.b64decode(credentials["value"]).decode("utf-8")
        headers = {"Connection": "keep-alive", "Content-Type": "application/json"}
        body = {"grant_type": "refresh_token", "client_id": configuration.client_id,
                "refresh_token": configuration.refresh_token}
        response = self.session.post(url=configuration.url_auth, json=body, headers=headers)
        logging.info(msg=f"HTTP Status Code {response.status_code}")
        if response.status_code == 200:
            configuration.token = json.loads(response.content.decode("utf-8"))["access_token"]
            return configuration

    def get_assets(self, save_as: str | None = None) -> Response:
        """
        Assets - Get all.
        Gets the UiPath Orchestrator assets.

        Args:
            save_as (str, optional): Name of the Json file that contains the
              request response.

        Returns:
            Response (dataclass): Asset information and response status.
        """
        logging.info(msg="Gets a list of all assets")
        headers = {"Content-Type": "application/json", "Authorization": "Bearer " + self.__configuration.token,
                   "X-UIPATH-OrganizationUnitID": f"{self.__configuration.unit_id}"}
        url_query = f"{self.__configuration.url_base}odata/Assets"

        class DataStructure(BaseModel):
            Id: int = Field(alias="Id")
            Name: str = Field(alias="Name")
            ExternalName: str | None = Field(alias="ExternalName", default=None)
            HasDefaultValue: bool = Field(alias="HasDefaultValue")
            Value: str = Field(alias="Value")
            ValueScope: str = Field(alias="ValueScope")
            ValueType: str = Field(alias="ValueType")
            IntValue: int = Field(alias="IntValue")
            StringValue: str = Field(alias="StringValue")
            BoolValue: bool = Field(alias="BoolValue")
            CredentialUsername: str = Field(alias="CredentialUsername")
            CredentialStoreId: int | None = Field(alias="CredentialStoreId", default=None)
            CanBeDeleted: bool = Field(alias="CanBeDeleted")
            Description: str | None = Field(alias="Description", default=None)

        alias_list = [field.alias if field.alias is not None else field_name for field_name, field in
                      DataStructure.model_fields.items()]
        params = {"$select": ",".join(alias_list)}
        response = self.session.get(url=url_query, headers=headers, params=params, verify=True)
        logging.info(msg=f"HTTP Status Code {response.status_code}")
        content = None
        if response.status_code == 200:
            if save_as is not None:
                with open(file=save_as, mode="wb") as file:
                    file.write(response.content)
            content_raw = response.json()["value"]
            content = [dict(data) for data in TypeAdapter(list[DataStructure]).validate_python(content_raw)]

        return Response(status_code=response.status_code, content=content)