实现登录功能

Signed-off-by: luorijun <luorijun@outlook.com>
2025-12-29 14:09:13 +08:00
parent f9a8df0fe5
commit bb8aec8ce5
19 changed files with 1079 additions and 110 deletions
--- a/src/actions/base.ts
+++ b/src/actions/base.ts
@@ -0,0 +1,184 @@
+"use server"
+import { cookies, headers } from "next/headers"
+import { redirect } from "next/navigation"
+import { cache } from "react"
+import {
+  API_BASE_URL,
+  type ApiResponse,
+  CLIENT_ID,
+  CLIENT_SECRET,
+} from "@/lib/api"
+
+// ======================
+// public
+// ======================
+
+async function callPublic<R = undefined>(
+  endpoint: string,
+  data?: unknown,
+): Promise<ApiResponse<R>> {
+  return _callPublic(endpoint, data ? JSON.stringify(data) : undefined)
+}
+
+const _callPublic = cache(
+  async <R = undefined>(
+    endpoint: string,
+    data?: string,
+  ): Promise<ApiResponse<R>> => {
+    return call(`${API_BASE_URL}${endpoint}`, data)
+  },
+)
+
+// ======================
+// device
+// ======================
+
+async function callByDevice<R = undefined>(
+  endpoint: string,
+  data: unknown,
+): Promise<ApiResponse<R>> {
+  return _callByDevice(endpoint, data ? JSON.stringify(data) : undefined)
+}
+
+const _callByDevice = cache(
+  async <R = undefined>(
+    endpoint: string,
+    data?: string,
+  ): Promise<ApiResponse<R>> => {
+    // 获取设备令牌
+    if (!CLIENT_ID || !CLIENT_SECRET) {
+      return {
+        success: false,
+        status: 401,
+        message: "未配置 CLIENT_ID 或 CLIENT_SECRET",
+      }
+    }
+    const token = Buffer.from(`${CLIENT_ID}:${CLIENT_SECRET}`).toString(
+      "base64url",
+    )
+
+    // 发起请求
+    return call(`${API_BASE_URL}${endpoint}`, data, `Basic ${token}`)
+  },
+)
+
+// ======================
+// user
+// ======================
+
+async function callByUser<R = undefined>(
+  endpoint: string,
+  data?: unknown,
+): Promise<ApiResponse<R>> {
+  return _callByUser(endpoint, data ? JSON.stringify(data) : undefined)
+}
+
+const _callByUser = cache(
+  async <R = undefined>(
+    endpoint: string,
+    data?: string,
+  ): Promise<ApiResponse<R>> => {
+    // 获取用户令牌
+    const cookie = await cookies()
+    const token = cookie.get("auth_token")?.value
+    if (!token) {
+      return {
+        success: false,
+        status: 401,
+        message: "会话已失效",
+      }
+    }
+
+    // 发起请求
+    return await call<R>(`${API_BASE_URL}${endpoint}`, data, `Bearer ${token}`)
+  },
+)
+
+// ======================
+// call
+// ======================
+
+async function call<R = undefined>(
+  url: string,
+  body: RequestInit["body"],
+  auth?: string,
+): Promise<ApiResponse<R>> {
+  let response: Response
+  try {
+    const reqHeaders = await headers()
+    const reqIP = reqHeaders.get("x-forwarded-for")
+    const reqUA = reqHeaders.get("user-agent")
+    const callHeaders: RequestInit["headers"] = {
+      "Content-Type": "application/json",
+    }
+    if (auth) callHeaders["Authorization"] = auth
+    if (reqIP) callHeaders["X-Forwarded-For"] = reqIP
+    if (reqUA) callHeaders["User-Agent"] = reqUA
+
+    response = await fetch(url, {
+      method: "POST",
+      headers: callHeaders,
+      body,
+    })
+  } catch (e) {
+    console.error("后端请求失败", url, (e as Error).message)
+    throw new Error(`请求失败，网络错误`)
+  }
+
+  const type = response.headers.get("Content-Type") ?? "text/plain"
+  if (type.indexOf("text/plain") !== -1) {
+    const text = await response.text()
+    if (!response.ok) {
+      console.log("后端请求失败", url, `status=${response.status}`, text)
+      return {
+        success: false,
+        status: response.status,
+        message: text || "请求失败",
+      }
+    }
+
+    if (text?.trim()?.length) {
+      console.log("未处理的响应成功", `type=text`, `text=${text}`)
+    }
+    return {
+      success: true,
+      data: undefined as R, // 强转类型，考虑优化
+    }
+  } else if (type.indexOf("application/json") !== -1) {
+    const json = await response.json()
+    if (!response.ok) {
+      console.log("后端请求失败", url, `status=${response.status}`, json)
+
+      return {
+        success: false,
+        status: response.status,
+        message: json.message || json.error_description || "请求失败", // 业务错误（message）或者 oauth 错误（error_description）
+      }
+    }
+    return {
+      success: true,
+      data: json,
+    }
+  }
+
+  throw new Error(`无法解析响应数据，未处理的 Content-Type: ${type}`)
+}
+
+async function postCall<R = undefined>(rawResp: Promise<ApiResponse<R>>) {
+  const header = await headers()
+  const pathname = header.get("x-pathname") || "/"
+  const resp = await rawResp
+
+  // 重定向到登录页
+  const match = [/^\/admin.*/].some(item => item.test(pathname))
+
+  if (match && !resp.success && resp.status === 401) {
+    console.log("🚗🚗🚗🚗🚗 非正常重定向 🚗🚗🚗🚗🚗")
+    redirect("/login?force=true")
+  }
+
+  return resp
+}
+
+// 导出
+export { callPublic, callByDevice, callByUser }