@shawnstack/quickforge 1.1.0 → 1.2.1

package/server/routes/storage.mjs

@@ -1,6 +1,6 @@
 import path from 'node:path'
 import { sendJson, readJsonBody, decodeSegment } from '../utils/response.mjs'
-import { readStore, writeStore, getComparable, dataDir, configDir, storageDir, cacheDir, logsDir } from '../storage.mjs'
+import { readStore, writeStore, atomicUpdate, getComparable, readSessionStoreScoped, readSessionValue, writeSessionValue, deleteSessionValue, ensureStorage, dataDir, configDir, storageDir, cacheDir, logsDir } from '../storage.mjs'
 import { directorySize } from '../utils/workspace.mjs'
 
 export async function handleStorageApi(req, res, url) {
@@ -37,8 +37,24 @@ export async function handleStorageApi(req, res, url) {
   if (req.method === 'GET' && parts[3] === 'index') {
     const indexName = decodeSegment(parts[4])
     const direction = url.searchParams.get('direction') === 'desc' ? 'desc' : 'asc'
-    const data = await readStore(store)
-    const values = Object.values(data)
+    const scope = url.searchParams.get('scope')
+    const projectId = url.searchParams.get('projectId')
+    const limitParam = url.searchParams.get('limit')
+    const offsetParam = url.searchParams.get('offset')
+
+    await ensureStorage()
+
+    let data
+    if (scope && (store === 'sessions' || store === 'sessions-metadata')) {
+      data = await readSessionStoreScoped(store, scope, scope === 'project' ? projectId : undefined)
+    } else {
+      data = await readStore(store)
+    }
+
+    let values = Object.values(data)
+    if (store === 'sessions-metadata') {
+      values = values.filter((value) => value?.messageCount !== 0)
+    }
     values.sort((a, b) => {
       const left = getComparable(a, indexName)
       const right = getComparable(b, indexName)
@@ -48,7 +64,16 @@ export async function handleStorageApi(req, res, url) {
       const result = String(left).localeCompare(String(right))
       return direction === 'desc' ? -result : result
     })
-    sendJson(res, 200, { values })
+
+    const total = values.length
+    const limit = limitParam ? parseInt(limitParam, 10) : undefined
+    const offset = offsetParam ? parseInt(offsetParam, 10) : 0
+
+    if (limit && limit > 0) {
+      sendJson(res, 200, { values: values.slice(offset, offset + limit), total })
+    } else {
+      sendJson(res, 200, { values, total })
+    }
     return
   }
 
@@ -74,6 +99,11 @@ export async function handleStorageApi(req, res, url) {
     }
 
     if (req.method === 'GET') {
+      if (store === 'sessions') {
+        sendJson(res, 200, { value: await readSessionValue(key) })
+        return
+      }
+
       const data = await readStore(store)
       sendJson(res, 200, { value: Object.prototype.hasOwnProperty.call(data, key) ? data[key] : null })
       return
@@ -81,17 +111,35 @@ export async function handleStorageApi(req, res, url) {
 
     if (req.method === 'PUT') {
       const body = await readJsonBody(req)
-      const data = await readStore(store)
-      data[key] = body?.value
-      await writeStore(store, data)
+      if (store === 'sessions') {
+        await writeSessionValue(key, body?.value)
+        sendJson(res, 200, { ok: true })
+        return
+      }
+
+      await atomicUpdate(store, (data) => {
+        data[key] = body?.value
+        return data
+      })
       sendJson(res, 200, { ok: true })
       return
     }
 
     if (req.method === 'DELETE') {
-      const data = await readStore(store)
-      delete data[key]
-      await writeStore(store, data)
+      if (store === 'sessions') {
+        await deleteSessionValue(key)
+        await atomicUpdate('sessions-metadata', (data) => {
+          delete data[key]
+          return data
+        })
+        sendJson(res, 200, { ok: true })
+        return
+      }
+
+      await atomicUpdate(store, (data) => {
+        delete data[key]
+        return data
+      })
       sendJson(res, 200, { ok: true })
       return
     }

package/server/routes/system.mjs

@@ -0,0 +1,35 @@
+import { sendJson } from '../utils/response.mjs'
+import { getLanUrls } from '../utils/network.mjs'
+
+export async function handleSystemApi(req, res, url, context) {
+  if (req.method === 'POST' && url.pathname === '/api/system/restart') {
+    if (req.headers['x-quickforge-action'] !== 'restart') {
+      const error = new Error('Forbidden action')
+      error.statusCode = 403
+      throw error
+    }
+
+    const result = await context.requestRestart()
+    sendJson(res, 202, result)
+    return
+  }
+
+  if (req.method === 'GET' && url.pathname === '/api/system/status') {
+    sendJson(res, 200, await context.getSystemStatus())
+    return
+  }
+
+  if (req.method === 'GET' && url.pathname === '/api/system/network') {
+    sendJson(res, 200, {
+      host: context.host,
+      port: context.port,
+      lanUrls: getLanUrls(context.port),
+      remoteEnabled: context.remoteEnabled === true,
+    })
+    return
+  }
+
+  const error = new Error('Not found')
+  error.statusCode = 404
+  throw error
+}

package/server/routes/tools.mjs

@@ -1,6 +1,36 @@
 import { sendJson, readJsonBody, decodeSegment } from '../utils/response.mjs'
-import { toolHandlers } from '../tools/index.mjs'
-import { projectContextFromId } from '../project-config.mjs'
+import { readStore } from '../storage.mjs'
+import { toolHandlers, loadSkillToolContext } from '../tools/index.mjs'
+import { createSkillTools, workspaceTools } from '../tools/definitions.mjs'
+import { projectContextFromId, readProjectConfig } from '../project-config.mjs'
+
+/**
+ * GET /api/tools — returns canonical tool definitions (no project context needed).
+ */
+export async function handleGetTools(_req, res) {
+  const config = await readProjectConfig()
+  const activeProject = config.projects.find((project) => project.id === config.activeProjectId) || config.projects[0]
+  const skillTools = await createSkillTools({
+    globalSkillNames: config.globalSkills,
+    projectSkillNames: activeProject?.skills,
+    workspaceRoot: activeProject?.path,
+  })
+  sendJson(res, 200, { tools: [...skillTools, ...workspaceTools] })
+}
+
+const dangerousTools = new Set(['write_file', 'edit_file', 'run_command'])
+
+async function assertYoloEnabledForTool(name) {
+  if (!dangerousTools.has(name)) return
+
+  const settings = await readStore('settings')
+  const yoloMode = settings?.['yolo-mode'] === true || settings?.['yolo-mode'] === 'true'
+  if (!yoloMode) {
+    const error = new Error('YOLO mode is disabled. Enable it to use this tool.')
+    error.statusCode = 403
+    throw error
+  }
+}
 
 export async function handleToolApi(req, res, url) {
   if (req.method !== 'POST') {
@@ -13,8 +43,27 @@ export async function handleToolApi(req, res, url) {
   let name = decodeSegment(parts[2])
   let context
 
+  if (name === 'activate_skill' || name === 'read_skill_resource') {
+    const config = await readProjectConfig()
+    const activeProject = config.projects.find((project) => project.id === config.activeProjectId) || config.projects[0]
+    context = await loadSkillToolContext({
+      globalSkillNames: config.globalSkills,
+      projectSkillNames: activeProject?.skills,
+      workspaceRoot: activeProject?.path,
+    })
+  }
+
   if (parts[1] === 'projects' && parts[3] === 'tools') {
     context = await projectContextFromId(decodeSegment(parts[2]))
+    const config = await readProjectConfig()
+    context = {
+      ...context,
+      ...(await loadSkillToolContext({
+        globalSkillNames: config.globalSkills,
+        projectSkillNames: context.project?.skills,
+        workspaceRoot: context.workspaceRoot,
+      })),
+    }
     name = decodeSegment(parts[4])
   }
 
@@ -25,6 +74,8 @@ export async function handleToolApi(req, res, url) {
     throw error
   }
 
+  await assertYoloEnabledForTool(name)
+
   const params = await readJsonBody(req)
   const result = await handler(params || {}, context)
   sendJson(res, 200, result)

package/server/session-utils.mjs

@@ -0,0 +1,102 @@
+import { streamSimple } from '@mariozechner/pi-ai'
+import { buildInstructionsPayload } from './project-config.mjs'
+import { composeSystemPrompt } from './system-prompt.mjs'
+
+// ---------------------------------------------------------------------------
+// System prompt
+// ---------------------------------------------------------------------------
+
+export async function buildSystemPrompt(projectId) {
+  return composeSystemPrompt(await buildInstructionsPayload(projectId))
+}
+
+// ---------------------------------------------------------------------------
+// Simple title generation (from first user message)
+// ---------------------------------------------------------------------------
+
+export function generateTitle(messages) {
+  const firstUser = messages.find(
+    (m) => m.role === 'user' || m.role === 'user-with-attachments',
+  )
+  if (!firstUser) return 'New chat'
+  const content = firstUser.content
+  const text = typeof content === 'string' ? content : Array.isArray(content)
+    ? content.filter((b) => b.type === 'text').map((b) => b.text ?? '').join(' ')
+    : ''
+  const normalized = text.trim().replace(/\s+/g, ' ')
+  if (!normalized) return 'New chat'
+  return normalized.length > 46 ? `${normalized.slice(0, 43)}...` : normalized
+}
+
+// ---------------------------------------------------------------------------
+// AI title generation
+// ---------------------------------------------------------------------------
+
+function normalizeAiTitle(value) {
+  return value
+    .trim()
+    .replace(/^[[\s"'""''`]+|[\]`\s"'""''.。,！!？?，,:：;；]+$/g, '')
+    .replace(/\s+/g, ' ')
+    .slice(0, 80)
+}
+
+export async function generateAiTitle(messages, model, thinkingLevel, getApiKey) {
+  const firstUser = messages.find((m) => m.role === 'user' || m.role === 'user-with-attachments')
+  if (!firstUser) return null
+
+  const userText = typeof firstUser.content === 'string'
+    ? firstUser.content
+    : Array.isArray(firstUser.content)
+      ? firstUser.content.filter((b) => b.type === 'text').map((b) => b.text ?? '').join(' ')
+      : ''
+
+  if (!userText.trim()) return null
+
+  const firstAssistant = messages.find((m) => m.role === 'assistant')
+  let assistantReply = ''
+  if (firstAssistant) {
+    const content = firstAssistant.content
+    if (typeof content === 'string') {
+      assistantReply = content.slice(0, 2000)
+    } else if (Array.isArray(content)) {
+      assistantReply = content
+        .filter((b) => b.type === 'text')
+        .map((b) => b.text ?? '')
+        .join(' ')
+        .slice(0, 2000)
+    }
+  }
+
+  const conversationText = assistantReply
+    ? `User: ${userText.trim()}\n\nAssistant: ${assistantReply}`
+    : `User: ${userText.trim()}`
+
+  try {
+    const apiKey = getApiKey ? await getApiKey(model.provider) : undefined
+    const stream = streamSimple(
+      model,
+      {
+        systemPrompt: '你是对话标题生成器。请用和用户相同的语言，根据对话主题生成 3 到 5 个词的短标题。只输出标题，不要解释，不要标点。',
+        messages: [{ role: 'user', content: conversationText, timestamp: Date.now() }],
+        tools: [],
+      },
+      {
+        apiKey,
+        maxTokens: 160,
+        temperature: 0.2,
+        reasoning: thinkingLevel === 'off' ? undefined : 'medium',
+        maxRetryDelayMs: 60000,
+      },
+    )
+    const titleMessage = await stream.result()
+    const titleText = Array.isArray(titleMessage.content)
+      ? titleMessage.content.filter((b) => b.type === 'text').map((b) => b.text ?? '').join(' ').trim()
+      : ''
+    if (!titleText) return null
+    const title = normalizeAiTitle(titleText)
+    return title || null
+  } catch (error) {
+    console.warn('Failed to generate AI title:', error.message || error)
+    return null
+  }
+}