feat: new admin panel

Author: Rieranthony

apps/api/src/lib/ai-credits/polar-meter.test.ts

@@ -1,5 +1,9 @@
 import { beforeEach, describe, expect, it, mock } from "bun:test";
-import { getAiCreditMeterState, ingestAiCreditUsage } from "./polar-meter";
+import {
+	getAiCreditMeterState,
+	grantAiCreditUsage,
+	ingestAiCreditUsage,
+} from "./polar-meter";
 
 class FakeRedis {
 	store = new Map<string, string>();
@@ -232,6 +236,60 @@ describe("ai credit Polar meter gateway", () => {
 		expect(updatedCache?.consumedUnits).toBe(12.5);
 	});
 
+	it("ingests negative credits and optimistically updates cached balance after admin grant", async () => {
+		await redis.set(
+			"ai-credit:meter:org-1",
+			buildCachedMeterState({
+				organizationId: "org-1",
+				lastSyncedAt: new Date(nowMs - 5000).toISOString(),
+				balance: 10,
+			})
+		);
+
+		const grantResult = await grantAiCreditUsage(
+			{
+				organizationId: "org-1",
+				websiteId: "site-1",
+				amount: 7.5,
+				adminUserId: "admin-1",
+			},
+			{
+				deps: {
+					redis,
+					polar,
+					now: () => nowMs,
+					meterId: "meter-ai-1",
+					eventName: "ai_usage",
+					cacheTtlSeconds: 15,
+					staleTtlSeconds: 300,
+				},
+			}
+		);
+
+		expect(grantResult.status).toBe("ingested");
+		expect(polar.events.ingest).toHaveBeenCalledWith({
+			events: [
+				{
+					name: "ai_usage",
+					externalCustomerId: "org-1",
+					metadata: {
+						credits: -7.5,
+						websiteId: "site-1",
+						organizationId: "org-1",
+						adminUserId: "admin-1",
+						kind: "admin_grant",
+					},
+				},
+			],
+		});
+
+		const updatedCacheRaw = await redis.get("ai-credit:meter:org-1");
+		const updatedCache = updatedCacheRaw ? JSON.parse(updatedCacheRaw) : null;
+		expect(updatedCache?.balance).toBe(17.5);
+		expect(updatedCache?.creditedUnits).toBe(42.5);
+		expect(updatedCache?.consumedUnits).toBe(10);
+	});
+
 	it("skips ingest while backoff cooldown is active", async () => {
 		await redis.set("ai-credit:ingest-backoff:org-1", "active-backoff");
 

apps/api/src/lib/ai-credits/polar-meter.ts

@@ -48,17 +48,33 @@ export type IngestAiCreditUsageInput = {
 	totalToolCount: number;
 };
 
+export type GrantAiCreditUsageInput = {
+	organizationId: string;
+	websiteId: string;
+	amount: number;
+	adminUserId: string;
+};
+
 export type IngestAiCreditUsageStatus =
 	| "ingested"
 	| "failed"
 	| "skipped_backoff"
 	| "skipped_zero"
 	| "skipped_disabled";
 
+export type GrantAiCreditUsageStatus =
+	| "ingested"
+	| "failed"
+	| "skipped_disabled";
+
 export type IngestAiCreditUsageResult = {
 	status: IngestAiCreditUsageStatus;
 };
 
+export type GrantAiCreditUsageResult = {
+	status: GrantAiCreditUsageStatus;
+};
+
 type EventMetadataValue = string | number | boolean;
 
 type RedisLike = {
@@ -300,6 +316,28 @@ function optimisticConsumeCredits(params: {
 	};
 }
 
+function optimisticGrantCredits(params: {
+	entry: CachedMeterEntry;
+	amount: number;
+	nowMs: number;
+}): CachedMeterEntry {
+	const nextBalance =
+		typeof params.entry.balance === "number"
+			? params.entry.balance + params.amount
+			: null;
+	const nextCredited =
+		typeof params.entry.creditedUnits === "number"
+			? params.entry.creditedUnits + params.amount
+			: null;
+
+	return {
+		...params.entry,
+		balance: nextBalance,
+		creditedUnits: nextCredited,
+		lastSyncedAt: toTimestamp(params.nowMs),
+	};
+}
+
 export async function getAiCreditMeterState(
 	organizationId: string,
 	options?: GatewayOptions
@@ -572,3 +610,70 @@ export async function ingestAiCreditUsage(
 
 	return { status: "ingested" };
 }
+
+export async function grantAiCreditUsage(
+	input: GrantAiCreditUsageInput,
+	options?: GatewayOptions
+): Promise<GrantAiCreditUsageResult> {
+	const deps = resolveDeps(options?.deps);
+
+	if (!deps.billingEnabled) {
+		return { status: "skipped_disabled" };
+	}
+
+	try {
+		await deps.polar.events.ingest({
+			events: [
+				{
+					name: deps.eventName,
+					externalCustomerId: input.organizationId,
+					metadata: {
+						credits: -input.amount,
+						websiteId: input.websiteId,
+						organizationId: input.organizationId,
+						adminUserId: input.adminUserId,
+						kind: "admin_grant",
+					},
+				},
+			],
+		});
+	} catch (error) {
+		console.error(
+			`[ai-credits] Failed to grant usage credits for org=${input.organizationId}:`,
+			error
+		);
+		return { status: "failed" };
+	}
+
+	const cacheKey = buildMeterCacheKey(input.organizationId);
+	let rawCache: string | null = null;
+	try {
+		rawCache = await deps.redis.get(cacheKey);
+	} catch {
+		return { status: "ingested" };
+	}
+	const cached = parseCachedEntry(rawCache);
+
+	if (!cached?.meterBacked) {
+		return { status: "ingested" };
+	}
+
+	const updated = optimisticGrantCredits({
+		entry: cached,
+		amount: input.amount,
+		nowMs: deps.now(),
+	});
+
+	try {
+		await writeCachedEntry({
+			redis: deps.redis,
+			key: cacheKey,
+			entry: updated,
+			staleTtlSeconds: deps.staleTtlSeconds,
+		});
+	} catch {
+		// Cache update is best-effort only.
+	}
+
+	return { status: "ingested" };
+}