import json

import pytest
from playwright.sync_api import BrowserContext, Page

from pytest_abra.dir_manager import DirManager
from pytest_abra.utils import BaseUrl


@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()
    config = DIR.get_config("authentik")
    base_url = BaseUrl(config["DOMAIN"])
    page.goto(base_url.get())
    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()
    config = DIR.get_config("authentik")
    base_url = BaseUrl(config["DOMAIN"])
    page.goto(base_url.get())
    return page