import requests
from fake_useragent import UserAgent

class AVConnectAPI:
    _BASE_URL = "https://www.avconnect.it"

    def __init__(self, username: str, password: str, session_id: str | None = None):
        self._ua = UserAgent(browsers=["Chrome Mobile"], os=["Android"], platforms=["mobile"]).random
        self._username = username
        self._password = password
        self._session = requests.Session()
        self._authenticated = False

        if session_id:
            self._session.cookies.set("PHPSESSID", session_id)
            self._authenticated = True

    def _authenticate(self) -> bool:
        login_url = f"{self._BASE_URL}/loginone.php"
        headers = {
            "User-Agent": self._ua,
            "Content-Type": "application/x-www-form-urlencoded"
        }
        payload = f"userid={self._username}&password={self._password}&entra=Login"
        response = self._session.post(login_url, data=payload, headers=headers)
        if response.ok and "PHPSESSID" in self._session.cookies:
            self._authenticated = True
            print("Authenticated")
            return True
        return False
    
    def _check_sessionid(self) -> bool:
        if not self._authenticated or not self._session.cookies.get("PHPSESSID"):
            return False
        exec_url = f"{self._BASE_URL}/exemacrocom.php"
        headers = {
            "User-Agent": self._ua,
        }
        response = self._session.get(exec_url, headers=headers)
        print(response.ok)
        return response.ok

    def exec_gate_macro(self, id_macro) -> bool:
        if (not self._authenticated or not self._check_sessionid()) and not self._authenticate():
            raise Exception("Authentication failed.")
        exec_url = f"{self._BASE_URL}/exemacrocom.php"
        headers = {
            "User-Agent": self._ua,
            "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8"
        }
        payload = f"idmacrocom={id_macro}&nome=16"
        response = self._session.post(exec_url, data=payload, headers=headers)
        return response.ok