feat: add feature flag for AI Chat

Author: santosral

.env.example

@@ -31,3 +31,6 @@ OPENAI_BASE_URL=xxx
 GOOGLE_CLIENT_ID=xxx
 GOOGLE_CLIENT_SECRET=xxx
 GOOGLE_HOSTED_DOMAIN=
+
+# The feature flags.
+FEATURE_AI_CHAT=true

src/routes/(main)/(protected)/+layout.server.ts

@@ -1,5 +1,6 @@
 import { error, redirect } from '@sveltejs/kit';
 
+import { env } from '$env/dynamic/private';
 import { db, type UserFindUniqueArgs, type UserGetPayload } from '$lib/server/db';
 
 import type { LayoutServerLoad } from './$types';
@@ -39,5 +40,8 @@ export const load: LayoutServerLoad = async (event) => {
     username: user.name,
     csrfToken: event.locals.session.csrfToken(),
     onboarded,
+    featureFlags: {
+      aiChat: env.FEATURE_AI_CHAT === 'true',
+    },
   };
 };

src/routes/(main)/(protected)/+layout.svelte

@@ -198,7 +198,9 @@
         />
       {/if}
 
-      <ChatWidget onclick={handleChatWidgetClick} />
+      {#if page.data.featureFlags.aiChat}
+        <ChatWidget onclick={handleChatWidgetClick} />
+      {/if}
     </div>
   </div>
 {/if}
@@ -327,6 +329,8 @@
   </Modal>
 {/if}
 
-<ChatView isopen={isChatViewOpen} onclose={handleChatViewClose} />
+{#if page.data.featureFlags.aiChat}
+  <ChatView isopen={isChatViewOpen} onclose={handleChatViewClose} />
+{/if}
 
 <OnboardingView isopen={isOnboardingViewOpen} onclose={handleOnboardingViewClose} />

src/routes/(main)/api/messages/+server.ts

@@ -1,5 +1,6 @@
 import { json, type RequestHandler } from '@sveltejs/kit';
 
+import { env } from '$env/dynamic/private';
 import { learnerAuth } from '$lib/server/auth';
 import { db, type MessageFindManyArgs, type MessageGetPayload, Role } from '$lib/server/db.js';
 import { DEVELOPER_MESSAGE, openAI } from '$lib/server/openai.js';
@@ -8,6 +9,10 @@ import { search } from '$lib/server/weaviate';
 import type { JSONObject } from '../types';
 
 export const GET: RequestHandler = async (event) => {
+  if (env.FEATURE_AI_CHAT !== 'true') {
+    return json(null, { status: 403 });
+  }
+
   const logger = event.locals.logger.child({ handler: 'api_get_messages' });
 
   const { user } = event.locals.session;
@@ -36,6 +41,10 @@ export const GET: RequestHandler = async (event) => {
 };
 
 export const POST: RequestHandler = async (event) => {
+  if (env.FEATURE_AI_CHAT !== 'true') {
+    return json(null, { status: 403 });
+  }
+
   const logger = event.locals.logger.child({ handler: 'api_create_message' });
 
   if (event.request.headers.get('content-type')?.split(';')[0] !== 'application/json') {
@@ -266,6 +275,10 @@ export const POST: RequestHandler = async (event) => {
 };
 
 export const DELETE: RequestHandler = async (event) => {
+  if (env.FEATURE_AI_CHAT !== 'true') {
+    return json(null, { status: 403 });
+  }
+
   const logger = event.locals.logger.child({ handler: 'api_delete_messages' });
 
   const { user } = event.locals.session;