himalaya-kit 0.2.0__py3-none-any.whl

himalaya_kit/__init__.py

@@ -0,0 +1,14 @@
+"""himalaya-kit: himalaya 邮件 CLI 的增强工具箱"""
+
+__version__ = "0.2.0"
+
+from himalaya_kit.config import KitConfig, load_kit_config
+from himalaya_kit.contacts import ContactStore, load_contact_store
+
+__all__ = [
+    "__version__",
+    "KitConfig",
+    "load_kit_config",
+    "ContactStore",
+    "load_contact_store",
+]

himalaya_kit/_lib/mail_compose.py

@@ -0,0 +1,139 @@
+#!/usr/bin/env python3
+"""邮件正文解析与签名（send / reply / forward 共用）。"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+from himalaya_kit._lib.mail_send import MailConfigError, render_markdown_to_html
+from himalaya_kit.i18n import t
+
+
+def get_body_source_epilog() -> str:
+    """获取正文来源说明（支持 i18n）。"""
+    return t("body_source_epilog")
+
+
+# 保持向后兼容
+BODY_SOURCE_EPILOG = get_body_source_epilog()
+
+
+def process_body(body: str) -> str:
+    return body.replace("\\n", "\n").replace("\\t", "\t")
+
+
+def read_stdin_body() -> str | None:
+    if sys.stdin.isatty():
+        return None
+    data = sys.stdin.read()
+    return data if data else None
+
+
+def resolve_body_content(args: argparse.Namespace, *, allow_empty: bool = False) -> tuple[str, bool]:
+    """解析正文，返回 (body, is_html)。优先级: --html > --file > -b/--body/--note > stdin。"""
+    body_text = getattr(args, "body", None)
+    if body_text is None and getattr(args, "note", None):
+        body_text = args.note
+
+    if args.html:
+        if args.markdown:
+            print("警告: --markdown 与 --html 同用时忽略 --markdown", file=sys.stderr)
+        return args.html, True
+
+    source: str | None = None
+    is_html = False
+
+    if args.file:
+        path = Path(args.file)
+        if not path.is_file():
+            raise FileNotFoundError(f"正文文件不存在: {path}")
+        source = path.read_text(encoding="utf-8")
+        if path.suffix.lower() in (".html", ".htm"):
+            is_html = True
+    elif body_text is not None:
+        source = process_body(body_text)
+    else:
+        source = read_stdin_body()
+        if source is None:
+            if allow_empty:
+                return "", False
+            raise MailConfigError(
+                "必须指定正文来源之一: -b / --html / --file，或通过管道传入 stdin\n"
+                "示例: cat note.md | email-reply.py <id> --markdown"
+            )
+
+    if args.markdown and not is_html:
+        source = render_markdown_to_html(source)
+        is_html = True
+
+    return source, is_html
+
+
+def append_signature(
+    body: str,
+    is_html: bool,
+    no_sig: bool,
+    signature_text: str = "",
+    signature_html: str = "",
+) -> str:
+    """追加签名到正文。签名内容由调用方从 KitConfig 传入。"""
+    if no_sig:
+        return body
+    sig = signature_html if is_html else signature_text
+    if not sig:
+        return body
+    return body + sig
+
+
+def add_body_arguments(parser: argparse.ArgumentParser, *, allow_empty: bool = False) -> None:
+    if allow_empty:
+        parser.add_argument("-b", "--body", default=None, help=t("arg.send.body_optional"))
+    else:
+        parser.add_argument("-b", "--body", default=None, help=t("arg.send.body"))
+    parser.add_argument("--html", default=None, help=t("arg.send.html"))
+    parser.add_argument("--file", default=None, help=t("arg.send.file"))
+    parser.add_argument(
+        "--markdown",
+        action="store_true",
+        help=t("arg.send.markdown"),
+    )
+    parser.add_argument(
+        "--attach",
+        action="append",
+        default=[],
+        help=t("arg.send.attach"),
+    )
+    parser.add_argument("--no-sig", action="store_true", help=t("arg.send.no_sig"))
+
+
+def build_email_content(
+    from_addr: str,
+    to: str,
+    subject: str,
+    body: str,
+    cc: str | None,
+    is_html: bool,
+    attachments: list[str],
+) -> str:
+    from himalaya_kit._lib.mail_send import build_multipart_email, build_simple_email
+
+    if attachments:
+        return build_multipart_email(
+            from_addr=from_addr,
+            to=to,
+            subject=subject,
+            body=body,
+            cc=cc,
+            is_html=is_html,
+            attachments=attachments,
+        )
+    return build_simple_email(
+        from_addr=from_addr,
+        to=to,
+        subject=subject,
+        body=body,
+        cc=cc,
+        is_html=is_html,
+    )

himalaya_kit/_lib/mail_message.py

@@ -0,0 +1,320 @@
+#!/usr/bin/env python3
+"""邮件读/解析公共库：导出、解码、正文提取（供 reply / forward 共用）。"""
+
+from __future__ import annotations
+
+import html as html_module
+import email
+import re
+import subprocess
+import sys
+import tempfile
+from email.header import decode_header
+from email.utils import getaddresses, parseaddr, parsedate_to_datetime
+from pathlib import Path
+
+from himalaya_kit.i18n import t
+
+
+def run(cmd: list[str]) -> subprocess.CompletedProcess:
+    return subprocess.run(cmd, capture_output=True, text=True)
+
+
+def export_message(msg_id: str, account: str | None = None) -> str:
+    tmp = tempfile.NamedTemporaryFile(suffix=".eml", delete=False)
+    tmp.close()
+    cmd = ["himalaya", "message", "export", msg_id, "--full", "-d", tmp.name]
+    if account:
+        cmd.extend(["-a", account])
+    result = run(cmd)
+    if result.returncode != 0:
+        print(t("error.export_failed", stderr=result.stderr), file=sys.stderr)
+        raise SystemExit(1)
+    return tmp.name
+
+
+def decode_mime_header(raw: str) -> str:
+    parts = decode_header(raw or "")
+    decoded: list[str] = []
+    for part, charset in parts:
+        if isinstance(part, bytes):
+            decoded.append(part.decode(charset or "utf-8", errors="replace"))
+        else:
+            decoded.append(part)
+    return "".join(decoded)
+
+
+def extract_text_body(msg: email.message.Message) -> str:
+    if msg.is_multipart():
+        for part in msg.walk():
+            if part.get_content_type() == "text/plain":
+                payload = part.get_payload(decode=True)
+                charset = part.get_content_charset() or "utf-8"
+                return payload.decode(charset, errors="replace")
+        for part in msg.walk():
+            if part.get_content_type() == "text/html":
+                payload = part.get_payload(decode=True)
+                charset = part.get_content_charset() or "utf-8"
+                html = payload.decode(charset, errors="replace")
+                return re.sub(r"<[^>]+>", "", html).strip()
+    else:
+        payload = msg.get_payload(decode=True)
+        if payload:
+            charset = msg.get_content_charset() or "utf-8"
+            return payload.decode(charset, errors="replace")
+    return ""
+
+
+def parse_original_email(eml_path: str) -> dict:
+    with open(eml_path, "r", encoding="utf-8", errors="replace") as fp:
+        msg = email.message_from_file(fp)
+
+    raw_subject = msg.get("Subject", t("output.no_subject"))
+    from_raw = msg.get("From", "")
+    to_raw = msg.get("To", "")
+    cc_raw = msg.get("Cc", "")
+    date_str = msg.get("Date", "")
+
+    try:
+        dt = parsedate_to_datetime(date_str)
+        formatted_date = dt.strftime("%Y年%m月%d日 %H:%M")
+    except Exception:
+        formatted_date = date_str
+
+    return {
+        "subject": decode_mime_header(raw_subject),
+        "from": decode_mime_header(from_raw),
+        "from_raw": from_raw,
+        "to": decode_mime_header(to_raw),
+        "to_raw": to_raw,
+        "cc": decode_mime_header(cc_raw),
+        "cc_raw": cc_raw,
+        "date": formatted_date,
+        "body": extract_text_body(msg).strip(),
+    }
+
+
+def reply_subject(subject: str) -> str:
+    s = subject.strip()
+    lowered = s.lower()
+    if lowered.startswith("re:") or s.startswith("回复:") or s.startswith("答复:"):
+        return s
+    return f"Re: {s}"
+
+
+def _normalize_email(email_addr: str) -> str:
+    return email_addr.strip().lower()
+
+
+def primary_from_address(from_header: str) -> str:
+    """从 From 头提取首选回复地址（姓名 <email> 或纯 email）。"""
+    addrs = getaddresses([from_header])
+    for _name, addr in addrs:
+        if addr:
+            return addr
+    _name, addr = parseaddr(from_header)
+    return addr
+
+
+def format_address_list(addrs: list[tuple[str, str]]) -> str:
+    parts = []
+    for name, addr in addrs:
+        if not addr:
+            continue
+        if name:
+            parts.append(f"{name} <{addr}>")
+        else:
+            parts.append(addr)
+    return ", ".join(parts)
+
+
+def resolve_reply_recipients(
+    orig: dict,
+    own_email: str,
+    reply_all: bool = False,
+    to_override: str | None = None,
+    cc_override: str | None = None,
+) -> tuple[str, str | None]:
+    """计算回复收件人。返回 (to, cc)。"""
+    if to_override:
+        to_addr = to_override
+    else:
+        to_addr = primary_from_address(orig["from_raw"])
+        if not to_addr:
+            raise ValueError(t("error.no_reply_address"))
+
+    if cc_override is not None:
+        return to_addr, cc_override or None
+
+    if not reply_all:
+        return to_addr, None
+
+    own = _normalize_email(own_email)
+    reply_to = _normalize_email(to_addr)
+    seen = {reply_to, own}
+    cc_addrs: list[tuple[str, str]] = []
+
+    for raw in (orig.get("to_raw") or "", orig.get("cc_raw") or ""):
+        for name, addr in getaddresses([raw]):
+            if not addr or _normalize_email(addr) in seen:
+                continue
+            seen.add(_normalize_email(addr))
+            cc_addrs.append((name, addr))
+
+    cc = format_address_list(cc_addrs) if cc_addrs else None
+    return to_addr, cc
+
+
+def forward_subject(subject: str) -> str:
+    s = subject.strip()
+    lowered = s.lower()
+    if lowered.startswith("fwd:") or s.startswith("转发:"):
+        return s
+    return f"Fwd: {s}"
+
+
+def _quote_block_plain(orig: dict, title: str) -> list[str]:
+    if title == t("output.original_mail"):
+        return [
+            f"---------- {title} ----------",
+            t("output.from", from_=orig['from']),
+            t("output.send_time", date=orig['date']),
+            t("output.to", to=orig['to']),
+            t("output.subject_label", subject=orig['subject']),
+            "",
+            orig["body"],
+        ]
+    return [
+        f"---------- {title} ----------",
+        t("output.from", from_=orig['from']),
+        t("output.to", to=orig['to']),
+        t("output.date", date=orig['date']),
+        t("output.subject_label", subject=orig['subject']),
+        "",
+        orig["body"],
+    ]
+
+
+def _quote_block_html(orig: dict, title: str) -> str:
+    esc = html_module.escape
+    body = esc(orig["body"])
+    if title == t("output.original_mail"):
+        meta = (
+            f"<p><strong>{t('output.from', from_='')}</strong>{esc(orig['from'])}</p>"
+            f"<p><strong>{t('output.send_time', date='')}</strong>{esc(orig['date'])}</p>"
+            f"<p><strong>{t('output.to', to='')}</strong>{esc(orig['to'])}</p>"
+            f"<p><strong>{t('output.subject_label', subject='')}</strong>{esc(orig['subject'])}</p>"
+        )
+    else:
+        meta = (
+            f"<p><strong>{t('output.from', from_='')}</strong>{esc(orig['from'])}</p>"
+            f"<p><strong>{t('output.to', to='')}</strong>{esc(orig['to'])}</p>"
+            f"<p><strong>{t('output.date', date='')}</strong>{esc(orig['date'])}</p>"
+            f"<p><strong>{t('output.subject_label', subject='')}</strong>{esc(orig['subject'])}</p>"
+        )
+    return (
+        '<div style="border-top: 1px solid #ccc; margin-top: 16px; padding-top: 8px; '
+        'color: #666; font-size: 12px;">'
+        f"<p><strong>{esc(title)}</strong></p>{meta}"
+        f'<pre style="white-space: pre-wrap; font-family: inherit; color: #333;">{body}</pre>'
+        "</div>"
+    )
+
+
+def build_reply_body(reply_text: str, orig: dict, is_html: bool = False) -> str:
+    if is_html:
+        parts = [reply_text] if reply_text else []
+        parts.append(_quote_block_html(orig, t("output.original_mail")))
+        return "".join(parts)
+    lines: list[str] = []
+    if reply_text:
+        lines.extend([reply_text, ""])
+    lines.extend(_quote_block_plain(orig, t("output.original_mail")))
+    return "\n".join(lines)
+
+
+def build_forward_body(note: str, orig: dict, is_html: bool = False) -> str:
+    if is_html:
+        parts = [note] if note else []
+        parts.append(_quote_block_html(orig, t("output.forwarded_mail")))
+        return "".join(parts)
+    lines: list[str] = []
+    if note:
+        lines.extend([note, ""])
+    lines.extend(_quote_block_plain(orig, t("output.forwarded_mail")))
+    return "\n".join(lines)
+
+
+def preview_outgoing(to: str, cc: str | None, subject: str, body: str, title: str | None = None) -> None:
+    if title is None:
+        title = t("output.forward_preview_title")
+    print("=" * 60)
+    print(f"【{title}】")
+    print("=" * 60)
+    print(f"  {t('output.recipient', to=to)}")
+    if cc:
+        print(f"  {t('output.cc', cc=cc)}")
+    print(f"  {t('output.subject', subject=subject)}")
+    print("-" * 60)
+    print(body)
+    print("=" * 60)
+
+
+def extract_attachments(eml_path: str) -> tuple[list[tuple[str, str]], str]:
+    """从 .eml 文件中提取真正的附件。
+
+    返回 (附件列表, 临时目录路径)。附件列表元素为 (临时文件路径, 原始文件名)。
+    调用方负责在完成后删除临时目录。
+    """
+    attachments: list[tuple[str, str]] = []
+    tmp_dir = tempfile.mkdtemp(prefix="email-fwd-attach-")
+
+    with open(eml_path, "rb") as fp:
+        msg = email.message_from_binary_file(fp)
+
+    for part in msg.walk():
+        if part.is_multipart():
+            continue
+
+        content_disposition = part.get("Content-Disposition", "")
+        # 仅转发明确标记为 attachment 的部分；跳过 inline 图片等内嵌资源
+        if "attachment" not in content_disposition.lower():
+            continue
+
+        filename = part.get_filename()
+        if not filename:
+            continue
+
+        payload = part.get_payload(decode=True)
+        if not payload:
+            continue
+
+        safe_name = re.sub(r'[\\/:*?"<>|]', "_", filename)
+        tmp_path = Path(tmp_dir) / safe_name
+        counter = 1
+        stem = tmp_path.stem
+        suffix = tmp_path.suffix
+        while tmp_path.exists():
+            tmp_path = Path(tmp_dir) / f"{stem}_{counter}{suffix}"
+            counter += 1
+
+        tmp_path.write_bytes(payload)
+        attachments.append((str(tmp_path), filename))
+
+    return attachments, tmp_dir
+
+
+def confirm_send() -> bool:
+    try:
+        answer = input(f"\n{t('output.confirm_send')}").strip().lower()
+        return answer == "y"
+    except (EOFError, KeyboardInterrupt):
+        return False
+
+
+def confirm_forward_with_attachments(attachments: list[tuple[str, str]]) -> bool:
+    """转发含附件邮件时的确认：先列清单，再询问是否发送。"""
+    print(f"\n{t('output.forward_attach_confirm')}")
+    for _path, filename in attachments:
+        print(f"  • {filename}")
+    return confirm_send()