import os
import requests
import msal
import base64

# ==============================================================================
# KONFIGURÁCIÓ
# ==============================================================================
# Az Azure App Registration-ból kapott adatok
TENANT_ID = "TENANT_ID_MASOLD_BE_IDE"
CLIENT_ID = "CLIENT_ID_MASOLD_BE_IDE"
CLIENT_SECRET = "CLIENT_SECRET_MASOLD_BE_IDE" # A "Value", nem a "Secret ID"

# A postafiók, amit figyelni kell
USER_EMAIL = "Bestellung-Fax-Eingang@aps-hh.de"

# Hova mentsük a letöltött PDF-eket
DOWNLOAD_DIR = r"\\aps-nb090\test"

# A mappa neve, ahova a feldolgozott leveleket helyezzük
PROCESSED_FOLDER_NAME = "erledigt"
# ==============================================================================

# Microsoft Graph API végpontok
GRAPH_API_ENDPOINT = "https://graph.microsoft.com/v1.0"
AUTHORITY_URL = f"https://login.microsoftonline.com/{TENANT_ID}"
SCOPES = ["https://graph.microsoft.com/.default"]


def get_graph_api_token():
    """Beszerzi a hozzáférési tokent a Microsoft Graph API-hoz."""
    app = msal.ConfidentialClientApplication(
        client_id=CLIENT_ID,
        authority=AUTHORITY_URL,
        client_credential=CLIENT_SECRET
    )
    result = app.acquire_token_silent(scopes=SCOPES, account=None)
    if not result:
        result = app.acquire_token_for_client(scopes=SCOPES)
    
    if "access_token" in result:
        return result["access_token"]
    else:
        print("Hiba a token beszerzése során!")
        print(result.get("error"))
        print(result.get("error_description"))
        return None

def get_folder_id(access_token, folder_name):
    """Megkeresi egy mappa ID-ját a neve alapján."""
    headers = {"Authorization": f"Bearer {access_token}"}
    url = f"{GRAPH_API_ENDPOINT}/users/{USER_EMAIL}/mailFolders"
    response = requests.get(url, headers=headers)
    response.raise_for_status()
    folders = response.json().get("value", [])
    
    for folder in folders:
        if folder["displayName"].lower() == folder_name.lower():
            return folder["id"]
    
    # Kezelhetnénk azt az esetet is, ha a mappa nem létezik, és létrehoznánk.
    # Most egyszerűen hibát dobunk.
    raise ValueError(f"A '{folder_name}' mappa nem található.")


def main():
    """Fő feldolgozó funkció."""
    print("Graph API token beszerzése...")
    access_token = get_graph_api_token()
    if not access_token:
        return

    headers = {"Authorization": f"Bearer {access_token}"}
    
    try:
        print(f"'{PROCESSED_FOLDER_NAME}' mappa ID-jának keresése...")
        processed_folder_id = get_folder_id(access_token, PROCESSED_FOLDER_NAME)
        print("Mappa ID sikeresen lekérve.")
    except (requests.HTTPError, ValueError) as e:
        print(f"Hiba a mappa ID lekérése közben: {e}")
        return

    # Csak az olvasatlan, csatolmánnyal rendelkező levelek lekérdezése
    # $select=id,subject -> Csak a szükséges mezőket kérjük le a hatékonyságért
    messages_url = (
        f"{GRAPH_API_ENDPOINT}/users/{USER_EMAIL}/mailFolders/inbox/messages?"
        "$filter=isRead eq false and hasAttachments eq true&"
        "$select=id,subject"
    )
    
    response = requests.get(messages_url, headers=headers)
    response.raise_for_status()
    messages = response.json().get("value", [])

    if not messages:
        print("Nincsenek új, feldolgozandó e-mailek.")
        return

    print(f"{len(messages)} új e-mail található csatolmánnyal.")

    for message in messages:
        msg_id = message["id"]
        subject = message.get("subject", "N/A")
        print(f"\n--- Feldolgozás alatt: '{subject}' (ID: {msg_id}) ---")
        
        attachments_url = f"{GRAPH_API_ENDPOINT}/users/{USER_EMAIL}/messages/{msg_id}/attachments"
        response = requests.get(attachments_url, headers=headers)
        if response.status_code != 200:
            print(f"  Hiba a csatolmányok lekérésekor: {response.json()}")
            continue
            
        attachments = response.json().get("value", [])
        pdf_found = False

        for att in attachments:
            filename = att.get("name", "unknown")
            content_type = att.get("contentType", "")
            
            if filename.lower().endswith(".pdf") or content_type == "application/pdf":
                print(f"  PDF csatolmány található: {filename}")
                
                # A csatolmány tartalma a 'contentBytes' mezőben van, Base64 kódolással
                file_content = base64.b64decode(att["contentBytes"])
                
                # Fájlnév "tisztítása"
                safe_filename = "".join(c if c.isalnum() or c in (" ", ".", "_", "-") else "_" for c in filename)
                filepath = os.path.join(DOWNLOAD_DIR, safe_filename)

                try:
                    with open(filepath, "wb") as f:
                        f.write(file_content)
                    print(f"  PDF sikeresen mentve: {filepath}")
                    pdf_found = True
                except Exception as e:
                    print(f"  Hiba a fájl mentése közben: {e}")

        if pdf_found:
            print(f"  E-mail áthelyezése a '{PROCESSED_FOLDER_NAME}' mappába...")
            move_url = f"{GRAPH_API_ENDPOINT}/users/{USER_EMAIL}/messages/{msg_id}/move"
            move_payload = {"destinationId": processed_folder_id}
            
            response = requests.post(move_url, headers=headers, json=move_payload)
            if response.status_code == 201:
                print("  E-mail sikeresen áthelyezve.")
            else:
                # Az áthelyezéshez Mail.ReadWrite jogosultság kell!
                print(f"  Hiba az e-mail áthelyezésekor: {response.status_code} - {response.text}")

    print("\nFeldolgozás befejezve.")


if __name__ == "__main__":
    if not os.path.exists(DOWNLOAD_DIR):
        print(f"Hiba: A letöltési mappa nem létezik: {DOWNLOAD_DIR}")
    else:
        main()