chore: redirect platform users to dashboard (#15361)

Ryukemeister · ThyMinimalDev · Rajiv Sahal · web-flow · commit 62e3f9587c1b · 2024-06-19T18:41:46.000Z
* chore: redirect platform users to dashboard

* update AppProps typing to include isPlatformUser

* fixup

* get isPlatformUser boolean from user and then redirect user accordingly

* fix type checks

* minor refactor

* AppProps type does not change

* helper function for redirecting on server side

* revert: AppProps type does not change

* replace helper function to redirect platform users with a custom hook

* add comments stating why we use this hook

* fixup! add comments stating why we use this hook

---------

Co-authored-by: Morgan Vernay &lt;morgan@cal.com&gt;
Co-authored-by: Rajiv Sahal &lt;rajivsahal@Rajivs-MacBook-Pro.local&gt;
Co-authored-by: Morgan &lt;33722304+ThyMinimalDev@users.noreply.github.com&gt;
Co-authored-by: Peer Richelsen &lt;peeroke@gmail.com&gt;
diff --git a/apps/web/components/PageWrapper.tsx b/apps/web/components/PageWrapper.tsx
@@ -1,8 +1,11 @@
+import { useSession } from "next-auth/react";
 import { DefaultSeo } from "next-seo";
 import { Inter } from "next/font/google";
 import localFont from "next/font/local";
 import Head from "next/head";
+import { useRouter, usePathname } from "next/navigation";
 import Script from "next/script";
+import { useEffect } from "react";
 
 import "@calcom/embed-core/src/embed-iframe";
 import LicenseRequired from "@calcom/features/ee/common/components/LicenseRequired";
@@ -27,6 +30,23 @@ const calFont = localFont({
   weight: "600",
 });
 
+const Redirects = () => {
+  const session = useSession();
+  const router = useRouter();
+  const pathName = usePathname();
+  // is the connected user in a platform organization or not
+  const isPlatformUser = session?.data?.isPlatformUser ?? false;
+
+  useEffect(() => {
+    // is platform user trying to access a non-platform page, redirect to platform dashboard
+    if (isPlatformUser && pathName && !pathName.startsWith("/settings")) {
+      return router.replace("/settings/platform");
+    }
+  }, [isPlatformUser, pathName, router]);
+
+  return <></>;
+};
+
 function PageWrapper(props: AppProps) {
   const { Component, pageProps, err, router } = props;
   let pageStatus = "200";
@@ -56,6 +76,7 @@ function PageWrapper(props: AppProps) {
 
   return (
     <AppProviders {...providerProps}>
+      <Redirects />
       <Head>
         <meta
           name="viewport"
diff --git a/apps/web/pages/_app.tsx b/apps/web/pages/_app.tsx
@@ -1,4 +1,5 @@
 import type { IncomingMessage } from "http";
+import type { NextApiRequest } from "next";
 import type { AppContextType } from "next/dist/shared/lib/utils";
 import React from "react";
 
@@ -22,14 +23,23 @@ declare global {
 }
 
 MyApp.getInitialProps = async (ctx: AppContextType) => {
-  const { req } = ctx.ctx;
+  const { req, res } = ctx.ctx;
 
   let newLocale = "en";
 
   if (req) {
     const { getLocale } = await import("@calcom/features/auth/lib/getLocale");
+    const { getServerSession } = await import("@calcom/features/auth/lib/getServerSession");
+    const session = await getServerSession({ req: req as NextApiRequest, res });
+
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     newLocale = await getLocale(req as IncomingMessage & { cookies: Record<string, any> });
+    return {
+      pageProps: {
+        newLocale,
+        session,
+      },
+    };
   } else if (typeof window !== "undefined" && window.calNewLocale) {
     newLocale = window.calNewLocale;
   }
diff --git a/packages/features/auth/lib/getServerSession.ts b/packages/features/auth/lib/getServerSession.ts
@@ -58,6 +58,13 @@ export async function getServerSession(options: {
     where: {
       email: token.email.toLowerCase(),
     },
+    include: {
+      organization: {
+        select: {
+          isPlatform: true,
+        },
+      },
+    },
   });
 
   if (!userFromDb) {
@@ -103,6 +110,7 @@ export async function getServerSession(options: {
       profile: user.profile,
     },
     profileId: token.profileId,
+    isPlatformUser: user.profile.organization?.isPlatform ?? false,
     upId,
   };
 
diff --git a/packages/types/next-auth.d.ts b/packages/types/next-auth.d.ts
@@ -14,6 +14,7 @@ declare module "next-auth" {
     profileId?: number | null;
     upId: string;
     user: User;
+    isPlatformUser?: boolean;
   }
 
   interface User extends Omit<DefaultUser, "id"> {

Original file line number	Diff line number	Diff line change
`@@ -14,6 +14,7 @@ declare module "next-auth" {`
`14`	`14`	`profileId?: number \| null;`
`15`	`15`	`upId: string;`
`16`	`16`	`user: User;`
	`17`	`+ isPlatformUser?: boolean;`
`17`	`18`	`}`
`18`	`19`
`19`	`20`	`interface User extends Omit<DefaultUser, "id"> {`