import traceback

import bpy
import importlib
from ..core import login_manager
from ..core import library_manager
from ..core import live_data_manager


class LoginButton(bpy.types.Operator):
    bl_idname = "rsl.login_login"
    bl_label = "Sign in"
    bl_description = "Sign into your Rokoko account with your browser"
    bl_options = {'INTERNAL'}

    @classmethod
    def poll(cls, context):
        return not login_manager.user.logging_in

    def execute(self, context):
        login = login_manager.Login()
        login.start()
        self.report({'INFO'}, 'Opened Rokoko ID website in your browser.')
        return {'FINISHED'}


class LogoutButton(bpy.types.Operator):
    bl_idname = "rsl.login_logout"
    bl_label = "Sign out"
    bl_description = "Sign out of your Rokoko account"
    bl_options = {'INTERNAL'}

    def execute(self, context):
        login_manager.user.logout()
        self.report({'INFO'}, 'Logout successful.')
        return {'FINISHED'}


class InstallLibsButton(bpy.types.Operator):
    bl_idname = 'rsl.login_install_libs'
    bl_label = 'Install Required Libraries'
    bl_description = 'Installs the required libraries for this plugin to work'
    bl_options = {'INTERNAL'}

    def execute(self, context):
        # Install the libraries
        try:
            self.install_libs()
        except ImportError as e:
            self.report({'ERROR'}, str(e))
            return {'CANCELLED'}
        except Exception as e:
            trace = traceback.format_exc()
            error_str = f"Unable to install the libraries!" \
                        f"\nTry running Blender as an admin and install the libraries again." \
                        f"\n\nFull Error: \n\n{trace}"
            self.report({'ERROR'}, error_str)
            return {'CANCELLED'}

        # Try importing all the libraries to see if they are installed correctly
        try:
            print("Verifying library installation by importing them...")
            import boto3
            from cryptography.fernet import Fernet
            from gql import gql, Client
            from gql.transport.appsync_websockets import AppSyncWebsocketsTransport
            from gql.transport.appsync_auth import AppSyncApiKeyAuthentication
            from gql.transport.websockets import log as websockets_logger
            from lz4 import frame
        except Exception as e:
            trace = traceback.format_exc()
            error_str = f"Library installation failed! Please restart Blender and try again." \
                        f"\nIf the error persists, let us know." \
                        f"\n\nFull Error: \n\n{trace}"
            print(error_str)
            library_manager.lib_manager.reset_current_library_installation()
            self.report({'ERROR'}, error_str)
            return {'CANCELLED'}

        # Save login manager data
        classes_logged_in = login_manager.user.classes_logged_in
        classes_logged_out = login_manager.user.classes_logged_out
        version_str = login_manager.user.version_str

        # Reload files to load the libraries
        importlib.reload(login_manager)
        importlib.reload(live_data_manager)

        # Load the login manager data
        login_manager.user.classes_logged_in = classes_logged_in
        login_manager.user.classes_logged_out = classes_logged_out
        login_manager.user.version_str = version_str

        # Attempt to auto login the user
        login_manager.user.auto_login()

        self.report({'INFO'}, 'Installed libraries successfully!')
        return {'FINISHED'}

    def install_libs(self):
        missing = library_manager.lib_manager.install_libraries(["websockets", "gql==3.5.3", "cryptography", "boto3"])
        if missing:
            raise ImportError("The following libraries could not be installed: "
                              "\n- " + " \n- ".join(missing) +
                              "  \n\nTry running Blender as an admin and install the libraries again."
                              "  \nSee console for more information.")
        library_manager.lib_manager.install_libraries(["lz4"])
        importlib.invalidate_caches()