import { tool } from '@kevisual/router';
import { app, cnbManager, notCNBCheck } from '../../app.ts';
import { addKeepAliveData, KeepAliveData, removeKeepAliveData, createLiveData } from '../../../src/workspace/keep-file-live.ts';
import { useKey } from '@kevisual/context';

// 保持工作空间存活技能
app.route({
  path: 'cnb',
  key: 'keep-workspace-alive',
  description: '保持工作空间存活技能，参数repo:代码仓库路径，例如 user/repo，pipelineId:流水线ID，例如 cnb-708-1ji9sog7o-001',
  middleware: ['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 cnb = await cnbManager.getContext(ctx);
  const repo = ctx.query?.repo as string;
  const pipelineId = ctx.query?.pipelineId as string;
  if (notCNBCheck(ctx)) return;
  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);
  if (res.code !== 200 || !res.data?.cookie) {
    ctx.throw(500, `获取工作空间 Cookie 失败: ${res.message}`);
  }

  // 添加保活数据
  const liveData = createLiveData({
    repo,
    pipelineId,
    cookie: res.data.cookie
  });
  addKeepAliveData(liveData);
  console.log('已添加 keep-alive 数据');

  ctx.body = { content: `已启动保持工作空间 ${repo}/${pipelineId} 存活的任务`, data: liveData };
}).addTo(app);

// 停止保持工作空间存活技能
app.route({
  path: 'cnb',
  key: 'stop-keep-workspace-alive',
  description: '停止保持工作空间存活技能, 参数repo:代码仓库路径，例如 user/repo，pipelineId:流水线ID，例如 cnb-708-1ji9sog7o-001',
  middleware: ['auth'],
  metadata: {
    tags: [],
    ...({
      args: {
        repo: tool.schema.string().describe('代码仓库路径，例如 user/repo'),
        pipelineId: tool.schema.string().describe('流水线ID，例如 cnb-708-1ji9sog7o-001'),
      }
    })
  }
}).define(async (ctx) => {
  if (notCNBCheck(ctx)) return;
  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);




app.route({
  path: 'cnb',
  key: 'keep-alive-current-workspace',
  description: '保持当前工作空间存活技能',
  middleware: ['auth'],
  metadata: {
    tags: ['opencode'],
    skill: 'keep-alive-current-workspace',
    title: '保持当前工作空间存活',
    summary: '保持当前工作空间存活，防止被关闭或释放资源',
  }
}).define(async (ctx) => {
  if (notCNBCheck(ctx)) return;
  const pipelineId = useKey('CNB_PIPELINE_ID');
  const repo = useKey('CNB_REPO_SLUG_LOWERCASE');
  if (!pipelineId || !repo) {
    ctx.throw(400, '当前环境缺少 CNB_PIPELINE_ID 或 CNB_REPO_SLUG_LOWERCASE 环境变量，无法保持工作空间存活');
  }
  const res = await app.run({ path: 'cnb', key: 'keep-workspace-alive', payload: { repo, pipelineId } }, ctx);
  ctx.forward(res);
}).addTo(app);