memsave 0.2.0__py3-none-any.whl

memsave/__init__.py

@@ -0,0 +1,12 @@
+"""memsave — read your memories without spending credits.
+
+Retrieval runs entirely on local CPU: SQLite FTS5/BM25 for keywords, an
+optional 24MB ONNX model for semantics. Only the handful of tokens that
+actually matched get handed to the model.
+"""
+
+from .store import Memory, Hit, Store
+from .budget import estimate_tokens
+
+__version__ = "0.2.0"
+__all__ = ["Memory", "Hit", "Store", "estimate_tokens"]

memsave/__main__.py

@@ -0,0 +1,3 @@
+from .cli import main
+
+raise SystemExit(main())

memsave/afdian.py

@@ -0,0 +1,310 @@
+"""爱发电（afdian.com）收款通道。
+
+## 这里有个必须先说清楚的事实
+
+**爱发电的 webhook 没有任何签名。** 官方文档里的签名机制只用于「你主动调用 API」，
+不用于「平台 POST 给你的 webhook」。也就是说：
+
+> 任何人只要拿到你的 webhook URL，就能 POST 一个假订单，骗走一把 Pro 授权码。
+
+URL 保密不算防御 —— 它会出现在服务器日志、CDN 日志、浏览器历史里。
+
+所以本模块的设计核心是：**收到 webhook 后用带签名的 API 反向回查订单。**
+伪造者 POST 得再像真的，他也过不了 API 回查那一关，因为回查要你的 api token。
+
+官方文档自己也建议这么做：「建议结合 API 一起使用」。
+
+## 另一个必须挡住的坑
+
+创作者通常还有一个低价的月度订阅方案（比如「请作者喝杯咖啡 ¥5/月」）。
+**订阅订单也会推送到同一个 webhook。** 不做区分的话，
+一个每月付 5 元的咖啡党，会收到一把 ¥99 的 Pro 授权码。
+
+所以每个订单都要过三道闸：类型、商品名、金额。
+"""
+
+from __future__ import annotations
+
+import hashlib
+import json
+import os
+import re
+import time
+import urllib.error
+import urllib.request
+
+API_BASE = "https://afdian.com/api/open"
+
+# 订单必须命中这些商品名才会发码。小写匹配，顺序无关。
+# 在 billing.env 里用 AFDIAN_SKU_PRO / AFDIAN_SKU_TEAM 覆盖。
+DEFAULT_SKUS = {"pro": ("pro", "终身", "lifetime"), "team": ("team", "团队")}
+
+# 各 tier 的最低成交金额（元）。低于此值不发码 —— 挡住 0 元/折扣/咖啡订单。
+# 在 billing.env 里用 AFDIAN_MIN_AMOUNT_PRO / AFDIAN_MIN_AMOUNT_TEAM 覆盖。
+DEFAULT_MIN_AMOUNT = {"pro": 79.0, "team": 199.0}
+
+EMAIL_RE = re.compile(r"[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Za-z]{2,}")
+
+# 订单状态。文档说目前仅推送 status=2（交易成功），但保险起见还是要查。
+STATUS_PAID = 2
+
+
+class AfdianError(Exception):
+    pass
+
+
+def _env(name: str, default: str = "") -> str:
+    return os.environ.get(name, default).strip()
+
+
+def _skus_for(tier: str) -> tuple[str, ...]:
+    raw = _env(f"AFDIAN_SKU_{tier.upper()}")
+    if raw:
+        return tuple(s.strip().lower() for s in raw.split(",") if s.strip())
+    return DEFAULT_SKUS.get(tier, ())
+
+
+def _min_amount(tier: str) -> float:
+    raw = _env(f"AFDIAN_MIN_AMOUNT_{tier.upper()}")
+    if raw:
+        try:
+            return float(raw)
+        except ValueError:
+            pass
+    return DEFAULT_MIN_AMOUNT.get(tier, 0.0)
+
+
+# --------------------------------------------------------------------- 签名
+
+def sign(token: str, params: str, ts: int, user_id: str) -> str:
+    """爱发电 API 签名。
+
+    sign = md5(token + "params" + params + "ts" + ts + "user_id" + user_id)
+
+    注意 key 和 value 直接拼接，没有任何连接符，value 也要原样带上 key 名。
+    文档给出的向量：
+        sign('123', '{"a":333}', 1624339905, 'abc')
+        == 'a4acc28b81598b7e5d84ebdc3e91710c'
+    """
+    raw = f"{token}params{params}ts{ts}user_id{user_id}"
+    return hashlib.md5(raw.encode("utf-8")).hexdigest()
+
+
+def signed_body(user_id: str, token: str, params: dict, ts: int | None = None) -> dict:
+    """构造一个可直接 POST 的请求体。params 会被序列化成紧凑 JSON。"""
+    s = json.dumps(params, separators=(",", ":"), ensure_ascii=False)
+    ts = int(ts if ts is not None else time.time())
+    return {
+        "user_id": user_id,
+        "params": s,
+        "ts": ts,
+        "sign": sign(token, s, ts, user_id),
+    }
+
+
+def _api(endpoint: str, body: dict, timeout: float = 15.0) -> dict:
+    req = urllib.request.Request(
+        f"{API_BASE}/{endpoint}",
+        data=json.dumps(body, separators=(",", ":"), ensure_ascii=False).encode("utf-8"),
+        headers={
+            "Content-Type": "application/json",
+            "User-Agent": "memsave-billing/0.2",
+        },
+        method="POST",
+    )
+    try:
+        with urllib.request.urlopen(req, timeout=timeout) as resp:
+            return json.loads(resp.read().decode("utf-8"))
+    except (urllib.error.URLError, TimeoutError, json.JSONDecodeError) as exc:
+        raise AfdianError(f"{endpoint} 请求失败: {exc}") from exc
+
+
+def ping() -> dict:
+    """校验 user_id + token 是否配对。返回平台原始响应。"""
+    user_id, token = credentials()
+    return _api("ping", signed_body(user_id, token, {"a": 333}))
+
+
+def credentials() -> tuple[str, str]:
+    user_id = _env("AFDIAN_USER_ID")
+    token = _env("AFDIAN_API_TOKEN")
+    if not user_id or not token:
+        raise AfdianError("未配置 AFDIAN_USER_ID / AFDIAN_API_TOKEN")
+    return user_id, token
+
+
+def query_order(out_trade_no: str) -> dict | None:
+    """按订单号回查。三种结果必须分清楚，否则整个防伪造机制失效：
+
+    - 找到了 -> 返回订单 dict
+    - 确实没这个订单（ec=200 但列表为空）-> None，调用方当作伪造拦截
+    - **平台报错 / 签名被拒 / 网络故障** -> 抛 AfdianError
+
+    第三点是最关键的。早期版本把「API 报错」和「订单不存在」都压成 None，
+    后果是：token 配错了没人知道，客户付了钱、码没发、日志干干净净。
+    现在报错必须向上抛，让调用方返回 500 触发平台重试，并在日志里炸出来。
+    """
+    user_id, token = credentials()
+    resp = _api("query-order", signed_body(user_id, token, {"out_trade_no": out_trade_no}))
+    ec = resp.get("ec")
+    if ec != 200:
+        raise AfdianError(f"query-order 返回 ec={ec} em={resp.get('em', '')!r}"
+                          f"（检查 AFDIAN_USER_ID / AFDIAN_API_TOKEN 是否配对）")
+    data = resp.get("data") or {}
+    for order in data.get("list") or []:
+        if order.get("out_trade_no") == out_trade_no:
+            return order
+    return None
+
+
+def fetch_orders(pages: int = 3) -> list[dict]:
+    """按分页拉取订单列表（不传 out_trade_no 就是查列表）。
+
+    存在的理由：webhook 是"推送"，推送的前提是我们的服务当时在线。
+    本机跑 + 免费隧道 = 电脑关机就收不到。所以对账必须能反过来
+    「主动去平台把订单拉回来」，否则漏单就是漏钱，而且客户不知道自己漏了。
+
+    配合 billing.reconcile 使用：拉取 → 逐单走同一套闸门和幂等账本 →
+    只补发本地没有的那些。
+    """
+    user_id, token = credentials()
+    out: list[dict] = []
+    total: int | None = None
+    for page in range(1, max(1, pages) + 1):
+        resp = _api("query-order", signed_body(user_id, token, {"page": page}))
+        if resp.get("ec") != 200:
+            raise AfdianError(
+                f"query-order 返回 ec={resp.get('ec')} em={resp.get('em', '')!r}"
+                f"（检查 AFDIAN_USER_ID / AFDIAN_API_TOKEN 是否配对）"
+            )
+        data = resp.get("data") or {}
+        if total is None:
+            try:
+                total = int(data.get("total_count") or 0)
+            except (TypeError, ValueError):
+                total = 0
+        lst = data.get("list") or []
+        out.extend(lst)
+        if not lst or (total and len(out) >= total):
+            break
+    return out
+
+
+# ----------------------------------------------------------------- webhook
+
+def parse_webhook(payload: dict) -> dict | None:
+    """从 webhook body 里抽出我们要的东西。
+
+    不是订单事件（文档说目前只有 order，但以后可能扩展）就返回 None。
+    """
+    data = payload.get("data") or {}
+    if data.get("type") != "order":
+        return None
+    order = data.get("order") or {}
+    if not order.get("out_trade_no"):
+        return None
+
+    skus = [
+        (s.get("name") or "").strip()
+        for s in (order.get("sku_detail") or [])
+        if isinstance(s, dict)
+    ]
+    return {
+        "order_id": str(order.get("out_trade_no")),
+        "buyer_id": str(order.get("user_id") or ""),
+        "remark": str(order.get("remark") or ""),
+        "amount": _to_float(order.get("total_amount")),
+        "status": int(order.get("status") or 0),
+        "product_type": int(order.get("product_type") or 0),
+        "skus": skus,
+        # 自定义信息收集（下单必填的「邮箱」字段）落在哪，官方文档没写死。
+        # 社区实测它出现在 custom_order_id；保险起见把所有可能藏文本的
+        # 字段都带回去，extract_email 逐个试。
+        "custom_order_id": str(order.get("custom_order_id") or ""),
+        "address_person": str(order.get("address_person") or ""),
+        "address_phone": str(order.get("address_phone") or ""),
+        "address_address": str(order.get("address_address") or ""),
+    }
+
+
+def extract_email_from(order: dict) -> str | None:
+    """按可靠度排序，从订单的多个文本字段里抠邮箱。
+
+    优先级：custom_order_id（自定义信息收集的必填邮箱，最可靠）
+         > remark（留言，商品说明里要求填的兜底）
+         > 收件人字段（几乎不会有人把邮箱填这里，但看一眼不花钱）
+    """
+    for key in ("custom_order_id", "remark", "address_person", "address_address"):
+        email = extract_email(order.get(key, ""))
+        if email:
+            return email
+    return None
+
+
+def _to_float(v) -> float:
+    try:
+        return float(v)
+    except (TypeError, ValueError):
+        return 0.0
+
+
+def extract_email(text: str) -> str | None:
+    """爱发电订单里没有邮箱字段，只有昵称。
+
+    所以商品的购买说明里必须写清楚「下单请在留言栏填邮箱」，
+    然后这里从留言里把它抠出来。这是社区里验证过的做法。
+    """
+    m = EMAIL_RE.search(text or "")
+    return m.group(0).lower() if m else None
+
+
+def tier_for(skus: list[str]) -> str | None:
+    """按 sku 商品名匹配 tier。命中多个时返回价高的那个（Pro 优先于终身字样）。"""
+    hit: list[str] = []
+    joined = " ".join(skus).lower()
+    for tier in ("team", "pro"):
+        if any(k in joined for k in _skus_for(tier)):
+            hit.append(tier)
+    if not hit:
+        return None
+    # 同时命中两档时按高价档发，宁可多发也别少发
+    return "team" if "team" in hit else hit[0]
+
+
+def gate(order: dict, skus: list[str]) -> tuple[bool, str]:
+    """三道闸：类型 → 商品名 → 金额。返回 (是否放行, 原因)。
+
+    挡住两类误发：
+      - 月度订阅的「咖啡党」订单（product_type=0，没有 sku）
+      - 折扣/兑换码导致的低价单
+    """
+    if order.get("status", 0) != STATUS_PAID:
+        return False, f"订单状态 {order.get('status')} 非已支付"
+
+    tier = tier_for(skus)
+    if tier is None:
+        return False, "商品名未命中任何授权档位（可能是月度订阅或无关商品）"
+
+    if not skus:
+        return False, "无 sku_detail，视为订阅订单，不发码"
+
+    floor = _min_amount(tier)
+    if order.get("amount", 0.0) < floor:
+        return False, f"金额 {order.get('amount')} 低于 {tier} 档下限 {floor}"
+
+    return True, tier
+
+
+def ok_response() -> dict:
+    """平台要求的成功响应格式。不返回 ec=200 它会当失败并重推。"""
+    return {"ec": 200, "em": ""}
+
+
+def fail_response() -> dict:
+    """明确告诉平台这次失败了，让它重试。
+
+    只在「我们确实没处理成功且重试有希望」时用。
+    对于伪造订单，应返回 200 —— 否则伪造者能靠反复重推把服务器打满，
+    而重推一万次也变不成真订单。
+    """
+    return {"ec": 400, "em": "retry"}