fix rebalance feature get route, moved to top level nav

api/src/server.ts

@@ -3347,6 +3347,15 @@ const CatBody = z.object({
   isSavings: z.boolean(),
   priority: z.number().int().min(0),
 });
+const ManualRebalanceBody = z.object({
+  targets: z.array(
+    z.object({
+      id: z.string().min(1),
+      targetCents: z.number().int().min(0),
+    })
+  ),
+  forceLowerSavings: z.boolean().optional(),
+});
 
 async function assertPercentTotal(
   tx: PrismaClient | Prisma.TransactionClient,
@@ -3379,6 +3388,13 @@ async function assertPercentTotal(
   }
 }
 
+async function getLatestBudgetSession(app: any, userId: string) {
+  return app.prisma.budgetSession.findFirst({
+    where: { userId },
+    orderBy: { periodStart: "desc" },
+  });
+}
+
 app.post("/variable-categories", mutationRateLimit, async (req, reply) => {
   const parsed = CatBody.safeParse(req.body);
   if (!parsed.success) {
@@ -3509,6 +3525,110 @@ app.post("/variable-categories/rebalance", mutationRateLimit, async (req, reply)
   return { ok: true, applied: true, totalBalance };
 });
 
+app.get("/variable-categories/manual-rebalance", async (req, reply) => {
+  const userId = req.userId;
+  const session = await getLatestBudgetSession(app, userId);
+  if (!session) return reply.code(400).send({ ok: false, code: "NO_BUDGET_SESSION" });
+
+  const cats = await app.prisma.variableCategory.findMany({
+    where: { userId },
+    orderBy: [{ priority: "asc" }, { name: "asc" }],
+    select: { id: true, name: true, percent: true, isSavings: true, balanceCents: true },
+  });
+
+  return reply.send({
+    ok: true,
+    availableCents: Number(session.availableCents ?? 0n),
+    categories: cats.map((c) => ({ ...c, balanceCents: Number(c.balanceCents ?? 0n) })),
+  });
+});
+
+app.post("/variable-categories/manual-rebalance", async (req, reply) => {
+  const userId = req.userId;
+  const parsed = ManualRebalanceBody.safeParse(req.body);
+  if (!parsed.success || parsed.data.targets.length === 0) {
+    return reply.code(400).send({ ok: false, code: "INVALID_BODY" });
+  }
+
+  const session = await getLatestBudgetSession(app, userId);
+  if (!session) return reply.code(400).send({ ok: false, code: "NO_BUDGET_SESSION" });
+  const availableCents = Number(session.availableCents ?? 0n);
+
+  const cats = await app.prisma.variableCategory.findMany({
+    where: { userId },
+    orderBy: [{ priority: "asc" }, { name: "asc" }],
+    select: { id: true, name: true, percent: true, isSavings: true, balanceCents: true },
+  });
+  if (cats.length === 0) return reply.code(400).send({ ok: false, code: "NO_CATEGORIES" });
+
+  const targetMap = new Map<string, number>();
+  for (const t of parsed.data.targets) {
+    if (targetMap.has(t.id)) return reply.code(400).send({ ok: false, code: "DUPLICATE_ID" });
+    targetMap.set(t.id, t.targetCents);
+  }
+  if (targetMap.size !== cats.length || cats.some((c) => !targetMap.has(c.id))) {
+    return reply.code(400).send({ ok: false, code: "MISSING_CATEGORY" });
+  }
+
+  const targets = cats.map((c) => ({
+    ...c,
+    target: targetMap.get(c.id)!,
+    currentBalance: Number(c.balanceCents ?? 0n),
+  }));
+
+  if (targets.some((t) => t.target < 0)) {
+    return reply.code(400).send({ ok: false, code: "NEGATIVE_TARGET" });
+  }
+  const sumTargets = targets.reduce((s, t) => s + t.target, 0);
+  if (sumTargets !== availableCents) {
+    return reply
+      .code(400)
+      .send({ ok: false, code: "SUM_MISMATCH", message: `Targets must sum to available (${availableCents}).` });
+  }
+  const maxAllowed = Math.floor(availableCents * 0.8);
+  if (availableCents > 0 && targets.some((t) => t.target > maxAllowed)) {
+    return reply.code(400).send({ ok: false, code: "OVER_80_PERCENT", message: "No category can exceed 80% of available." });
+  }
+  const totalSavingsBefore = targets.filter((t) => t.isSavings).reduce((s, t) => s + t.currentBalance, 0);
+  const totalSavingsAfter = targets.filter((t) => t.isSavings).reduce((s, t) => s + t.target, 0);
+  const savingsFloor = Math.floor(availableCents * 0.2);
+  const loweringSavings = totalSavingsAfter < totalSavingsBefore;
+  const belowFloor = totalSavingsAfter < savingsFloor;
+  if ((loweringSavings || belowFloor) && !parsed.data.forceLowerSavings) {
+    return reply.code(400).send({ ok: false, code: "SAVINGS_FLOOR", message: "Lowering savings requires confirmation." });
+  }
+
+  await app.prisma.$transaction(async (tx) => {
+    for (const t of targets) {
+      await tx.variableCategory.update({
+        where: { id: t.id },
+        data: { balanceCents: BigInt(t.target) },
+      });
+    }
+    await tx.transaction.create({
+      data: {
+        userId,
+        kind: "rebalance",
+        amountCents: 0n,
+        occurredAt: new Date(),
+        note: JSON.stringify({
+          availableCents,
+          before: targets.map((t) => ({ id: t.id, balanceCents: t.currentBalance })),
+          after: targets.map((t) => ({ id: t.id, balanceCents: t.target })),
+          totalSavingsBefore,
+          totalSavingsAfter,
+        }),
+      },
+    });
+  });
+
+  return reply.send({
+    ok: true,
+    availableCents,
+    categories: targets.map((t) => ({ id: t.id, balanceCents: t.target })),
+  });
+});
+
 // ----- Fixed plans -----
 const PlanBody = z.object({
   name: z.string().trim().min(1),