Added @ splat route

Author: matt-aitken

apps/webapp/app/models/admin.server.ts

@@ -1,5 +1,12 @@
+import { redirect } from "@remix-run/server-runtime";
 import { prisma } from "~/db.server";
 import { SearchParams } from "~/routes/admin._index";
+import {
+  clearImpersonationId,
+  commitImpersonationSession,
+  setImpersonationId,
+} from "~/services/impersonation.server";
+import { requireUser } from "~/services/session.server";
 
 const pageSize = 20;
 
@@ -219,3 +226,26 @@ export async function setV3Enabled(userId: string, id: string, v3Enabled: boolea
     },
   });
 }
+
+export async function redirectWithImpersonation(request: Request, userId: string, path: string) {
+  const user = await requireUser(request);
+  if (!user.admin) {
+    throw new Error("Unauthorized");
+  }
+
+  const session = await setImpersonationId(userId, request);
+
+  return redirect(path, {
+    headers: { "Set-Cookie": await commitImpersonationSession(session) },
+  });
+}
+
+export async function clearImpersonation(request: Request, path: string) {
+  const session = await clearImpersonationId(request);
+
+  return redirect(path, {
+    headers: {
+      "Set-Cookie": await commitImpersonationSession(session),
+    },
+  });
+}

apps/webapp/app/routes/@.ts

@@ -0,0 +1,6 @@
+import type { LoaderFunctionArgs } from "@remix-run/server-runtime";
+import { clearImpersonation } from "~/models/admin.server";
+
+export async function loader({ request, params }: LoaderFunctionArgs) {
+  return clearImpersonation(request, "/admin");
+}

apps/webapp/app/routes/_app.@.orgs.$organizationSlug.$.ts

@@ -0,0 +1,60 @@
+import type { LoaderFunctionArgs } from "@remix-run/server-runtime";
+import { redirect } from "remix-typedjson";
+import { $replica } from "~/db.server";
+import { clearImpersonation, redirectWithImpersonation } from "~/models/admin.server";
+import { logger } from "~/services/logger.server";
+import { requireUser } from "~/services/session.server";
+
+export async function loader({ request, params }: LoaderFunctionArgs) {
+  const user = await requireUser(request);
+  if (!user.admin) {
+    return redirect("/");
+  }
+
+  const path = params["*"];
+  const organizationSlug = params.organizationSlug;
+
+  logger.debug("Impersonating user", { path, organizationSlug });
+
+  if (!organizationSlug) {
+    logger.debug("Exiting impersonation mode");
+    return clearImpersonation(request, "/admin");
+  }
+
+  const org = await $replica.organization.findFirst({
+    where: {
+      slug: organizationSlug,
+      deletedAt: null,
+    },
+    select: {
+      members: {
+        select: {
+          user: {
+            select: {
+              id: true,
+              confirmedBasicDetails: true,
+            },
+          },
+        },
+      },
+    },
+  });
+
+  if (!org) {
+    logger.debug("Organization not found", { organizationSlug });
+    return clearImpersonation(request, "/admin");
+  }
+
+  const firstValidMember = org.members.find((m) => m.user.confirmedBasicDetails);
+
+  if (!firstValidMember) {
+    logger.debug("No valid members found", { organizationSlug });
+    return clearImpersonation(request, "/admin");
+  }
+
+  return redirectWithImpersonation(
+    request,
+    firstValidMember.user.id,
+    `/orgs/${organizationSlug}/${path}`
+  );
+}

apps/webapp/app/routes/admin._index.tsx

@@ -19,7 +19,7 @@ import {
   TableRow,
 } from "~/components/primitives/Table";
 import { useUser } from "~/hooks/useUser";
-import { adminGetUsers } from "~/models/admin.server";
+import { adminGetUsers, redirectWithImpersonation } from "~/models/admin.server";
 import { commitImpersonationSession, setImpersonationId } from "~/services/impersonation.server";
 import { requireUserId } from "~/services/session.server";
 import { createSearchParams } from "~/utils/searchParams";
@@ -53,12 +53,9 @@ export async function action({ request }: ActionFunctionArgs) {
   const payload = Object.fromEntries(await request.formData());
   const { id } = FormSchema.parse(payload);
 
-  const session = await setImpersonationId(id, request);
-
-  return redirect("/", {
-    headers: { "Set-Cookie": await commitImpersonationSession(session) },
-  });
+  return redirectWithImpersonation(request, id, "/");
 }
+
 export default function AdminDashboardRoute() {
   const user = useUser();
   const { users, filters, page, pageCount } = useTypedLoaderData<typeof loader>();

apps/webapp/app/routes/admin.orgs.tsx

@@ -104,7 +104,15 @@ export default function AdminDashboardRoute() {
                     <TableCell>{org.v2Enabled ? "✅" : ""}</TableCell>
                     <TableCell>{org.v3Enabled ? "✅" : ""}</TableCell>
                     <TableCell>{org.deletedAt ? "☠️" : ""}</TableCell>
-                    <TableCell isSticky={true}> </TableCell>
+                    <TableCell isSticky={true}>
+                      <LinkButton
+                        to={`/@/orgs/${org.slug}`}
+                        className="mr-2"
+                        variant="tertiary/small"
+                      >
+                        Impersonate
+                      </LinkButton>
+                    </TableCell>
                   </TableRow>
                 );
               })

apps/webapp/app/routes/resources.impersonation.ts

@@ -1,13 +1,6 @@
 import type { ActionFunctionArgs } from "@remix-run/server-runtime";
-import { redirect } from "remix-typedjson";
-import { clearImpersonationId, commitImpersonationSession } from "~/services/impersonation.server";
+import { clearImpersonation } from "~/models/admin.server";
 
 export async function action({ request }: ActionFunctionArgs) {
-  const session = await clearImpersonationId(request);
-
-  return redirect("/admin", {
-    headers: {
-      "Set-Cookie": await commitImpersonationSession(session),
-    },
-  });
+  return clearImpersonation(request, "/admin");
 }