"""
main file for browsertrix-api system
supports docker and kubernetes based deployments of multiple browsertrix-crawlers
"""

import os

from fastapi import FastAPI, Request, HTTPException

from db import init_db

from users import init_users_api, UserDB
from archives import init_archives_api
from crawlconfigs import init_crawl_config_api
from crawls import init_crawls_api
from emailsender import EmailSender

app = FastAPI()


# ============================================================================
class BrowsertrixAPI:
    """
    Main class for BrowsertrixAPI
    """

    # pylint: disable=too-many-instance-attributes
    def __init__(self, _app):
        self.app = _app

        self.default_storage_endpoint_url = os.environ.get(
            "STORE_ENDPOINT_URL", "http://minio:9000/test-bucket/"
        )

        self.default_storage_access_key = os.environ.get("STORE_ACCESS_KEY", "access")
        self.default_storage_secret_key = os.environ.get("STORE_SECRET_KEY", "secret")

        self.email = EmailSender()
        self.crawl_manager = None

        self.default_crawl_params = [
            "--timeout",
            "90",
            "--logging",
            "behaviors,debug",
            "--generateWACZ",
        ]

        self.mdb = init_db()

        self.fastapi_users = init_users_api(
            self.app,
            self.mdb,
            self.on_after_register,
            self.on_after_forgot_password,
            self.on_after_verification_request,
        )

        current_active_user = self.fastapi_users.current_user(active=True)

        self.archive_ops = init_archives_api(
            self.app, self.mdb, self.fastapi_users, self.email, current_active_user
        )

        # pylint: disable=import-outside-toplevel
        if os.environ.get("KUBERNETES_SERVICE_HOST"):
            from k8sman import K8SManager

            self.crawl_manager = K8SManager(self.default_crawl_params)
        else:
            from dockerman import DockerManager

            self.crawl_manager = DockerManager(
                self.archive_ops, self.default_crawl_params
            )

        self.crawl_config_ops = init_crawl_config_api(
            self.app,
            self.mdb,
            current_active_user,
            self.archive_ops,
            self.crawl_manager,
        )

        init_crawls_api(
            self.app,
            self.mdb,
            self.crawl_manager,
            self.archive_ops,
        )

        self.app.include_router(self.archive_ops.router)

        # @app.get("/")
        # async def root():
        #    return {"message": "Hello World"}

    # pylint: disable=no-self-use, unused-argument
    async def on_after_register(self, user: UserDB, request: Request):
        """callback after registeration"""

        print(f"User {user.id} has registered.")

        req_data = await request.json()

        if req_data.get("newArchive"):
            print(f"Creating new archive for {user.id}")

            archive_name = req_data.get("name") or f"{user.email} Archive"

            await self.archive_ops.create_new_archive_for_user(
                archive_name=archive_name,
                base_endpoint_url=self.default_storage_endpoint_url,
                access_key=self.default_storage_access_key,
                secret_key=self.default_storage_secret_key,
                user=user,
            )

        if req_data.get("inviteToken"):
            try:
                await self.archive_ops.handle_new_user_invite(
                    req_data.get("inviteToken"), user
                )
            except HTTPException as exc:
                print(exc)

    # pylint: disable=no-self-use, unused-argument
    def on_after_forgot_password(self, user: UserDB, token: str, request: Request):
        """callback after password forgot"""
        print(f"User {user.id} has forgot their password. Reset token: {token}")

    # pylint: disable=no-self-use, unused-argument
    def on_after_verification_request(self, user: UserDB, token: str, request: Request):
        """callback after verification request"""

        self.email.send_user_validation(token, user.email)


# ============================================================================
@app.on_event("startup")
async def startup():
    """init on startup"""
    BrowsertrixAPI(app)