fix: add preferred team middleware

apps/remix/app/routes/_index.tsx

@@ -1,14 +1,47 @@
 import { redirect } from 'react-router';
 
+import { extractCookieFromHeaders } from '@documenso/auth/server/lib/utils/cookies';
 import { getOptionalSession } from '@documenso/auth/server/lib/utils/get-session';
+import { getTeams } from '@documenso/lib/server-only/team/get-teams';
+import { formatDocumentsPath } from '@documenso/lib/utils/teams';
+import { ZTeamUrlSchema } from '@documenso/trpc/server/team-router/schema';
 
 import type { Route } from './+types/_index';
 
 export async function loader({ request }: Route.LoaderArgs) {
-  const { isAuthenticated } = await getOptionalSession(request);
+  const session = await getOptionalSession(request);
 
-  if (isAuthenticated) {
-    throw redirect('/documents');
+  if (session.isAuthenticated) {
+    const teamUrlCookie = extractCookieFromHeaders('preferred-team-url', request.headers);
+
+    const referrer = request.headers.get('referer');
+    let isReferrerFromTeamUrl = false;
+
+    if (referrer) {
+      const referrerUrl = new URL(referrer);
+
+      if (referrerUrl.pathname.startsWith('/t/')) {
+        isReferrerFromTeamUrl = true;
+      }
+    }
+
+    const preferredTeamUrl =
+      teamUrlCookie && ZTeamUrlSchema.safeParse(teamUrlCookie).success ? teamUrlCookie : undefined;
+
+    // Early return for no preferred team.
+    if (!preferredTeamUrl || isReferrerFromTeamUrl) {
+      throw redirect('/documents');
+    }
+
+    const teams = await getTeams({ userId: session.user.id });
+
+    const currentTeam = teams.find((team) => team.url === preferredTeamUrl);
+
+    if (!currentTeam) {
+      throw redirect('/documents');
+    }
+
+    throw redirect(formatDocumentsPath(currentTeam.url));
   }
 
   throw redirect('/signin');

apps/remix/server/middleware.ts

@@ -0,0 +1,73 @@
+import type { Context, Next } from 'hono';
+import { deleteCookie, setCookie } from 'hono/cookie';
+
+import { AppDebugger } from '@documenso/lib/utils/debugger';
+
+const debug = new AppDebugger('Middleware');
+
+/**
+ * Middleware for initial page loads.
+ *
+ * You won't be able to easily handle sequential page loads because they will be
+ * called under `path.data`
+ *
+ * Example an initial page load would be `/documents` then if the user click templates
+ * the path here would be `/templates.data`.
+ */
+export const appMiddleware = async (c: Context, next: Next) => {
+  const { req } = c;
+  const { path } = req;
+
+  // Paths to ignore.
+  if (nonPagePathRegex.test(path)) {
+    return next();
+  }
+
+  // PRE-HANDLER CODE: Place code here to execute BEFORE the route handler runs.
+
+  await next();
+
+  // POST-HANDLER CODE: Place code here to execute AFTER the route handler completes.
+  // This is useful for:
+  // - Setting cookies
+  // - Any operations that should happen after all route handlers but before sending the response
+
+  debug.log('Path', path);
+
+  const pathname = path.replace('.data', '');
+  const referrer = c.req.header('referer');
+  const referrerUrl = referrer ? new URL(referrer) : null;
+  const referrerPathname = referrerUrl ? referrerUrl.pathname : null;
+
+  // Whether to reset the preferred team url cookie if the user accesses a non team page from a team page.
+  const resetPreferredTeamUrl =
+    referrerPathname &&
+    referrerPathname.startsWith('/t/') &&
+    (!pathname.startsWith('/t/') || pathname === '/');
+
+  // Set the preferred team url cookie if user accesses a team page.
+  if (pathname.startsWith('/t/')) {
+    debug.log('Setting preferred team url cookie');
+
+    setCookie(c, 'preferred-team-url', pathname.split('/')[2]);
+
+    return;
+  }
+
+  // Clear preferred team url cookie if user accesses a non team page from a team page.
+  if (resetPreferredTeamUrl || pathname === '/documents') {
+    debug.log('Deleting preferred team url cookie');
+
+    deleteCookie(c, 'preferred-team-url');
+
+    return;
+  }
+};
+
+// This regex matches any path that:
+// 1. Starts with /api/, /ingest/, /__manifest/, or /assets/
+// 2. Starts with /apple- (like /apple-touch-icon.png)
+// 3. Starts with /favicon (like /favicon.ico)
+// The ^ ensures matching from the beginning of the string
+// The | acts as OR operator between different patterns
+const nonPagePathRegex = /^(\/api\/|\/ingest\/|\/__manifest|\/assets\/|\/apple-.*|\/favicon.*)/;

apps/remix/server/router.ts

@@ -9,6 +9,7 @@ import { openApiDocument } from '@documenso/trpc/server/open-api';
 
 import { filesRoute } from './api/files';
 import { type AppContext, appContext } from './context';
+import { appMiddleware } from './middleware';
 import { openApiTrpcServerHandler } from './trpc/hono-trpc-open-api';
 import { reactRouterTrpcServer } from './trpc/hono-trpc-remix';
 
@@ -26,6 +27,11 @@ const app = new Hono<HonoEnv>();
 app.use(contextStorage());
 app.use(appContext);
 
+/**
+ * RR7 app middleware.
+ */
+app.use('*', appMiddleware);
+
 // Auth server.
 app.route('/api/auth', auth);