25:05:05 10:41:39 v0.3.7

.gitattributes

@@ -0,0 +1 @@
+*.png filter=lfs diff=lfs merge=lfs -text

.github/workflows/docker-publish.yml

@@ -0,0 +1,80 @@
+name: Build and Publish Docker Image
+
+on:
+  release:
+    types: [published]
+  pull_request:
+    branches: [main]
+
+env:
+  REGISTRY: ghcr.io
+  IMAGE_NAME: ${{ github.repository }}
+  DOCKERHUB_REGISTRY: docker.io
+  DOCKERHUB_IMAGE_NAME: variantconst/openwebui-monitor
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+      # 检出代码
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      # 设置 QEMU（支持多平台）
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+
+      # 设置 Docker Buildx（支持多平台构建）
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      # 登录到 GitHub Container Registry (GHCR)
+      - name: Log into GHCR
+        if: github.event_name != 'pull_request'
+        uses: docker/login-action@v3
+        with:
+          registry: ${{ env.REGISTRY }}
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      # 登录到 Docker Hub
+      - name: Log into Docker Hub
+        if: github.event_name != 'pull_request'
+        uses: docker/login-action@v3
+        with:
+          registry: ${{ env.DOCKERHUB_REGISTRY }}
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+
+      # 提取 Docker 元数据
+      - name: Extract Docker metadata
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          images: |
+            ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
+            ${{ env.DOCKERHUB_REGISTRY }}/${{ env.DOCKERHUB_IMAGE_NAME }}
+          tags: |
+            type=ref,event=branch
+            type=ref,event=pr
+            type=semver,pattern={{version}}
+            type=semver,pattern={{major}}.{{minor}}
+            type=sha
+            type=raw,value=latest,enable={{is_default_branch}}
+
+      # 构建并推送到 GHCR 和 Docker Hub
+      - name: Build and push Docker image
+        uses: docker/build-push-action@v5
+        with:
+          context: .
+          platforms: linux/amd64,linux/arm64
+          push: ${{ github.event_name != 'pull_request' }}
+          tags: ${{ steps.meta.outputs.tags }}
+          labels: ${{ steps.meta.outputs.labels }}
+          cache-from: type=gha
+          cache-to: type=gha,mode=max
+          provenance: false

VERSION

@@ -0,0 +1 @@
+v0.3.7

app/api/config/route.ts

@@ -1,23 +0,0 @@
-import { NextResponse } from "next/server";
-import { headers } from "next/headers";
-
-export async function GET() {
-  const headersList = headers();
-  const token = headersList.get("authorization")?.split(" ")[1];
-  const expectedToken = process.env.ACCESS_TOKEN;
-
-  if (!token || token !== expectedToken) {
-    return NextResponse.json(
-      {
-        apiKey: "Unauthorized",
-        status: 401,
-      },
-      { status: 401 }
-    );
-  }
-
-  return NextResponse.json({
-    apiKey: process.env.API_KEY || "Unconfigured",
-    status: 200,
-  });
-}

app/api/init/route.ts

@@ -0,0 +1,22 @@
+import { NextResponse } from "next/server";
+import { initDatabase } from "@/lib/db/client";
+
+let initialized = false;
+
+export async function GET() {
+  if (!initialized) {
+    try {
+      await initDatabase();
+      initialized = true;
+      return NextResponse.json({ success: true, message: "数据库初始化成功" });
+    } catch (error) {
+      console.error("数据库初始化失败:", error);
+      return NextResponse.json(
+        { success: false, error: "数据库初始化失败" },
+        { status: 500 }
+      );
+    }
+  } else {
+    return NextResponse.json({ success: true, message: "数据库已初始化" });
+  }
+}

app/api/{config → v1/config}/key/route.ts

@@ -1,7 +1,12 @@
 import { NextResponse } from "next/server";
-import { cookies } from "next/headers";
+import { verifyApiToken } from "@/lib/auth";
+
+export async function GET(req: Request) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
 
-export async function GET() {
   const apiKey = process.env.API_KEY;
 
   if (!apiKey) {

app/api/v1/config/route.ts

@@ -0,0 +1,14 @@
+import { NextResponse } from "next/server";
+import { verifyApiToken } from "@/lib/auth";
+
+export async function GET(req: Request) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
+
+  return NextResponse.json({
+    apiKey: process.env.API_KEY || "Unconfigured",
+    status: 200,
+  });
+}

app/api/v1/inlet/route.ts

@@ -9,7 +9,6 @@ export async function POST(req: Request) {
     const user = await getOrCreateUser(data.user);
     const modelId = data.body?.model;
 
-    // 如果用户被拉黑,返回余额为 -1
     if (user.deleted) {
       return NextResponse.json({
         success: true,
@@ -18,10 +17,8 @@ export async function POST(req: Request) {
       });
     }
 
-    // 获取预扣费金额
     const inletCost = getModelInletCost(modelId);
 
-    // 预扣费
     if (inletCost > 0) {
       const userResult = await query(
         `UPDATE users 

app/api/v1/models/price/route.ts

@@ -1,5 +1,6 @@
 import { NextRequest, NextResponse } from "next/server";
-import { updateModelPrice } from "@/lib/db";
+import { updateModelPrice } from "@/lib/db/client";
+import { verifyApiToken } from "@/lib/auth";
 
 interface PriceUpdate {
   id: string;
@@ -9,11 +10,15 @@ interface PriceUpdate {
 }
 
 export async function POST(request: NextRequest) {
+  const authError = verifyApiToken(request);
+  if (authError) {
+    return authError;
+  }
+
   try {
     const data = await request.json();
     console.log("Raw data received:", data);
 
-    // 从对象中提取模型数组
     const updates = data.updates || data;
     if (!Array.isArray(updates)) {
       console.error("Invalid data format - expected array:", updates);
@@ -23,7 +28,6 @@ export async function POST(request: NextRequest) {
       );
     }
 
-    // 验证并转换数据格式
     const validUpdates = updates
       .map((update: any) => ({
         id: update.id,
@@ -52,7 +56,6 @@ export async function POST(request: NextRequest) {
       `Successfully verified price updating requests of ${validUpdates.length} models`
     );
 
-    // 执行批量更新并收集结果
     const results = await Promise.all(
       validUpdates.map(async (update: PriceUpdate) => {
         try {

app/api/v1/models/route.ts

@@ -1,5 +1,6 @@
 import { NextResponse } from "next/server";
-import { ensureTablesExist, getOrCreateModelPrices } from "@/lib/db";
+import { ensureTablesExist, getOrCreateModelPrices } from "@/lib/db/client";
+import { verifyApiToken } from "@/lib/auth";
 
 interface ModelInfo {
   id: string;
@@ -21,9 +22,13 @@ interface ModelResponse {
   }[];
 }
 
-export async function GET() {
+export async function GET(req: Request) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
+
   try {
-    // Ensure database is initialized
     await ensureTablesExist();
 
     const domain = process.env.OPENWEBUI_DOMAIN;
@@ -31,7 +36,6 @@ export async function GET() {
       throw new Error("OPENWEBUI_DOMAIN environment variable is not set.");
     }
 
-    // Normalize API URL
     const apiUrl = domain.replace(/\/+$/, "") + "/api/models";
 
     const response = await fetch(apiUrl, {
@@ -47,9 +51,7 @@ export async function GET() {
       throw new Error(`Failed to fetch models: ${response.status}`);
     }
 
-    // Get response text for debugging
     const responseText = await response.text();
-    // console.log("API response:", responseText);
 
     let data: ModelResponse;
     try {
@@ -59,20 +61,25 @@ export async function GET() {
       throw new Error("Invalid JSON response from API");
     }
 
-    console.log("data:", data);
-
     if (!data || !Array.isArray(data.data)) {
       console.error("Unexpected API response structure:", data);
       throw new Error("Unexpected API response structure");
     }
 
-    // Get price information for all models
+    const apiModelsMap = new Map();
+    data.data.forEach((item) => {
+      apiModelsMap.set(String(item.id), {
+        name: String(item.name),
+        base_model_id: item.info?.base_model_id || "",
+        imageUrl: item.info?.meta?.profile_image_url || "/static/favicon.png",
+        system_prompt: item.info?.params?.system || "",
+      });
+    });
+
     const modelsWithPrices = await getOrCreateModelPrices(
       data.data.map((item) => {
-        // 处理形如 gemini_search.gemini-2.0-flash 的派生模型ID
         let baseModelId = item.info?.base_model_id;
 
-        // 如果没有明确的base_model_id，尝试从ID中提取
         if (!baseModelId && item.id) {
           const idParts = String(item.id).split(".");
           if (idParts.length > 1) {
@@ -88,30 +95,46 @@ export async function GET() {
       })
     );
 
-    const validModels = data.data.map((item, index) => {
-      // 处理形如 gemini_search.gemini-2.0-flash 的派生模型ID
-      let baseModelId = item.info?.base_model_id || "";
+    const dbModelsMap = new Map();
+    modelsWithPrices.forEach((model) => {
+      dbModelsMap.set(model.id, {
+        input_price: model.input_price,
+        output_price: model.output_price,
+        per_msg_price: model.per_msg_price,
+        updated_at: model.updated_at,
+      });
+    });
+
+    const validModels = Array.from(apiModelsMap.entries()).map(
+      ([id, apiModel]) => {
+        const dbModel = dbModelsMap.get(id) || {
+          input_price: 60,
+          output_price: 60,
+          per_msg_price: -1,
+          updated_at: new Date(),
+        };
 
-      // 如果没有明确的base_model_id，尝试从ID中提取
-      if (!baseModelId && item.id) {
-        const idParts = String(item.id).split(".");
-        if (idParts.length > 1) {
-          baseModelId = idParts[idParts.length - 1];
+        let baseModelId = apiModel.base_model_id;
+        if (!baseModelId && id) {
+          const idParts = String(id).split(".");
+          if (idParts.length > 1) {
+            baseModelId = idParts[idParts.length - 1];
+          }
         }
-      }
 
-      return {
-        id: modelsWithPrices[index].id,
-        base_model_id: baseModelId,
-        name: modelsWithPrices[index].name,
-        imageUrl: item.info?.meta?.profile_image_url || "/static/favicon.png",
-        system_prompt: item.info?.params?.system || "",
-        input_price: modelsWithPrices[index].input_price,
-        output_price: modelsWithPrices[index].output_price,
-        per_msg_price: modelsWithPrices[index].per_msg_price,
-        updated_at: modelsWithPrices[index].updated_at,
-      };
-    });
+        return {
+          id: id,
+          base_model_id: baseModelId,
+          name: apiModel.name,
+          imageUrl: apiModel.imageUrl,
+          system_prompt: apiModel.system_prompt,
+          input_price: dbModel.input_price,
+          output_price: dbModel.output_price,
+          per_msg_price: dbModel.per_msg_price,
+          updated_at: dbModel.updated_at,
+        };
+      }
+    );
 
     return NextResponse.json(validModels);
   } catch (error) {
@@ -126,8 +149,12 @@ export async function GET() {
   }
 }
 
-// Add inlet endpoint
 export async function POST(req: Request) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
+
   const data = await req.json();
 
   return new Response("Inlet placeholder response", {
@@ -135,10 +162,13 @@ export async function POST(req: Request) {
   });
 }
 
-// Add outlet endpoint
 export async function PUT(req: Request) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
+
   const data = await req.json();
-  // console.log("Outlet received:", JSON.stringify(data, null, 2));
 
   return new Response("Outlet placeholder response", {
     headers: { "Content-Type": "application/json" },

app/api/v1/models/sync-all-prices/route.ts

@@ -1,11 +1,16 @@
 import { NextRequest, NextResponse } from "next/server";
-import { pool } from "@/lib/db";
+import { pool } from "@/lib/db/client";
+import { verifyApiToken } from "@/lib/auth";
 
 export async function POST(request: NextRequest) {
+  const authError = verifyApiToken(request);
+  if (authError) {
+    return authError;
+  }
+
   try {
     const client = await pool.connect();
     try {
-      // 1. 获取所有有效的派生模型（base_model_id 存在且在数据库中有对应记录）
       const derivedModelsResult = await client.query(`
         SELECT d.id, d.name, d.base_model_id 
         FROM model_prices d
@@ -24,10 +29,8 @@ export async function POST(request: NextRequest) {
       const derivedModels = derivedModelsResult.rows;
       const syncResults = [];
 
-      // 2. 为每个派生模型同步价格
       for (const derivedModel of derivedModels) {
         try {
-          // 获取上游模型价格
           const baseModelResult = await client.query(
             `SELECT input_price, output_price, per_msg_price FROM model_prices WHERE id = $1`,
             [derivedModel.base_model_id]
@@ -45,7 +48,6 @@ export async function POST(request: NextRequest) {
 
           const baseModel = baseModelResult.rows[0];
 
-          // 更新派生模型价格
           const updateResult = await client.query(
             `UPDATE model_prices 
              SET 

app/api/v1/models/sync-price/route.ts

@@ -1,7 +1,13 @@
 import { NextRequest, NextResponse } from "next/server";
-import { pool } from "@/lib/db";
+import { pool } from "@/lib/db/client";
+import { verifyApiToken } from "@/lib/auth";
 
 export async function POST(request: NextRequest) {
+  const authError = verifyApiToken(request);
+  if (authError) {
+    return authError;
+  }
+
   try {
     const data = await request.json();
     const { modelId } = data;
@@ -15,7 +21,6 @@ export async function POST(request: NextRequest) {
 
     const client = await pool.connect();
     try {
-      // 1. 获取派生模型信息
       const derivedModelResult = await client.query(
         `SELECT id, name, base_model_id FROM model_prices WHERE id = $1`,
         [modelId]
@@ -28,13 +33,11 @@ export async function POST(request: NextRequest) {
       const derivedModel = derivedModelResult.rows[0];
       let baseModelId = derivedModel.base_model_id;
 
-      // 如果数据库中没有base_model_id，尝试从ID中提取
       if (!baseModelId) {
         const idParts = modelId.split(".");
         if (idParts.length > 1) {
           baseModelId = idParts[idParts.length - 1];
 
-          // 更新数据库中的base_model_id
           await client.query(
             `UPDATE model_prices SET base_model_id = $2 WHERE id = $1`,
             [modelId, baseModelId]
@@ -49,7 +52,6 @@ export async function POST(request: NextRequest) {
         );
       }
 
-      // 2. 获取上游模型价格
       const baseModelResult = await client.query(
         `SELECT input_price, output_price, per_msg_price FROM model_prices WHERE id = $1`,
         [baseModelId]
@@ -64,7 +66,6 @@ export async function POST(request: NextRequest) {
 
       const baseModel = baseModelResult.rows[0];
 
-      // 3. 更新派生模型价格
       const updateResult = await client.query(
         `UPDATE model_prices 
          SET 

app/api/v1/models/test/route.ts

@@ -1,6 +1,12 @@
 import { NextResponse } from "next/server";
+import { verifyApiToken } from "@/lib/auth";
 
 export async function POST(req: Request) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
+
   try {
     const { modelId } = await req.json();
 

app/api/v1/outlet/route.ts

@@ -34,7 +34,6 @@ async function getModelPrice(modelId: string): Promise<ModelPrice | null> {
     return result.rows[0];
   }
 
-  // If no price is found in the database, use the default price
   const defaultInputPrice = parseFloat(
     process.env.DEFAULT_MODEL_INPUT_PRICE || "60"
   );
@@ -42,7 +41,6 @@ async function getModelPrice(modelId: string): Promise<ModelPrice | null> {
     process.env.DEFAULT_MODEL_OUTPUT_PRICE || "60"
   );
 
-  // Verify that the default price is a valid non-negative number
   if (
     isNaN(defaultInputPrice) ||
     defaultInputPrice < 0 ||
@@ -57,7 +55,7 @@ async function getModelPrice(modelId: string): Promise<ModelPrice | null> {
     name: modelId,
     input_price: defaultInputPrice,
     output_price: defaultOutputPrice,
-    per_msg_price: -1, // Default to token-based pricing
+    per_msg_price: -1,
   };
 }
 
@@ -66,7 +64,6 @@ export async function POST(req: Request) {
   let pgClient: DbClient | null = null;
 
   try {
-    // Get a dedicated transaction client
     if (isVercel) {
       pgClient = client;
     } else {
@@ -74,21 +71,18 @@ export async function POST(req: Request) {
     }
 
     const data = await req.json();
-    console.log("请求数据:", JSON.stringify(data, null, 2));
+    console.log("Request data:", JSON.stringify(data, null, 2));
     const modelId = data.body.model;
     const userId = data.user.id;
     const userName = data.user.name || "Unknown User";
 
-    // Start a transaction
     await query("BEGIN");
 
-    // Get model price
     const modelPrice = await getModelPrice(modelId);
     if (!modelPrice) {
       throw new Error(`Fail to fetch price info of model ${modelId}`);
     }
 
-    // Calculate tokens
     const lastMessage = data.body.messages[data.body.messages.length - 1];
 
     let inputTokens: number;
@@ -109,32 +103,25 @@ export async function POST(req: Request) {
       inputTokens = totalTokens - outputTokens;
     }
 
-    // Calculate total cost
     let totalCost: number;
     if (outputTokens === 0) {
-      // If output tokens are 0, no charge
       totalCost = 0;
       console.log("No charge for zero output tokens");
     } else if (modelPrice.per_msg_price >= 0) {
-      // If fixed pricing is set, use it directly
       totalCost = Number(modelPrice.per_msg_price);
       console.log(
         `Using fixed pricing: ${totalCost} (${modelPrice.per_msg_price} per message)`
       );
     } else {
-      // Otherwise, calculate price by token quantity
       const inputCost = (inputTokens / 1_000_000) * modelPrice.input_price;
       const outputCost = (outputTokens / 1_000_000) * modelPrice.output_price;
       totalCost = inputCost + outputCost;
     }
 
-    // Get the pre-deducted cost when getting inlet
     const inletCost = getModelInletCost(modelId);
 
-    // The actual cost to be deducted = total cost - pre-deducted cost
     const actualCost = totalCost - inletCost;
 
-    // Get and update user balance
     const userResult = await query(
       `UPDATE users 
        SET balance = LEAST(
@@ -156,7 +143,6 @@ export async function POST(req: Request) {
       throw new Error("Balance exceeds maximum allowed value");
     }
 
-    // Record usage
     await query(
       `INSERT INTO user_usage_records (
         user_id, nickname, model_name, 
@@ -208,7 +194,6 @@ export async function POST(req: Request) {
       { status: 500 }
     );
   } finally {
-    // Only release connection in non-Vercel environment
     if (!isVercel && pgClient && "release" in pgClient) {
       pgClient.release();
     }

app/api/v1/panel/database/export/route.ts

@@ -1,24 +1,18 @@
-import { pool } from "@/lib/db";
+import { query } from "@/lib/db/client";
 import { NextResponse } from "next/server";
-import { PoolClient } from "pg";
+import { verifyApiToken } from "@/lib/auth";
 
-export async function GET() {
-  let client: PoolClient | null = null;
+export async function GET(req: Request) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
 
   try {
-    // 获取数据库连接
-    client = await pool.connect();
-
-    // 获取所有表的数据
-    const users = await client.query("SELECT * FROM users ORDER BY id");
-    const modelPrices = await client.query(
-      "SELECT * FROM model_prices ORDER BY id"
-    );
-    const records = await client.query(
-      "SELECT * FROM user_usage_records ORDER BY id"
-    );
+    const users = await query("SELECT * FROM users ORDER BY id");
+    const modelPrices = await query("SELECT * FROM model_prices ORDER BY id");
+    const records = await query("SELECT * FROM user_usage_records ORDER BY id");
 
-    // 构建导出数据结构
     const exportData = {
       version: "1.0",
       timestamp: new Date().toISOString(),
@@ -29,7 +23,6 @@ export async function GET() {
       },
     };
 
-    // 设置响应头
     const headers = new Headers();
     headers.set("Content-Type", "application/json");
     headers.set(
@@ -48,9 +41,5 @@ export async function GET() {
       { error: "Fail to export database" },
       { status: 500 }
     );
-  } finally {
-    if (client) {
-      client.release();
-    }
   }
 }

app/api/v1/panel/database/import/route.ts

@@ -1,34 +1,30 @@
-import { pool } from "@/lib/db";
+import { query } from "@/lib/db/client";
 import { NextResponse } from "next/server";
-import { PoolClient } from "pg";
+import { verifyApiToken } from "@/lib/auth";
 
 export async function POST(req: Request) {
-  let client: PoolClient | null = null;
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
 
   try {
     const data = await req.json();
 
-    // 验证数据格式
     if (!data.version || !data.data) {
       throw new Error("Invalid import data format");
     }
 
-    // 获取数据库连接
-    client = await pool.connect();
-
-    // 开启事务
-    await client.query("BEGIN");
-
     try {
-      // 清空现有数据
-      await client.query("TRUNCATE TABLE user_usage_records CASCADE");
-      await client.query("TRUNCATE TABLE model_prices CASCADE");
-      await client.query("TRUNCATE TABLE users CASCADE");
+      await query("BEGIN");
+
+      await query("TRUNCATE TABLE user_usage_records CASCADE");
+      await query("TRUNCATE TABLE model_prices CASCADE");
+      await query("TRUNCATE TABLE users CASCADE");
 
-      // 导入用户数据
       if (data.data.users?.length) {
         for (const user of data.data.users) {
-          await client.query(
+          await query(
             `INSERT INTO users (id, email, name, role, balance)
              VALUES ($1, $2, $3, $4, $5)`,
             [user.id, user.email, user.name, user.role, user.balance]
@@ -36,10 +32,9 @@ export async function POST(req: Request) {
         }
       }
 
-      // 导入模型价格
       if (data.data.model_prices?.length) {
         for (const price of data.data.model_prices) {
-          await client.query(
+          await query(
             `INSERT INTO model_prices (id, name, input_price, output_price)
              VALUES ($1, $2, $3, $4)`,
             [price.id, price.name, price.input_price, price.output_price]
@@ -47,10 +42,9 @@ export async function POST(req: Request) {
         }
       }
 
-      // 导入使用记录
       if (data.data.user_usage_records?.length) {
         for (const record of data.data.user_usage_records) {
-          await client.query(
+          await query(
             `INSERT INTO user_usage_records (
               user_id, nickname, use_time, model_name, 
               input_tokens, output_tokens, cost, balance_after
@@ -69,14 +63,14 @@ export async function POST(req: Request) {
         }
       }
 
-      await client.query("COMMIT");
+      await query("COMMIT");
 
       return NextResponse.json({
         success: true,
         message: "Data import successful",
       });
     } catch (error) {
-      await client.query("ROLLBACK");
+      await query("ROLLBACK");
       throw error;
     }
   } catch (error) {
@@ -89,9 +83,5 @@ export async function POST(req: Request) {
       },
       { status: 500 }
     );
-  } finally {
-    if (client) {
-      client.release();
-    }
   }
 }

app/api/v1/panel/records/export/route.ts

@@ -1,13 +1,15 @@
-import { pool } from "@/lib/db";
+import { query } from "@/lib/db/client";
 import { NextResponse } from "next/server";
-import { PoolClient } from "pg";
+import { verifyApiToken } from "@/lib/auth";
 
-export async function GET() {
-  let client: PoolClient | null = null;
-  try {
-    client = await pool.connect();
+export async function GET(req: Request) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
 
-    const records = await client.query(`
+  try {
+    const records = await query(`
       SELECT 
         nickname,
         use_time,
@@ -20,7 +22,6 @@ export async function GET() {
       ORDER BY use_time DESC
     `);
 
-    // 生成 CSV 内容
     const csvHeaders = [
       "User",
       "Time",
@@ -45,7 +46,6 @@ export async function GET() {
       ...rows.map((row) => row.join(",")),
     ].join("\n");
 
-    // 设置响应头
     const responseHeaders = new Headers();
     responseHeaders.set("Content-Type", "text/csv; charset=utf-8");
     responseHeaders.set(
@@ -62,9 +62,5 @@ export async function GET() {
       { error: "Fail to export records" },
       { status: 500 }
     );
-  } finally {
-    if (client) {
-      client.release();
-    }
   }
 }

app/api/v1/panel/records/route.ts

@@ -1,9 +1,13 @@
-import { pool } from "@/lib/db";
+import { query } from "@/lib/db/client";
 import { NextResponse } from "next/server";
-import { PoolClient } from "pg";
+import { verifyApiToken } from "@/lib/auth";
 
 export async function GET(req: Request) {
-  let client: PoolClient | null = null;
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
+
   try {
     const { searchParams } = new URL(req.url);
     const page = parseInt(searchParams.get("page") || "1");
@@ -12,10 +16,9 @@ export async function GET(req: Request) {
     const sortOrder = searchParams.get("sortOrder");
     const users = searchParams.get("users")?.split(",") || [];
     const models = searchParams.get("models")?.split(",") || [];
+    const startDate = searchParams.get("startDate");
+    const endDate = searchParams.get("endDate");
 
-    client = await pool.connect();
-
-    // 构建查询条件
     const conditions = [];
     const params = [];
     let paramIndex = 1;
@@ -32,23 +35,29 @@ export async function GET(req: Request) {
       paramIndex++;
     }
 
+    if (startDate && endDate) {
+      conditions.push(
+        `use_time >= $${paramIndex} AND use_time <= $${paramIndex + 1}`
+      );
+      params.push(startDate);
+      params.push(endDate);
+      paramIndex += 2;
+    }
+
     const whereClause =
       conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
 
-    // 构建排序
     const orderClause = sortField
       ? `ORDER BY ${sortField} ${sortOrder === "descend" ? "DESC" : "ASC"}`
       : "ORDER BY use_time DESC";
 
-    // 获取总记录数
     const countQuery = `
       SELECT COUNT(*) 
       FROM user_usage_records 
       ${whereClause}
     `;
-    const countResult = await client.query(countQuery, params);
+    const countResult = await query(countQuery, params);
 
-    // 获取分页数据
     const offset = (page - 1) * pageSize;
     const dataQuery = `
       SELECT 
@@ -67,7 +76,7 @@ export async function GET(req: Request) {
     `;
 
     const dataParams = [...params, pageSize, offset];
-    const records = await client.query(dataQuery, dataParams);
+    const records = await query(dataQuery, dataParams);
 
     const total = parseInt(countResult.rows[0].count);
 
@@ -81,9 +90,5 @@ export async function GET(req: Request) {
       { error: "Fail to fetch usage records" },
       { status: 500 }
     );
-  } finally {
-    if (client) {
-      client.release();
-    }
   }
 }

app/api/v1/panel/usage/route.ts

@@ -1,12 +1,20 @@
 import { NextResponse } from "next/server";
-import { pool } from "@/lib/db";
+import { query } from "@/lib/db/client";
+import { verifyApiToken } from "@/lib/auth";
 
 export async function GET(request: Request) {
+  const authError = verifyApiToken(request);
+  if (authError) {
+    return authError;
+  }
+
   try {
     const { searchParams } = new URL(request.url);
     const startTime = searchParams.get("startTime");
     const endTime = searchParams.get("endTime");
 
+    console.log("Query params:", [startTime, endTime]);
+
     const timeFilter =
       startTime && endTime ? `WHERE use_time >= $1 AND use_time <= $2` : "";
 
@@ -14,7 +22,7 @@ export async function GET(request: Request) {
 
     const [modelResult, userResult, timeRangeResult, statsResult] =
       await Promise.all([
-        pool.query(
+        query(
           `
         SELECT 
           model_name,
@@ -27,7 +35,7 @@ export async function GET(request: Request) {
       `,
           params
         ),
-        pool.query(
+        query(
           `
         SELECT 
           nickname,
@@ -40,13 +48,13 @@ export async function GET(request: Request) {
       `,
           params
         ),
-        pool.query(`
+        query(`
         SELECT 
           MIN(use_time) as min_time,
           MAX(use_time) as max_time
         FROM user_usage_records
       `),
-        pool.query(
+        query(
           `
         SELECT 
           COALESCE(SUM(input_tokens + output_tokens), 0) as total_tokens,
@@ -82,6 +90,9 @@ export async function GET(request: Request) {
     return NextResponse.json(formattedData);
   } catch (error) {
     console.error("Fail to fetch usage records:", error);
+    if (error instanceof Error) {
+      console.error("[DB Query Error]", error);
+    }
     return NextResponse.json(
       { error: "Fail to fetch usage records" },
       { status: 500 }

app/api/{users → v1/users}/[id]/balance/route.ts

@@ -1,17 +1,25 @@
 import { query } from "@/lib/db/client";
 import { NextResponse } from "next/server";
+import { verifyApiToken } from "@/lib/auth";
 
 export async function PUT(
   req: Request,
   { params }: { params: { id: string } }
 ) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
+
   try {
     const { balance } = await req.json();
     const userId = params.id;
 
+    console.log(`Updating balance for user ${userId} to ${balance}`);
+
     if (typeof balance !== "number") {
       return NextResponse.json(
-        { error: "Balance must be positive" },
+        { error: "Balance must be a number" },
         { status: 400 }
       );
     }
@@ -24,6 +32,8 @@ export async function PUT(
       [balance, userId]
     );
 
+    console.log(`Update result:`, result);
+
     if (result.rows.length === 0) {
       return NextResponse.json(
         { error: "User does not exist" },

app/api/{users → v1/users}/[id]/route.ts

@@ -1,11 +1,17 @@
 import { NextRequest, NextResponse } from "next/server";
 import { deleteUser } from "@/lib/db/users";
 import { query } from "@/lib/db/client";
+import { verifyApiToken } from "@/lib/auth";
 
 export async function DELETE(
   req: NextRequest,
   { params }: { params: { id: string } }
 ) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
+
   try {
     await deleteUser(params.id);
     return NextResponse.json({ success: true });
@@ -19,6 +25,11 @@ export async function PATCH(
   req: NextRequest,
   { params }: { params: { id: string } }
 ) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
+
   try {
     const { deleted } = await req.json();
 

app/api/{users → v1/users}/route.ts

@@ -1,10 +1,15 @@
 import { NextRequest, NextResponse } from "next/server";
 import { query } from "@/lib/db/client";
 import { ensureUserTableExists } from "@/lib/db/users";
+import { verifyApiToken } from "@/lib/auth";
 
 export async function GET(req: NextRequest) {
+  const authError = verifyApiToken(req);
+  if (authError) {
+    return authError;
+  }
+
   try {
-    // 确保表结构正确
     await ensureUserTableExists();
 
     const { searchParams } = new URL(req.url);
@@ -15,7 +20,6 @@ export async function GET(req: NextRequest) {
     const search = searchParams.get("search");
     const deleted = searchParams.get("deleted") === "true";
 
-    // 构建查询条件
     const conditions = [`deleted = ${deleted}`];
     const params = [];
     let paramIndex = 1;
@@ -30,14 +34,12 @@ export async function GET(req: NextRequest) {
 
     const whereClause = `WHERE ${conditions.join(" AND ")}`;
 
-    // 获取总记录数
     const countResult = await query(
       `SELECT COUNT(*) FROM users ${whereClause}`,
       params
     );
     const total = parseInt(countResult.rows[0].count);
 
-    // 获取分页数据
     const result = await query(
       `SELECT id, email, name, role, balance, deleted, created_at 
        FROM users 

app/globals.css

@@ -80,7 +80,6 @@ body {
   display: none;
 }
 
-/* 更新模态框样式 */
 .update-modal .ant-modal-content {
   padding: 24px;
   border-radius: 16px;
@@ -222,7 +221,6 @@ body {
   background-size: 24px 24px;
 }
 
-/* 添加以下样式 */
 @media (max-width: 640px) {
   .toaster-group {
     --viewport-padding: 16px;
@@ -240,7 +238,6 @@ body {
   }
 }
 
-/* 自定义日期选择器样式 */
 .custom-date-picker {
   border-radius: 0.5rem;
   box-shadow: 0 10px 15px -3px rgb(0 0 0 / 0.1);

app/models/page.tsx

@@ -263,7 +263,12 @@ export default function ModelsPage() {
   useEffect(() => {
     const fetchModels = async () => {
       try {
-        const response = await fetch("/api/v1/models");
+        const token = localStorage.getItem("access_token");
+        const response = await fetch("/api/v1/models", {
+          headers: {
+            Authorization: `Bearer ${token}`,
+          },
+        });
         if (!response.ok) {
           throw new Error(t("error.model.failToFetchModels"));
         }
@@ -291,7 +296,12 @@ export default function ModelsPage() {
   useEffect(() => {
     const fetchApiKey = async () => {
       try {
-        const response = await fetch("/api/config/key");
+        const token = localStorage.getItem("access_token");
+        const response = await fetch("/api/v1/config/key", {
+          headers: {
+            Authorization: `Bearer ${token}`,
+          },
+        });
         if (!response.ok) {
           throw new Error(
             `${t("error.model.failToFetchApiKey")}: ${response.status}`
@@ -342,9 +352,13 @@ export default function ModelsPage() {
       const per_msg_price =
         field === "per_msg_price" ? validValue : model.per_msg_price;
 
+      const token = localStorage.getItem("access_token");
       const response = await fetch("/api/v1/models/price", {
         method: "POST",
-        headers: { "Content-Type": "application/json" },
+        headers: {
+          "Content-Type": "application/json",
+          Authorization: `Bearer ${token}`,
+        },
         body: JSON.stringify({
           updates: [
             {
@@ -416,10 +430,12 @@ export default function ModelsPage() {
     try {
       setSyncing(true);
 
+      const token = localStorage.getItem("access_token");
       const response = await fetch("/api/v1/models/sync-all-prices", {
         method: "POST",
         headers: {
           "Content-Type": "application/json",
+          Authorization: `Bearer ${token}`,
         },
       });
 
@@ -429,7 +445,6 @@ export default function ModelsPage() {
         throw new Error(data.error || t("models.syncFail"));
       }
 
-      // 更新模型数据
       if (data.syncedModels && data.syncedModels.length > 0) {
         setModels((prev) =>
           prev.map((model) => {
@@ -655,11 +670,12 @@ export default function ModelsPage() {
     const timeoutId = setTimeout(() => controller.abort(), 30000);
 
     try {
+      const token = localStorage.getItem("access_token");
       const response = await fetch("/api/v1/models/test", {
         method: "POST",
         headers: {
           "Content-Type": "application/json",
-          Authorization: `Bearer ${apiKey}`,
+          Authorization: `Bearer ${token}`,
         },
         body: JSON.stringify({
           modelId: model.id,
@@ -725,7 +741,6 @@ export default function ModelsPage() {
     }
   };
 
-  // 修改表格样式
   const tableClassName = `
     [&_.ant-table]:!border-b-0 
     [&_.ant-table-container]:!rounded-xl 
@@ -746,7 +761,6 @@ export default function ModelsPage() {
     [&_.ant-table-cell:last-child]:!pr-6
   `;
 
-  // 修改移动端卡片组件
   const MobileCard = ({ record }: { record: Model }) => {
     const isPerMsgEnabled = record.per_msg_price >= 0;
 
@@ -821,7 +835,6 @@ export default function ModelsPage() {
     );
   };
 
-  // 将 renderPriceCell 修改为接收一个额外的 showTooltip 参数
   const renderPriceCell = (
     field: "input_price" | "output_price" | "per_msg_price",
     record: Model,
@@ -841,9 +854,7 @@ export default function ModelsPage() {
           try {
             await handlePriceUpdate(record.id, field, value);
             setEditingCell(null);
-          } catch {
-            // 错误已经在 handlePriceUpdate 中处理
-          }
+          } catch {}
         }}
         t={t}
         disabled={isDisabled}
@@ -875,7 +886,6 @@ export default function ModelsPage() {
 
   return (
     <div className="max-w-6xl mx-auto px-4 sm:px-6 lg:px-8 py-8 pt-24 space-y-8">
-      {/* 添加 Toaster 组件 */}
       <Toaster
         richColors
         position="top-center"
@@ -884,7 +894,6 @@ export default function ModelsPage() {
         duration={1500}
       />
 
-      {/* 页面标题部分 */}
       <div className="space-y-4">
         <h1 className="text-3xl font-bold tracking-tight">
           {t("models.title")}
@@ -892,7 +901,6 @@ export default function ModelsPage() {
         <p className="text-muted-foreground">{t("models.description")}</p>
       </div>
 
-      {/* 操作按钮组 */}
       <div className="flex flex-wrap gap-4">
         <Button
           variant="default"
@@ -1005,7 +1013,6 @@ export default function ModelsPage() {
         />
       </div>
 
-      {/* 替换原有的TestProgress组件 */}
       <TestProgressPanel
         isVisible={testing || isTestComplete}
         models={models}
@@ -1013,7 +1020,6 @@ export default function ModelsPage() {
         t={t}
       />
 
-      {/* 桌面端表格视图 */}
       <div className="hidden sm:block">
         <div className="rounded-xl border border-border/40 bg-card shadow-sm overflow-hidden">
           {loading ? (
@@ -1034,7 +1040,6 @@ export default function ModelsPage() {
         </div>
       </div>
 
-      {/* 移动端卡片视图 */}
       <div className="sm:hidden">
         {loading ? (
           <LoadingState t={t} />

app/page.tsx

@@ -31,7 +31,6 @@ export default function HomePage() {
         const currentVer = APP_VERSION.replace(/^v/, "");
         const newVer = latestVer.replace(/^v/, "");
 
-        // 检查是否有更新且用户未禁用该版本的提示
         const ignoredVersion = localStorage.getItem("ignoredVersion");
         if (currentVer !== newVer && ignoredVersion !== latestVer) {
           setLatestVersion(latestVer);
@@ -61,7 +60,6 @@ export default function HomePage() {
 
   return (
     <main className="relative min-h-screen w-full overflow-hidden bg-gradient-to-br from-rose-50 via-slate-50 to-teal-50 pt-16">
-      {/* 新增动态网格背景 */}
       <AnimatedGridPattern
         numSquares={30}
         maxOpacity={0.03}
@@ -73,18 +71,15 @@ export default function HomePage() {
         )}
       />
 
-      {/* 装饰性背景模糊圆 */}
       <div className="absolute top-1/2 left-1/2 -translate-x-1/2 -translate-y-1/2 w-[1000px] h-[1000px] bg-gradient-to-br from-rose-100/20 via-slate-100/20 to-teal-100/20 rounded-full blur-3xl opacity-40" />
       <div className="absolute top-1/4 left-1/4 w-[500px] h-[500px] bg-gradient-to-br from-pink-100/10 to-indigo-100/10 rounded-full blur-3xl opacity-30" />
       <div className="absolute bottom-1/4 right-1/4 w-[700px] h-[700px] bg-gradient-to-br from-teal-100/10 to-slate-100/10 rounded-full blur-3xl opacity-30" />
 
-      {/* 修改主要内容容器，使用flex布局固定GitHub在底部 */}
       <motion.div
         initial={{ opacity: 0 }}
         animate={{ opacity: 1 }}
         className="min-h-[calc(100vh-4rem)] flex flex-col relative z-10"
       >
-        {/* 标题区域保持不变 */}
         <motion.div className="flex-1 flex flex-col items-center justify-center">
           <motion.div
             initial={{ opacity: 0, y: 20 }}
@@ -111,13 +106,10 @@ export default function HomePage() {
             </div>
           </motion.div>
 
-          {/* 重新设计的导航区域 */}
           <div className="w-full max-w-2xl px-4 sm:px-6">
             <div className="relative">
-              {/* 装饰性背景 */}
               <div className="absolute inset-0 bg-gradient-to-r from-rose-100/20 via-slate-100/20 to-teal-100/20 blur-3xl -z-10" />
 
-              {/* 新的垂直导航设计 */}
               <div className="space-y-4">
                 {[
                   {
@@ -164,15 +156,12 @@ export default function HomePage() {
                         shadow-[0_4px_20px_-4px_rgba(0,0,0,0.05)] 
                         hover:shadow-[0_8px_30px_-4px_rgba(0,0,0,0.12)]"
                     >
-                      {/* 移除边框设计，只保留渐变背景 */}
                       <div
                         className={`absolute inset-0 bg-gradient-to-r ${item.gradient} opacity-0 group-hover:opacity-100 transition-opacity duration-500`}
                       />
 
-                      {/* 内容区域 */}
                       <div className="relative p-6">
                         <div className="flex items-center gap-4">
-                          {/* 图标容器 - 使用投影替代边框 */}
                           <div
                             className={cn(
                               "p-3 rounded-xl transition-all duration-500",
@@ -185,7 +174,6 @@ export default function HomePage() {
                             {item.icon}
                           </div>
 
-                          {/* 文字内容 */}
                           <div className="flex-1 min-w-0">
                             <h3 className="text-lg font-medium text-slate-800 group-hover:text-white transition-colors mb-1">
                               {item.title}
@@ -195,7 +183,6 @@ export default function HomePage() {
                             </p>
                           </div>
 
-                          {/* 箭头 */}
                           <div
                             className={cn(
                               "transform transition-all duration-300",
@@ -227,7 +214,6 @@ export default function HomePage() {
           </div>
         </motion.div>
 
-        {/* GitHub 图标固定在底部 */}
         <motion.div
           initial={{ opacity: 0 }}
           animate={{ opacity: 1 }}
@@ -245,7 +231,6 @@ export default function HomePage() {
         </motion.div>
       </motion.div>
 
-      {/* 更新提示框样式修改 */}
       {isUpdateVisible && (
         <motion.div
           initial={{ opacity: 0, y: 20 }}

app/panel/page.tsx

@@ -2,7 +2,7 @@
 
 import { useState, useEffect } from "react";
 import Head from "next/head";
-import dayjs from "dayjs";
+import dayjs from "@/lib/dayjs";
 import type { TablePaginationConfig } from "antd/es/table";
 import type { SorterResult } from "antd/es/table/interface";
 import type { FilterValue } from "antd/es/table/interface";
@@ -72,7 +72,7 @@ export default function PanelPage() {
   const [loading, setLoading] = useState(true);
   const [tableLoading, setTableLoading] = useState(true);
   const [dateRange, setDateRange] = useState<[Date, Date]>([
-    new Date(), // 将在加载数据后更新
+    new Date(),
     new Date(),
   ]);
   const [availableTimeRange, setAvailableTimeRange] = useState<{
@@ -109,18 +109,26 @@ export default function PanelPage() {
   const fetchUsageData = async (range: [Date, Date]) => {
     setLoading(true);
     try {
-      const startTime = dayjs(range[0]).startOf("day").toISOString();
-      const endTime = dayjs(range[1]).endOf("day").toISOString();
+      const startTime = dayjs(range[0])
+        .startOf("day")
+        .format("YYYY-MM-DDTHH:mm:ssZ");
+      const endTime = dayjs(range[1])
+        .endOf("day")
+        .format("YYYY-MM-DDTHH:mm:ssZ");
 
       const url = `/api/v1/panel/usage?startTime=${startTime}&endTime=${endTime}`;
-      const response = await fetch(url);
+      const token = localStorage.getItem("access_token");
+      const response = await fetch(url, {
+        headers: {
+          Authorization: `Bearer ${token}`,
+        },
+      });
 
       if (!response.ok) throw new Error("Failed to fetch data");
 
       const data = await response.json();
       setUsageData(data);
 
-      // 如果是全部时间范围,更新可用时间范围
       if (timeRangeType === "all") {
         const minTime = new Date(data.timeRange.minTime);
         const maxTime = new Date(data.timeRange.maxTime);
@@ -143,7 +151,6 @@ export default function PanelPage() {
         params.pagination.pageSize?.toString() || "10"
       );
 
-      // 添加排序和过滤参数
       if (params.sortField) {
         searchParams.append("sortField", params.sortField);
         searchParams.append("sortOrder", params.sortOrder || "ascend");
@@ -155,18 +162,23 @@ export default function PanelPage() {
         searchParams.append("models", params.filters.model_name.join(","));
       }
 
-      // 添加日期范围
       searchParams.append(
         "startDate",
-        dayjs(range[0]).startOf("day").format("YYYY-MM-DD")
+        dayjs(range[0]).startOf("day").format("YYYY-MM-DDTHH:mm:ssZ")
       );
       searchParams.append(
         "endDate",
-        dayjs(range[1]).endOf("day").format("YYYY-MM-DD")
+        dayjs(range[1]).endOf("day").format("YYYY-MM-DDTHH:mm:ssZ")
       );
 
+      const token = localStorage.getItem("access_token");
       const response = await fetch(
-        `/api/v1/panel/records?${searchParams.toString()}`
+        `/api/v1/panel/records?${searchParams.toString()}`,
+        {
+          headers: {
+            Authorization: `Bearer ${token}`,
+          },
+        }
       );
       const data = await response.json();
 
@@ -187,15 +199,18 @@ export default function PanelPage() {
 
   useEffect(() => {
     const loadInitialData = async () => {
-      // 获取全部时间范围的数据
-      const response = await fetch("/api/v1/panel/usage");
+      const token = localStorage.getItem("access_token");
+      const response = await fetch("/api/v1/panel/usage", {
+        headers: {
+          Authorization: `Bearer ${token}`,
+        },
+      });
       const data = await response.json();
 
       const minTime = dayjs(data.timeRange.minTime).startOf("day").toDate();
       const maxTime = dayjs(data.timeRange.maxTime).endOf("day").toDate();
       setAvailableTimeRange({ minTime, maxTime });
 
-      // 设置为全部时间范围
       const allTimeRange: [Date, Date] = [minTime, maxTime];
       setDateRange(allTimeRange);
       setTimeRangeType("all");
@@ -237,11 +252,9 @@ export default function PanelPage() {
   };
 
   const renderDateRangeLabel = () => {
-    // 如果是同一天，只显示一个日期
     if (dayjs(dateRange[0]).isSame(dateRange[1], "day")) {
       return dayjs(dateRange[0]).format("YYYY-MM-DD");
     }
-    // 否则显示日期范围
     return `${dayjs(dateRange[0]).format("YYYY-MM-DD")} ~ ${dayjs(
       dateRange[1]
     ).format("YYYY-MM-DD")}`;

app/records/page.tsx

@@ -177,7 +177,6 @@ export default function RecordsPage() {
         },
       });
 
-      // 设置筛选选项
       setUsers(data.users as string[]);
       setModels(data.models as string[]);
     } catch (error) {
@@ -230,7 +229,6 @@ export default function RecordsPage() {
     }
   };
 
-  // 修改表格样式
   const tableClassName = `
     [&_.ant-table]:!border-b-0 
     [&_.ant-table-container]:!rounded-xl 

app/token/page.tsx

@@ -34,7 +34,7 @@ export default function TokenPage() {
     setLoading(true);
     try {
       localStorage.setItem("access_token", token);
-      const res = await fetch("/api/config", {
+      const res = await fetch("/api/v1/config", {
         headers: {
           Authorization: `Bearer ${token}`,
         },
@@ -76,7 +76,6 @@ export default function TokenPage() {
         )}
       />
 
-      {/* 装饰性背景模糊圆 */}
       <div className="absolute top-1/2 left-1/2 -translate-x-1/2 -translate-y-1/2 w-[1000px] h-[1000px] bg-gradient-to-br from-rose-100/20 via-slate-100/20 to-teal-100/20 rounded-full blur-3xl opacity-40" />
       <div className="absolute top-1/4 left-1/4 w-[500px] h-[500px] bg-gradient-to-br from-pink-100/10 to-indigo-100/10 rounded-full blur-3xl opacity-30" />
       <div className="absolute bottom-1/4 right-1/4 w-[700px] h-[700px] bg-gradient-to-br from-teal-100/10 to-slate-100/10 rounded-full blur-3xl opacity-30" />
@@ -112,7 +111,6 @@ export default function TokenPage() {
           className="backdrop-blur-[20px] bg-white/[0.08] p-10 rounded-[2.5rem] border border-white/20 shadow-2xl relative overflow-hidden 
             hover:shadow-[0_8px_60px_rgba(120,119,198,0.15)] transition-shadow duration-300 group"
         >
-          {/* 新增流光边框效果 */}
           <div
             className="absolute inset-0 rounded-[2.5rem] p-[2px] 
             bg-gradient-to-br from-white/30 via-transparent to-transparent 
@@ -120,7 +118,6 @@ export default function TokenPage() {
             [mask-composite:xor] opacity-30 group-hover:opacity-50 transition-opacity"
           ></div>
 
-          {/* 新增动态粒子背景 */}
           <div className="absolute inset-0 bg-[radial-gradient(ellipse_at_center,_var(--tw-gradient-stops))] from-slate-100/5 via-transparent to-transparent opacity-20" />
 
           <motion.form

app/users/page.tsx

@@ -275,7 +275,6 @@ const BlockConfirmModal = ({
   );
 };
 
-// 修改 LoadingState 组件,添加 t 参数
 const LoadingState = ({ t }: { t: TFunction }) => (
   <div className="flex flex-col items-center justify-center py-12 px-4">
     <div className="h-12 w-12 rounded-full border-4 border-primary/10 border-t-primary animate-spin mb-4" />
@@ -310,7 +309,7 @@ export default function UsersPage() {
   const fetchUsers = async (page: number, isBlacklist: boolean = false) => {
     setLoading(true);
     try {
-      let url = `/api/users?page=${page}&deleted=${isBlacklist}`;
+      let url = `/api/v1/users?page=${page}&deleted=${isBlacklist}`;
       if (sortInfo.field && sortInfo.order) {
         url += `&sortField=${sortInfo.field}&sortOrder=${sortInfo.order}`;
       }
@@ -318,7 +317,12 @@ export default function UsersPage() {
         url += `&search=${encodeURIComponent(searchText)}`;
       }
 
-      const res = await fetch(url);
+      const token = localStorage.getItem("access_token");
+      const res = await fetch(url, {
+        headers: {
+          Authorization: `Bearer ${token}`,
+        },
+      });
       const data = await res.json();
       if (!res.ok) throw new Error(data.error);
 
@@ -339,7 +343,12 @@ export default function UsersPage() {
 
   const fetchBlacklistTotal = async () => {
     try {
-      const res = await fetch(`/api/users?page=1&deleted=true&pageSize=1`);
+      const token = localStorage.getItem("access_token");
+      const res = await fetch(`/api/v1/users?page=1&deleted=true&pageSize=1`, {
+        headers: {
+          Authorization: `Bearer ${token}`,
+        },
+      });
       const data = await res.json();
       if (!res.ok) throw new Error(data.error);
       setBlacklistTotal(data.total);
@@ -361,29 +370,41 @@ export default function UsersPage() {
 
   const handleUpdateBalance = async (userId: string, newBalance: number) => {
     try {
-      const res = await fetch(`/api/users/${userId}/balance`, {
+      console.log(`Updating balance for user ${userId} to ${newBalance}`);
+
+      const token = localStorage.getItem("access_token");
+      if (!token) {
+        throw new Error(t("auth.unauthorized"));
+      }
+
+      const res = await fetch(`/api/v1/users/${userId}/balance`, {
         method: "PUT",
-        headers: { "Content-Type": "application/json" },
+        headers: {
+          "Content-Type": "application/json",
+          Authorization: `Bearer ${token}`,
+        },
         body: JSON.stringify({ balance: newBalance }),
       });
 
       const data = await res.json();
-      if (!res.ok) throw new Error(data.error);
+      console.log("Update balance response:", data);
+
+      if (!res.ok) {
+        throw new Error(data.error || t("users.message.updateBalance.error"));
+      }
 
-      // 立即更新本地数据
       setUsers(
         users.map((user) =>
           user.id === userId ? { ...user, balance: newBalance } : user
         )
       );
 
-      // 然后再显示成功提示并清除编辑状态
       toast.success(t("users.message.updateBalance.success"));
       setEditingKey("");
 
-      // 最后再重新获取完整列表
       fetchUsers(currentPage, false);
     } catch (err) {
+      console.error("Failed to update balance:", err);
       toast.error(
         err instanceof Error
           ? err.message
@@ -396,10 +417,12 @@ export default function UsersPage() {
     if (!userToDelete) return;
 
     try {
-      const res = await fetch(`/api/users/${userToDelete.id}`, {
+      const token = localStorage.getItem("access_token");
+      const res = await fetch(`/api/v1/users/${userToDelete.id}`, {
         method: "PATCH",
         headers: {
           "Content-Type": "application/json",
+          Authorization: `Bearer ${token}`,
         },
         body: JSON.stringify({
           deleted: !userToDelete.deleted,
@@ -782,7 +805,6 @@ export default function UsersPage() {
     );
   };
 
-  // 添加空状态组件
   const EmptyState = ({ searchText }: { searchText: string }) => (
     <div className="flex flex-col items-center justify-center py-12 px-4">
       <div className="h-12 w-12 rounded-full bg-muted/40 flex items-center justify-center mb-4">

components/AuthCheck.tsx

@@ -9,9 +9,20 @@ export default function AuthCheck({ children }: { children: React.ReactNode }) {
   const router = useRouter();
   const pathname = usePathname();
 
+  useEffect(() => {
+    const initDb = async () => {
+      try {
+        await fetch("/api/init");
+      } catch (error) {
+        console.error("初始化数据库失败:", error);
+      }
+    };
+
+    initDb();
+  }, []);
+
   useEffect(() => {
     const checkAuth = async () => {
-      // 如果已经在token页面，不需要检查
       if (pathname === "/token") {
         setIsLoading(false);
         setIsAuthorized(true);
@@ -25,7 +36,7 @@ export default function AuthCheck({ children }: { children: React.ReactNode }) {
       }
 
       try {
-        const res = await fetch("/api/config", {
+        const res = await fetch("/api/v1/config", {
           headers: {
             Authorization: `Bearer ${token}`,
           },
@@ -49,7 +60,6 @@ export default function AuthCheck({ children }: { children: React.ReactNode }) {
     checkAuth();
   }, [router, pathname]);
 
-  // 显示加载状态或空白页面
   if (isLoading || !isAuthorized) {
     return null;
   }

components/Header.tsx

@@ -40,15 +40,12 @@ export default function Header() {
   const [isBackupModalOpen, setIsBackupModalOpen] = useState(false);
   const [isCheckingUpdate, setIsCheckingUpdate] = useState(false);
   const [accessToken, setAccessToken] = useState<string | null>(null);
-  const [isSettingsExpanded, setIsSettingsExpanded] = useState(false);
 
-  // 将函数声明移到前面
   const handleLanguageChange = async (newLang: string) => {
     await i18n.changeLanguage(newLang);
     localStorage.setItem("language", newLang);
   };
 
-  // 如果是token页面，只显示语言切换按钮
   const isTokenPage = pathname === "/token";
 
   if (isTokenPage) {
@@ -89,15 +86,13 @@ export default function Header() {
       return;
     }
 
-    // 验证token的有效性
-    fetch("/api/config", {
+    fetch("/api/v1/config", {
       headers: {
         Authorization: `Bearer ${token}`,
       },
     })
       .then((res) => {
         if (!res.ok) {
-          // 如果token无效，清除token并重定向
           localStorage.removeItem("access_token");
           router.push("/token");
           return;
@@ -111,7 +106,6 @@ export default function Header() {
       })
       .catch(() => {
         setApiKey(t("common.error"));
-        // 发生错误时也清除token并重定向
         localStorage.removeItem("access_token");
         router.push("/token");
       });
@@ -292,11 +286,10 @@ export default function Header() {
   ];
 
   const menuItems = [
-    // 在小屏幕上将导航项添加到菜单中，但需要特殊处理
     ...(!isTokenPage
       ? navigationItems.map((item) => ({
           ...item,
-          onClick: () => router.push(item.path), // 为导航项添加 onClick 处理
+          onClick: () => router.push(item.path),
         }))
       : []),
     {
@@ -325,7 +318,6 @@ export default function Header() {
     },
   ];
 
-  // 在navigationItems数组后添加
   const actionItems = [
     {
       icon: <Globe className="w-5 h-5" />,
@@ -372,9 +364,7 @@ export default function Header() {
               </Link>
             </motion.div>
 
-            {/* 右侧内容 */}
             <div className="flex items-center gap-4">
-              {/* 导航项 - 仅在大屏幕显示 */}
               {!isTokenPage && (
                 <div className="hidden md:flex items-center gap-3">
                   {navigationItems.map((item) => (
@@ -401,7 +391,6 @@ export default function Header() {
                 </div>
               )}
 
-              {/* 语言切换和菜单按钮 */}
               <div className="flex items-center gap-3">
                 {actionItems.map((item, index) => (
                   <button
@@ -433,7 +422,6 @@ export default function Header() {
       <AnimatePresence>
         {isMenuOpen && (
           <>
-            {/* 背景遮罩 */}
             <motion.div
               initial={{ opacity: 0 }}
               animate={{ opacity: 1 }}
@@ -443,7 +431,6 @@ export default function Header() {
               onClick={() => setIsMenuOpen(false)}
             />
 
-            {/* 菜单面板 - 响应式布局 */}
             <motion.div
               initial={{ x: "100%" }}
               animate={{ x: 0 }}
@@ -451,9 +438,7 @@ export default function Header() {
               transition={{ type: "spring", damping: 25, stiffness: 200 }}
               className="fixed top-0 right-0 h-full bg-white/95 backdrop-blur-xl z-50 w-full max-w-[480px] md:top-[calc(4rem+0.5rem)] md:h-auto md:mr-6 md:rounded-2xl md:border md:shadow-xl md:max-h-[calc(100vh-5rem)] overflow-hidden shadow-lg border-l border-gray-100/50"
             >
-              {/* 内容容器 */}
               <div className="relative h-full flex flex-col">
-                {/* 顶部栏 */}
                 <div className="flex items-center justify-between p-4 border-b border-gray-100/50">
                   <h2 className="text-lg font-medium bg-gradient-to-r from-gray-900 via-gray-800 to-gray-900 bg-clip-text text-transparent">
                     {t("header.menu.settings")}
@@ -466,10 +451,8 @@ export default function Header() {
                   </button>
                 </div>
 
-                {/* 菜单项列表 */}
                 <div className="flex-1 overflow-y-auto p-2">
                   <div className="space-y-1">
-                    {/* 导航项 - 仅在移动端显示 */}
                     <div className="md:hidden space-y-1">
                       {navigationItems.map((item, index) => (
                         <motion.button
@@ -497,7 +480,6 @@ export default function Header() {
                       ))}
                     </div>
 
-                    {/* 设置选项 - 直接显示所有选项 */}
                     <div className="border-t border-gray-100/50 pt-2 md:border-t-0 md:pt-0">
                       {settingsItems.map((item, index) => (
                         <motion.button

components/editable-cell.tsx

@@ -83,7 +83,6 @@ export function EditableCell({
 
       await onSubmit(numValue);
     } catch (err) {
-      // 错误已在父组件中处理
     } finally {
       setIsSaving(false);
     }

components/panel/TimeRangeSelector.tsx

@@ -1,7 +1,7 @@
 "use client";
 
 import { useState, useEffect } from "react";
-import dayjs from "dayjs";
+import dayjs from "@/lib/dayjs";
 import { Button } from "@/components/ui/button";
 import { useTranslation } from "react-i18next";
 import { motion, AnimatePresence } from "framer-motion";

components/panel/UsageRecordsTable.tsx

@@ -4,7 +4,7 @@ import { useState } from "react";
 import { Table, TablePaginationConfig, Select } from "antd";
 import type { FilterValue } from "antd/es/table/interface";
 import type { SorterResult } from "antd/es/table/interface";
-import dayjs from "dayjs";
+import dayjs from "@/lib/dayjs";
 import { useTranslation } from "react-i18next";
 interface UsageRecord {
   id: number;
@@ -200,7 +200,6 @@ export default function UsageRecordsTable({
         />
       </div>
 
-      {/* 桌面设备表格 */}
       <div className="hidden sm:block">
         <Table
           columns={columns}
@@ -235,7 +234,6 @@ export default function UsageRecordsTable({
         />
       </div>
 
-      {/* 移动设备卡片列表 */}
       <div className="sm:hidden space-y-4">
         {loading ? (
           <div className="flex justify-center py-8">

components/panel/UserRankingChart.tsx

@@ -225,13 +225,12 @@ export default function UserRankingChart({
   const onChartReady = (instance: ECharts) => {
     chartRef.current = instance;
     const zoomSize = 6;
-    let isZoomed = false; // 增加一个状态变量
+    let isZoomed = false;
 
     instance.on("click", (params) => {
       const dataLength = users.length;
 
       if (!isZoomed) {
-        // 第一次点击，放大区域
         instance.dispatchAction({
           type: "dataZoom",
           startValue:
@@ -242,7 +241,6 @@ export default function UserRankingChart({
         });
         isZoomed = true;
       } else {
-        // 第二次点击，还原缩放
         instance.dispatchAction({
           type: "dataZoom",
           start: 0,

components/ui/animated-grid-pattern.tsx

@@ -43,7 +43,6 @@ export function AnimatedGridPattern({
     ];
   }
 
-  // Adjust the generateSquares function to return objects with an id, x, and y
   function generateSquares(count: number) {
     return Array.from({ length: count }, (_, i) => ({
       id: i,
@@ -51,7 +50,6 @@ export function AnimatedGridPattern({
     }));
   }
 
-  // Function to update a single square's position
   const updateSquarePosition = (id: number) => {
     setSquares((currentSquares) =>
       currentSquares.map((sq) =>
@@ -65,14 +63,12 @@ export function AnimatedGridPattern({
     );
   };
 
-  // Update squares to animate in
   useEffect(() => {
     if (dimensions.width && dimensions.height) {
       setSquares(generateSquares(numSquares));
     }
   }, [dimensions, numSquares]);
 
-  // Resize observer to update container dimensions
   useEffect(() => {
     const resizeObserver = new ResizeObserver((entries) => {
       for (let entry of entries) {

components/ui/chart.tsx

@@ -1,55 +1,49 @@
-"use client"
+"use client";
 
-import * as React from "react"
-import * as RechartsPrimitive from "recharts"
-import {
-  NameType,
-  Payload,
-  ValueType,
-} from "recharts/types/component/DefaultTooltipContent"
+import * as React from "react";
+import * as RechartsPrimitive from "recharts";
 
-import { cn } from "@/lib/utils"
+import { cn } from "@/lib/utils";
 
-// Format: { THEME_NAME: CSS_SELECTOR }
-const THEMES = { light: "", dark: ".dark" } as const
+const THEMES = { light: "", dark: ".dark" } as const;
 
 export type ChartConfig = {
   [k in string]: {
-    label?: React.ReactNode
-    icon?: React.ComponentType
+    label?: React.ReactNode;
+    icon?: React.ComponentType;
   } & (
     | { color?: string; theme?: never }
     | { color?: never; theme: Record<keyof typeof THEMES, string> }
-  )
-}
+  );
+};
 
 type ChartContextProps = {
-  config: ChartConfig
-}
+  config: ChartConfig;
+};
 
-const ChartContext = React.createContext<ChartContextProps | null>(null)
+const ChartContext = React.createContext<ChartContextProps | null>(null);
 
 function useChart() {
-  const context = React.useContext(ChartContext)
+  const context = React.useContext(ChartContext);
 
   if (!context) {
-    throw new Error("useChart must be used within a <ChartContainer />")
+    throw new Error("useChart must be used within a <ChartContainer />");
   }
 
-  return context
+  return context;
 }
 
 const ChartContainer = React.forwardRef<
   HTMLDivElement,
   React.ComponentProps<"div"> & {
-    config: ChartConfig
+    config: ChartConfig;
     children: React.ComponentProps<
       typeof RechartsPrimitive.ResponsiveContainer
-    >["children"]
+    >["children"];
   }
 >(({ id, className, children, config, ...props }, ref) => {
-  const uniqueId = React.useId()
-  const chartId = `chart-${id || uniqueId.replace(/:/g, "")}`
+  const uniqueId = React.useId();
+  const chartId = `chart-${id || uniqueId.replace(/:/g, "")}`;
 
   return (
     <ChartContext.Provider value={{ config }}>
@@ -68,17 +62,17 @@ const ChartContainer = React.forwardRef<
         </RechartsPrimitive.ResponsiveContainer>
       </div>
     </ChartContext.Provider>
-  )
-})
-ChartContainer.displayName = "Chart"
+  );
+});
+ChartContainer.displayName = "Chart";
 
 const ChartStyle = ({ id, config }: { id: string; config: ChartConfig }) => {
   const colorConfig = Object.entries(config).filter(
     ([_, config]) => config.theme || config.color
-  )
+  );
 
   if (!colorConfig.length) {
-    return null
+    return null;
   }
 
   return (
@@ -92,8 +86,8 @@ ${colorConfig
   .map(([key, itemConfig]) => {
     const color =
       itemConfig.theme?.[theme as keyof typeof itemConfig.theme] ||
-      itemConfig.color
-    return color ? `  --color-${key}: ${color};` : null
+      itemConfig.color;
+    return color ? `  --color-${key}: ${color};` : null;
   })
   .join("\n")}
 }
@@ -102,20 +96,20 @@ ${colorConfig
           .join("\n"),
       }}
     />
-  )
-}
+  );
+};
 
-const ChartTooltip = RechartsPrimitive.Tooltip
+const ChartTooltip = RechartsPrimitive.Tooltip;
 
 const ChartTooltipContent = React.forwardRef<
   HTMLDivElement,
   React.ComponentProps<typeof RechartsPrimitive.Tooltip> &
     React.ComponentProps<"div"> & {
-      hideLabel?: boolean
-      hideIndicator?: boolean
-      indicator?: "line" | "dot" | "dashed"
-      nameKey?: string
-      labelKey?: string
+      hideLabel?: boolean;
+      hideIndicator?: boolean;
+      indicator?: "line" | "dot" | "dashed";
+      nameKey?: string;
+      labelKey?: string;
     }
 >(
   (
@@ -136,34 +130,34 @@ const ChartTooltipContent = React.forwardRef<
     },
     ref
   ) => {
-    const { config } = useChart()
+    const { config } = useChart();
 
     const tooltipLabel = React.useMemo(() => {
       if (hideLabel || !payload?.length) {
-        return null
+        return null;
       }
 
-      const [item] = payload
-      const key = `${labelKey || item.dataKey || item.name || "value"}`
-      const itemConfig = getPayloadConfigFromPayload(config, item, key)
+      const [item] = payload;
+      const key = `${labelKey || item.dataKey || item.name || "value"}`;
+      const itemConfig = getPayloadConfigFromPayload(config, item, key);
       const value =
         !labelKey && typeof label === "string"
           ? config[label as keyof typeof config]?.label || label
-          : itemConfig?.label
+          : itemConfig?.label;
 
       if (labelFormatter) {
         return (
           <div className={cn("font-medium", labelClassName)}>
             {labelFormatter(value, payload)}
           </div>
-        )
+        );
       }
 
       if (!value) {
-        return null
+        return null;
       }
 
-      return <div className={cn("font-medium", labelClassName)}>{value}</div>
+      return <div className={cn("font-medium", labelClassName)}>{value}</div>;
     }, [
       label,
       labelFormatter,
@@ -172,13 +166,13 @@ const ChartTooltipContent = React.forwardRef<
       labelClassName,
       config,
       labelKey,
-    ])
+    ]);
 
     if (!active || !payload?.length) {
-      return null
+      return null;
     }
 
-    const nestLabel = payload.length === 1 && indicator !== "dot"
+    const nestLabel = payload.length === 1 && indicator !== "dot";
 
     return (
       <div
@@ -191,9 +185,9 @@ const ChartTooltipContent = React.forwardRef<
         {!nestLabel ? tooltipLabel : null}
         <div className="grid gap-1.5">
           {payload.map((item, index) => {
-            const key = `${nameKey || item.name || item.dataKey || "value"}`
-            const itemConfig = getPayloadConfigFromPayload(config, item, key)
-            const indicatorColor = color || item.payload.fill || item.color
+            const key = `${nameKey || item.name || item.dataKey || "value"}`;
+            const itemConfig = getPayloadConfigFromPayload(config, item, key);
+            const indicatorColor = color || item.payload.fill || item.color;
 
             return (
               <div
@@ -252,33 +246,33 @@ const ChartTooltipContent = React.forwardRef<
                   </>
                 )}
               </div>
-            )
+            );
           })}
         </div>
       </div>
-    )
+    );
   }
-)
-ChartTooltipContent.displayName = "ChartTooltip"
+);
+ChartTooltipContent.displayName = "ChartTooltip";
 
-const ChartLegend = RechartsPrimitive.Legend
+const ChartLegend = RechartsPrimitive.Legend;
 
 const ChartLegendContent = React.forwardRef<
   HTMLDivElement,
   React.ComponentProps<"div"> &
     Pick<RechartsPrimitive.LegendProps, "payload" | "verticalAlign"> & {
-      hideIcon?: boolean
-      nameKey?: string
+      hideIcon?: boolean;
+      nameKey?: string;
     }
 >(
   (
     { className, hideIcon = false, payload, verticalAlign = "bottom", nameKey },
     ref
   ) => {
-    const { config } = useChart()
+    const { config } = useChart();
 
     if (!payload?.length) {
-      return null
+      return null;
     }
 
     return (
@@ -291,8 +285,8 @@ const ChartLegendContent = React.forwardRef<
         )}
       >
         {payload.map((item) => {
-          const key = `${nameKey || item.dataKey || "value"}`
-          const itemConfig = getPayloadConfigFromPayload(config, item, key)
+          const key = `${nameKey || item.dataKey || "value"}`;
+          const itemConfig = getPayloadConfigFromPayload(config, item, key);
 
           return (
             <div
@@ -313,22 +307,21 @@ const ChartLegendContent = React.forwardRef<
               )}
               {itemConfig?.label}
             </div>
-          )
+          );
         })}
       </div>
-    )
+    );
   }
-)
-ChartLegendContent.displayName = "ChartLegend"
+);
+ChartLegendContent.displayName = "ChartLegend";
 
-// Helper to extract item config from a payload.
 function getPayloadConfigFromPayload(
   config: ChartConfig,
   payload: unknown,
   key: string
 ) {
   if (typeof payload !== "object" || payload === null) {
-    return undefined
+    return undefined;
   }
 
   const payloadPayload =
@@ -336,15 +329,15 @@ function getPayloadConfigFromPayload(
     typeof payload.payload === "object" &&
     payload.payload !== null
       ? payload.payload
-      : undefined
+      : undefined;
 
-  let configLabelKey: string = key
+  let configLabelKey: string = key;
 
   if (
     key in payload &&
     typeof payload[key as keyof typeof payload] === "string"
   ) {
-    configLabelKey = payload[key as keyof typeof payload] as string
+    configLabelKey = payload[key as keyof typeof payload] as string;
   } else if (
     payloadPayload &&
     key in payloadPayload &&
@@ -352,12 +345,12 @@ function getPayloadConfigFromPayload(
   ) {
     configLabelKey = payloadPayload[
       key as keyof typeof payloadPayload
-    ] as string
+    ] as string;
   }
 
   return configLabelKey in config
     ? config[configLabelKey]
-    : config[key as keyof typeof config]
+    : config[key as keyof typeof config];
 }
 
 export {
@@ -367,4 +360,4 @@ export {
   ChartLegend,
   ChartLegendContent,
   ChartStyle,
-}
+};

components/ui/sidebar.tsx

@@ -70,8 +70,6 @@ const SidebarProvider = React.forwardRef<
     const isMobile = useIsMobile();
     const [openMobile, setOpenMobile] = React.useState(false);
 
-    // This is the internal state of the sidebar.
-    // We use openProp and setOpenProp for control from outside the component.
     const [_open, _setOpen] = React.useState(defaultOpen);
     const open = openProp ?? _open;
     const setOpen = React.useCallback(
@@ -83,20 +81,17 @@ const SidebarProvider = React.forwardRef<
           _setOpen(openState);
         }
 
-        // This sets the cookie to keep the sidebar state.
         document.cookie = `${SIDEBAR_COOKIE_NAME}=${openState}; path=/; max-age=${SIDEBAR_COOKIE_MAX_AGE}`;
       },
       [setOpenProp, open]
     );
 
-    // Helper to toggle the sidebar.
     const toggleSidebar = React.useCallback(() => {
       return isMobile
         ? setOpenMobile((open) => !open)
         : setOpen((open) => !open);
     }, [isMobile, setOpen, setOpenMobile]);
 
-    // Adds a keyboard shortcut to toggle the sidebar.
     React.useEffect(() => {
       const handleKeyDown = (event: KeyboardEvent) => {
         if (
@@ -112,8 +107,6 @@ const SidebarProvider = React.forwardRef<
       return () => window.removeEventListener("keydown", handleKeyDown);
     }, [toggleSidebar]);
 
-    // We add a state so that we can do data-state="expanded" or "collapsed".
-    // This makes it easier to style the sidebar with Tailwind classes.
     const state = open ? "expanded" : "collapsed";
 
     const contextValue = React.useMemo<SidebarContext>(
@@ -221,7 +214,6 @@ const Sidebar = React.forwardRef<
         data-variant={variant}
         data-side={side}
       >
-        {/* This is what handles the sidebar gap on desktop */}
         <div
           className={cn(
             "duration-200 relative h-svh w-[--sidebar-width] bg-transparent transition-[width] ease-linear",
@@ -238,7 +230,6 @@ const Sidebar = React.forwardRef<
             side === "left"
               ? "left-0 group-data-[collapsible=offcanvas]:left-[calc(var(--sidebar-width)*-1)]"
               : "right-0 group-data-[collapsible=offcanvas]:right-[calc(var(--sidebar-width)*-1)]",
-            // Adjust the padding for floating and inset variants.
             variant === "floating" || variant === "inset"
               ? "p-2 group-data-[collapsible=icon]:w-[calc(var(--sidebar-width-icon)_+_theme(spacing.4)_+2px)]"
               : "group-data-[collapsible=icon]:w-[--sidebar-width-icon] group-data-[side=left]:border-r group-data-[side=right]:border-l",
@@ -650,7 +641,6 @@ const SidebarMenuSkeleton = React.forwardRef<
     showIcon?: boolean;
   }
 >(({ className, showIcon = false, ...props }, ref) => {
-  // Random width between 50 to 90%.
   const width = React.useMemo(() => {
     return `${Math.floor(Math.random() * 40) + 50}%`;
   }, []);

hooks/use-toast.ts

@@ -1,78 +1,74 @@
-"use client"
+"use client";
 
-// Inspired by react-hot-toast library
-import * as React from "react"
+import * as React from "react";
 
-import type {
-  ToastActionElement,
-  ToastProps,
-} from "@/components/ui/toast"
+import type { ToastActionElement, ToastProps } from "@/components/ui/toast";
 
-const TOAST_LIMIT = 1
-const TOAST_REMOVE_DELAY = 1000000
+const TOAST_LIMIT = 1;
+const TOAST_REMOVE_DELAY = 1000000;
 
 type ToasterToast = ToastProps & {
-  id: string
-  title?: React.ReactNode
-  description?: React.ReactNode
-  action?: ToastActionElement
-}
+  id: string;
+  title?: React.ReactNode;
+  description?: React.ReactNode;
+  action?: ToastActionElement;
+};
 
 const actionTypes = {
   ADD_TOAST: "ADD_TOAST",
   UPDATE_TOAST: "UPDATE_TOAST",
   DISMISS_TOAST: "DISMISS_TOAST",
   REMOVE_TOAST: "REMOVE_TOAST",
-} as const
+} as const;
 
-let count = 0
+let count = 0;
 
 function genId() {
-  count = (count + 1) % Number.MAX_SAFE_INTEGER
-  return count.toString()
+  count = (count + 1) % Number.MAX_SAFE_INTEGER;
+  return count.toString();
 }
 
-type ActionType = typeof actionTypes
+type ActionType = typeof actionTypes;
 
 type Action =
   | {
-      type: ActionType["ADD_TOAST"]
-      toast: ToasterToast
+      type: ActionType["ADD_TOAST"];
+      toast: ToasterToast;
     }
   | {
-      type: ActionType["UPDATE_TOAST"]
-      toast: Partial<ToasterToast>
+      type: ActionType["UPDATE_TOAST"];
+      toast: Partial<ToasterToast>;
     }
   | {
-      type: ActionType["DISMISS_TOAST"]
-      toastId?: ToasterToast["id"]
+      type: ActionType["DISMISS_TOAST"];
+      toastId?: ToasterToast["id"];
     }
   | {
-      type: ActionType["REMOVE_TOAST"]
-      toastId?: ToasterToast["id"]
-    }
+      type: ActionType["REMOVE_TOAST"];
+      toastId?: ToasterToast["id"];
+    };
 
 interface State {
-  toasts: ToasterToast[]
+  toasts: ToasterToast[];
 }
 
-const toastTimeouts = new Map<string, ReturnType<typeof setTimeout>>()
+const toastTimeouts = new Map<string, ReturnType<typeof setTimeout>>();
 
 const addToRemoveQueue = (toastId: string) => {
   if (toastTimeouts.has(toastId)) {
-    return
+    return;
   }
 
   const timeout = setTimeout(() => {
-    toastTimeouts.delete(toastId)
+    toastTimeouts.delete(toastId);
     dispatch({
       type: "REMOVE_TOAST",
       toastId: toastId,
-    })
-  }, TOAST_REMOVE_DELAY)
+    });
+  }, TOAST_REMOVE_DELAY);
 
-  toastTimeouts.set(toastId, timeout)
-}
+  toastTimeouts.set(toastId, timeout);
+};
 
 export const reducer = (state: State, action: Action): State => {
   switch (action.type) {
@@ -80,7 +76,7 @@ export const reducer = (state: State, action: Action): State => {
       return {
         ...state,
         toasts: [action.toast, ...state.toasts].slice(0, TOAST_LIMIT),
-      }
+      };
 
     case "UPDATE_TOAST":
       return {
@@ -88,19 +84,17 @@ export const reducer = (state: State, action: Action): State => {
         toasts: state.toasts.map((t) =>
           t.id === action.toast.id ? { ...t, ...action.toast } : t
         ),
-      }
+      };
 
     case "DISMISS_TOAST": {
-      const { toastId } = action
+      const { toastId } = action;
 
-      // ! Side effects ! - This could be extracted into a dismissToast() action,
-      // but I'll keep it here for simplicity
       if (toastId) {
-        addToRemoveQueue(toastId)
+        addToRemoveQueue(toastId);
       } else {
         state.toasts.forEach((toast) => {
-          addToRemoveQueue(toast.id)
-        })
+          addToRemoveQueue(toast.id);
+        });
       }
 
       return {
@@ -113,44 +107,44 @@ export const reducer = (state: State, action: Action): State => {
               }
             : t
         ),
-      }
+      };
     }
     case "REMOVE_TOAST":
       if (action.toastId === undefined) {
         return {
           ...state,
           toasts: [],
-        }
+        };
       }
       return {
         ...state,
         toasts: state.toasts.filter((t) => t.id !== action.toastId),
-      }
+      };
   }
-}
+};
 
-const listeners: Array<(state: State) => void> = []
+const listeners: Array<(state: State) => void> = [];
 
-let memoryState: State = { toasts: [] }
+let memoryState: State = { toasts: [] };
 
 function dispatch(action: Action) {
-  memoryState = reducer(memoryState, action)
+  memoryState = reducer(memoryState, action);
   listeners.forEach((listener) => {
-    listener(memoryState)
-  })
+    listener(memoryState);
+  });
 }
 
-type Toast = Omit<ToasterToast, "id">
+type Toast = Omit<ToasterToast, "id">;
 
 function toast({ ...props }: Toast) {
-  const id = genId()
+  const id = genId();
 
   const update = (props: ToasterToast) =>
     dispatch({
       type: "UPDATE_TOAST",
       toast: { ...props, id },
-    })
-  const dismiss = () => dispatch({ type: "DISMISS_TOAST", toastId: id })
+    });
+  const dismiss = () => dispatch({ type: "DISMISS_TOAST", toastId: id });
 
   dispatch({
     type: "ADD_TOAST",
@@ -159,36 +153,36 @@ function toast({ ...props }: Toast) {
       id,
       open: true,
       onOpenChange: (open) => {
-        if (!open) dismiss()
+        if (!open) dismiss();
       },
     },
-  })
+  });
 
   return {
     id: id,
     dismiss,
     update,
-  }
+  };
 }
 
 function useToast() {
-  const [state, setState] = React.useState<State>(memoryState)
+  const [state, setState] = React.useState<State>(memoryState);
 
   React.useEffect(() => {
-    listeners.push(setState)
+    listeners.push(setState);
     return () => {
-      const index = listeners.indexOf(setState)
+      const index = listeners.indexOf(setState);
       if (index > -1) {
-        listeners.splice(index, 1)
+        listeners.splice(index, 1);
       }
-    }
-  }, [state])
+    };
+  }, [state]);
 
   return {
     ...state,
     toast,
     dismiss: (toastId?: string) => dispatch({ type: "DISMISS_TOAST", toastId }),
-  }
+  };
 }
 
-export { useToast, toast }
+export { useToast, toast };

lib/auth.ts

@@ -0,0 +1,23 @@
+import { NextResponse } from "next/server";
+
+const ACCESS_TOKEN = process.env.ACCESS_TOKEN;
+
+export function verifyApiToken(req: Request) {
+  if (!ACCESS_TOKEN) {
+    console.error("ACCESS_TOKEN is not set");
+    return NextResponse.json(
+      { error: "Server configuration error" },
+      { status: 500 }
+    );
+  }
+
+  const authHeader = req.headers.get("authorization");
+  const token = authHeader?.replace("Bearer ", "");
+
+  if (!token || token !== ACCESS_TOKEN) {
+    console.log("Unauthorized access attempt");
+    return NextResponse.json({ error: "Unauthorized" }, { status: 401 });
+  }
+
+  return null;
+}

lib/dayjs.ts

@@ -0,0 +1,19 @@
+import dayjs from "dayjs";
+import utc from "dayjs/plugin/utc";
+import timezone from "dayjs/plugin/timezone";
+
+dayjs.extend(utc);
+dayjs.extend(timezone);
+
+const localTimezone = Intl.DateTimeFormat().resolvedOptions().timeZone;
+dayjs.tz.setDefault(localTimezone);
+
+const originalFormat = dayjs.prototype.format;
+dayjs.prototype.format = function (template: string) {
+  if (template === "YYYY-MM-DDTHH:mm:ssZ") {
+    return this.toISOString();
+  }
+  return originalFormat.call(this, template);
+};
+
+export default dayjs;

lib/db.ts

@@ -1,352 +0,0 @@
-import { Pool, PoolClient } from "pg";
-
-// 构建数据库连接配置
-const dbConfig = process.env.POSTGRES_URL
-  ? {
-      // 远程数据库配置
-      connectionString: process.env.POSTGRES_URL,
-      ssl: {
-        rejectUnauthorized: false, // 允许自签名证书
-      },
-    }
-  : {
-      // 本地 Docker 数据库配置
-      host: process.env.POSTGRES_HOST || "localhost",
-      user: process.env.POSTGRES_USER || "postgres",
-      password: process.env.POSTGRES_PASSWORD,
-      database: process.env.POSTGRES_DATABASE || "openwebui_monitor",
-      ssl: false,
-    };
-
-// 创建连接池
-export const pool = new Pool(dbConfig);
-
-// 测试连接
-pool.on("error", (err) => {
-  console.error("Unexpected error on idle client", err);
-  process.exit(-1);
-});
-
-// 数据库行的类型定义
-interface ModelPriceRow {
-  id: string;
-  name: string;
-  input_price: string | number;
-  output_price: string | number;
-  per_msg_price: string | number;
-  updated_at: Date;
-}
-
-export interface ModelPrice {
-  id: string;
-  name: string;
-  input_price: number;
-  output_price: number;
-  per_msg_price: number;
-  updated_at: Date;
-}
-
-export interface UserUsageRecord {
-  id: number;
-  userId: number;
-  nickname: string;
-  useTime: Date;
-  modelName: string;
-  inputTokens: number;
-  outputTokens: number;
-  cost: number;
-  balanceAfter: number;
-}
-
-// 确保表存在
-export async function ensureTablesExist() {
-  let client: PoolClient | null = null;
-  try {
-    client = await pool.connect();
-
-    // 首先创建 users 表
-    await client.query(`
-      CREATE TABLE IF NOT EXISTS users (
-        id TEXT PRIMARY KEY,
-        email TEXT NOT NULL,
-        name TEXT NOT NULL,
-        role TEXT NOT NULL DEFAULT 'user',
-        balance DECIMAL(16, 6) NOT NULL DEFAULT 0,
-        created_at TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP
-      );
-    `);
-
-    // 获取默认价格
-    const defaultInputPrice = parseFloat(
-      process.env.DEFAULT_MODEL_INPUT_PRICE || "60"
-    );
-    const defaultOutputPrice = parseFloat(
-      process.env.DEFAULT_MODEL_OUTPUT_PRICE || "60"
-    );
-    const defaultPerMsgPrice = parseFloat(
-      process.env.DEFAULT_MODEL_PER_MSG_PRICE || "-1"
-    );
-
-    // 然后创建 model_prices 表，使用具体的默认值而不是参数绑定
-    await client.query(`
-      CREATE TABLE IF NOT EXISTS model_prices (
-        id TEXT PRIMARY KEY,
-        name TEXT NOT NULL,
-        base_model_id TEXT,
-        input_price NUMERIC(10, 6) DEFAULT ${defaultInputPrice},
-        output_price NUMERIC(10, 6) DEFAULT ${defaultOutputPrice},
-        per_msg_price NUMERIC(10, 6) DEFAULT ${defaultPerMsgPrice},
-        updated_at TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP
-      );
-    `);
-
-    // 检查并添加 per_msg_price 列（如果不存在）
-    await client.query(`
-      DO $$ 
-      BEGIN 
-        BEGIN
-          ALTER TABLE model_prices 
-          ADD COLUMN per_msg_price NUMERIC(10, 6) DEFAULT ${defaultPerMsgPrice};
-        EXCEPTION 
-          WHEN duplicate_column THEN NULL;
-        END;
-      END $$;
-    `);
-
-    // 检查并添加 base_model_id 列（如果不存在）
-    await client.query(`
-      DO $$ 
-      BEGIN 
-        BEGIN
-          ALTER TABLE model_prices 
-          ADD COLUMN base_model_id TEXT;
-        EXCEPTION 
-          WHEN duplicate_column THEN NULL;
-        END;
-      END $$;
-    `);
-
-    // 最后创建 user_usage_records 表
-    await client.query(`
-      CREATE TABLE IF NOT EXISTS user_usage_records (
-        id SERIAL PRIMARY KEY,
-        user_id TEXT NOT NULL,
-        nickname VARCHAR(255) NOT NULL,
-        use_time TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP,
-        model_name VARCHAR(255) NOT NULL,
-        input_tokens INTEGER NOT NULL,
-        output_tokens INTEGER NOT NULL,
-        cost DECIMAL(10, 4) NOT NULL,
-        balance_after DECIMAL(10, 4) NOT NULL,
-        FOREIGN KEY (user_id) REFERENCES users(id)
-      );
-    `);
-  } catch (error) {
-    console.error("Database connection/initialization error:", error);
-    throw error;
-  } finally {
-    if (client) {
-      client.release();
-    }
-  }
-}
-
-// 获取模型价格，如果不存在则创建默认值
-export async function getOrCreateModelPrices(
-  models: Array<{ id: string; name: string; base_model_id?: string }>
-): Promise<ModelPrice[]> {
-  let client: PoolClient | null = null;
-  try {
-    client = await pool.connect();
-
-    // 获取默认价格
-    const defaultInputPrice = parseFloat(
-      process.env.DEFAULT_MODEL_INPUT_PRICE || "60"
-    );
-    const defaultOutputPrice = parseFloat(
-      process.env.DEFAULT_MODEL_OUTPUT_PRICE || "60"
-    );
-    const defaultPerMsgPrice = parseFloat(
-      process.env.DEFAULT_MODEL_PER_MSG_PRICE || "-1"
-    );
-
-    // 1. 首先获取所有已存在的模型价格
-    const modelIds = models.map((m) => m.id);
-    const baseModelIds = models.map((m) => m.base_model_id).filter((id) => id);
-
-    const existingModelsResult = await client.query<ModelPriceRow>(
-      `SELECT * FROM model_prices WHERE id = ANY($1::text[])`,
-      [modelIds]
-    );
-
-    // 2. 获取所有基础模型的价格
-    const baseModelsResult = await client.query<ModelPriceRow>(
-      `SELECT * FROM model_prices WHERE id = ANY($1::text[])`,
-      [baseModelIds]
-    );
-
-    const existingModels = new Map(
-      existingModelsResult.rows.map((row) => [row.id, row])
-    );
-    const baseModels = new Map(
-      baseModelsResult.rows.map((row) => [row.id, row])
-    );
-
-    // 3. 更新所有模型的名称并插入缺失的模型
-    const modelsToUpdate = models.filter((m) => existingModels.has(m.id));
-    const missingModels = models.filter((m) => !existingModels.has(m.id));
-
-    // 3.1 更新现有模型的名称
-    if (modelsToUpdate.length > 0) {
-      for (const model of modelsToUpdate) {
-        await client.query(`UPDATE model_prices SET name = $2 WHERE id = $1`, [
-          model.id,
-          model.name,
-        ]);
-      }
-    }
-
-    // 3.2 插入缺失的模型
-    if (missingModels.length > 0) {
-      const values = missingModels.map((m) => {
-        const baseModel = m.base_model_id
-          ? baseModels.get(m.base_model_id)
-          : null;
-        return [
-          m.id,
-          m.name,
-          baseModel?.input_price ?? defaultInputPrice,
-          baseModel?.output_price ?? defaultOutputPrice,
-          baseModel?.per_msg_price ?? defaultPerMsgPrice,
-        ];
-      });
-
-      const placeholders = values
-        .map(
-          (_, i) =>
-            `($${i * 5 + 1}, $${i * 5 + 2}, $${i * 5 + 3}, $${i * 5 + 4}, $${
-              i * 5 + 5
-            })`
-        )
-        .join(",");
-
-      const result = await client.query<ModelPriceRow>(
-        `INSERT INTO model_prices (id, name, input_price, output_price, per_msg_price)
-         VALUES ${placeholders}
-         RETURNING *`,
-        values.flat()
-      );
-
-      result.rows.forEach((row) => existingModels.set(row.id, row));
-    }
-
-    // 4. 重新获取所有模型的最新数据
-    const updatedModelsResult = await client.query<ModelPriceRow>(
-      `SELECT * FROM model_prices WHERE id = ANY($1::text[])`,
-      [modelIds]
-    );
-
-    const updatedModels = new Map(
-      updatedModelsResult.rows.map((row) => [row.id, row])
-    );
-
-    return models.map((m) => {
-      const row = updatedModels.get(m.id)!;
-      return {
-        id: row.id,
-        name: row.name,
-        input_price: Number(row.input_price),
-        output_price: Number(row.output_price),
-        per_msg_price: Number(row.per_msg_price),
-        updated_at: row.updated_at,
-      };
-    });
-  } catch (error) {
-    console.error("Error in getOrCreateModelPrices:", error);
-    throw error;
-  } finally {
-    if (client) {
-      client.release();
-    }
-  }
-}
-
-// 更新模型价格
-export async function updateModelPrice(
-  id: string,
-  input_price: number,
-  output_price: number,
-  per_msg_price: number
-): Promise<ModelPrice | null> {
-  let client: PoolClient | null = null;
-  try {
-    client = await pool.connect();
-
-    // 使用 CAST 确保数据类型正确
-    const result = await client.query<ModelPriceRow>(
-      `UPDATE model_prices 
-       SET 
-         input_price = CAST($2 AS NUMERIC(10,6)),
-         output_price = CAST($3 AS NUMERIC(10,6)),
-         per_msg_price = CAST($4 AS NUMERIC(10,6)),
-         updated_at = CURRENT_TIMESTAMP
-       WHERE id = $1
-       RETURNING *`,
-      [id, input_price, output_price, per_msg_price]
-    );
-
-    if (result.rows[0]) {
-      return {
-        id: result.rows[0].id,
-        name: result.rows[0].name,
-        input_price: Number(result.rows[0].input_price),
-        output_price: Number(result.rows[0].output_price),
-        per_msg_price: Number(result.rows[0].per_msg_price),
-        updated_at: result.rows[0].updated_at,
-      };
-    }
-    return null;
-  } catch (error) {
-    console.error(`Failed to update ${id} price:`, error);
-    throw error;
-  } finally {
-    if (client) {
-      client.release();
-    }
-  }
-}
-
-// 添加一个始化函数
-export async function initDatabase() {
-  try {
-    await ensureTablesExist();
-    // console.log("Database initialized successfully");
-  } catch (error) {
-    console.error("Failed to initialize database:", error);
-    throw error;
-  }
-}
-
-// 更新用户余额
-export async function updateUserBalance(userId: string, balance: number) {
-  let client: PoolClient | null = null;
-  try {
-    client = await pool.connect();
-    const result = await client.query(
-      `UPDATE users
-       SET balance = $2
-       WHERE id = $1
-       RETURNING id, email, balance`,
-      [userId, balance]
-    );
-
-    return result.rows[0];
-  } catch (error) {
-    console.error("Error in updateUserBalance:", error);
-    throw error;
-  } finally {
-    if (client) {
-      client.release();
-    }
-  }
-}

lib/db/client.ts

@@ -6,7 +6,6 @@ import { Pool, PoolClient } from "pg";
 
 const isVercel = process.env.VERCEL === "1";
 
-// 为 Vercel 环境添加连接池
 let vercelPool: {
   client: ReturnType<typeof createClient>;
   isConnected: boolean;
@@ -21,7 +20,6 @@ async function getVercelClient() {
     };
   }
 
-  // 如果还没连接，则建立连接
   if (!vercelPool.isConnected) {
     try {
       await vercelPool.client.connect();
@@ -48,7 +46,8 @@ function getClient() {
         port: parseInt(process.env.POSTGRES_PORT || "5432"),
         max: 20,
         idleTimeoutMillis: 30000,
-        connectionTimeoutMillis: 2000,
+        connectionTimeoutMillis: 30000,
+        statement_timeout: 30000,
       };
 
       if (process.env.POSTGRES_URL) {
@@ -59,7 +58,8 @@ function getClient() {
           },
           max: 20,
           idleTimeoutMillis: 30000,
-          connectionTimeoutMillis: 2000,
+          connectionTimeoutMillis: 30000,
+          statement_timeout: 30000,
         });
       } else {
         pgPool = new Pool(config);
@@ -74,13 +74,11 @@ function getClient() {
   }
 }
 
-// 定义一个通用的查询结果类型
 type CommonQueryResult<T = any> = {
   rows: T[];
   rowCount: number;
 };
 
-// 导出一个通用的查询函数
 export async function query<T = any>(
   text: string,
   params?: any[]
@@ -100,7 +98,6 @@ export async function query<T = any>(
       };
     } catch (error) {
       console.error("[DB Query Error]", error);
-      // 如果连接出错，重置连接状态
       if (vercelPool) {
         vercelPool.isConnected = false;
       }
@@ -128,7 +125,6 @@ export async function query<T = any>(
   }
 }
 
-// 确保在应用关闭时清理连接
 if (typeof window === "undefined") {
   process.on("SIGTERM", async () => {
     console.log("SIGTERM received, closing database connections");
@@ -143,3 +139,347 @@ if (typeof window === "undefined") {
 }
 
 export { getClient };
+
+export async function ensureTablesExist() {
+  try {
+    const usersTableExists = await query(`
+      SELECT EXISTS (
+        SELECT FROM information_schema.tables 
+        WHERE table_name = 'users'
+      );
+    `);
+
+    if (!usersTableExists.rows[0].exists) {
+      await query(`
+        CREATE TABLE IF NOT EXISTS users (
+          id TEXT PRIMARY KEY,
+          email TEXT NOT NULL,
+          name TEXT NOT NULL,
+          role TEXT NOT NULL DEFAULT 'user',
+          balance DECIMAL(16, 6) NOT NULL DEFAULT 0,
+          created_at TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP,
+          deleted BOOLEAN DEFAULT FALSE
+        );
+      `);
+    } else {
+      try {
+        await query(`
+          DO $$ 
+          BEGIN 
+            BEGIN
+              ALTER TABLE users 
+              ADD COLUMN deleted BOOLEAN DEFAULT FALSE;
+            EXCEPTION 
+              WHEN duplicate_column THEN NULL;
+            END;
+          END $$;
+        `);
+      } catch (error) {
+        console.error("Error adding deleted column to users table:", error);
+      }
+    }
+
+    const modelPricesTableExists = await query(`
+      SELECT EXISTS (
+        SELECT FROM information_schema.tables 
+        WHERE table_name = 'model_prices'
+      );
+    `);
+
+    const defaultInputPrice = parseFloat(
+      process.env.DEFAULT_MODEL_INPUT_PRICE || "60"
+    );
+    const defaultOutputPrice = parseFloat(
+      process.env.DEFAULT_MODEL_OUTPUT_PRICE || "60"
+    );
+    const defaultPerMsgPrice = parseFloat(
+      process.env.DEFAULT_MODEL_PER_MSG_PRICE || "-1"
+    );
+
+    if (!modelPricesTableExists.rows[0].exists) {
+      await query(`
+        CREATE TABLE IF NOT EXISTS model_prices (
+          id TEXT PRIMARY KEY,
+          name TEXT NOT NULL,
+          base_model_id TEXT,
+          input_price NUMERIC(10, 6) DEFAULT ${defaultInputPrice},
+          output_price NUMERIC(10, 6) DEFAULT ${defaultOutputPrice},
+          per_msg_price NUMERIC(10, 6) DEFAULT ${defaultPerMsgPrice},
+          updated_at TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP
+        );
+      `);
+    } else {
+      try {
+        await query(`
+          DO $$ 
+          BEGIN 
+            BEGIN
+              ALTER TABLE model_prices 
+              ADD COLUMN per_msg_price NUMERIC(10, 6) DEFAULT ${defaultPerMsgPrice};
+            EXCEPTION 
+              WHEN duplicate_column THEN NULL;
+            END;
+          END $$;
+        `);
+      } catch (error) {
+        console.error("Error adding per_msg_price column:", error);
+      }
+
+      try {
+        await query(`
+          DO $$ 
+          BEGIN 
+            BEGIN
+              ALTER TABLE model_prices 
+              ADD COLUMN base_model_id TEXT;
+            EXCEPTION 
+              WHEN duplicate_column THEN NULL;
+            END;
+          END $$;
+        `);
+      } catch (error) {
+        console.error("Error adding base_model_id column:", error);
+      }
+    }
+
+    const userUsageRecordsTableExists = await query(`
+      SELECT EXISTS (
+        SELECT FROM information_schema.tables 
+        WHERE table_name = 'user_usage_records'
+      );
+    `);
+
+    if (!userUsageRecordsTableExists.rows[0].exists) {
+      await query(`
+        CREATE TABLE IF NOT EXISTS user_usage_records (
+          id SERIAL PRIMARY KEY,
+          user_id TEXT NOT NULL,
+          nickname VARCHAR(255) NOT NULL,
+          use_time TIMESTAMP WITH TIME ZONE DEFAULT CURRENT_TIMESTAMP,
+          model_name VARCHAR(255) NOT NULL,
+          input_tokens INTEGER NOT NULL,
+          output_tokens INTEGER NOT NULL,
+          cost DECIMAL(10, 4) NOT NULL,
+          balance_after DECIMAL(10, 4) NOT NULL,
+          FOREIGN KEY (user_id) REFERENCES users(id)
+        );
+      `);
+    }
+
+    console.log("Database tables initialized successfully");
+  } catch (error) {
+    console.error("Failed to initialize database tables:", error);
+    throw error;
+  }
+}
+
+export async function initDatabase() {
+  try {
+    await ensureTablesExist();
+    console.log("Database initialized successfully");
+  } catch (error) {
+    console.error("Failed to initialize database:", error);
+    throw error;
+  }
+}
+
+export interface ModelPrice {
+  id: string;
+  name: string;
+  input_price: number;
+  output_price: number;
+  per_msg_price: number;
+  updated_at: Date;
+}
+
+export interface UserUsageRecord {
+  id: number;
+  userId: number;
+  nickname: string;
+  useTime: Date;
+  modelName: string;
+  inputTokens: number;
+  outputTokens: number;
+  cost: number;
+  balanceAfter: number;
+}
+
+export async function getOrCreateModelPrices(
+  models: Array<{ id: string; name: string; base_model_id?: string }>
+): Promise<ModelPrice[]> {
+  try {
+    const defaultInputPrice = parseFloat(
+      process.env.DEFAULT_MODEL_INPUT_PRICE || "60"
+    );
+    const defaultOutputPrice = parseFloat(
+      process.env.DEFAULT_MODEL_OUTPUT_PRICE || "60"
+    );
+    const defaultPerMsgPrice = parseFloat(
+      process.env.DEFAULT_MODEL_PER_MSG_PRICE || "-1"
+    );
+
+    const modelIds = models.map((m) => m.id);
+    const baseModelIds = models.map((m) => m.base_model_id).filter((id) => id);
+
+    const existingModelsResult = await query(
+      `SELECT * FROM model_prices WHERE id = ANY($1::text[])`,
+      [modelIds]
+    );
+
+    const baseModelsResult = await query(
+      `SELECT * FROM model_prices WHERE id = ANY($1::text[])`,
+      [baseModelIds]
+    );
+
+    const existingModels = new Map(
+      existingModelsResult.rows.map((row) => [row.id, row])
+    );
+    const baseModels = new Map(
+      baseModelsResult.rows.map((row) => [row.id, row])
+    );
+
+    const modelsToUpdate = models.filter((m) => existingModels.has(m.id));
+    const missingModels = models.filter((m) => !existingModels.has(m.id));
+
+    if (modelsToUpdate.length > 0) {
+      for (const model of modelsToUpdate) {
+        await query(`UPDATE model_prices SET name = $2 WHERE id = $1`, [
+          model.id,
+          model.name,
+        ]);
+      }
+    }
+
+    if (missingModels.length > 0) {
+      for (const model of missingModels) {
+        const baseModel = model.base_model_id
+          ? baseModels.get(model.base_model_id)
+          : null;
+
+        await query(
+          `INSERT INTO model_prices (id, name, input_price, output_price, per_msg_price)
+           VALUES ($1, $2, $3, $4, $5)
+           RETURNING *`,
+          [
+            model.id,
+            model.name,
+            baseModel?.input_price ?? defaultInputPrice,
+            baseModel?.output_price ?? defaultOutputPrice,
+            baseModel?.per_msg_price ?? defaultPerMsgPrice,
+          ]
+        );
+      }
+    }
+
+    const updatedModelsResult = await query(
+      `SELECT * FROM model_prices WHERE id = ANY($1::text[])`,
+      [modelIds]
+    );
+
+    return updatedModelsResult.rows.map((row) => ({
+      id: row.id,
+      name: row.name,
+      input_price: Number(row.input_price),
+      output_price: Number(row.output_price),
+      per_msg_price: Number(row.per_msg_price),
+      updated_at: row.updated_at,
+    }));
+  } catch (error) {
+    console.error("Error in getOrCreateModelPrices:", error);
+    throw error;
+  }
+}
+
+export async function updateModelPrice(
+  id: string,
+  input_price: number,
+  output_price: number,
+  per_msg_price: number
+): Promise<ModelPrice | null> {
+  try {
+    const result = await query(
+      `UPDATE model_prices 
+       SET 
+         input_price = CAST($2 AS NUMERIC(10,6)),
+         output_price = CAST($3 AS NUMERIC(10,6)),
+         per_msg_price = CAST($4 AS NUMERIC(10,6)),
+         updated_at = CURRENT_TIMESTAMP
+       WHERE id = $1
+       RETURNING *`,
+      [id, input_price, output_price, per_msg_price]
+    );
+
+    if (result.rows[0]) {
+      return {
+        id: result.rows[0].id,
+        name: result.rows[0].model_name,
+        input_price: Number(result.rows[0].input_price),
+        output_price: Number(result.rows[0].output_price),
+        per_msg_price: Number(result.rows[0].per_msg_price),
+        updated_at: result.rows[0].updated_at,
+      };
+    }
+    return null;
+  } catch (error) {
+    console.error("Error updating model price:", error);
+    throw error;
+  }
+}
+
+export async function updateUserBalance(userId: string, balance: number) {
+  try {
+    const result = await query(
+      `UPDATE users
+       SET balance = $2
+       WHERE id = $1
+       RETURNING id, email, balance`,
+      [userId, balance]
+    );
+
+    return result.rows[0];
+  } catch (error) {
+    console.error("Error in updateUserBalance:", error);
+    throw error;
+  }
+}
+
+export const pool = {
+  connect: async () => {
+    if (isVercel) {
+      return {
+        query: async (text: string, params?: any[]) => {
+          const client = await getVercelClient();
+          const result = await client.query({
+            text,
+            values: params || [],
+          });
+          return result;
+        },
+        release: () => {},
+      };
+    } else {
+      return (pgPool || (getClient() as Pool)).connect();
+    }
+  },
+  query: async (text: string, params?: any[]) => {
+    if (isVercel) {
+      const client = await getVercelClient();
+      return client.query({
+        text,
+        values: params || [],
+      });
+    } else {
+      return (pgPool || (getClient() as Pool)).query(text, params);
+    }
+  },
+  end: async () => {
+    if (isVercel) {
+      if (vercelPool?.client) {
+        await vercelPool.client.end();
+        vercelPool.isConnected = false;
+      }
+    } else if (pgPool) {
+      await pgPool.end();
+    }
+  },
+};

lib/db/index.ts

@@ -1,8 +1,7 @@
 import { query } from "./client";
 import { ensureUserTableExists } from "./users";
-import { ModelPrice } from "../db";
+import { ModelPrice, updateModelPrice } from "./client";
 
-// 创建模型价格表
 async function ensureModelPricesTableExists() {
   const defaultInputPrice = parseFloat(
     process.env.DEFAULT_MODEL_INPUT_PRICE || "60"
@@ -16,7 +15,7 @@ async function ensureModelPricesTableExists() {
 
   await query(
     `CREATE TABLE IF NOT EXISTS model_prices (
-      model_id TEXT PRIMARY KEY,
+      id TEXT PRIMARY KEY,
       model_name TEXT NOT NULL,
       input_price DECIMAL(10, 6) DEFAULT CAST($1 AS DECIMAL(10, 6)),
       output_price DECIMAL(10, 6) DEFAULT CAST($2 AS DECIMAL(10, 6)),
@@ -26,7 +25,6 @@ async function ensureModelPricesTableExists() {
     [defaultInputPrice, defaultOutputPrice, defaultPerMsgPrice]
   );
 
-  // 为现有记录添加 per_msg_price 字段（如果不存在）
   await query(
     `DO $$ 
     BEGIN 
@@ -56,16 +54,16 @@ export async function getOrCreateModelPrice(
     );
 
     const result = await query(
-      `INSERT INTO model_prices (model_id, model_name, per_msg_price, updated_at)
+      `INSERT INTO model_prices (id, model_name, per_msg_price, updated_at)
        VALUES ($1, $2, CAST($3 AS DECIMAL(10, 6)), CURRENT_TIMESTAMP)
-       ON CONFLICT (model_id) DO UPDATE 
+       ON CONFLICT (id) DO UPDATE 
        SET model_name = $2, updated_at = CURRENT_TIMESTAMP
        RETURNING *`,
       [id, name, defaultPerMsgPrice]
     );
 
     return {
-      id: result.rows[0].model_id,
+      id: result.rows[0].id,
       name: result.rows[0].model_name,
       input_price: Number(result.rows[0].input_price),
       output_price: Number(result.rows[0].output_price),
@@ -83,37 +81,6 @@ export async function getOrCreateModelPrice(
   }
 }
 
-export async function updateModelPrice(
-  modelId: string,
-  input_price: number,
-  output_price: number,
-  per_msg_price: number
-): Promise<ModelPrice | null> {
-  const result = await query(
-    `UPDATE model_prices 
-    SET 
-      input_price = CAST($2 AS DECIMAL(10,6)),
-      output_price = CAST($3 AS DECIMAL(10,6)),
-      per_msg_price = CAST($4 AS DECIMAL(10,6)),
-      updated_at = CURRENT_TIMESTAMP
-    WHERE model_id = $1
-    RETURNING *;`,
-    [modelId, input_price, output_price, per_msg_price]
-  );
-
-  if (result.rows.length === 0) {
-    return null;
-  }
-
-  return {
-    id: result.rows[0].model_id,
-    name: result.rows[0].model_name,
-    input_price: Number(result.rows[0].input_price),
-    output_price: Number(result.rows[0].output_price),
-    per_msg_price: Number(result.rows[0].per_msg_price),
-    updated_at: result.rows[0].updated_at,
-  };
-}
 export {
   getUsers,
   getOrCreateUser,

lib/utils/inlet-cost.ts

@@ -5,13 +5,11 @@ interface ModelInletCost {
 function parseInletCostConfig(config: string | undefined): ModelInletCost {
   if (!config) return {};
 
-  // 如果配置是一个数字，对所有模型使用相同的预扣费
   const numericValue = Number(config);
   if (!isNaN(numericValue)) {
     return { default: numericValue };
   }
 
-  // 否则解析 model1:0.32,model2:0.01 格式
   try {
     const costs: ModelInletCost = {};
     config.split(",").forEach((pair) => {

lib/version.ts

@@ -1,2 +1 @@
-// 从环境变量获取版本号
 export const APP_VERSION = process.env.NEXT_PUBLIC_APP_VERSION || "0.0.1";

middleware.ts

@@ -7,15 +7,24 @@ const ACCESS_TOKEN = process.env.ACCESS_TOKEN;
 export async function middleware(request: NextRequest) {
   const { pathname } = request.nextUrl;
 
-  // 只验证 inlet/outlet/test API 请求
   if (
     pathname.startsWith("/api/v1/inlet") ||
     pathname.startsWith("/api/v1/outlet") ||
-    pathname.startsWith("/api/v1/models/test")
+    pathname.startsWith("/api/v1/models") ||
+    pathname.startsWith("/api/v1/panel") ||
+    pathname.startsWith("/api/v1/config") ||
+    pathname.startsWith("/api/v1/users")
   ) {
-    // API 请求验证
-    if (!API_KEY) {
-      console.error("API Key is not set");
+    const token =
+      pathname.startsWith("/api/v1/panel") ||
+      pathname.startsWith("/api/v1/config") ||
+      pathname.startsWith("/api/v1/users") ||
+      pathname.startsWith("/api/v1/models")
+        ? ACCESS_TOKEN
+        : API_KEY;
+
+    if (!token) {
+      console.error("API Key or Access Token is not set");
       return NextResponse.json(
         { error: "Server configuration error" },
         { status: 500 }
@@ -25,14 +34,13 @@ export async function middleware(request: NextRequest) {
     const authHeader = request.headers.get("authorization");
     const providedKey = authHeader?.replace("Bearer ", "");
 
-    if (!providedKey || providedKey !== API_KEY) {
-      console.log("Invalid API key");
-      return NextResponse.json({ error: "Invalid API key" }, { status: 401 });
+    if (!providedKey || providedKey !== token) {
+      console.log("Invalid API key or token");
+      return NextResponse.json({ error: "Unauthorized" }, { status: 401 });
     }
 
     return NextResponse.next();
   } else if (!pathname.startsWith("/api/")) {
-    // 页面访问验证
     if (!ACCESS_TOKEN) {
       console.error("ACCESS_TOKEN is not set");
       return NextResponse.json(
@@ -41,12 +49,10 @@ export async function middleware(request: NextRequest) {
       );
     }
 
-    // 如果是令牌验证页面，直接允许访问
     if (pathname === "/token") {
       return NextResponse.next();
     }
 
-    // 添加 no-store 和 no-cache 头，防止 Cloudflare 缓存
     const response = NextResponse.next();
     response.headers.set(
       "Cache-Control",
@@ -57,14 +63,14 @@ export async function middleware(request: NextRequest) {
 
     return response;
   } else if (pathname.startsWith("/api/config/key")) {
-    // 确保这个路径不被中间件拦截
+    return NextResponse.next();
+  } else if (pathname.startsWith("/api/init")) {
     return NextResponse.next();
   }
 
   return NextResponse.next();
 }
 
-// 配置中间件匹配的路由
 export const config = {
   matcher: ["/((?!_next/static|_next/image|favicon.ico).*)"],
 };

package.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://json.schemastore.org/package.json",
   "name": "openwebui-usage-monitor",
-  "version": "0.3.5",
+  "version": "0.3.7",
   "private": true,
   "scripts": {
     "dev": "next dev",