From b3764a793806dd7ee366b4e3ab6bce5a103b7822 Mon Sep 17 00:00:00 2001
From: mtvpls <mtvpls@redaaa.ip-ddns.com>
Date: Mon, 26 Jan 2026 08:58:04 +0800
Subject: [PATCH] =?UTF-8?q?middleware=E6=87=92=E5=8A=A0=E8=BD=BDstorage?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/lib/refresh-token.ts | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/src/lib/refresh-token.ts b/src/lib/refresh-token.ts
index ee7b580..ada24ca 100644
--- a/src/lib/refresh-token.ts
+++ b/src/lib/refresh-token.ts
@@ -1,6 +1,15 @@
 /* eslint-disable no-console */
 
-import { getStorage } from './db';
+// Lazy import to avoid Edge Runtime issues in middleware
+let getStorage: (() => any) | null = null;
+
+async function loadStorage() {
+  if (!getStorage) {
+    const db = await import('./db');
+    getStorage = db.getStorage;
+  }
+  return getStorage();
+}
 
 // Token 配置
 export const TOKEN_CONFIG = {
@@ -38,7 +47,7 @@ export async function storeRefreshToken(
   tokenData: TokenData
 ): Promise<void> {
   const hashKey = `user_tokens:${username}`;
-  const storage = getStorage();
+  const storage = await loadStorage();
 
   if (!storage || typeof (storage as any).adapter?.hSet !== 'function') {
     console.warn('Redis Hash not supported, skipping token storage');
@@ -65,7 +74,7 @@ export async function verifyRefreshToken(
   refreshToken: string
 ): Promise<boolean> {
   const hashKey = `user_tokens:${username}`;
-  const storage = getStorage();
+  const storage = await loadStorage();
 
   if (!storage || typeof (storage as any).adapter?.hGet !== 'function') {
     console.warn('Redis Hash not supported');
@@ -114,7 +123,7 @@ export async function revokeRefreshToken(
   tokenId: string
 ): Promise<void> {
   const hashKey = `user_tokens:${username}`;
-  const storage = getStorage();
+  const storage = await loadStorage();
 
   if (!storage || typeof (storage as any).adapter?.hDel !== 'function') {
     console.warn('Redis Hash not supported');
@@ -138,7 +147,7 @@ export async function getUserDevices(username: string): Promise<Array<{
   expiresAt: number;
 }>> {
   const hashKey = `user_tokens:${username}`;
-  const storage = getStorage();
+  const storage = await loadStorage();
 
   if (!storage || typeof (storage as any).adapter?.hGetAll !== 'function') {
     console.warn('Redis Hash not supported');
@@ -188,7 +197,7 @@ export async function getUserDevices(username: string): Promise<Array<{
 // 撤销所有 Token
 export async function revokeAllRefreshTokens(username: string): Promise<void> {
   const hashKey = `user_tokens:${username}`;
-  const storage = getStorage();
+  const storage = await loadStorage();
 
   if (!storage || typeof (storage as any).adapter?.del !== 'function') {
     console.warn('Redis Hash not supported');
@@ -206,7 +215,7 @@ export async function revokeAllRefreshTokens(username: string): Promise<void> {
 // 清理过期的 Token（定期任务）
 export async function cleanupExpiredTokens(username: string): Promise<number> {
   const hashKey = `user_tokens:${username}`;
-  const storage = getStorage();
+  const storage = await loadStorage();
 
   if (!storage || typeof (storage as any).adapter?.hGetAll !== 'function') {
     return 0;