Admin API endpoint that adds v3 staging environments, and updates their concurrency

2024-07-11 10:22:06 +01:00
parent 946efd1fb4
commit 3d6cfe5ead
1 changed files with 69 additions and 0 deletions
@@ -0,0 +1,69 @@
+import { ActionFunctionArgs, json } from "@remix-run/server-runtime";
+import { z } from "zod";
+import { prisma } from "~/db.server";
+import { createEnvironment } from "~/models/organization.server";
+import { authenticateApiRequestWithPersonalAccessToken } from "~/services/personalAccessToken.server";
+import { marqs } from "~/v3/marqs/index.server";
+
+const ParamsSchema = z.object({
+  organizationId: z.string(),
+});
+
+/**
+ * It will create a staging environment for all the projects where there isn't one already
+ */
+export async function action({ request, params }: 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 });
+  }
+
+  const { organizationId } = ParamsSchema.parse(params);
+
+  const organization = await prisma.organization.findUnique({
+    where: {
+      id: organizationId,
+    },
+    include: {
+      projects: {
+        include: { environments: true },
+      },
+    },
+  });
+
+  if (!organization) {
+    return json({ error: "Organization not found" }, { status: 404 });
+  }
+
+  let created = 0;
+
+  for (const project of organization.projects) {
+    const stagingEnvironment = project.environments.find((env) => env.type === "STAGING");
+
+    if (!stagingEnvironment) {
+      const staging = await createEnvironment(organization, project, "STAGING");
+      await marqs?.updateEnvConcurrencyLimits({ ...staging, organization, project });
+      created++;
+    } else {
+      await marqs?.updateEnvConcurrencyLimits({ ...stagingEnvironment, organization, project });
+    }
+  }
+
+  return json({ success: true, created, total: organization.projects.length });
+}