Add user agent header

Add user agent header

Author: afrinxnahar

apps/api/src/main.ts

@@ -1,10 +1,24 @@
 import { NestFactory } from '@nestjs/core';
 import { DocumentBuilder, SwaggerModule } from '@nestjs/swagger';
+import axios from 'axios';
+import {
+  CREATOR_AI_USER_AGENT,
+  installCreatorAiAxiosDefaults,
+  installCreatorAiFetchDefaults,
+} from '@repo/validation';
 import { AppModule } from './app.module';
 
+installCreatorAiFetchDefaults();
+installCreatorAiAxiosDefaults(axios);
+
 async function bootstrap() {
   const app = await NestFactory.create(AppModule, { rawBody: true });
 
+  app.use((_req, res, next) => {
+    res.setHeader('Server', CREATOR_AI_USER_AGENT);
+    next();
+  });
+
   const isWorker = process.env.NODE_ENV === 'worker';
 
   if (isWorker) {

apps/api/src/youtube/youtube.controller.ts

@@ -10,7 +10,7 @@ import { getUserId } from '../common/get-user-id';
 @Controller('youtube')
 @UseGuards(SupabaseAuthGuard)
 export class YoutubeController {
-  constructor(private readonly youtubeService: YoutubeService) {}
+  constructor(private readonly youtubeService: YoutubeService) { }
 
   @Get('video-metadata')
   @ApiOperation({ summary: 'Resolve YouTube video metadata from URL' })

apps/api/src/youtube/youtube.service.ts

@@ -169,7 +169,7 @@ export class YoutubeService {
       if (!usageData.allowed) {
         throw new BadRequestException(usageData.message);
       }
-     
+
       const { data: channel, error } = await this.supabase
         .from('youtube_channels')
         .select('channel_id, provider_token, refresh_token')

apps/web/app/api/auth/callback/route.ts

@@ -2,6 +2,7 @@ import { type EmailOtpType } from '@supabase/supabase-js';
 import { type NextRequest } from 'next/server';
 import { redirect } from 'next/navigation';
 import { getSupabaseServer } from '@/lib/supabase/server';
+import { creatorAiFetch } from '@/lib/creator-ai-http';
 import { Resend } from 'resend';
 import { BACKEND_URL } from '@/lib/constants';
 
@@ -104,7 +105,7 @@ async function processReferral(
   userEmail: string,
 ) {
   try {
-    const response = await fetch(`${BACKEND_URL}/api/v1/referral/track`, {
+    const response = await creatorAiFetch(`${BACKEND_URL}/api/v1/referral/track`, {
       method: 'POST',
       headers: { 'Content-Type': 'application/json' },
       body: JSON.stringify({ referralCode, userEmail }),

apps/web/instrumentation.ts

@@ -0,0 +1,7 @@
+export async function register() {
+  if (process.env.NEXT_RUNTIME !== 'nodejs') return;
+
+  const axios = (await import('axios')).default;
+  const { installCreatorAiAxiosDefaults } = await import('@repo/validation');
+  installCreatorAiAxiosDefaults(axios);
+}

apps/web/lib/api-client.ts

@@ -10,6 +10,7 @@ import axios, {
   AxiosProgressEvent
 } from 'axios';
 import { createClient } from "@/lib/supabase/client";
+import { CREATOR_AI_USER_AGENT } from "@repo/validation";
 
 
 export interface ApiErrorResponse {
@@ -88,6 +89,8 @@ const axiosInstance = axios.create({
   baseURL: getBackendUrl(),
   headers: {
     'Content-Type': 'application/json',
+    'User-Agent': CREATOR_AI_USER_AGENT,
+    'X-Creator-AI-Agent': CREATOR_AI_USER_AGENT,
   },
 });
 

apps/web/lib/creator-ai-http.ts

@@ -0,0 +1,16 @@
+import {
+  CREATOR_AI_USER_AGENT,
+  mergeCreatorAiHeaders,
+} from '@repo/validation';
+
+export { CREATOR_AI_USER_AGENT, mergeCreatorAiHeaders };
+
+export function creatorAiFetch(
+  input: RequestInfo | URL,
+  init?: RequestInit,
+): Promise<Response> {
+  return fetch(input, {
+    ...init,
+    headers: mergeCreatorAiHeaders(init?.headers),
+  });
+}

apps/web/middleware.ts

@@ -1,4 +1,5 @@
 import { createServerClient } from "@supabase/ssr"
+import { CREATOR_AI_USER_AGENT, type UserRole } from "@repo/validation"
 import { NextResponse } from "next/server"
 import type { NextRequest } from "next/server"
 
@@ -10,14 +11,19 @@ const BYPASS_PATHS = [
   "/api/youtube/callback",
 ]
 
-async function getUserRole(supabase: ReturnType<typeof createServerClient>, userId: string) {
+async function getUserRole(
+  supabase: ReturnType<typeof createServerClient>,
+  userId: string,
+): Promise<UserRole | undefined> {
   try {
-    const { data } = await supabase
+    const { data, error } = await supabase
       .from("profiles")
       .select("role")
       .eq("user_id", userId)
-      .single()
-    return data?.role as string | undefined
+      .single<{ role: UserRole }>()
+
+    if (error || !data) return undefined
+    return data.role
   } catch {
     return undefined
   }
@@ -27,10 +33,13 @@ export async function middleware(request: NextRequest) {
   const { pathname } = request.nextUrl
 
   if (BYPASS_PATHS.includes(pathname)) {
-    return NextResponse.next()
+    const response = NextResponse.next()
+    response.headers.set("Server", CREATOR_AI_USER_AGENT)
+    return response
   }
 
   const response = NextResponse.next()
+  response.headers.set("Server", CREATOR_AI_USER_AGENT)
 
   const supabase = createServerClient(
     process.env.NEXT_PUBLIC_SUPABASE_URL!,

apps/web/next.config.mjs

@@ -33,6 +33,7 @@ const nextConfig = {
       {
         source: "/(.*)",
         headers: [
+          { key: "Server", value: "CreatorAI/1.0 (+https://tryscriptai.com)" },
           { key: "X-Content-Type-Options", value: "nosniff" },
           { key: "X-Frame-Options", value: "SAMEORIGIN" },
           { key: "Referrer-Policy", value: "strict-origin-when-cross-origin" },

packages/train-ai-worker/src/main.ts

@@ -1,7 +1,15 @@
 import { NestFactory } from '@nestjs/core';
 import { DocumentBuilder, SwaggerModule } from '@nestjs/swagger';
+import axios from 'axios';
+import {
+  installCreatorAiAxiosDefaults,
+  installCreatorAiFetchDefaults,
+} from '@repo/validation';
 import { WorkerModule } from './worker.module';
 
+installCreatorAiFetchDefaults();
+installCreatorAiAxiosDefaults(axios);
+
 async function bootstrap() {
   const app = await NestFactory.create(WorkerModule);