from datetime import datetime

from httpx import AsyncClient, Headers

from wolf_comm import constants
from wolf_comm.constants import SESSION_ID, TIMESTAMP
from wolf_comm.helpers import bearer_header


async def create_session(client: AsyncClient, token: str):
    data = {
        TIMESTAMP: datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    }
    resp = await client.post(
        constants.BASE_URL_PORTAL + "/api/portal/CreateSession2",
        headers=Headers({
            **bearer_header(token),
            **{"Content-Type": "application/json"}
            }
        ),
        json=data
    )

    return resp.json()['BrowserSessionId']


async def update_session(client: AsyncClient, token: str, session_id: str):
    data = {
        SESSION_ID: session_id
    }
    await client.post(
        constants.BASE_URL_PORTAL + "/api/portal/UpdateSession",
        headers=Headers({
            **bearer_header(token),
            **{"Content-Type": "application/json"}
            }
        ),
        json=data)