feat: add Authorization header support (#170)

Author: JounQin

.changeset/metal-adults-greet.md

@@ -0,0 +1,5 @@
+---
+"@alauda/doom": patch
+---
+
+feat: add Authorization header support

packages/doom/src/cli/load-config.ts

@@ -344,7 +344,6 @@ const getCommonConfig = async ({
       server: {
         open,
         proxy: {
-          '/api/v1': 'https://cloud.alauda.cn',
           '/smart/api': 'https://docs-dev.alauda.cn',
         },
       },

packages/doom/src/global/Intelligence/AIAssistant/Preamble/LoginForm/index.tsx

@@ -147,9 +147,9 @@ export const LoginForm = ({ onSubmit, ...props }: LoginFormProps) => {
           {CLOUD_AUTH_ORIGINS.map(({ name, value }) => (
             <Radio
               key={name}
-              label={name === 'local' ? 'Local' : t(`customer_portal_${name}`)}
+              label={t(`customer_portal_${name}`)}
               value={value}
-            ></Radio>
+            />
           ))}
         </RadioGroup>
       </FormItem>

packages/doom/src/global/Intelligence/AIAssistant/index.tsx

@@ -1,7 +1,13 @@
 import { clsx } from 'clsx'
-import { useRef, useState } from 'react'
+import { useEffect, useRef, useState } from 'react'
 import { Tooltip } from 'react-tooltip'
-import { ApiMethod, xfetch } from 'x-fetch'
+import {
+  ApiMethod,
+  interceptors,
+  ResponseError,
+  xfetch,
+  type ApiInterceptor,
+} from 'x-fetch'
 
 import { type CloudAuth, useCloudAuth } from '../context.tsx'
 import type { AuthInfo } from '../types.ts'
@@ -54,6 +60,36 @@ export const AIAssistant = ({ open, onOpenChange }: AIAssistantProps) => {
     onNewChat()
   })
 
+  useEffect(() => {
+    const interceptor: ApiInterceptor = async (req, next) => {
+      if (!req.url.startsWith('/smart/')) {
+        return next(req)
+      }
+      if (!req.headers.has('Authorization')) {
+        req.headers.set('Authorization', `Bearer ${authInfo!.token}`)
+      }
+      if (!req.headers.has('CLOUD_AUTH_ORIGIN')) {
+        req.headers.set('CLOUD_AUTH_ORIGIN', authInfo!.origin)
+      }
+      try {
+        return await next(req)
+      } catch (err) {
+        if (
+          err instanceof ResponseError &&
+          // type-coverage:ignore-next-line -- no idea
+          err.response.status === 401
+        ) {
+          onLogout()
+        }
+        throw err
+      }
+    }
+    interceptors.use(interceptor)
+    return () => {
+      interceptors.eject(interceptor)
+    }
+  }, [authInfo, onLogout])
+
   const onClose = useMemoizedFn(() => {
     onOpenChange(false)
   })
@@ -93,7 +129,6 @@ export const AIAssistant = ({ open, onOpenChange }: AIAssistantProps) => {
         '/smart/api/new_session',
         {
           method: ApiMethod.POST,
-          headers: { username: authInfo!.detail!.user.name },
         },
       )
       sessionIdRef.current = session_id
@@ -152,6 +187,9 @@ export const AIAssistant = ({ open, onOpenChange }: AIAssistantProps) => {
     try {
       await onSend_(content)
     } catch {
+      if (!sessionIdRef.current) {
+        return
+      }
       flushMessages((messages) => {
         const index = assistantMessageIndexRef.current
         return [

packages/doom/src/global/Intelligence/constants.ts

@@ -1,5 +1,3 @@
-import { isProduction } from '@rspress/shared'
-
 import type { CloudAuthRegion } from './types.js'
 
 export const CLOUD_AUTH_ORIGIN_KEY = '__CLOUD_AUTH_ORIGIN__'
@@ -16,13 +14,6 @@ const CLOUD_AUTH_ORIGINS: CloudAuthRegion[] = [
   },
 ]
 
-if (!isProduction()) {
-  CLOUD_AUTH_ORIGINS.unshift({
-    name: 'local',
-    value: '',
-  })
-}
-
 export const CLOUD_AUTH_ORIGIN_VALUES = CLOUD_AUTH_ORIGINS.map(
   ({ value }) => value,
 )

packages/doom/src/global/Intelligence/context.tsx

@@ -6,6 +6,7 @@ import { getAuthInfoFromToken, setLocalStorage } from './utils.js'
 
 export interface CloudAuth {
   origin: string
+  token?: string
   detail?: AuthInfo
 }
 
@@ -30,7 +31,7 @@ const getCloudAuth = (): CloudAuth | null => {
     return { origin }
   }
 
-  return { origin, detail: getAuthInfoFromToken(token) }
+  return { origin, token, detail: getAuthInfoFromToken(token) }
 }
 
 export const CloudAuthProvider = ({ children }: { children: ReactNode }) => {

packages/doom/src/global/Intelligence/types.ts

@@ -34,6 +34,6 @@ export interface AuthInfo {
 }
 
 export interface CloudAuthRegion {
-  name: 'global' | 'china' | 'local'
+  name: 'global' | 'china'
   value: string
 }