@downcity/services 0.1.67 → 0.1.69

package/bin/usage/service.js

@@ -0,0 +1,140 @@
+/**
+ * Downcity 官方 Usage 服务。
+ *
+ * 通过全局 hook 记录 service 调用事件。
+ * 兼容 Node.js 和 Cloudflare Workers（使用 Web Crypto API）。
+ */
+import { sqliteTable, text } from "drizzle-orm/sqlite-core";
+import { InstallableService } from "@downcity/city";
+/**
+ * Usage 事件表。
+ */
+export const usageEvents = sqliteTable("service_usage_events", {
+    event_id: text("event_id").primaryKey(),
+    town_id: text("town_id").notNull(),
+    user_id: text("user_id").notNull(),
+    service: text("service").notNull(),
+    model_id: text("model_id").notNull(),
+    status: text("status").notNull(),
+    metadata_json: text("metadata_json").notNull(),
+    created_at: text("created_at").notNull(),
+});
+/**
+ * Usage 服务实例。
+ */
+export class UsageService extends InstallableService {
+    options;
+    id = "usage";
+    name = "Usage";
+    version = "0.1.0";
+    schema = { events: usageEvents };
+    constructor(options = {}) {
+        super();
+        this.options = options;
+        this.instruction = [
+            "通过全局 hook 记录真实用户侧 service 调用事件。",
+            "默认只记录成功调用；record_errors=true 时也会记录失败调用。",
+            "常用读取方式是管理端查看 events/summary，用户侧查看 me。",
+        ].join("\n");
+    }
+    install(ctx) {
+        const events = ctx.table("events");
+        ctx.hook.after(async (serviceCtx) => {
+            if (!shouldRecordUsage(serviceCtx))
+                return;
+            await events.insert(createUsageEvent(serviceCtx, "success"));
+        });
+        if (this.options.record_errors) {
+            ctx.hook.onError(async (serviceCtx) => {
+                if (!shouldRecordUsage(serviceCtx))
+                    return;
+                await events.insert(createUsageEvent(serviceCtx, "error"));
+            });
+        }
+        ctx.route({
+            method: "GET",
+            path: "/events",
+            auth: ["admin"],
+            async handler(requestCtx) {
+                return requestCtx.jsonResponse({ items: await events.select() });
+            },
+        });
+        ctx.route({
+            method: "GET",
+            path: "/summary",
+            auth: ["admin"],
+            async handler(requestCtx) {
+                return requestCtx.jsonResponse({
+                    items: summarizeUsage(await events.select()),
+                });
+            },
+        });
+        ctx.route({
+            method: "GET",
+            path: "/me",
+            auth: ["user"],
+            async handler(requestCtx) {
+                return requestCtx.jsonResponse({
+                    items: await events.select({
+                        user_id: requestCtx.user?.user_id ?? "",
+                        town_id: requestCtx.town?.town_id ?? "",
+                    }),
+                });
+            },
+        });
+    }
+}
+/**
+ * 只记录真实用户侧调用。
+ *
+ * 管理端操作没有 user/town 上下文，usage 服务自己的查询也不应反过来
+ * 产生 usage 事件，否则统计接口会污染自身结果。
+ */
+function shouldRecordUsage(ctx) {
+    return Boolean(ctx.user?.user_id && ctx.town?.town_id && ctx.service?.id !== "usage");
+}
+/**
+ * 创建 usage 事件记录。
+ */
+function createUsageEvent(ctx, status) {
+    return {
+        event_id: `usage_${randomId()}`,
+        town_id: ctx.town?.town_id ?? "",
+        user_id: ctx.user?.user_id ?? "",
+        service: ctx.service?.id ?? "",
+        model_id: ctx.variant?.id ?? "",
+        status,
+        metadata_json: JSON.stringify({
+            variant: ctx.variant?.id,
+            started_at: ctx.started_at?.toISOString(),
+            ended_at: ctx.ended_at?.toISOString(),
+            error: ctx.error?.message,
+        }),
+        created_at: new Date().toISOString(),
+    };
+}
+/**
+ * 汇总 usage 事件。
+ */
+function summarizeUsage(rows) {
+    const byKey = new Map();
+    for (const row of rows) {
+        const key = `${row.town_id}\u0000${row.service}\u0000${row.status}`;
+        const current = byKey.get(key) ?? { town_id: row.town_id, service: row.service, status: row.status, count: 0 };
+        current.count += 1;
+        byKey.set(key, current);
+    }
+    return [...byKey.values()].sort((a, b) => `${a.town_id}:${a.service}:${a.status}`.localeCompare(`${b.town_id}:${b.service}:${b.status}`));
+}
+/**
+ * 生成随机 ID（兼容 Node 和 Workers）。
+ */
+function randomId() {
+    const buf = new Uint8Array(12);
+    crypto.getRandomValues(buf);
+    return btoa(String.fromCharCode(...buf))
+        .replace(/\+/g, "-")
+        .replace(/\//g, "_")
+        .replace(/=+$/u, "");
+}
+//# sourceMappingURL=service.js.map

package/bin/usage/service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.js","sourceRoot":"","sources":["../../src/usage/service.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAA4C,MAAM,gBAAgB,CAAC;AAc9F;;GAEG;AACH,MAAM,CAAC,MAAM,WAAW,GAAG,WAAW,CAAC,sBAAsB,EAAE;IAC7D,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,UAAU,EAAE;IACvC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;IAClC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;IAClC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;IAClC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE;IACpC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE;IAChC,aAAa,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,EAAE;IAC9C,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE;CACzC,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,OAAO,YAAa,SAAQ,kBAAkB;IAMrB;IALpB,EAAE,GAAG,OAAO,CAAC;IACb,IAAI,GAAG,OAAO,CAAC;IACf,OAAO,GAAG,OAAO,CAAC;IAClB,MAAM,GAAG,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;IAE1C,YAA6B,UAA+B,EAAE;QAC5D,KAAK,EAAE,CAAC;QADmB,YAAO,GAAP,OAAO,CAA0B;QAE5D,IAAI,CAAC,WAAW,GAAG;YACjB,iCAAiC;YACjC,yCAAyC;YACzC,uCAAuC;SACxC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC;IAED,OAAO,CAAC,GAA0B;QAChC,MAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAgB,QAAQ,CAAC,CAAC;QAElD,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE;YAClC,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC;gBAAE,OAAO;YAC3C,MAAM,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YAC/B,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE;gBACpC,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC;oBAAE,OAAO;gBAC3C,MAAM,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;YAC7D,CAAC,CAAC,CAAC;QACL,CAAC;QAED,GAAG,CAAC,KAAK,CAAC;YACR,MAAM,EAAE,KAAK;YACb,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,CAAC,OAAO,CAAC;YACf,KAAK,CAAC,OAAO,CAAC,UAAU;gBACtB,OAAO,UAAU,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACnE,CAAC;SACF,CAAC,CAAC;QAEH,GAAG,CAAC,KAAK,CAAC;YACR,MAAM,EAAE,KAAK;YACb,IAAI,EAAE,UAAU;YAChB,IAAI,EAAE,CAAC,OAAO,CAAC;YACf,KAAK,CAAC,OAAO,CAAC,UAAU;gBACtB,OAAO,UAAU,CAAC,YAAY,CAAC;oBAC7B,KAAK,EAAE,cAAc,CAAC,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;iBAC7C,CAAC,CAAC;YACL,CAAC;SACF,CAAC,CAAC;QAEH,GAAG,CAAC,KAAK,CAAC;YACR,MAAM,EAAE,KAAK;YACb,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,CAAC,MAAM,CAAC;YACd,KAAK,CAAC,OAAO,CAAC,UAAU;gBACtB,OAAO,UAAU,CAAC,YAAY,CAAC;oBAC7B,KAAK,EAAE,MAAM,MAAM,CAAC,MAAM,CAAC;wBACzB,OAAO,EAAE,UAAU,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;wBACvC,OAAO,EAAE,UAAU,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;qBACxC,CAAC;iBACH,CAAC,CAAC;YACL,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAED;;;;;GAKG;AACH,SAAS,iBAAiB,CAAC,GAAY;IACrC,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,IAAI,GAAG,CAAC,IAAI,EAAE,OAAO,IAAI,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,OAAO,CAAC,CAAC;AACxF,CAAC;AAgBD;;GAEG;AACH,SAAS,gBAAgB,CAAC,GAAY,EAAE,MAA2B;IACjE,OAAO;QACL,QAAQ,EAAE,SAAS,QAAQ,EAAE,EAAE;QAC/B,OAAO,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;QAChC,OAAO,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,IAAI,EAAE;QAChC,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE;QAC9B,QAAQ,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE;QAC/B,MAAM;QACN,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC;YAC5B,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE;YACxB,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,WAAW,EAAE;YACzC,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,WAAW,EAAE;YACrC,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,OAAO;SAC1B,CAAC;QACF,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACrC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,IAAqB;IAC3C,MAAM,KAAK,GAAG,IAAI,GAAG,EAA+E,CAAC;IACrG,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,OAAO,SAAS,GAAG,CAAC,OAAO,SAAS,GAAG,CAAC,MAAM,EAAE,CAAC;QACpE,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QAC/G,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC;QACnB,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC;IACD,OAAO,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACvC,GAAG,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC,CAC/F,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,QAAQ;IACf,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,EAAE,CAAC,CAAC;IAC/B,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;IAC5B,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,GAAG,CAAC,CAAC;SACrC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;SACnB,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;SACnB,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;AACzB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@downcity/services",
-  "version": "0.1.67",
+  "version": "0.1.69",
   "description": "Downcity public services package for accounts, balance, usage, Stripe, Creem, Dodo, and Waffo payment flows.",
   "type": "module",
   "main": "./bin/index.js",
@@ -50,7 +50,7 @@
   },
   "dependencies": {
     "@waffo/pancake-ts": "^0.11.0",
-    "@downcity/city": "0.2.75",
+    "@downcity/city": "0.2.78",
     "better-auth": "^1.6.12",
     "dodopayments": "^2.36.0",
     "drizzle-orm": "^0.45.2"