Turn ideas in a sheet into approved LinkedIn posts

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

AUTHOR, SHEET_ID, LINK = "urn:li:person:abc123", "your-spreadsheet-id", "https://run.clickruntime.example/t/abc123"

def sheet(): return build("sheets", "v4", credentials=Credentials(credentials.connection("google", "work")), cache_discovery=False).spreadsheets().values()

@on_trigger
def run(event):
    if state.get("pending"):
        return {"waiting": True}
    rows = sheet().get(spreadsheetId=SHEET_ID, range="Ideas!A2:B").execute().get("values", [])
    row = next((i + 2 for i, r in enumerate(rows) if r and len(r) < 2), None)
    if not row:
        return {"ideas": 0}
    draft = ai.generate(f"Write a LinkedIn post from this idea, first person, no hashtags spam: {rows[row - 2][0]}", max_output_tokens=500).text
    state.set("pending", {"row": row, "text": draft})
    notify.email_me("LinkedIn draft to approve", f"{draft}\n\nApprove: {LINK}?approve=1")
    return {"drafted": row}

@on_event
def approve(event):
    p = state.get("pending")
    r = http.post("https://api.linkedin.com/rest/posts", headers={"Authorization": f"Bearer {credentials.connection('linkedin', 'work')}", "LinkedIn-Version": "202409"},
                  json={"author": AUTHOR, "commentary": p["text"], "visibility": "PUBLIC", "lifecycleState": "PUBLISHED", "distribution": {"feedDistribution": "MAIN_FEED"}})
    url = f"https://www.linkedin.com/feed/update/{r.headers.get('x-restli-id')}"
    sheet().update(spreadsheetId=SHEET_ID, range=f"Ideas!B{p['row']}", valueInputOption="RAW", body={"values": [[url]]}).execute()
    state.delete("pending")
    return {"posted": url}