@wanghaopeng1148/deskpet 2.0.0

package/server/db/task-repository.ts

@@ -0,0 +1,585 @@
+/**
+ * 任务仓储 — tasks / executions 表的读写映射
+ */
+import type { DatabaseSync } from 'node:sqlite'
+import type { ExecutionRecord, TaskConfig, TaskDto, TaskStatus, TaskTrigger } from '../../shared/types.ts'
+import { TaskStatus as TS } from '../../shared/types.ts'
+
+export interface TaskRow {
+  id: string
+  name: string
+  type: string
+  action: string
+  action_params: string
+  trigger_type: string
+  trigger_params: string
+  script_path: string
+  interpreter: string | null
+  script_args: string
+  script_timeout: number
+  work_dir: string | null
+  priority: number
+  retry_count: number
+  enabled: number
+  notify_channels: string
+  chain_next: string | null
+  chain_condition: string | null
+  one_time: number
+  status: string
+  execution_count: number
+  created_at: string
+  updated_at: string
+}
+
+/** 每个任务保留的执行记录条数（超出即删除最旧的） */
+export const EXECUTION_KEEP_PER_TASK = 10
+
+/** updateExecution 允许更新的列（camelCase → snake_case 白名单） */
+const EXEC_UPDATE_COLUMNS: Record<string, string> = {  taskId: 'task_id',
+  taskName: 'task_name',
+  status: 'status',
+  startedAt: 'started_at',
+  finishedAt: 'finished_at',
+  exitCode: 'exit_code',
+  durationMs: 'duration_ms',
+  /** 子进程 pid（服务重启后用于判断脚本是否仍在后台运行） */
+  pid: 'pid',
+  stdout: 'stdout',
+  stderr: 'stderr',
+  error: 'error'
+}
+
+export const TASK_COLUMNS =
+  'id, name, type, action, action_params, trigger_type, trigger_params, ' +
+  'script_path, interpreter, script_args, script_timeout, work_dir, priority, ' +
+  'retry_count, enabled, notify_channels, chain_next, chain_condition, one_time, ' +
+  'status, execution_count, created_at, updated_at'
+
+function defaultTrigger(): TaskTrigger {
+  return {
+    type: 'manual',
+    datetime: null,
+    cron: null,
+    delaySeconds: null,
+    intervalMinutes: null,
+    idleMinutes: null,
+    maxExecutions: -1,
+    holidayCheck: false
+  }
+}
+
+export function defaultTaskConfig(): TaskConfig {
+  return {
+    name: '',
+    type: 'manual',
+    trigger: defaultTrigger(),
+    action: 'python_script',
+    actionParams: {},
+    scriptPath: '',
+    interpreter: null,
+    scriptArgs: [],
+    scriptTimeout: 120,
+    scriptWorkDir: null,
+    retryCount: 0,
+    enabled: true,
+    notifyChannels: ['desktop'],
+    chainNext: null,
+    chainCondition: null,
+    oneTime: false
+  }
+}
+
+export function taskConfigToRow(config: TaskConfig, now: string): Omit<TaskRow, 'id' | 'status' | 'execution_count' | 'created_at'> {
+  return {
+    name: config.name,
+    type: config.type,
+    action: config.action,
+    action_params: JSON.stringify(config.actionParams ?? {}),
+    trigger_type: config.trigger?.type ?? 'manual',
+    trigger_params: JSON.stringify(config.trigger ?? defaultTrigger()),
+    script_path: config.scriptPath ?? '',
+    interpreter: config.interpreter ?? null,
+    script_args: JSON.stringify(config.scriptArgs ?? []),
+    script_timeout: config.scriptTimeout ?? 120,
+    work_dir: config.scriptWorkDir ?? null,
+    // priority 列已废弃（不再可配置），保留常量以兼容既有数据库表结构
+    priority: 1,
+    retry_count: config.retryCount ?? 0,
+    enabled: config.enabled ? 1 : 0,
+    notify_channels: JSON.stringify(config.notifyChannels ?? ['desktop']),
+    chain_next: config.chainNext ?? null,
+    chain_condition: config.chainCondition ?? null,
+    one_time: config.oneTime ? 1 : 0,
+    updated_at: now
+  }
+}
+
+export function rowToTaskConfig(row: TaskRow): TaskConfig {
+  let trigger: TaskTrigger = defaultTrigger()
+  try {
+    const parsed = JSON.parse(row.trigger_params ?? '{}')
+    trigger = { ...defaultTrigger(), ...parsed, type: row.trigger_type || parsed.type || 'manual' }
+  } catch {
+    trigger.type = row.trigger_type || 'manual'
+  }
+  return {
+    name: row.name,
+    type: row.type as TaskConfig['type'],
+    trigger,
+    action: row.action,
+    actionParams: safeJson(row.action_params, {}),
+    scriptPath: row.script_path ?? '',
+    interpreter: row.interpreter ?? null,
+    scriptArgs: safeJson(row.script_args, []),
+    scriptTimeout: row.script_timeout ?? 120,
+    scriptWorkDir: row.work_dir ?? null,
+    retryCount: row.retry_count ?? 0,
+    enabled: row.enabled === 1,
+    notifyChannels: safeJson(row.notify_channels, ['desktop']),
+    chainNext: row.chain_next ?? null,
+    chainCondition: row.chain_condition ?? null,
+    oneTime: row.one_time === 1
+  }
+}
+
+export function rowToTaskDto(row: TaskRow): TaskDto {
+  return {
+    id: row.id,
+    config: rowToTaskConfig(row),
+    status: row.status as TaskStatus,
+    createdAt: row.created_at,
+    executionCount: row.execution_count ?? 0,
+    lastRunAt: null, // 由调度器内存态补充
+    lastResult: null,
+    lastError: null,
+    nextRunAt: null,
+    currentRetry: 0
+  }
+}
+
+function safeJson<T>(s: string | null | undefined, fallback: T): T {
+  if (!s) return fallback
+  try {
+    return JSON.parse(s) as T
+  } catch {
+    return fallback
+  }
+}
+
+export class TaskRepository {
+  private db: DatabaseSync
+
+  constructor(db: DatabaseSync) {
+    this.db = db
+  }
+
+  // ── tasks ────────────────────────────────────────────────
+
+  listAll(): TaskDto[] {
+    const rows = this.db
+      .prepare(`SELECT ${TASK_COLUMNS} FROM tasks ORDER BY created_at`)
+      .all() as unknown as TaskRow[]
+    return rows.map(rowToTaskDto)
+  }
+
+  get(id: string): TaskDto | null {
+    const row = this.db.prepare(`SELECT ${TASK_COLUMNS} FROM tasks WHERE id = ?`).get(id) as
+      | TaskRow
+      | undefined
+    return row ? rowToTaskDto(row) : null
+  }
+
+  count(): number {
+    const row = this.db.prepare('SELECT COUNT(*) AS c FROM tasks').get() as { c: number }
+    return row.c
+  }
+
+  insert(dto: TaskDto): void {
+    const now = dto.createdAt
+    const r = taskConfigToRow(dto.config, now)
+    this.db
+      .prepare(
+        `INSERT INTO tasks (${TASK_COLUMNS}) VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)`
+      )
+      .run(
+        dto.id,
+        r.name,
+        r.type,
+        r.action,
+        r.action_params,
+        r.trigger_type,
+        r.trigger_params,
+        r.script_path,
+        r.interpreter,
+        r.script_args,
+        r.script_timeout,
+        r.work_dir,
+        r.priority,
+        r.retry_count,
+        r.enabled,
+        r.notify_channels,
+        r.chain_next,
+        r.chain_condition,
+        r.one_time,
+        dto.status,
+        dto.executionCount,
+        now,
+        r.updated_at
+      )
+  }
+
+  update(id: string, config: TaskConfig, status?: TaskStatus, executionCount?: number): boolean {
+    const r = taskConfigToRow(config, new Date().toISOString())
+    const statusVal = status ?? undefined
+    const execVal = executionCount ?? undefined
+    const result = this.db
+      .prepare(
+        `UPDATE tasks SET
+          name=?, type=?, action=?, action_params=?, trigger_type=?, trigger_params=?,
+          script_path=?, interpreter=?, script_args=?, script_timeout=?, work_dir=?,
+          priority=?, retry_count=?, enabled=?, notify_channels=?, chain_next=?,
+          chain_condition=?, one_time=?, status=COALESCE(?, status), execution_count=COALESCE(?, execution_count),
+          updated_at=?
+         WHERE id=?`
+      )
+      .run(
+        r.name,
+        r.type,
+        r.action,
+        r.action_params,
+        r.trigger_type,
+        r.trigger_params,
+        r.script_path,
+        r.interpreter,
+        r.script_args,
+        r.script_timeout,
+        r.work_dir,
+        r.priority,
+        r.retry_count,
+        r.enabled,
+        r.notify_channels,
+        r.chain_next,
+        r.chain_condition,
+        r.one_time,
+        statusVal ?? null,
+        execVal ?? null,
+        r.updated_at,
+        id
+      )
+    return result.changes > 0
+  }
+
+  delete(id: string): boolean {
+    return this.db.prepare('DELETE FROM tasks WHERE id = ?').run(id).changes > 0
+  }
+
+  /** 整体替换任务集（快照恢复 / 配置包导入用） */
+  replaceAll(tasks: Array<{ id: string; config: TaskConfig }>): void {
+    const now = new Date().toISOString()
+    const insert = this.db.prepare(
+      `INSERT INTO tasks (${TASK_COLUMNS}) VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)`
+    )
+    this.db.exec('BEGIN')
+    try {
+      this.db.exec('DELETE FROM tasks')
+      for (const t of tasks) {
+        const r = taskConfigToRow(t.config, now)
+        insert.run(
+          t.id,
+          r.name,
+          r.type,
+          r.action,
+          r.action_params,
+          r.trigger_type,
+          r.trigger_params,
+          r.script_path,
+          r.interpreter,
+          r.script_args,
+          r.script_timeout,
+          r.work_dir,
+          r.priority,
+          r.retry_count,
+          r.enabled,
+          r.notify_channels,
+          r.chain_next,
+          r.chain_condition,
+          r.one_time,
+          'idle',
+          0,
+          now,
+          r.updated_at
+        )
+      }
+      this.db.exec('COMMIT')
+    } catch (err) {
+      this.db.exec('ROLLBACK')
+      throw err
+    }
+  }
+
+  // ── executions ───────────────────────────────────────────
+
+  insertExecution(rec: Omit<ExecutionRecord, 'id'>): number {
+    const result = this.db
+      .prepare(
+        `INSERT INTO executions (task_id, task_name, status, started_at, finished_at, exit_code, duration_ms, pid, stdout, stderr, error)
+         VALUES (?,?,?,?,?,?,?,?,?,?,?)`
+      )
+      .run(
+        rec.taskId,
+        rec.taskName,
+        rec.status,
+        rec.startedAt,
+        rec.finishedAt,
+        rec.exitCode,
+        rec.durationMs,
+        rec.pid ?? null,
+        rec.stdout,
+        rec.stderr,
+        rec.error
+      )
+    const id = Number(result.lastInsertRowid)
+    // 只保留每个任务最近的 N 条执行记录（超出即删除最旧的）。
+    // 这里是执行热路径，裁剪失败绝不能影响任务本身 → 静默忽略错误。
+    try {
+      this.pruneExecutionsToLimit(rec.taskId)
+    } catch {
+      /* 清理失败可忽略（下次执行或启动时还会再清） */
+    }
+    return id
+  }
+
+  updateExecution(id: number, patch: Partial<Omit<ExecutionRecord, 'id'>>): void {
+    const sets: string[] = []
+    const values: (string | number | bigint | null)[] = []
+    for (const [k, v] of Object.entries(patch)) {
+      const col = EXEC_UPDATE_COLUMNS[k]
+      if (!col) continue
+      sets.push(`${col} = ?`)
+      values.push(typeof v === 'boolean' ? (v ? 1 : 0) : (v ?? null))
+    }
+    if (sets.length === 0) return
+    values.push(id)
+    this.db.prepare(`UPDATE executions SET ${sets.join(', ')} WHERE id = ?`).run(...values)
+  }
+
+  listExecutions(taskId?: string, limit = 100, offset = 0): ExecutionRecord[] {
+    const where = taskId ? 'WHERE task_id = ?' : ''
+    const params = taskId ? [taskId, limit, offset] : [limit, offset]
+    interface ExecRow {
+      id: number
+      task_id: string
+      task_name: string
+      status: string
+      started_at: string
+      finished_at: string | null
+      exit_code: number | null
+      duration_ms: number | null
+      pid: number | null
+      stdout: string
+      stderr: string
+      error: string
+    }
+    const rows = this.db
+      .prepare(
+        `SELECT id, task_id, task_name, status, started_at, finished_at, exit_code, duration_ms, pid, stdout, stderr, error
+         FROM executions ${where} ORDER BY started_at DESC LIMIT ? OFFSET ?`
+      )
+      .all(...params) as unknown as ExecRow[]
+    return rows.map(rowToExecution)
+  }
+
+  getRecentExecutions(limit = 50): ExecutionRecord[] {
+    return this.listExecutions(undefined, limit, 0)
+  }
+
+  /**
+   * 收尾「进程被强杀」遗留的 running 执行记录（启动时调用）
+   *
+   * 脚本执行中若进程被 Ctrl+C / 强杀，走不到 updateExecution，
+   * 记录会永远停在 running：历史里显示「运行中」、完成时间与耗时都是空。
+   * @returns 被收尾的记录条数
+   */
+  markRunningExecutionsInterrupted(
+    reason = '服务重启，执行被中断',
+    now = new Date().toISOString(),
+    onlyIds?: number[]
+  ): number {
+    // 明确指定「没有任何一条要处理」时直接返回，避免退化成全表收尾
+    if (onlyIds && onlyIds.length === 0) return 0
+    const scope = onlyIds?.length
+      ? ` AND id IN (${onlyIds.map(() => '?').join(',')})`
+      : ''
+    const result = this.db
+      .prepare(
+        `UPDATE executions
+            SET status = 'failed',
+                finished_at = COALESCE(finished_at, ?),
+                duration_ms = COALESCE(
+                  duration_ms,
+                  (julianday(?) - julianday(started_at)) * 86400000
+                ),
+                error = CASE
+                  WHEN error IS NULL OR error = '' THEN ?
+                  ELSE error || char(10) || ?
+                END
+          WHERE status = 'running'${scope}`
+      )
+      .run(now, now, reason, reason, ...(onlyIds ?? []))
+    return Number(result.changes ?? 0)
+  }
+
+  /**
+   * 仍未结束（status='running'）的执行记录，含子进程 pid。
+   *
+   * 启动恢复时用它区分两种情况：
+   *   · pid 已不存在 → 进程真的没了，按「执行被中断」收尾
+   *   · pid 还活着   → 脚本仍在后台跑（Windows 上父进程退出并不带走子进程），
+   *                    必须如实标注，否则用户会以为失败了而重复触发部署
+   */
+  listRunningExecutions(): Array<{
+    id: number
+    taskId: string
+    taskName: string
+    pid: number | null
+    startedAt: string
+  }> {
+    const rows = this.db
+      .prepare(
+        `SELECT id, task_id, task_name, pid, started_at
+           FROM executions WHERE status = 'running' ORDER BY id ASC`
+      )
+      .all() as unknown as Array<{
+      id: number
+      task_id: string
+      task_name: string
+      pid: number | null
+      started_at: string
+    }>
+    return rows.map((r) => ({
+      id: Number(r.id),
+      taskId: r.task_id,
+      taskName: r.task_name,
+      pid: r.pid == null ? null : Number(r.pid),
+      startedAt: r.started_at
+    }))
+  }
+
+  /**
+   * 按「每个任务只保留最近 N 条」清理执行记录
+   * @param taskId 只清理指定任务；不传则对所有任务各清理一遍（启动时用）
+   * @returns 删除的条数
+   */
+  pruneExecutionsToLimit(taskId?: string, keep = EXECUTION_KEEP_PER_TASK): number {
+    const keepCount = Math.max(0, Math.floor(keep))
+    interface ExecIdRow {
+      id: number
+      task_id: string
+    }
+    const rows = (
+      taskId
+        ? this.db
+            .prepare(
+              'SELECT id, task_id FROM executions WHERE task_id = ? ORDER BY started_at DESC, id DESC'
+            )
+            .all(taskId)
+        : this.db
+            .prepare(
+              'SELECT id, task_id FROM executions ORDER BY task_id ASC, started_at DESC, id DESC'
+            )
+            .all()
+    ) as unknown as ExecIdRow[]
+
+    // 行已按「新 → 旧」排序，每个任务计数超过 keepCount 的都是要删的
+    const seen = new Map<string, number>()
+    const doomed: number[] = []
+    for (const r of rows) {
+      const n = (seen.get(r.task_id) ?? 0) + 1
+      seen.set(r.task_id, n)
+      if (n > keepCount) doomed.push(r.id)
+    }
+    if (doomed.length === 0) return 0
+
+    const del = this.db.prepare('DELETE FROM executions WHERE id = ?')
+    this.db.exec('BEGIN')
+    try {
+      for (const id of doomed) del.run(id)
+      this.db.exec('COMMIT')
+    } catch (err) {
+      this.db.exec('ROLLBACK')
+      throw err
+    }
+    return doomed.length
+  }
+
+  // ── 微信消息记录（wechat_messages 表） ───────────────────
+
+  insertWechatMessage(direction: 'in' | 'out', content: string): void {
+    this.db
+      .prepare('INSERT INTO wechat_messages (direction, content, created_at) VALUES (?, ?, ?)')
+      .run(direction, content.slice(0, 4000), new Date().toISOString())
+  }
+
+  listWechatMessages(limit = 100): Array<{ id: number; direction: string; content: string; createdAt: string }> {
+    const rows = this.db
+      .prepare(
+        'SELECT id, direction, content, created_at FROM wechat_messages ORDER BY id DESC LIMIT ?'
+      )
+      .all(limit) as unknown as Array<{
+      id: number
+      direction: string
+      content: string
+      created_at: string
+    }>
+    return rows.map((r) => ({
+      id: Number(r.id),
+      direction: r.direction,
+      content: r.content,
+      createdAt: r.created_at
+    }))
+  }
+}
+
+/** executions 表行 → ExecutionRecord（snake_case → camelCase） */
+function rowToExecution(row: {
+  id: number
+  task_id: string
+  task_name: string
+  status: string
+  started_at: string
+  finished_at: string | null
+  exit_code: number | null
+  duration_ms: number | null
+  pid?: number | null
+  stdout: string
+  stderr: string
+  error: string
+}): ExecutionRecord {
+  return {
+    id: Number(row.id),
+    taskId: row.task_id,
+    taskName: row.task_name,
+    status: row.status,
+    startedAt: row.started_at,
+    finishedAt: row.finished_at,
+    exitCode: row.exit_code == null ? null : Number(row.exit_code),
+    durationMs: row.duration_ms == null ? null : Number(row.duration_ms),
+    pid: row.pid == null ? null : Number(row.pid),
+    stdout: row.stdout ?? '',
+    stderr: row.stderr ?? '',
+    error: row.error ?? ''
+  }
+}
+
+/** 任务 ID 生成 */
+export function newTaskId(): string {
+  return `task_${Math.random().toString(16).slice(2, 10)}`
+}
+
+/** 状态名 → 执行记录状态 */
+export function executionStatusFor(status: TaskStatus): string {
+  if (status === TS.RUNNING) return 'running'
+  if (status === TS.FAILED) return 'failed'
+  if (status === TS.CANCELLED) return 'cancelled'
+  return 'completed'
+}