78 lines
2.7 KiB
TypeScript
78 lines
2.7 KiB
TypeScript
import { tool } from '@kevisual/router';
|
||
import { app, cnb } from '../../app.ts';
|
||
import { addKeepAliveData, KeepAliveData, removeKeepAliveData, createLiveData } from '../../../src/workspace/keep-file-live.ts';
|
||
|
||
// 保持工作空间存活技能
|
||
app.route({
|
||
path: 'cnb',
|
||
key: 'keep-workspace-alive',
|
||
description: '保持工作空间存活技能，参数repo:代码仓库路径，例如 user/repo，pipelineId:流水线ID，例如 cnb-708-1ji9sog7o-001',
|
||
middleware: ['admin-auth'],
|
||
metadata: {
|
||
tags: [],
|
||
...({
|
||
args: {
|
||
repo: tool.schema.string().describe('代码仓库路径，例如 user/repo'),
|
||
pipelineId: tool.schema.string().describe('流水线ID，例如 cnb-708-1ji9sog7o-001'),
|
||
}
|
||
})
|
||
}
|
||
}).define(async (ctx) => {
|
||
const repo = ctx.query?.repo as string;
|
||
const pipelineId = ctx.query?.pipelineId as string;
|
||
|
||
if (!repo || !pipelineId) {
|
||
ctx.throw(400, '缺少参数 repo 或 pipelineId');
|
||
}
|
||
const validCookie = await cnb.user.checkCookieValid()
|
||
if (validCookie.code !== 200) {
|
||
ctx.throw(401, 'CNB_COOKIE 环境变量无效或已过期，请重新登录获取新的cookie');
|
||
}
|
||
const res = await cnb.workspace.getWorkspaceCookie(repo, pipelineId);
|
||
let wsUrl = `wss://${pipelineId}.cnb.space:443?skipWebSocketFrames=false`;
|
||
let cookie = '';
|
||
if (res.code === 200) {
|
||
cookie = res.data.value;
|
||
console.log(`启动保持工作空间 ${wsUrl} 存活的任务`);
|
||
} else {
|
||
ctx.throw(500, `获取工作空间访问cookie失败: ${res.message}`);
|
||
}
|
||
|
||
console.log(`启动保持工作空间 ${wsUrl} 存活的任务`);
|
||
|
||
const config: KeepAliveData = createLiveData({ cookie, repo, pipelineId });
|
||
addKeepAliveData(config);
|
||
|
||
ctx.body = { content: `已启动保持工作空间 ${wsUrl} 存活的任务`, data: config };
|
||
}).addTo(app);
|
||
|
||
// 停止保持工作空间存活技能
|
||
app.route({
|
||
path: 'cnb',
|
||
key: 'stop-keep-workspace-alive',
|
||
description: '停止保持工作空间存活技能, 参数repo:代码仓库路径，例如 user/repo，pipelineId:流水线ID，例如 cnb-708-1ji9sog7o-001',
|
||
middleware: ['admin-auth'],
|
||
metadata: {
|
||
tags: [],
|
||
...({
|
||
args: {
|
||
repo: tool.schema.string().describe('代码仓库路径，例如 user/repo'),
|
||
pipelineId: tool.schema.string().describe('流水线ID，例如 cnb-708-1ji9sog7o-001'),
|
||
}
|
||
})
|
||
}
|
||
}).define(async (ctx) => {
|
||
const repo = ctx.query?.repo as string;
|
||
const pipelineId = ctx.query?.pipelineId as string;
|
||
|
||
if (!repo || !pipelineId) {
|
||
ctx.throw(400, '缺少参数 repo 或 pipelineId');
|
||
}
|
||
removeKeepAliveData(repo, pipelineId);
|
||
ctx.body = { content: `已停止保持工作空间 ${repo}/${pipelineId} 存活的任务` };
|
||
}).addTo(app);
|
||
|
||
|
||
|
||
|