initial budget ui on the provider

src/lib/aiBudgetScope.ts

@@ -0,0 +1,70 @@
+import type { AiBudget } from "@server/db";
+
+export type AiBudgetScopeType =
+    | "provider"
+    | "model"
+    | "resource"
+    | "siteResource"
+    | "role";
+
+export type AiBudgetScope = {
+    type: AiBudgetScopeType;
+    id: number;
+};
+
+export type AiBudgetScopeBodyField =
+    | "providerId"
+    | "modelId"
+    | "resourceId"
+    | "siteResourceId"
+    | "roleId";
+
+const scopeConfig: Record<
+    AiBudgetScopeType,
+    { listPath: (id: number) => string; bodyField: AiBudgetScopeBodyField }
+> = {
+    provider: {
+        listPath: (id) => `/ai-provider/${id}/ai-budgets`,
+        bodyField: "providerId"
+    },
+    model: {
+        listPath: (id) => `/ai-model/${id}/ai-budgets`,
+        bodyField: "modelId"
+    },
+    resource: {
+        listPath: (id) => `/resource/${id}/ai-budgets`,
+        bodyField: "resourceId"
+    },
+    siteResource: {
+        listPath: (id) => `/site-resource/${id}/ai-budgets`,
+        bodyField: "siteResourceId"
+    },
+    role: {
+        listPath: (id) => `/role/${id}/ai-budgets`,
+        bodyField: "roleId"
+    }
+};
+
+export function getAiBudgetScopeListPath(scope: AiBudgetScope): string {
+    return scopeConfig[scope.type].listPath(scope.id);
+}
+
+export function getAiBudgetScopeBodyField(
+    scope: AiBudgetScope
+): AiBudgetScopeBodyField {
+    return scopeConfig[scope.type].bodyField;
+}
+
+export type AiBudgetUnit = AiBudget["unit"];
+export type AiBudgetPeriod = AiBudget["period"];
+
+export const AI_BUDGET_UNITS: AiBudgetUnit[] = ["usd", "tokens"];
+
+export const AI_BUDGET_PERIODS: AiBudgetPeriod[] = [
+    "hourly",
+    "daily",
+    "weekly",
+    "monthly",
+    "yearly",
+    "lifetime"
+];