Search your Gmail by messaging a Telegram bot

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_event, credentials, http, ai

MY_CHAT_ID = 123456789

@on_event
def handle(event):
    msg = event.event.get("message", {})
    if msg.get("chat", {}).get("id") != MY_CHAT_ID or not msg.get("text"):
        return {"ignored": True}
    query = ai.generate(f"Turn this request into a single Gmail search query. Reply with the query only.\n{msg['text']}",
                        max_output_tokens=60).text.strip()
    gmail = build("gmail", "v1", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False)
    ids = gmail.users().messages().list(userId="me", q=query, maxResults=5).execute().get("messages", [])
    lines = []
    for m in ids:
        full = gmail.users().messages().get(userId="me", id=m["id"], format="metadata", metadataHeaders=["From", "Date", "Subject"]).execute()
        h = {x["name"]: x["value"] for x in full["payload"]["headers"]}
        lines.append(f"{h.get('Date', '')[:16]} | {h.get('From')} | {h.get('Subject')}")
    text = f"Search: {query}\n" + ("\n".join(lines) or "No matches.")
    http.post(f"https://api.telegram.org/bot{credentials.secret('TELEGRAM_BOT_TOKEN')}/sendMessage",
              json={"chat_id": MY_CHAT_ID, "text": text})
    return {"results": len(lines)}