feat: Signup query params tracking (#2098)

* feat: Add tracking of signup query params

* fix: Headers already sent to client

* fix: OAuth error wipes previously written query params cookie

server/auth/providers/email.js

@@ -101,7 +101,7 @@ router.get("email.callback", async (ctx) => {
     await user.update({ lastActiveAt: new Date() });
 
     // set cookies on response and redirect to team subdomain
-    signIn(ctx, user, user.team, "email", false);
+    await signIn(ctx, user, user.team, "email", false, false);
   } catch (err) {
     ctx.redirect(`/?notice=expired-token`);
   }

server/middlewares/passport.js

@@ -9,7 +9,7 @@ export default function createMiddleware(providerName: string) {
     return passport.authorize(
       providerName,
       { session: false },
-      (err, _, result: AccountProvisionerResult) => {
+      async (err, _, result: AccountProvisionerResult) => {
         if (err) {
           console.error(err);
 
@@ -39,7 +39,14 @@ export default function createMiddleware(providerName: string) {
           return ctx.redirect("/?notice=suspended");
         }
 
-        signIn(ctx, result.user, result.team, providerName, result.isNewUser);
+        await signIn(
+          ctx,
+          result.user,
+          result.team,
+          providerName,
+          result.isNewUser,
+          result.isNewTeam
+        );
       }
     )(ctx);
   };

server/migrations/20210430024222-marketing-tracking.js

@@ -0,0 +1,14 @@
+'use strict';
+
+module.exports = {
+  up: async (queryInterface, Sequelize) => {
+     await queryInterface.addColumn("teams", "signupQueryParams", {
+      type: Sequelize.JSONB,
+      allowNull: true,
+    });
+  },
+
+  down: async (queryInterface, Sequelize) => {
+    await queryInterface.removeColumn("teams", "signupQueryParams");
+  }
+};

server/models/Team.js

@@ -55,6 +55,10 @@ const Team = sequelize.define(
     googleId: { type: DataTypes.STRING, allowNull: true },
     avatarUrl: { type: DataTypes.STRING, allowNull: true },
     sharing: { type: DataTypes.BOOLEAN, allowNull: false, defaultValue: true },
+    signupQueryParams: {
+      type: DataTypes.JSONB,
+      allowNull: true,
+    },
     guestSignin: {
       type: DataTypes.BOOLEAN,
       allowNull: false,

server/utils/authentication.js

@@ -1,6 +1,9 @@
 // @flow
+import querystring from "querystring";
+import * as Sentry from "@sentry/node";
 import addMonths from "date-fns/add_months";
 import { type Context } from "koa";
+import { pick } from "lodash";
 import { User, Event, Team } from "../models";
 import { getCookieDomain } from "../utils/domains";
 
@@ -10,17 +13,36 @@ export function getAllowedDomains(): string[] {
   return env ? env.split(",") : [];
 }
 
-export function signIn(
+export async function signIn(
   ctx: Context,
   user: User,
   team: Team,
   service: string,
-  isFirstSignin: boolean = false
+  isNewUser: boolean = false,
+  isNewTeam: boolean = false
 ) {
   if (user.isSuspended) {
     return ctx.redirect("/?notice=suspended");
   }
 
+  if (isNewTeam) {
+    // see: scenes/Login/index.js for where this cookie is written when
+    // viewing the /login or /create pages. It is a URI encoded JSON string.
+    const cookie = ctx.cookies.get("signupQueryParams");
+
+    if (cookie) {
+      try {
+        const signupQueryParams = pick(
+          JSON.parse(querystring.unescape(cookie)),
+          ["ref", "utm_content", "utm_medium", "utm_source", "utm_campaign"]
+        );
+        await team.update({ signupQueryParams });
+      } catch (err) {
+        Sentry.captureException(err);
+      }
+    }
+  }
+
   // update the database when the user last signed in
   user.updateSignedIn(ctx.request.ip);
 
@@ -77,6 +99,6 @@ export function signIn(
       httpOnly: false,
       expires,
     });
-    ctx.redirect(`${team.url}/home${isFirstSignin ? "?welcome" : ""}`);
+    ctx.redirect(`${team.url}/home${isNewUser ? "?welcome" : ""}`);
   }
 }