Gemini API のニッチ SaaS で「請求とコストが噛み合わない」を消す — 従量原価の計測と Stripe 突合の運用メモ

// 自分の契約している料金表から、100万トークンあたりの単価（円）を定数化する。
// 実際の数値は Google の料金ページで必ず確認し、改定があれば更新すること。
const PRICE_PER_MTOK = {
  input: 4.5,        // 通常の入力
  cachedInput: 1.1,  // キャッシュヒットした入力（大幅に安い）
  output: 18.0,      // 出力
} as const;
 
const JPY_PER_USD = 158; // 為替は月次でまとめて見直す
 
function costJpy(u: {
  promptTokenCount: number;
  cachedContentTokenCount?: number;
  candidatesTokenCount: number;
}): number {
  const fresh = u.promptTokenCount - (u.cachedContentTokenCount ?? 0);
  const usd =
    (fresh / 1_000_000) * PRICE_PER_MTOK.input +
    ((u.cachedContentTokenCount ?? 0) / 1_000_000) * PRICE_PER_MTOK.cachedInput +
    (u.candidatesTokenCount / 1_000_000) * PRICE_PER_MTOK.output;
  return usd * JPY_PER_USD;
}

async function recordCost(env: Env, userId: string, jpy: number, usage: object) {
  const month = new Date().toISOString().slice(0, 7); // "2026-06"
  const key = `cost:${userId}:${month}`;
  const prev = parseFloat((await env.KV.get(key)) ?? "0");
  const next = +(prev + jpy).toFixed(4);
  // 翌々月まで保持して、月次突合のときに参照できるようにする
  await env.KV.put(key, String(next), { expirationTtl: 60 * 60 * 24 * 70 });
 
  // 監査用に1リクエスト1行のログも残す（Workers Analytics Engine や R2 への追記でよい）
  await env.LEDGER.writeDataPoint?.({
    blobs: [userId, month],
    doubles: [jpy],
    indexes: [userId],
  });
}

const res = await ai.models.generateContent({ model: "gemini-2.5-flash", contents });
const u = res.usageMetadata;
const jpy = costJpy({
  promptTokenCount: u.promptTokenCount,
  cachedContentTokenCount: u.cachedContentTokenCount,
  candidatesTokenCount: u.candidatesTokenCount,
});
await recordCost(env, userId, jpy, u);

async function reconcile(env: Env, stripe: Stripe, month: string) {
  const rows: { userId: string; revenue: number; cost: number; margin: number }[] = [];
 
  // 当月に paid になった請求を集計（ページングは省略）
  const invoices = await stripe.invoices.list({ status: "paid", limit: 100 });
  const revenueByUser = new Map<string, number>();
  for (const inv of invoices.data) {
    const userId = inv.metadata?.app_user_id;
    if (!userId) continue;
    const jpy = inv.currency === "jpy" ? inv.amount_paid : inv.amount_paid / 100 * JPY_PER_USD;
    revenueByUser.set(userId, (revenueByUser.get(userId) ?? 0) + jpy);
  }
 
  for (const [userId, revenue] of revenueByUser) {
    const cost = parseFloat((await env.KV.get(`cost:${userId}:${month}`)) ?? "0");
    rows.push({ userId, revenue, cost, margin: revenue - cost });
  }
  rows.sort((a, b) => a.margin - b.margin); // 赤字（マージン最小）が先頭に来る
  return rows;
}

async function maybeAlert(env: Env, userId: string, monthlyPriceJpy: number) {
  const month = new Date().toISOString().slice(0, 7);
  const cost = parseFloat((await env.KV.get(`cost:${userId}:${month}`)) ?? "0");
  const ratio = cost / monthlyPriceJpy;
  if (ratio < 0.7) return;
 
  // 同じユーザーに同じ月で何度も鳴らさないよう、通知済みフラグを置く
  const flag = `alert:${userId}:${month}`;
  if (await env.KV.get(flag)) return;
  await env.KV.put(flag, "1", { expirationTtl: 60 * 60 * 24 * 35 });
 
  await notify(env, `⚠️ ${userId}: 原価比 ${(ratio * 100).toFixed(0)}%（¥${cost.toFixed(0)} / ¥${monthlyPriceJpy}）`);
}

async function handleStripeEvent(env: Env, event: Stripe.Event) {
  const seen = `evt:${event.id}`;
  // 既に見たイベントなら即終了（KV の存在チェックで冪等化）
  if (await env.KV.get(seen)) return;
  await env.KV.put(seen, "1", { expirationTtl: 60 * 60 * 24 * 30 });
 
  if (event.type === "invoice.paid") {
    const inv = event.data.object as Stripe.Invoice;
    const userId = inv.metadata?.app_user_id;
    if (userId) {
      await env.KV.put(
        `member:${userId}`,
        JSON.stringify({ plan: "pro", expiresAt: Date.now() + 35 * 864e5 }),
        { expirationTtl: 60 * 60 * 24 * 40 }
      );
    }
  }
}