import { Base, AIService } from "@visiblebase/core";
import { balancePlugin } from "@visiblebase/plugin-balance";

const balance = balancePlugin({
  init: 100,
  unit: "credits",
});

base.use(balance);

const ai = new AIService();

ai.hook.before(async (ctx) => {
  if (!ctx.user || ctx.action?.id !== "chat/completions") return;
  await balance.require(ctx.user.user_id, 10);
  ctx.locals.balance_amount = 10;
});

ai.hook.after(async (ctx) => {
  if (!ctx.user || ctx.action?.id !== "chat/completions") return;
  const amount = Number(ctx.locals.balance_amount ?? 0);
  if (amount <= 0) return;

  await balance.sub(ctx.user.user_id, amount, {
    note: "agent chat",
    meta: {
      product_id: ctx.product?.product_id,
      model_id: ctx.variant?.id,
    },
  });
});