更新 CNB 配置获取逻辑,修复未授权问题,添加测试文件
This commit is contained in:
@@ -4,7 +4,9 @@ import { useKey } from '@kevisual/context';
|
||||
import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
|
||||
export const getConfig = async (opts: { token?: string }) => {
|
||||
const kevisualEnv = useKey('KEVISUAL_ENV')
|
||||
const baseUrl = kevisualEnv === 'production' ? 'https://kevisual.cn/api/router' : 'https://kevisual.xiongxiao.me/api/router';
|
||||
const isCNB = useKey('CNB');
|
||||
let isProduction = kevisualEnv !== 'development' || (isCNB && !kevisualEnv);
|
||||
const baseUrl = isProduction ? 'https://kevisual.cn/api/router' : 'https://kevisual.xiongxiao.me/api/router';
|
||||
const res = await fetch(baseUrl, {
|
||||
method: 'POST',
|
||||
body: JSON.stringify({
|
||||
@@ -70,6 +72,8 @@ export class CNBManager {
|
||||
if (token) {
|
||||
return this.addCNB({ username, token, cookie })
|
||||
}
|
||||
} else {
|
||||
console.error('获取 CNB 配置失败', username, res)
|
||||
}
|
||||
return null
|
||||
}
|
||||
|
||||
@@ -9,7 +9,7 @@ app.route({
|
||||
middleware: ['auth'],
|
||||
|
||||
}).define(async (ctx) => {
|
||||
const tokenUser = ctx.tokenUser;
|
||||
const tokenUser = ctx.state?.tokenUser;
|
||||
if (!tokenUser) {
|
||||
ctx.throw(401, '未授权');
|
||||
}
|
||||
@@ -29,7 +29,8 @@ app.route({
|
||||
description: '获取我的cnb配置',
|
||||
middleware: ['auth'],
|
||||
}).define(async (ctx) => {
|
||||
const username = ctx.tokenUser?.username;
|
||||
const tokenUser = ctx.state?.tokenUser;
|
||||
const username = tokenUser?.username;
|
||||
const token = ctx.query?.token;
|
||||
if (!username) {
|
||||
ctx.throw(400, '未授权');
|
||||
|
||||
6
agent/routes/cnb-manager/test.ts
Normal file
6
agent/routes/cnb-manager/test.ts
Normal file
@@ -0,0 +1,6 @@
|
||||
const token = 'st_logh1b3ozq2resntxlnk4bccao0bon8e'
|
||||
import { CNBManager } from "../../modules/cnb-manager.ts"
|
||||
const cnbManager = new CNBManager()
|
||||
|
||||
const cnbItem = await cnbManager.getCNB({ username: 'root', kevisualToken: token });
|
||||
console.log('cnbItem', cnbItem)
|
||||
@@ -1,6 +1,6 @@
|
||||
{
|
||||
"name": "@kevisual/cnb",
|
||||
"version": "0.0.43",
|
||||
"version": "0.0.45",
|
||||
"description": "",
|
||||
"main": "index.js",
|
||||
"basename": "/root/cnb",
|
||||
|
||||
Reference in New Issue
Block a user