yeyinly/MoonTVPlus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

middleware懒加载storage

Browse Source
This commit is contained in:
mtvpls
2026-01-26 08:58:04 +08:00
parent 9b335da4ce
commit b3764a7938
1 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lib/refresh-token.ts
+16 -7
View File
@@ -1,6 +1,15 @@
/* eslint-disable no-console */ /* 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 配置 // Token 配置
export const TOKEN_CONFIG = { export const TOKEN_CONFIG = {
@@ -38,7 +47,7 @@ export async function storeRefreshToken(
tokenData: TokenData tokenData: TokenData
): Promise<void> { ): Promise<void> {
const hashKey = `user_tokens:${username}`; const hashKey = `user_tokens:${username}`;
const storage = getStorage(); const storage = await loadStorage();
if (!storage || typeof (storage as any).adapter?.hSet !== 'function') { if (!storage || typeof (storage as any).adapter?.hSet !== 'function') {
console.warn('Redis Hash not supported, skipping token storage'); console.warn('Redis Hash not supported, skipping token storage');
@@ -65,7 +74,7 @@ export async function verifyRefreshToken(
refreshToken: string refreshToken: string
): Promise<boolean> { ): Promise<boolean> {
const hashKey = `user_tokens:${username}`; const hashKey = `user_tokens:${username}`;
const storage = getStorage(); const storage = await loadStorage();
if (!storage || typeof (storage as any).adapter?.hGet !== 'function') { if (!storage || typeof (storage as any).adapter?.hGet !== 'function') {
console.warn('Redis Hash not supported'); console.warn('Redis Hash not supported');
@@ -114,7 +123,7 @@ export async function revokeRefreshToken(
tokenId: string tokenId: string
): Promise<void> { ): Promise<void> {
const hashKey = `user_tokens:${username}`; const hashKey = `user_tokens:${username}`;
const storage = getStorage(); const storage = await loadStorage();
if (!storage || typeof (storage as any).adapter?.hDel !== 'function') { if (!storage || typeof (storage as any).adapter?.hDel !== 'function') {
console.warn('Redis Hash not supported'); console.warn('Redis Hash not supported');
@@ -138,7 +147,7 @@ export async function getUserDevices(username: string): Promise<Array<{
expiresAt: number; expiresAt: number;
}>> { }>> {
const hashKey = `user_tokens:${username}`; const hashKey = `user_tokens:${username}`;
const storage = getStorage(); const storage = await loadStorage();
if (!storage || typeof (storage as any).adapter?.hGetAll !== 'function') { if (!storage || typeof (storage as any).adapter?.hGetAll !== 'function') {
console.warn('Redis Hash not supported'); console.warn('Redis Hash not supported');
@@ -188,7 +197,7 @@ export async function getUserDevices(username: string): Promise<Array<{
// 撤销所有 Token // 撤销所有 Token
export async function revokeAllRefreshTokens(username: string): Promise<void> { export async function revokeAllRefreshTokens(username: string): Promise<void> {
const hashKey = `user_tokens:${username}`; const hashKey = `user_tokens:${username}`;
const storage = getStorage(); const storage = await loadStorage();
if (!storage || typeof (storage as any).adapter?.del !== 'function') { if (!storage || typeof (storage as any).adapter?.del !== 'function') {
console.warn('Redis Hash not supported'); console.warn('Redis Hash not supported');
@@ -206,7 +215,7 @@ export async function revokeAllRefreshTokens(username: string): Promise<void> {
// 清理过期的 Token（定期任务） // 清理过期的 Token（定期任务）
export async function cleanupExpiredTokens(username: string): Promise<number> { export async function cleanupExpiredTokens(username: string): Promise<number> {
const hashKey = `user_tokens:${username}`; const hashKey = `user_tokens:${username}`;
const storage = getStorage(); const storage = await loadStorage();
if (!storage || typeof (storage as any).adapter?.hGetAll !== 'function') { if (!storage || typeof (storage as any).adapter?.hGetAll !== 'function') {
return 0; return 0;