import os
import pyotp
from urllib.parse import urlencode
from bs4 import BeautifulSoup
import requests
from dotenv import load_dotenv
import json

class MIBLogin:
    def __init__(self):
        self.session = requests.Session()
        self.base_url = "https://faisanet.mib.com.mv"
        self.headers = {
            "User-Agent": "Mozilla/5.0 (give-api) ismath-owes-me-real-api/give-real-api AppleWebKit/537.36 (KHTML, like Gecko) Safari/537.36",
            "Accept": "*/*"
        }
        
    def get_login_page(self):
        response = self.session.get(f"{self.base_url}/auth", headers=self.headers)
        soup = BeautifulSoup(response.text, 'html.parser')
        return soup.find('input', {'name': 'rTag'})['value']

    def get_auth_type(self, rtag, username, retain=1):
        data = {'rTag': rtag, 'pgf01': username, 'retain': retain}
        headers = {**self.headers, 'Content-Type': 'application/x-www-form-urlencoded'}
        response = self.session.post(f"{self.base_url}/aAuth/getAuthType", headers=headers, data=urlencode(data))
        return response.json()

    def login(self, username, password, retain=1):
        rtag = self.get_login_page()
        self.get_auth_type(rtag, username, retain)
        
        data = {'rTag': rtag, 'pgf01': username, 'pgf02': password, 'retain': retain}
        headers = {**self.headers, 'Content-Type': 'application/x-www-form-urlencoded'}
        self.session.post(f"{self.base_url}/aAuth", headers=headers, data=urlencode(data))

    def auth_2fa(self, totp_seed):
        self.session.get(f"{self.base_url}/auth2FA", headers=self.headers)
        totp = pyotp.TOTP(totp_seed)
        
        data = {'otpType': 3, 'otp': totp.now()}
        headers = {**self.headers, 'Content-Type': 'application/x-www-form-urlencoded'}
        self.session.post(f"{self.base_url}/aAuth2FA/verifyOTP", headers=headers, data=urlencode(data))
        
        return {cookie.name: cookie.value for cookie in self.session.cookies if cookie.name in ['IBSID', 'ql_0']}

def main():
    load_dotenv()
    mib = MIBLogin()
    mib.login(os.getenv('USERNAME'), os.getenv('PASSWORD'))
    cookies = mib.auth_2fa(os.getenv('TOTP_SEED'))

    # Print the cookies in JSON format
    print(json.dumps(cookies, indent=4))

if __name__ == "__main__":
    main()