Rork Max × Stripe サブスクリプションを本番運用へ — Webhook冪等性・状態遷移・照合の実装

-- Supabase の例
CREATE TABLE subscriptions (
  id UUID PRIMARY KEY DEFAULT uuid_generate_v4(),
  user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
  stripe_subscription_id TEXT UNIQUE NOT NULL,
  stripe_customer_id TEXT NOT NULL,
  plan_id TEXT NOT NULL,
  status TEXT NOT NULL, -- 'active','trialing','past_due','canceled','incomplete'
  current_period_end TIMESTAMP NOT NULL,
  cancel_at_period_end BOOLEAN DEFAULT FALSE,
  updated_at TIMESTAMP DEFAULT NOW()
);
 
-- イベント台帳: Webhookの冪等処理に使う
CREATE TABLE stripe_events (
  event_id TEXT PRIMARY KEY,        -- Stripe の evt_xxx
  type TEXT NOT NULL,
  processed_at TIMESTAMP DEFAULT NOW()
);

// Rork Max Webhook Handler（冪等版）
// エンドポイント: /webhooks/stripe
 
exports.handler = async (event) => {
  const sig = event.headers['stripe-signature'];
 
  // 1. 署名検証: 生ボディで検証する（パース済みJSONでは失敗する）
  let stripeEvent;
  try {
    stripeEvent = stripe.webhooks.constructEvent(
      event.body, sig, process.env.STRIPE_WEBHOOK_SECRET
    );
  } catch (err) {
    // 検証失敗は 400 を返す。ここで 200 を返すとStripeが再送を止めてしまう
    return { statusCode: 400, body: `Signature verification failed` };
  }
 
  // 2. 冪等ガード: 既に処理済みなら何もせず 200
  const inserted = await insertEventIfNew(stripeEvent.id, stripeEvent.type);
  if (!inserted) {
    return { statusCode: 200, body: JSON.stringify({ duplicate: true }) };
  }
 
  // 3. 種別ごとに処理（副作用はここだけ）
  try {
    switch (stripeEvent.type) {
      case 'customer.subscription.created':
      case 'customer.subscription.updated':
        await upsertSubscription(stripeEvent.data.object);
        break;
      case 'customer.subscription.deleted':
        await markCanceled(stripeEvent.data.object);
        break;
      case 'invoice.payment_failed':
        await handlePaymentFailed(stripeEvent.data.object);
        break;
    }
  } catch (e) {
    // 処理に失敗したら台帳から削除し、Stripeの再送に委ねる
    await deleteEvent(stripeEvent.id);
    return { statusCode: 500, body: 'processing failed' };
  }
 
  return { statusCode: 200, body: JSON.stringify({ received: true }) };
};
 
// insertEventIfNew: INSERT ... ON CONFLICT DO NOTHING の結果で真偽を返す
async function insertEventIfNew(id, type) {
  const res = await db.query(
    `INSERT INTO stripe_events (event_id, type) VALUES ($1, $2)
     ON CONFLICT (event_id) DO NOTHING RETURNING event_id`,
    [id, type]
  );
  return res.rowCount === 1; // 1なら新規、0なら重複
}

async function upsertSubscription(subObject) {
  // イベント内の値ではなく、最新をStripeから取得する
  const sub = await stripe.subscriptions.retrieve(subObject.id);
 
  // 期間フィールドは新しいAPIバージョンで items 側へ移動している
  const periodEnd = sub.items.data[0]?.current_period_end
    ?? sub.current_period_end;
 
  await db.query(
    `INSERT INTO subscriptions
       (user_id, stripe_subscription_id, stripe_customer_id, plan_id, status, current_period_end, cancel_at_period_end)
     VALUES ($1,$2,$3,$4,$5,$6,$7)
     ON CONFLICT (stripe_subscription_id) DO UPDATE
       SET status=$5, current_period_end=$6, cancel_at_period_end=$7, updated_at=NOW()`,
    [sub.metadata.userId, sub.id, sub.customer, sub.items.data[0].price.id,
     sub.status, new Date(periodEnd * 1000), sub.cancel_at_period_end]
  );
}

const GRANT_STATES = new Set(['trialing', 'active', 'past_due']);
 
function canAccess(subscription) {
  if (GRANT_STATES.has(subscription.status)) return true;
  // 解約予約中でも期末まではアクセス可
  if (subscription.status === 'canceled'
      && subscription.current_period_end > Date.now() / 1000) {
    return true;
  }
  return false;
}

// プラン変更（日割りあり）
async function updatePlan(subscriptionId, newPriceId) {
  const sub = await stripe.subscriptions.retrieve(subscriptionId);
  return stripe.subscriptions.update(subscriptionId, {
    items: [{ id: sub.items.data[0].id, price: newPriceId }],
    proration_behavior: 'create_prorations'
  });
}
 
// 期末解約（即時停止ではなく予約）
async function cancelAtPeriodEnd(subscriptionId) {
  return stripe.subscriptions.update(subscriptionId, {
    cancel_at_period_end: true
  });
}

# 手元のエンドポイントへイベントを転送（署名シークレットが払い出される）
stripe listen --forward-to localhost:3000/webhooks/stripe
 
# 別ターミナルで任意のイベントを発火
stripe trigger invoice.payment_failed
stripe trigger customer.subscription.deleted