Update feature flags API endpoint

Author: matt-aitken

apps/webapp/app/routes/admin.api.v1.feature-flags.ts

@@ -0,0 +1,71 @@
+import { ActionFunctionArgs, json } from "@remix-run/server-runtime";
+import { prisma } from "~/db.server";
+import { authenticateApiRequestWithPersonalAccessToken } from "~/services/personalAccessToken.server";
+import { getRunsReplicationGlobal } from "~/services/runsReplicationGlobal.server";
+import { runsReplicationInstance } from "~/services/runsReplicationInstance.server";
+import {
+  makeSetFlags,
+  setFlags,
+  FeatureFlagCatalogSchema,
+  validateAllFeatureFlags,
+  validatePartialFeatureFlags,
+  makeSetMultipleFlags,
+} from "~/v3/featureFlags.server";
+import { z } from "zod";
+
+export async function action({ request }: ActionFunctionArgs) {
+  // Next authenticate the request
+  const authenticationResult = await authenticateApiRequestWithPersonalAccessToken(request);
+
+  if (!authenticationResult) {
+    return json({ error: "Invalid or Missing API key" }, { status: 401 });
+  }
+
+  const user = await prisma.user.findUnique({
+    where: {
+      id: authenticationResult.userId,
+    },
+  });
+
+  if (!user) {
+    return json({ error: "Invalid or Missing API key" }, { status: 401 });
+  }
+
+  if (!user.admin) {
+    return json({ error: "You must be an admin to perform this action" }, { status: 403 });
+  }
+
+  try {
+    // Parse the request body
+    const body = await request.json();
+
+    // Validate the input using the partial schema
+    const validationResult = validatePartialFeatureFlags(body as Record<string, unknown>);
+    if (!validationResult.success) {
+      return json(
+        {
+          error: "Invalid feature flags data",
+          details: validationResult.error.issues,
+        },
+        { status: 400 }
+      );
+    }
+
+    const featureFlags = validationResult.data;
+    const setMultipleFlags = makeSetMultipleFlags(prisma);
+    const updatedFlags = await setMultipleFlags(featureFlags);
+
+    return json({
+      success: true,
+      updatedFlags,
+      message: `Updated ${updatedFlags.length} feature flag(s)`,
+    });
+  } catch (error) {
+    return json(
+      {
+        error: error instanceof Error ? error.message : String(error),
+      },
+      { status: 400 }
+    );
+  }
+}

apps/webapp/app/v3/featureFlags.server.ts

@@ -67,3 +67,46 @@ export function makeSetFlags(_prisma: PrismaClientOrTransaction = prisma) {
 
 export const flags = makeFlags();
 export const setFlags = makeSetFlags();
+
+// Create a Zod schema from the existing catalog
+export const FeatureFlagCatalogSchema = z.object(FeatureFlagCatalog);
+
+// Utility function to validate a feature flag value
+export function validateFeatureFlagValue<T extends FeatureFlagKey>(
+  key: T,
+  value: unknown
+): z.SafeParseReturnType<unknown, z.infer<(typeof FeatureFlagCatalog)[T]>> {
+  return FeatureFlagCatalog[key].safeParse(value);
+}
+
+// Utility function to validate all feature flags at once
+export function validateAllFeatureFlags(values: Record<string, unknown>) {
+  return FeatureFlagCatalogSchema.safeParse(values);
+}
+
+// Utility function to validate partial feature flags (all keys optional)
+export function validatePartialFeatureFlags(values: Record<string, unknown>) {
+  return FeatureFlagCatalogSchema.partial().safeParse(values);
+}
+
+// Utility function to set multiple feature flags at once
+export function makeSetMultipleFlags(_prisma: PrismaClientOrTransaction = prisma) {
+  return async function setMultipleFlags(
+    flags: Partial<z.infer<typeof FeatureFlagCatalogSchema>>
+  ): Promise<{ key: string; value: any }[]> {
+    const setFlag = makeSetFlags(_prisma);
+    const updatedFlags: { key: string; value: any }[] = [];
+
+    for (const [key, value] of Object.entries(flags)) {
+      if (value !== undefined) {
+        await setFlag({
+          key: key as any,
+          value: value as any,
+        });
+        updatedFlags.push({ key, value });
+      }
+    }
+
+    return updatedFlags;
+  };
+}