From 371d66b28929710d15752e2b632de55698b2c038 Mon Sep 17 00:00:00 2001 From: xiongxiao Date: Tue, 23 Dec 2025 13:15:16 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96=20getToken=20?= =?UTF-8?q?=E5=87=BD=E6=95=B0=EF=BC=8C=E7=A7=BB=E9=99=A4=E6=97=A0=E7=94=A8?= =?UTF-8?q?=E7=9A=84=E5=93=8D=E5=BA=94=E5=8F=82=E6=95=B0=EF=BC=8C=E8=B0=83?= =?UTF-8?q?=E6=95=B4=E6=8E=88=E6=9D=83=E8=BF=87=E6=BB=A4=E9=80=BB=E8=BE=91?= =?UTF-8?q?=E4=BB=A5=E6=94=AF=E6=8C=81=E6=96=B0=E8=B7=AF=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- assistant/src/module/http-token.ts | 10 +--------- assistant/src/server.ts | 4 ++-- assistant/src/services/proxy/proxy-page-index.ts | 7 +++++-- 3 files changed, 8 insertions(+), 13 deletions(-) diff --git a/assistant/src/module/http-token.ts b/assistant/src/module/http-token.ts index 27b7ea7..6f4f734 100644 --- a/assistant/src/module/http-token.ts +++ b/assistant/src/module/http-token.ts @@ -15,14 +15,9 @@ const cookie = { return cookies; } } -export const getToken = async (req: http.IncomingMessage, res: http.ServerResponse) => { +export const getToken = async (req: http.IncomingMessage) => { let token = (req.headers?.['authorization'] as string) || (req.headers?.['Authorization'] as string) || ''; const url = new URL(req.url || '', 'http://localhost'); - const resNoPermission = () => { - res.statusCode = 401; - res.end(error('Invalid authorization')); - return { tokenUser: null, token: null }; - }; if (!token) { token = url.searchParams.get('token') || ''; } @@ -30,9 +25,6 @@ export const getToken = async (req: http.IncomingMessage, res: http.ServerRespon const parsedCookies = cookie.parse(req.headers.cookie || ''); token = parsedCookies.token || ''; } - if (!token) { - return resNoPermission(); - } if (token) { token = token.replace('Bearer ', ''); } diff --git a/assistant/src/server.ts b/assistant/src/server.ts index 52338f7..8102c5c 100644 --- a/assistant/src/server.ts +++ b/assistant/src/server.ts @@ -10,7 +10,7 @@ import chalk from 'chalk'; import { AssistantApp } from './lib.ts'; import { getBunPath } from './module/get-bun-path.ts'; import { qwenAsr } from './services/asr/qwen-asr.ts'; -export const runServer = async (port: number = 51015, listenPath = '127.0.0.1') => { +export const runServer = async (port: number = 51515, listenPath = '127.0.0.1') => { let _port: number | undefined; if (port) { _port = await getPort({ port }); @@ -21,7 +21,7 @@ export const runServer = async (port: number = 51015, listenPath = '127.0.0.1') } if (!_port) { // 检车端口可用性 - const isPortAvailable = await getPort({ port: portNumbers(51015, 52000) }); + const isPortAvailable = await getPort({ port: portNumbers(51515, 52000) }); if (!isPortAvailable) { console.log(`Port ${isPortAvailable} is not available`); process.exit(1); diff --git a/assistant/src/services/proxy/proxy-page-index.ts b/assistant/src/services/proxy/proxy-page-index.ts index 100a048..81ed1b2 100644 --- a/assistant/src/services/proxy/proxy-page-index.ts +++ b/assistant/src/services/proxy/proxy-page-index.ts @@ -34,7 +34,7 @@ const authFilter = async (req: http.IncomingMessage, res: http.ServerResponse) = return false; } // 放开首页 - if (pathname.startsWith('/root/home')) { + if (pathname.startsWith('/root/home') || pathname === '/root/cli') { return false; } // 放开api, 以 /api, /v1, /client, /serve 开头的请求 @@ -47,8 +47,11 @@ const authFilter = async (req: http.IncomingMessage, res: http.ServerResponse) = if (share === 'public') { return false; } - const { token } = await getToken(req, res) + const { token } = await getToken(req) if (!token) { + // no token 转到登录页面 + res.writeHead(302, { Location: `/root/home/` }); + res.end(); return false; } const tokenUser = await getTokenUserCache(token);