from sys import argv
from os import environ
from dotenv import load_dotenv
from cloudscraper import CloudScraper

class Scraper:
    def __init__(self, pseudo, password, app, debug = False):
        self.debug = debug
        self.url = "https://forum.mobilism.org"
        self.requested_app = app
        self.loginData = {
            "username": pseudo,
            "password": password,
            "login": "Login",
            # "redirect": "./ucp.php?mode=login",
            "redirect": "index.php"
        }

    def errorFormat(self, code: int) -> str:
        return f"\Error: [{code}]"

    def connect(self):
        with CloudScraper(browser = {"browser": "chrome", "platform": "windows"}) as session:
            if self.debug: print("Retrieval of the login cookie...", end = " ")
            reponse = session.get(f"{self.url}/ucp.php?mode=login")
            if reponse.status_code != 200:
                raise ConnectionRefusedError(self.errorFormat(reponse.status_code))
            self.loginData["sid"] = reponse.cookies.get_dict()["ppcw_29d3s_sid"]
            if self.debug: print("Cookie retrieval done, connection attempt...", end = " ")
            reponse = session.post(f"{self.url}/ucp.php?mode=login", data = self.loginData)
            if reponse.status_code != 200:
                raise ConnectionRefusedError(self.errorFormat(reponse.status_code))
            if self.debug: print("Connection done.")
            if self.debug: print(reponse.status_code, reponse.url)
            with open("temp.html", "w") as f:
                f.writelines(reponse.text)
            link = "No link for your application was found."
            return link

if __name__ == "__main__":
    argv = argv[1:]
    if len(argv) == 3:
        print(Scraper(*argv).connect())
    else:
        try:
            load_dotenv()
            try:
                debug = environ["DEBUG_MOBILISM"].lower() in ("yes", "true", "1")
            except:
                debug = False
            print(Scraper(environ["PSEUDO_MOBILISM"], environ["PASSWORD_MOBILISM"], environ["APP_MOBILISM"], debug).connect())
        except:
            print('Please fill in the username and password (with ") by args or with .env file.')