agent-timecard/send_review.py at master · ElleNajt/agent-timecard · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
#!/usr/bin/env python3
"""Send review email (non-interactive, for cron jobs)."""

import base64
import smtplib
import sys
from email.mime.image import MIMEImage
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText

import markdown

from config import load_config

STYLE = """\
<style>
body { font-family: -apple-system, sans-serif; max-width: 700px; margin: 0 auto; padding: 20px; color: #333; }
h1 { font-size: 1.4em; border-bottom: 2px solid #333; padding-bottom: 6px; }
h2 { font-size: 1.1em; margin-top: 1.5em; color: #555; }
h3 { font-size: 1em; margin-top: 1.2em; }
ul { padding-left: 1.5em; }
li { margin-bottom: 4px; }
code { background: #f4f4f4; padding: 1px 4px; border-radius: 3px; font-size: 0.9em; }
hr { border: none; border-top: 1px solid #ddd; margin: 1.5em 0; }
</style>
"""


def md_to_html(body: str, html_prefix: str = "", html_suffix: str = "") -> str:
    """Convert markdown body to styled HTML, with optional pre/post HTML."""
    html_body = markdown.markdown(body, extensions=["fenced_code", "tables"])
    return f"<html><head>{STYLE}</head><body>{html_prefix}{html_body}{html_suffix}</body></html>"


def _build_message(
    to: str,
    subject: str,
    body: str,
    html_prefix: str = "",
    html_suffix: str = "",
    images: dict[str, bytes] | None = None,
) -> MIMEMultipart | MIMEText:
    """Build email message, with optional inline images.

    images: dict of {cid: png_bytes} — referenced in HTML as <img src="cid:name">
    """
    html = md_to_html(body, html_prefix, html_suffix)

    if not images:
        msg = MIMEText(html, "html")
        msg["to"] = to
        msg["subject"] = subject
        return msg

    msg = MIMEMultipart("related")
    msg["to"] = to
    msg["subject"] = subject
    msg.attach(MIMEText(html, "html"))

    for cid, png_bytes in images.items():
        img = MIMEImage(png_bytes, _subtype="png")
        img.add_header("Content-ID", f"<{cid}>")
        img.add_header("Content-Disposition", "inline", filename=f"{cid}.png")
        msg.attach(img)

    return msg


def send_gmail(
    to: str,
    subject: str,
    body: str,
    html_prefix: str = "",
    html_suffix: str = "",
    images: dict[str, bytes] | None = None,
):
    """Send email via Gmail API (macOS Keychain auth)."""
    from keychain_auth import get_gmail_service

    service = get_gmail_service()

    message = _build_message(to, subject, body, html_prefix, html_suffix, images)
    raw = base64.urlsafe_b64encode(message.as_bytes()).decode()
    result = service.users().messages().send(userId="me", body={"raw": raw}).execute()

    print(f"Email sent! Message ID: {result['id']}")
    return result


def send_smtp(
    to: str, subject: str, body: str, smtp_config: dict, html_prefix: str = ""
):
    """Send email via SMTP."""
    import os

    message = MIMEText(md_to_html(body, html_prefix), "html")
    message["to"] = to
    message["subject"] = subject
    message["from"] = smtp_config["username"]

    password = os.environ.get(smtp_config.get("password_env", "SMTP_PASSWORD"))
    if not password:
        raise RuntimeError(
            f"Set {smtp_config.get('password_env', 'SMTP_PASSWORD')} env var"
        )

    with smtplib.SMTP(smtp_config["host"], smtp_config.get("port", 587)) as server:
        server.starttls()
        server.login(smtp_config["username"], password)
        server.send_message(message)

    print(f"Email sent via SMTP to {to}")


def send_email(
    to: str,
    subject: str,
    body: str,
    html_prefix: str = "",
    html_suffix: str = "",
    images: dict[str, bytes] | None = None,
):
    """Send email using configured method."""
    cfg = load_config()
    method = cfg["email_method"]

    if method == "gmail":
        send_gmail(to, subject, body, html_prefix, html_suffix, images)
    elif method == "smtp":
        if not cfg.get("smtp"):
            raise RuntimeError(
                "smtp settings required in config.yaml for email_method: smtp"
            )
        send_smtp(to, subject, body, cfg["smtp"], html_prefix)
    else:
        raise RuntimeError(f"Unknown email_method: {method}")


if __name__ == "__main__":
    if len(sys.argv) < 4:
        print("Usage: send_review.py <to> <subject> <body>")
        sys.exit(1)

    send_email(sys.argv[1], sys.argv[2], sys.argv[3])