Turn action emails into Google Tasks and flag urgent ones in Slack

from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from clickruntime import on_trigger, report_fetched, credentials, state, notify, ai

SCHEMA = {"type": "object", "properties": {"action": {"type": "boolean"}, "urgent": {"type": "boolean"}, "task": {"type": "string"}}}

@on_trigger
def run(event):
    creds = Credentials(credentials.connection("google", "work"))
    gmail = build("gmail", "v1", credentials=creds, cache_discovery=False).users().messages()
    done = set(state.get("done") or [])
    new = [m for m in gmail.list(userId="me", q="in:inbox newer_than:1d").execute().get("messages", []) if m["id"] not in done]
    report_fetched(len(new))
    if not new:
        return {"new": 0}
    tasks = build("tasks", "v1", credentials=creds, cache_discovery=False).tasks()
    for m in new:
        msg = gmail.get(userId="me", id=m["id"], format="metadata", metadataHeaders=["From", "Subject"]).execute()
        h = {x["name"]: x["value"] for x in msg["payload"]["headers"]}
        out = ai.generate(f"Does this email ask me to do something? Is it urgent? If action, give a short task title.\n"
                          f"From: {h.get('From')}\nSubject: {h.get('Subject')}\n{msg['snippet']}",
                          json_schema=SCHEMA, max_output_tokens=80).json
        link = f"https://mail.google.com/mail/u/0/#all/{m['id']}"
        if out["action"]:
            tasks.insert(tasklist="@default", body={"title": out["task"], "notes": f"From {h.get('From')}\n{link}"}).execute()
        if out["action"] and out["urgent"]:
            notify.slack(f"🚨 {out['task']} (from {h.get('From')})\n{link}", channel="#urgent")
    state.set("done", sorted(done | {m["id"] for m in new})[-1000:])
    return {"processed": len(new)}