work-agent 0.1.0

package/lib/workflow.ts

@@ -0,0 +1,154 @@
+/**
+ * 工作流引擎 - 工单状态管理和审批流程
+ */
+
+import type { Ticket, Approval } from './db';
+import { getTicket, updateTicket, addApproval, createAuditLog } from './db';
+
+export type TicketStatus = Ticket['status'];
+
+// 状态转换规则
+const STATUS_TRANSITIONS: Record<TicketStatus, TicketStatus[]> = {
+	draft: ['pending', 'deleted'],
+	pending: ['approved', 'rejected'],
+	approved: ['executing', 'pending'],
+	rejected: ['pending', 'deleted'],
+	executing: ['completed', 'failed'],
+	completed: [],
+	failed: ['pending'],
+	deleted: [],
+};
+
+/**
+ * 检查状态转换是否允许
+ */
+export function canTransition(from: TicketStatus, to: TicketStatus): boolean {
+	if (from === 'deleted') return false;
+	return STATUS_TRANSITIONS[from]?.includes(to) ?? false;
+}
+
+/**
+ * 转换工单状态
+ */
+export async function transitionTicket(ticketId: string, newStatus: TicketStatus, userId: string): Promise<Ticket | null> {
+	const ticket = await getTicket(ticketId);
+	if (!ticket) {
+		throw new Error(`工单不存在: ${ticketId}`);
+	}
+
+	if (!canTransition(ticket.status, newStatus)) {
+		throw new Error(`不能将工单从 ${ticket.status} 转换为 ${newStatus}`);
+	}
+
+	const updated = await updateTicket(ticketId, { status: newStatus });
+
+	// 记录审计日志
+	await createAuditLog({
+		action: 'status_transition',
+		userId,
+		ticketId,
+		sessionId: ticket.aiSessionId || undefined,
+		details: { from: ticket.status, to: newStatus },
+		status: 'success',
+	});
+
+	return updated;
+}
+
+/**
+ * 提交工单审批
+ */
+export async function submitTicket(ticketId: string, userId: string): Promise<Ticket | null> {
+	return await transitionTicket(ticketId, 'pending', userId);
+}
+
+/**
+ * 批准工单
+ */
+export async function approveTicket(ticketId: string, approver: string, comment?: string): Promise<Ticket | null> {
+	const ticket = await getTicket(ticketId);
+	if (!ticket) {
+		throw new Error(`工单不存在: ${ticketId}`);
+	}
+
+	if (ticket.status !== 'pending') {
+		throw new Error(`只能批准待审核状态的工单，当前状态: ${ticket.status}`);
+	}
+
+	// 添加审批记录
+	await addApproval(ticketId, {
+		approver,
+		decision: 'approved',
+		comment,
+	});
+
+	// 转换状态
+	return await transitionTicket(ticketId, 'approved', approver);
+}
+
+/**
+ * 拒绝工单
+ */
+export async function rejectTicket(ticketId: string, approver: string, comment?: string): Promise<Ticket | null> {
+	const ticket = await getTicket(ticketId);
+	if (!ticket) {
+		throw new Error(`工单不存在: ${ticketId}`);
+	}
+
+	if (ticket.status !== 'pending') {
+		throw new Error(`只能拒绝待审核状态的工单，当前状态: ${ticket.status}`);
+	}
+
+	// 添加审批记录
+	await addApproval(ticketId, {
+		approver,
+		decision: 'rejected',
+		comment,
+	});
+
+	// 转换状态
+	return await transitionTicket(ticketId, 'rejected', approver);
+}
+
+/**
+ * 开始执行
+ */
+export async function startExecution(ticketId: string, userId: string): Promise<Ticket | null> {
+	return await transitionTicket(ticketId, 'executing', userId);
+}
+
+/**
+ * 完成执行
+ */
+export async function completeExecution(ticketId: string, userId: string): Promise<Ticket | null> {
+	return await transitionTicket(ticketId, 'completed', userId);
+}
+
+/**
+ * 执行失败
+ */
+export async function failExecution(ticketId: string, userId: string, error?: string): Promise<Ticket | null> {
+	const ticket = await getTicket(ticketId);
+	const updated = await transitionTicket(ticketId, 'failed', userId);
+
+	// 记录错误详情
+	if (error && ticket) {
+		await createAuditLog({
+			action: 'execution_failed',
+			userId,
+			ticketId,
+			sessionId: ticket.aiSessionId || undefined,
+			details: { error },
+			status: 'failure',
+		});
+	}
+
+	return updated;
+}
+
+/**
+ * 重新提交
+ */
+export async function resubmitTicket(ticketId: string, userId: string): Promise<Ticket | null> {
+	return await transitionTicket(ticketId, 'pending', userId);
+}

package/lib/zip-tool.ts

@@ -0,0 +1,191 @@
+/**
+ * ZIP打包和解压工具
+ * 用于技能市场的技能打包和安装
+ */
+
+import AdmZip from 'adm-zip';
+import { mkdirSync, existsSync, unlinkSync } from 'fs';
+import { join } from 'path';
+
+/**
+ * 将技能目录打包为ZIP
+ *
+ * @param skillPath 技能目录路径
+ * @param outputPath 输出ZIP文件路径
+ * @returns ZIP文件路径
+ */
+export function packSkillToZip(skillPath: string, outputPath: string): string {
+  try {
+    const zip = new AdmZip();
+
+    // 添加技能目录下的所有文件
+    zip.addLocalFolder(skillPath);
+
+    // 确保输出目录存在
+    const outputDir = join(outputPath, '..');
+    if (!existsSync(outputDir)) {
+      mkdirSync(outputDir, { recursive: true });
+    }
+
+    // 写入ZIP文件
+    zip.writeZip(outputPath);
+
+    console.log(`[zip-tool] Packed skill to ${outputPath}`);
+    return outputPath;
+  } catch (error) {
+    console.error('[zip-tool] Failed to pack skill:', error);
+    throw new Error(`Failed to pack skill: ${error}`);
+  }
+}
+
+/**
+ * 从ZIP解压技能
+ *
+ * @param zipPath ZIP文件路径
+ * @param outputPath 解压目标路径
+ * @returns 解压后的目录路径
+ */
+export function unpackSkillFromZip(zipPath: string, outputPath: string): string {
+  try {
+    const zip = new AdmZip(zipPath);
+
+    // 验证ZIP内容（防止路径穿越攻击）
+    const entries = zip.getEntries();
+    for (const entry of entries) {
+      const entryName = entry.entryName;
+
+      // 检查是否有路径穿越尝试
+      if (entryName.includes('..') || entryName.startsWith('/')) {
+        throw new Error(`Invalid zip entry: ${entryName}`);
+      }
+
+      // 检查是否包含必要的文件
+      const normalized = entryName.replace(/\\/g, '/');
+      if (normalized === 'SKILL.md' || normalized === 'SKILL.json') {
+        // 必要文件存在
+      }
+    }
+
+    // 确保输出目录存在
+    if (!existsSync(outputPath)) {
+      mkdirSync(outputPath, { recursive: true });
+    }
+
+    // 解压文件
+    zip.extractAllTo(outputPath, true);
+
+    console.log(`[zip-tool] Unpacked skill to ${outputPath}`);
+    return outputPath;
+  } catch (error) {
+    console.error('[zip-tool] Failed to unpack skill:', error);
+    throw new Error(`Failed to unpack skill: ${error}`);
+  }
+}
+
+/**
+ * 验证ZIP包是否是有效的技能包
+ *
+ * @param zipPath ZIP文件路径
+ * @returns 验证结果
+ */
+export function validateSkillZip(zipPath: string): { valid: boolean; error?: string } {
+  try {
+    const zip = new AdmZip(zipPath);
+    const entries = zip.getEntries();
+
+    let hasSkillJson = false;
+    let hasSkillMd = false;
+
+    for (const entry of entries) {
+      const normalized = entry.entryName.replace(/\\/g, '/');
+
+      if (normalized === 'SKILL.json') {
+        hasSkillJson = true;
+        // 验证JSON格式
+        try {
+          const content = entry.getData().toString('utf-8');
+          JSON.parse(content);
+        } catch {
+          return { valid: false, error: 'SKILL.json is not valid JSON' };
+        }
+      }
+
+      if (normalized === 'SKILL.md') {
+        hasSkillMd = true;
+      }
+    }
+
+    if (!hasSkillJson) {
+      return { valid: false, error: 'SKILL.json not found in zip' };
+    }
+
+    if (!hasSkillMd) {
+      return { valid: false, error: 'SKILL.md not found in zip' };
+    }
+
+    return { valid: true };
+  } catch (error: any) {
+    return { valid: false, error: error.message };
+  }
+}
+
+/**
+ * 从ZIP中读取SKILL.json
+ *
+ * @param zipPath ZIP文件路径
+ * @returns SKILL.json 内容
+ */
+export function readSkillJsonFromZip(zipPath: string): any {
+  try {
+    const zip = new AdmZip(zipPath);
+    const entry = zip.getEntries().find(e => e.entryName.replace(/\\/g, '/') === 'SKILL.json');
+
+    if (!entry) {
+      throw new Error('SKILL.json not found in zip');
+    }
+
+    const content = entry.getData().toString('utf-8');
+    return JSON.parse(content);
+  } catch (error) {
+    console.error('[zip-tool] Failed to read SKILL.json:', error);
+    throw error;
+  }
+}
+
+/**
+ * 从ZIP中读取SKILL.md
+ *
+ * @param zipPath ZIP文件路径
+ * @returns SKILL.md 内容
+ */
+export function readSkillMdFromZip(zipPath: string): string {
+  try {
+    const zip = new AdmZip(zipPath);
+    const entry = zip.getEntries().find(e => e.entryName.replace(/\\/g, '/') === 'SKILL.md');
+
+    if (!entry) {
+      throw new Error('SKILL.md not found in zip');
+    }
+
+    return entry.getData().toString('utf-8');
+  } catch (error) {
+    console.error('[zip-tool] Failed to read SKILL.md:', error);
+    throw error;
+  }
+}
+
+/**
+ * 删除临时ZIP文件
+ *
+ * @param zipPath ZIP文件路径
+ */
+export function cleanupZipFile(zipPath: string): void {
+  try {
+    if (existsSync(zipPath)) {
+      unlinkSync(zipPath);
+      console.log(`[zip-tool] Cleaned up zip file: ${zipPath}`);
+    }
+  } catch (error) {
+    console.error('[zip-tool] Failed to cleanup zip file:', error);
+  }
+}

package/next.config.js

@@ -0,0 +1,8 @@
+/** @type {import('next').NextConfig} */
+const nextConfig = {
+	serverExternalPackages: ['better-sqlite3', '@mariozechner/pi-coding-agent'],
+	// 启用 standalone 输出模式用于 Docker 部署
+	output: 'standalone',
+};
+
+module.exports = nextConfig;

package/package.json

@@ -0,0 +1,106 @@
+{
+	"name": "work-agent",
+	"version": "0.1.0",
+	"description": "AI work agent with local file system access, powered by Next.js",
+	"bin": {
+		"work-agent": "./bin/work-agent.js"
+	},
+	"scripts": {
+		"dev": "next dev",
+		"build": "next build && npm rebuild better-sqlite3 --prefix .next/standalone",
+		"start": "next start",
+		"lint": "next lint",
+		"test": "npm run test:sse",
+		"test:sse": "node tests/sse-heartbeat/test.js",
+		"postinstall": "node bin/postinstall.js"
+	},
+	"dependencies": {
+		"@kubernetes/client-node": "^0.20.0",
+		"@mariozechner/pi-coding-agent": "^0.53.1",
+		"@radix-ui/react-accordion": "^1.2.12",
+		"@radix-ui/react-avatar": "^1.1.11",
+		"@radix-ui/react-collapsible": "^1.1.12",
+		"@radix-ui/react-dialog": "^1.1.15",
+		"@radix-ui/react-dropdown-menu": "^2.1.16",
+		"@radix-ui/react-hover-card": "^1.1.15",
+		"@radix-ui/react-label": "^2.1.8",
+		"@radix-ui/react-popover": "^1.1.15",
+		"@radix-ui/react-progress": "^1.1.8",
+		"@radix-ui/react-scroll-area": "^1.2.10",
+		"@radix-ui/react-select": "^2.2.6",
+		"@radix-ui/react-separator": "^1.1.8",
+		"@radix-ui/react-slot": "^1.2.4",
+		"@radix-ui/react-switch": "^1.2.6",
+		"@radix-ui/react-tabs": "^1.1.13",
+		"@radix-ui/react-tooltip": "^1.2.8",
+		"@radix-ui/react-use-controllable-state": "^1.2.2",
+		"@rive-app/react-webgl2": "^4.27.0",
+		"@streamdown/cjk": "^1.0.2",
+		"@streamdown/code": "^1.0.3",
+		"@streamdown/math": "^1.0.2",
+		"@streamdown/mermaid": "^1.0.2",
+		"@types/better-sqlite3": "^7.6.13",
+		"@uiw/react-md-editor": "^4.0.11",
+		"@xyflow/react": "^12.10.1",
+		"adm-zip": "^0.5.16",
+		"ai": "^6.0.103",
+		"ai-elements": "^1.8.4",
+		"ansi-to-react": "^6.2.6",
+		"axios": "^1.13.5",
+		"better-sqlite3": "^12.6.2",
+		"class-variance-authority": "^0.7.1",
+		"clsx": "^2.1.1",
+		"cmdk": "^1.1.1",
+		"cron-parser": "^5.5.0",
+		"embla-carousel-react": "^8.6.0",
+		"highlight.js": "^11.11.1",
+		"js-yaml": "^4.1.0",
+		"lucide-react": "^0.563.0",
+		"media-chrome": "^4.18.0",
+		"motion": "^12.34.3",
+		"nanoid": "^5.1.6",
+		"next": "^15.0.0",
+		"open": "^10.2.0",
+		"pinyin-pro": "^3.24.5",
+		"react": "^19.0.0",
+		"react-dom": "^19.0.0",
+		"react-jsx-parser": "^2.4.1",
+		"recharts": "^3.7.0",
+		"shiki": "^3.23.0",
+		"streamdown": "^2.3.0",
+		"tailwind-merge": "^3.4.0",
+		"tailwindcss": "^3.4.19",
+		"tailwindcss-animate": "^1.0.7",
+		"tokenlens": "^1.3.1",
+		"use-stick-to-bottom": "^1.1.3"
+	},
+	"devDependencies": {
+		"@types/adm-zip": "^0.5.7",
+		"@types/js-yaml": "^4.0.9",
+		"@types/node": "^22.19.11",
+		"@types/react": "^19.0.0",
+		"@types/react-dom": "^19.0.0",
+		"@types/sql.js": "^1.4.9",
+		"autoprefixer": "^10.4.20",
+		"postcss": "^8.4.49",
+		"typescript": "^5.7.3",
+		"vitest": "^4.0.18"
+	},
+	"engines": {
+		"node": ">=18.18.0"
+	},
+	"files": [
+		"app",
+		"lib",
+		"public",
+		"bin",
+		"components",
+		"hooks",
+		"types",
+		"next.config.js",
+		"tailwind.config.ts",
+		"postcss.config.mjs",
+		"tsconfig.json",
+		"components.json"
+	]
+}

package/public/.gitkeep

@@ -0,0 +1 @@
+public directory placeholder

package/public/icon.svg

@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><rect width="18" height="14" x="3" y="4" rx="2"/><path d="M12 2v2"/><circle cx="12" cy="12" r="2"/><path d="M8 14v.01"/><path d="M16 14v.01"/><path d="M9 18v.01"/><path d="M15 18v.01"/></svg>

package/tsconfig.json

@@ -0,0 +1,42 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "lib": [
+      "ES2023",
+      "DOM",
+      "DOM.Iterable"
+    ],
+    "jsx": "preserve",
+    "module": "ESNext",
+    "moduleResolution": "bundler",
+    "resolveJsonModule": true,
+    "allowJs": true,
+    "strict": true,
+    "noEmit": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "incremental": true,
+    "paths": {
+      "@/*": [
+        "./*"
+      ]
+    },
+    "plugins": [
+      {
+        "name": "next"
+      }
+    ],
+    "isolatedModules": true
+  },
+  "include": [
+    "next-env.d.ts",
+    "**/*.ts",
+    "**/*.tsx",
+    ".next/types/**/*.ts"
+  ],
+  "exclude": [
+    "node_modules",
+    "tools"
+  ]
+}