Merge pull request #429 from pesickaa/feat/invitation-codes

Feat: Support invitation codes

Author: Koosha-Owji

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kinde-oss/kinde-auth-nextjs",
-  "version": "2.11.0",
+  "version": "2.11.1",
   "description": "Kinde Auth SDK for NextJS",
   "main": "dist/cjs/index.js",
   "module": "dist/index.js",
@@ -61,7 +61,8 @@
     "release": "release-it",
     "lint": "prettier --check .",
     "lint:fix": "prettier --write .",
-    "dev": "cd playground && pnpm run dev"
+    "dev": "cd playground && pnpm run dev",
+    "dev:prepare": "rm -rf playground && git clone --depth 1 https://github.com/kinde-starter-kits/kinde-nextjs-app-router-starter-kit playground && rm -rf playground/.git && mv playground/.env_sample playground/.env && cd playground && pnpm i && pnpm link .."
   },
   "author": {
     "name": "Kinde",

playground/next-env.d.ts

@@ -1,5 +1,6 @@
 /// <reference types="next" />
 /// <reference types="next/image-types/global" />
+/// <reference path="./.next/types/routes.d.ts" />
 
 // NOTE: This file should not be edited
-// see https://nextjs.org/docs/app/building-your-application/configuring/typescript for more information.
+// see https://nextjs.org/docs/app/api-reference/config/typescript for more information.

src/authMiddleware/authMiddleware.ts

@@ -14,9 +14,48 @@ import { getStandardCookieOptions } from "../utils/cookies/getStandardCookieOpti
 import { isPublicPathMatch } from "../utils/isPublicPathMatch";
 import { TWENTY_NINE_DAYS } from "src/utils/constants";
 
+/**
+ * Handles invitation code redirect logic.
+ * Redirects to the register page with the invitation code, or to login on error.
+ */
+const handleInvitationCodeRedirect = (
+  invitationCode: string,
+  registerPage: string,
+  loginRedirectUrl: string,
+  redirectURLBase: string | undefined,
+): NextResponse => {
+  try {
+    const params = new URLSearchParams();
+    params.set("invitation_code", invitationCode);
+    params.set("is_invitation", "true");
+
+    const registerWithInviteRedirectUrl = `${registerPage}?${params.toString()}`;
+
+    return NextResponse.redirect(
+      new URL(
+        registerWithInviteRedirectUrl,
+        redirectURLBase || config.redirectURL,
+      ),
+    );
+  } catch (error) {
+    if (config.isDebugMode) {
+      console.error(
+        "authMiddleware: error redirecting to register with invitation code",
+        error,
+      );
+    }
+    return NextResponse.redirect(
+      new URL(loginRedirectUrl, redirectURLBase || config.redirectURL),
+    );
+  }
+};
+
 const handleMiddleware = async (req, options, onSuccess) => {
   const { pathname, search } = req.nextUrl;
 
+  const params = new URLSearchParams(search);
+  const invitationCode = params.get("invitation_code");
+  const hasInvitationCode = !!invitationCode?.trim();
   const isReturnToCurrentPage = options?.isReturnToCurrentPage;
   const orgCode: string | undefined = options?.orgCode;
   const loginPage = options?.loginPage || `${config.apiPath}/${routes.login}`;
@@ -39,7 +78,6 @@ const handleMiddleware = async (req, options, onSuccess) => {
       publicPaths = options.publicPaths;
     }
   }
-
   const loginRedirectUrlParams = new URLSearchParams();
 
   if (orgCode) {
@@ -55,6 +93,15 @@ const handleMiddleware = async (req, options, onSuccess) => {
     ? `${loginPage}?${queryString}`
     : loginPage;
 
+  if (hasInvitationCode) {
+    return handleInvitationCodeRedirect(
+      invitationCode,
+      registerPage,
+      loginRedirectUrl,
+      options?.redirectURLBase,
+    );
+  }
+
   // Use extracted utility for public path matching
   // eslint-disable-next-line @typescript-eslint/no-var-requires
   const isPublicPath = isPublicPathMatch(