tvbox支持根据用户细分

2026-02-25 11:56:27 +08:00
parent 6a494f8a1b
commit 0402828c4c
16 changed files with 583 additions and 78 deletions
@@ -0,0 +1,46 @@
+import { NextRequest, NextResponse } from 'next/server';
+
+import { getAuthInfoFromCookie } from '@/lib/auth';
+import { db } from '@/lib/db';
+import { generateTvboxToken } from '@/lib/tvbox-token';
+
+export const runtime = 'nodejs';
+
+/**
+ * 重置用户的TVBox订阅token
+ * 旧token将失效
+ */
+export async function POST(request: NextRequest) {
+  try {
+    // 验证用户登录
+    const authInfo = getAuthInfoFromCookie(request);
+    if (!authInfo?.username) {
+      return NextResponse.json(
+        { error: '未登录' },
+        { status: 401 }
+      );
+    }
+
+    const username = authInfo.username;
+
+    // 生成新token
+    const newToken = generateTvboxToken();
+    await db.setTvboxSubscribeToken(username, newToken);
+
+    console.log(`用户 ${username} 重置了TVBox订阅token`);
+
+    return NextResponse.json({
+      token: newToken,
+      message: '订阅token已重置，旧链接已失效',
+    });
+  } catch (error) {
+    console.error('重置TVBox订阅token失败:', error);
+    return NextResponse.json(
+      {
+        error: '重置订阅token失败',
+        details: (error as Error).message,
+      },
+      { status: 500 }
+    );
+  }
+}
@@ -0,0 +1,47 @@
+import { NextRequest, NextResponse } from 'next/server';
+
+import { getAuthInfoFromCookie } from '@/lib/auth';
+import { db } from '@/lib/db';
+import { generateTvboxToken } from '@/lib/tvbox-token';
+
+export const runtime = 'nodejs';
+
+/**
+ * 获取用户的TVBox订阅token
+ * 如果用户没有token，自动生成一个
+ */
+export async function GET(request: NextRequest) {
+  try {
+    // 验证用户登录
+    const authInfo = getAuthInfoFromCookie(request);
+    if (!authInfo?.username) {
+      return NextResponse.json(
+        { error: '未登录' },
+        { status: 401 }
+      );
+    }
+
+    const username = authInfo.username;
+
+    // 获取token，如果没有则生成
+    let token = await db.getTvboxSubscribeToken(username);
+
+    if (!token) {
+      // 懒加载：首次访问时生成token
+      token = generateTvboxToken();
+      await db.setTvboxSubscribeToken(username, token);
+      console.log(`为用户 ${username} 生成TVBox订阅token`);
+    }
+
+    return NextResponse.json({ token });
+  } catch (error) {
+    console.error('获取TVBox订阅token失败:', error);
+    return NextResponse.json(
+      {
+        error: '获取订阅token失败',
+        details: (error as Error).message,
+      },
+      { status: 500 }
+    );
+  }
+}