From 66ae66639d24e4cebb5bc4152aef0140627b7372 Mon Sep 17 00:00:00 2001
From: Storm <storm@reacher.email>
Date: Sun, 15 Dec 2024 00:39:20 +0100
Subject: [PATCH] feat: Redirect on dashboard

---
 src/app/[lang]/dashboard/page.tsx | 18 ++++++++-
 src/supabase/supabaseServer.ts    | 65 +++++++++++++++++++++++++++++--
 2 files changed, 78 insertions(+), 5 deletions(-)

diff --git a/src/app/[lang]/dashboard/page.tsx b/src/app/[lang]/dashboard/page.tsx
index 9796753..0c080ee 100644
--- a/src/app/[lang]/dashboard/page.tsx
+++ b/src/app/[lang]/dashboard/page.tsx
@@ -1,4 +1,4 @@
-import { getSubscription } from "@/supabase/supabaseServer";
+import { getSubscription, getSupabaseUser } from "@/supabase/supabaseServer";
 import { ENABLE_BULK } from "@/util/helpers";
 import {
 	COMMERCIAL_LICENSE_PRODUCT_ID,
@@ -12,6 +12,7 @@ export default async function Dashboard({
 	params: { lang: string };
 }) {
 	const subscription = await getSubscription();
+	const user = await getSupabaseUser();
 
 	switch (subscription?.prices?.product_id) {
 		case COMMERCIAL_LICENSE_PRODUCT_ID:
@@ -25,6 +26,19 @@ export default async function Dashboard({
 				RedirectType.replace
 			);
 		default:
-			return redirect(`/${lang}/dashboard/verify`, RedirectType.replace);
+			if (
+				user.user_metadata?.emailVolume === undefined ||
+				user.user_metadata?.emailVolume === "SAAS_10K"
+			) {
+				return redirect(
+					`/${lang}/dashboard/verify`,
+					RedirectType.replace
+				);
+			} else {
+				return redirect(
+					`/${lang}/dashboard/commercial_license`,
+					RedirectType.replace
+				);
+			}
 	}
 }
diff --git a/src/supabase/supabaseServer.ts b/src/supabase/supabaseServer.ts
index 360ca09..a11cba4 100644
--- a/src/supabase/supabaseServer.ts
+++ b/src/supabase/supabaseServer.ts
@@ -5,7 +5,39 @@
 import { cookies } from "next/headers";
 import { createClient } from "./server";
 import { Tables } from "./database.types";
+import { User } from "@supabase/supabase-js";
 
+/**
+ * Gets the currently authenticated Supabase user.
+ *
+ * @returns Promise resolving to the authenticated user response
+ * @throws Error if no user is found or if there's an authentication error
+ */
+export async function getSupabaseUser(): Promise<User> {
+	const cookieStore = cookies();
+	const supabase = createClient(cookieStore);
+	const {
+		data: { user },
+		error,
+	} = await supabase.auth.getUser();
+
+	// Handle auth errors
+	if (error) {
+		throw error;
+	}
+	// Ensure user exists
+	if (!user) {
+		throw new Error("No user found.");
+	}
+
+	return user;
+}
+
+/**
+ * Gets the current Supabase session
+ * @returns Promise resolving to the current session or null if not authenticated
+ * @throws Error if there's an authentication error
+ */
 export async function getSession() {
 	const cookieStore = cookies();
 	const supabase = createClient(cookieStore);
@@ -20,8 +52,16 @@ export async function getSession() {
 	return session;
 }
 
+/**
+ * Type alias for the return type of getUserDetails function
+ */
 export type UserDetails = Awaited<ReturnType<typeof getUserDetails>>;
 
+/**
+ * Gets the details of the currently authenticated user from the users table
+ * @returns Promise resolving to the user details
+ * @throws Error if no user is found or if there's a database error
+ */
 export async function getUserDetails() {
 	const cookieStore = cookies();
 	const supabase = createClient(cookieStore);
@@ -33,10 +73,18 @@ export async function getUserDetails() {
 	return userDetails;
 }
 
+/**
+ * Interface extending the subscriptions table type to include the associated price
+ */
 export interface SubscriptionWithPrice extends Tables<"subscriptions"> {
 	prices: Tables<"prices">;
 }
 
+/**
+ * Gets the active subscription for the current user
+ * @returns Promise resolving to the subscription with price details, or undefined if no active subscription
+ * @throws Error if there's a database error
+ */
 export async function getSubscription() {
 	const cookieStore = cookies();
 	const supabase = createClient(cookieStore);
@@ -52,10 +100,18 @@ export async function getSubscription() {
 	return data[0] as SubscriptionWithPrice;
 }
 
+/**
+ * Interface extending the products table type to include associated prices
+ */
 export interface ProductWithPrice extends Tables<"products"> {
 	prices: Tables<"prices">[];
 }
 
+/**
+ * Gets all active products with their associated prices
+ * @returns Promise resolving to array of products with prices
+ * @throws Error if there's a database error
+ */
 export const getActiveProductsWithPrices = async () => {
 	const cookieStore = cookies();
 	const supabase = createClient(cookieStore);
@@ -70,9 +126,12 @@ export const getActiveProductsWithPrices = async () => {
 	return (data as unknown as ProductWithPrice[]) ?? [];
 };
 
-// Get the api calls of a user in the past month/billing period.
-// Once we upgrade to postgres 15, we can use the security_invoker = true
-// and skip the userId here.
+/**
+ * Gets the API call usage statistics for a user in their current billing period
+ * @param userId - The ID of the user to get stats for
+ * @returns Promise resolving to the subscription and calls statistics
+ * @throws Error if no stats found or if there's a database error
+ */
 export async function getSubAndCalls(
 	userId: string
 ): Promise<Tables<"sub_and_calls">> {