Rork の AI 画像生成アプリで「クレジットが合わない」を消す — 原子的な台帳消費とモデレーションの実装メモ

create or replace function consume_credits(
  p_user_id uuid,
  p_amount  integer,
  p_reason  text,
  p_idem_key text
) returns integer            -- 消費後の残高を返す
language plpgsql
as $$
declare
  v_balance integer;
begin
  -- 1) 冪等チェック: 同じキーが既にあれば再消費せず現在残高を返す
  if exists (
    select 1 from credit_ledger
    where user_id = p_user_id and idem_key = p_idem_key
  ) then
    return (select coalesce(sum(delta),0) from credit_ledger where user_id = p_user_id);
  end if;
 
  -- 2) 同一ユーザーの消費を直列化（advisory lock で行を奪い合わせない）
  perform pg_advisory_xact_lock(hashtext(p_user_id::text));
 
  v_balance := (select coalesce(sum(delta),0) from credit_ledger where user_id = p_user_id);
  if v_balance < p_amount then
    raise exception 'INSUFFICIENT_CREDITS' using errcode = 'P0001';
  end if;
 
  -- 3) 消費を 1 行記録（ここが課金の確定点）
  insert into credit_ledger (user_id, delta, reason, idem_key)
  values (p_user_id, -p_amount, p_reason, p_idem_key);
 
  return v_balance - p_amount;
end;
$$;

// src/lib/idempotency.ts
import * as Crypto from "expo-crypto";
 
// 1 回の生成試行に対して 1 つだけ発行し、リトライでは使い回す
export function newIdemKey(): string {
  return Crypto.randomUUID();
}

// supabase/functions/generate-image/index.ts（要点のみ）
const { prompt, negativePrompt, style, aspectRatio, idemKey } = await req.json();
const userId = await getUserIdFromJWT(req);      // 認証はサーバで確定する
 
// 1) 先に原子的に消費する（成功して初めて生成に進む）
const { data: balance, error: consumeErr } = await admin.rpc("consume_credits", {
  p_user_id: userId, p_amount: 1, p_reason: "image", p_idem_key: idemKey,
});
if (consumeErr) {
  const code = consumeErr.message.includes("INSUFFICIENT_CREDITS") ? 402 : 500;
  return json({ error: "credits" }, code);
}
 
// 2) 生成を実行する
try {
  const imageUrl = await runFalImage({ prompt, negativePrompt, style, aspectRatio });
  await saveGeneration(userId, idemKey, imageUrl);
  return json({ imageUrl, creditsRemaining: balance });
} catch (e) {
  // 3) 失敗したら同じ冪等キーで必ず返金する（次節）
  await admin.rpc("refund_credits", { p_user_id: userId, p_amount: 1, p_ref: idemKey });
  return json({ error: "generation_failed" }, 502);
}

create or replace function refund_credits(
  p_user_id uuid, p_amount integer, p_ref text
) returns void language plpgsql as $$
begin
  -- 同じ生成に対する二重返金を禁止
  if exists (
    select 1 from credit_ledger
    where user_id = p_user_id and reason = 'refund' and ref_id = p_ref
  ) then
    return;
  end if;
  insert into credit_ledger (user_id, delta, reason, ref_id)
  values (p_user_id, p_amount, 'refund', p_ref);
end;
$$;

// src/hooks/useImageGeneration.ts（キャンセル対応の要点）
const controller = useRef<AbortController | null>(null);
 
async function generate(params: GenerationParams) {
  const idemKey = newIdemKey();
  controller.current = new AbortController();
  try {
    const res = await fetch(GEN_URL, {
      method: "POST",
      signal: controller.current.signal,
      body: JSON.stringify({ ...params, idemKey }),
    });
    // ...結果処理
  } catch (e) {
    if ((e as Error).name === "AbortError") {
      // 表示は止めるが、課金の整合はサーバの返金に委ねる
      await requestServerCancel(idemKey);
    }
  }
}
 
function cancel() {
  controller.current?.abort();
}

// supabase/functions/_shared/moderate.ts
const BLOCK_PATTERNS = [
  /\b(nsfw|explicit|gore)\b/i,
  // 未成年・実在人物の不適切表現などのドメイン固有パターンを追加
];
 
export function screenPrompt(prompt: string): { ok: boolean; reason?: string } {
  for (const re of BLOCK_PATTERNS) {
    if (re.test(prompt)) return { ok: false, reason: "prompt_blocked" };
  }
  return { ok: true };
}

// 出力モデレーションでブロックした場合の整合
const verdict = await classifyImage(imageUrl);   // { nsfw: number }
if (verdict.nsfw > 0.8) {
  await admin.rpc("refund_credits", { p_user_id: userId, p_amount: 1, p_ref: idemKey });
  return json({ error: "blocked_output" }, 422);
}

-- 直近24時間の消費数（image/video）を数えて上限と比較
create or replace function daily_usage(p_user_id uuid) returns integer
language sql stable as $$
  select count(*)::int from credit_ledger
  where user_id = p_user_id
    and reason in ('image','video_hold')
    and created_at > now() - interval '24 hours';
$$;