Send UTM params when logging into dashboard

Author: MananTank

apps/dashboard/src/app/login/auth-actions.ts

@@ -44,6 +44,18 @@ export async function doLogin(payload: VerifyLoginPayloadParams) {
   }
 
   const cookieStore = await cookies();
+  const utmCookies = cookieStore
+    .getAll()
+    .filter((cookie) => {
+      return cookie.name.startsWith("utm_");
+    })
+    .reduce(
+      (acc, cookie) => {
+        acc[cookie.name] = cookie.value;
+        return acc;
+      },
+      {} as Record<string, string>,
+    );
 
   // forward the request to the API server
   const res = await fetch(`${API_SERVER_URL}/v2/siwe/login`, {
@@ -53,7 +65,7 @@ export async function doLogin(payload: VerifyLoginPayloadParams) {
       "x-service-api-key": THIRDWEB_API_SECRET,
     },
     // set the createAccount flag to true to create a new account if it does not exist
-    body: JSON.stringify({ ...payload, createAccount: true }),
+    body: JSON.stringify({ ...payload, createAccount: true, utm: utmCookies }),
   });
 
   // if the request failed, log the error and throw an error

apps/dashboard/src/middleware.ts

@@ -1,6 +1,7 @@
 import { getTeams } from "@/api/team";
 import { isLoginRequired } from "@/constants/auth";
 import { COOKIE_ACTIVE_ACCOUNT, COOKIE_PREFIX_TOKEN } from "@/constants/cookie";
+import { cookies } from "next/headers";
 import { type NextRequest, NextResponse } from "next/server";
 import { getAddress } from "thirdweb";
 import { getChainMetadata } from "thirdweb/chains";
@@ -22,12 +23,38 @@ export const config = {
 };
 
 export async function middleware(request: NextRequest) {
+  const defaultResponse = NextResponse.next();
   const { pathname } = request.nextUrl;
   const activeAccount = request.cookies.get(COOKIE_ACTIVE_ACCOUNT)?.value;
   const authCookie = activeAccount
     ? request.cookies.get(COOKIE_PREFIX_TOKEN + getAddress(activeAccount))
     : null;
 
+  // utm collection
+  // NOTE: this is not working for pages with rewrites in next.config.js - (framer pages)
+  // if user is already signed in - don't bother capturing utm params
+  if (!authCookie) {
+    const searchParamsEntires = request.nextUrl.searchParams.entries();
+    const utmParams: Map<string, string> = new Map();
+    for (const param of searchParamsEntires) {
+      if (param[0].startsWith("utm_")) {
+        utmParams.set(param[0], param[1]);
+      }
+    }
+
+    // if we have utm params, set them as cookies
+    if (utmParams.size) {
+      const currentCookies = await cookies();
+      for (const [key, value] of utmParams.entries()) {
+        // if its already set - don't set it again
+        if (!currentCookies.get(key)) {
+          currentCookies.set(key, value);
+          defaultResponse.cookies.set(key, value);
+        }
+      }
+    }
+  }
+
   // logged in paths
   if (isLoginRequired(pathname)) {
     // check if the user is logged in (has a valid auth cookie)
@@ -117,6 +144,8 @@ export async function middleware(request: NextRequest) {
   }
   // END /<address>/... case
   // all other cases are handled by the file system router so we just fall through
+
+  return defaultResponse;
 }
 
 function isPossibleEVMAddress(address: string) {