'use server' import {API_BASE_URL, CLIENT_ID, CLIENT_SECRET, ApiResponse, UnauthorizedError} from '@/lib/api' import {cookies, headers} from 'next/headers' import {redirect} from 'next/navigation' // OAuth令牌缓存 interface TokenCache { token: string expires: number // 过期时间戳 } // ====================== // region device token // ====================== let tokenCache: TokenCache | null = null // 获取OAuth2访问令牌 async function getDeviceToken(forceRefresh = false): Promise { try { // 检查缓存的令牌是否可用 if (!forceRefresh && tokenCache && tokenCache.expires > Date.now()) { return tokenCache.token } const addr = `${API_BASE_URL}/api/auth/token` const body = { client_id: CLIENT_ID, client_secret: CLIENT_SECRET, grant_type: 'client_credentials', } const response = await fetch(addr, { method: 'POST', headers: { 'Content-Type': 'application/json', }, body: JSON.stringify(body), }) if (!response.ok) { throw new Error(`OAuth token request failed: ${response.status} ${await response.text()}`) } const data = await response.json() // 缓存令牌和过期时间 // 通常后端会返回expires_in(秒为单位) tokenCache = { token: data.access_token, expires: Date.now() + data.expires_in * 1000, } return tokenCache.token } catch (error) { console.error('Failed to get access token:', error) throw new Error('认证服务暂时不可用') } } // 通用的API调用函数 async function callByDevice(endpoint: string, data: unknown): Promise> { try { // 发送请求 let accessToken = getDeviceToken() const requestOptions = { method: 'POST', headers: { 'Content-Type': 'application/json', 'Authorization': `Bearer ${await accessToken}`, }, body: JSON.stringify(data), } let response = await fetch(`${API_BASE_URL}${endpoint}`, requestOptions) // 如果返回401未授权,尝试刷新令牌并重试一次 if (response.status === 401) { accessToken = getDeviceToken(true) // 强制刷新令牌 // 使用新令牌重试请求 requestOptions.headers['Authorization'] = `Bearer ${await accessToken}` response = await fetch(`${API_BASE_URL}${endpoint}`, requestOptions) } // 检查响应状态 if (!response.ok) { console.log('响应不成功', `status=${response.status}`, await response.text()) return { success: false, status: response.status, message: '请求失败', } } // 检查响应状态 return handleResponse(response) } catch (e) { console.error('API call failed:', e) throw new Error('服务调用失败', {cause: e}) } } // endregion // ====================== // region user token // ====================== async function getUserToken(refresh = false): Promise { // 从 cookie 中获取用户令牌 const cookie = await cookies() const userToken = cookie.get('auth_token')?.value const userRefresh = cookie.get('auth_refresh')?.value // 检查缓存的令牌是否可用 if (!refresh && userToken) { return userToken } // 如果没有刷新令牌,抛出异常 if (!userRefresh) { throw UnauthorizedError } // 请求刷新访问令牌 const addr = `${API_BASE_URL}/api/auth/token` const body = { grant_type: 'refresh_token', client_id: CLIENT_ID, client_secret: CLIENT_SECRET, refresh_token: userRefresh, } const response = await fetch(addr, { method: 'POST', headers: { 'Content-Type': 'application/json', }, body: JSON.stringify(body), }) if (!response.ok) { console.log('刷新令牌失败', `status=${response.status}`, await response.text()) throw UnauthorizedError } // 保存新的用户令牌到 cookie const data = await response.json() const nextAccessToken = data.access_token const nextRefreshToken = data.refresh_token const expiresIn = data.expires_in cookie.set('auth_token', nextAccessToken, { httpOnly: true, sameSite: 'strict', maxAge: Math.max(expiresIn, 0), }) cookie.set('auth_refresh', nextRefreshToken, { httpOnly: true, sameSite: 'strict', maxAge: 7 * 24 * 3600, // 7天 }) return nextAccessToken } // 使用用户令牌的API调用函数 async function callByUser( endpoint: string, data?: unknown, ): Promise> { try { let token = await getUserToken() const header = await headers() // 获取客户端 IP const clientIp = header.get('x-forwarded-for') // 发送请求 const request = { method: 'POST', headers: { 'Content-Type': 'application/json', 'Authorization': `Bearer ${token}`, } as Record, body: data ? JSON.stringify(data) : undefined, } if (clientIp) { request.headers['X-Forwarded-For'] = clientIp } let response = await fetch(`${API_BASE_URL}${endpoint}`, request) if (response.status === 401) { token = await getUserToken(true) request.headers['Authorization'] = `Bearer ${token}` response = await fetch(`${API_BASE_URL}${endpoint}`, request) } // 检查响应状态 if (!response.ok) { const body = await response.text() console.log('响应不成功', `status=${response.status}`, body) if (response.status === 401) { return redirect('/login') } if (response.status >= 400 && response.status < 500) { return { status: response.status, success: false, message: body, } } if (response.status >= 500) { return { status: response.status, success: false, message: '服务器错误', } } return { status: response.status, success: false, message: `请求失败,status = ${response.status}`, } } return handleResponse(response) } catch (e) { console.error('API call with user token failed:', e) throw new Error('服务调用失败', {cause: e}) } } // endregion // ====================== // region no token // ====================== // 不需要令牌的公共API调用函数 async function callPublic( endpoint: string, data?: unknown, ): Promise> { try { // 发送请求 const requestOptions: RequestInit = { method: 'POST', headers: { 'Content-Type': 'application/json', }, body: data ? JSON.stringify(data) : undefined, } const response = await fetch(`${API_BASE_URL}${endpoint}`, requestOptions) // 检查响应状态 if (!response.ok) { console.log('公共接口响应不成功', `status=${response.status}`, await response.text()) return { status: response.status, success: false, message: response.status >= 500 ? '服务器错误' : '请求失败', } } return handleResponse(response) } catch (e) { console.error('Public API call failed:', e) throw new Error('服务调用失败', { cause: e }) } } // endregion // 统一响应解析 async function handleResponse(response: Response): Promise> { // 解析响应数据 const type = response.headers.get('Content-Type') ?? 'text/plain' if (type.indexOf('application/json') !== -1) { const json = await response.json() if (!response.ok) { console.log('响应不成功', `status=${response.status}`, json) return { success: false, status: response.status, message: json.message || '请求失败', } } return { success: true, data: json, } } else if (type.indexOf('text/plain') !== -1) { const text = await response.text() if (!response.ok) { console.log('响应不成功', `status=${response.status}`, text) return { success: false, status: response.status, message: text || '请求失败', } } return { success: true, data: undefined as unknown as R, // 强转类型,考虑优化 } } else { throw new Error(`无法解析响应数据,未处理的 Content-Type: ${type}`) } } // 预定义错误 // 导出 export { getDeviceToken, getUserToken, callByDevice, callByUser, callPublic, }