import json

import pytest
from dotenv import dotenv_values
from playwright.sync_api import BrowserContext, Page

from src.dir_manager import DirManager


@pytest.fixture
def authentik_admin_context(context: BrowserContext, DIR: DirManager) -> BrowserContext:
    state_file = DIR.STATES / "authentik_admin_state.json"
    storage_state = json.loads(state_file.read_bytes())
    context.add_cookies(storage_state["cookies"])
    return context


@pytest.fixture
def authentik_admin_page(authentik_admin_context: BrowserContext, DIR: DirManager) -> Page:
    page = authentik_admin_context.new_page()
    env_file = DIR.ENV_FILES / "authentik"
    config: dict[str, str] = dotenv_values(env_file)  # type: ignore
    url = "https://" + config["DOMAIN"]
    page.goto(url)
    return page


@pytest.fixture
def authentik_user_context(context: BrowserContext, DIR: DirManager) -> BrowserContext:
    state_file = DIR.STATES / "authentik_user_state.json"
    storage_state = json.loads(state_file.read_bytes())
    context.add_cookies(storage_state["cookies"])
    return context


@pytest.fixture
def authentik_user_page(authentik_user_context: BrowserContext, DIR: DirManager) -> Page:
    page = authentik_user_context.new_page()
    env_file = DIR.ENV_FILES / "authentik"
    config: dict[str, str] = dotenv_values(env_file)  # type: ignore
    url = "https://" + config["DOMAIN"]
    page.goto(url)
    return page