dotfiles/bin/.bin/checkworkmail.py

#!/usr/bin/env -S uv run --script
#
# /// script
# dependencies = [
#   "requests",
#   "beautifulsoup4",
#   "imaplib2",
# ]
# ///

import imaplib
import email
from email.header import decode_header
import requests
import time
import os

# ----------------------
# CONFIGURATION
# ----------------------
GMAIL_USER = os.getenv("GMAIL_USER", "your-email@gmail.com")
GMAIL_APP_PASSWORD = os.getenv("GMAIL_APP_PASSWORD", "")  # Replace or use env vars
NTFY_TOPIC = os.getenv("NTFY_TOPIC", "KKddGQxVm2LoP0JF")
NTFY_SERVER = os.getenv("NTFY_SERVER", "https://ntfy.unbl.ink")
CHECK_INTERVAL = int(os.getenv("CHECK_INTERVAL", "60"))  # seconds

def clean_subject(subject):
    """Decode and clean up the email subject"""
    decoded, encoding = decode_header(subject)[0]
    if isinstance(decoded, bytes):
        decoded = decoded.decode(encoding or 'utf-8', errors='ignore')
    # Remove problematic characters for HTTP headers
    return decoded.replace('\r', '').replace('\n', '').strip()

def send_ntfy_notification(title, message):
    url = f"{NTFY_SERVER.rstrip('/')}/{NTFY_TOPIC}"
    requests.post(url, data=message.encode("utf-8"), headers={"Title": title})

def check_github_notifications():
    print("Connecting to Gmail...")
    mail = imaplib.IMAP4_SSL("imap.gmail.com")
    mail.login(GMAIL_USER, GMAIL_APP_PASSWORD)
    mail.select("inbox")

    result, data = mail.search(None, '(UNSEEN FROM "notifications@github.com")')
    mail_ids = data[0].split()

    if not mail_ids:
        print("No new GitHub notifications.")
        mail.logout()
        return

    print(f"Found {len(mail_ids)} new GitHub notifications")

    for num in mail_ids:
        result, msg_data = mail.fetch(num, '(RFC822)')
        raw_email = msg_data[0][1]
        msg = email.message_from_bytes(raw_email)

        subject = clean_subject(msg["Subject"])
        from_email = msg.get("From")

        if msg.is_multipart():
            for part in msg.walk():
                if part.get_content_type() == "text/plain":
                    body = part.get_payload(decode=True).decode(errors="ignore")
                    break
            else:
                body = "(No plain text content found)"
        else:
            body = msg.get_payload(decode=True).decode(errors="ignore")

        print(f"Sending: {subject.replace('\n', '').replace('\r', '').strip()}")
        send_ntfy_notification(subject, body[:300])

    mail.logout()

if __name__ == "__main__":
    while True:
        try:
            check_github_notifications()
        except Exception as e:
            print(f"Error: {e}")
        time.sleep(CHECK_INTERVAL)