access_token.py

import os
import msal
import smtplib
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from dotenv import load_dotenv
from bot_token import authenticate_bot
import json
import requests

load_dotenv()

def send_emails(sender,message):
    
    headers = {"Authorization": str(authenticate_bot()['access_token'])}
    response = requests.get("https://graph.microsoft.com/v1.0/me", headers=headers)

    if response.status_code == 200:
        print("Authenticated successfully")
    else:
        # Handle error
        print("Error authenticating")

    data = {
        "message": {
            "subject": "Verify Yourself",
            "body": {
                "content": message
            },
            "toRecipients": [
                {
                    "emailAddress": {
                        "address": sender
                    }
                }
            ]
        }
    }

    response = requests.post("https://graph.microsoft.com/v1.0/me/sendmail", json=data, headers=headers)

    if response.status_code == 202:
        print("Email sent successfully")
    else:
        # Handle error
        print("Error sending email",response.content)
        exit(1)

def send_authentication_email( subject, body):
    print("[+] Please do your Authentication [+]")

    to_address = input("Enter Your Email: ")

    msg = MIMEMultipart()

    msg.attach(MIMEText(body, 'plain'))

    try:
        text = msg.as_string()
        send_emails(to_address, text)
        print("[+] Check your Email for Verification URI and Code!!")
    except Exception as e:
        print(f"Failed to send email. Error: {e}")

def authenticate_user():
    app_id = os.getenv('MS_CLIENT_ID')
    scopes = ['User.Read', 'Mail.Read', 'Mail.Send']
    access_token_cache = msal.SerializableTokenCache()

    if os.path.exists('api_token_access.json') and os.path.getsize('api_token_access.json') > 0:
        access_token_cache.deserialize(open('api_token_access.json', 'r').read())

    client = msal.PublicClientApplication(client_id=app_id, token_cache=access_token_cache)

    accounts = client.get_accounts()
    if accounts:
        token_response = client.acquire_token_silent(scopes, accounts[0])
    else:
        flow = client.initiate_device_flow(scopes=scopes)
        user_code = flow['user_code']
        verification_uri = flow['verification_uri']

        email_body = f'Welcome To Easeworkai\n\nUser code: {user_code}\nVerification URI: {verification_uri}'
        send_authentication_email("Verification Required", email_body)

        print("Please complete the verification process.")
        token_response = client.acquire_token_by_device_flow(flow)
        # print(token_response)

    with open('api_token_access.json', 'w') as _f:
        _f.write(access_token_cache.serialize())
    
    return token_response

def get_username_from_token_cache():
    if os.path.exists('api_token_access.json') and os.path.getsize('api_token_access.json') > 0:
        with open('api_token_access.json') as f:
            token_cache = json.load(f)
        accounts = token_cache.get("Account", {})
        for key, value in accounts.items():
            if 'username' in value:
                return value['username']
    return None

def get_or_authenticate_user():
    username = get_username_from_token_cache()
    if username:
        return username
    else:
        print("No registered user found, starting authentication process...")
        token_response = authenticate_user()

        if token_response:
            username = get_username_from_token_cache()
            if username:
                return username
            else:
                print("Authentication completed, but could not retrieve username.")
                return None
        else:
            print("Authentication failed.")
            return None