md-annotator 0.5.8 → 0.7.0

package/client/src/utils/parser.js

@@ -61,7 +61,7 @@ export function parseMarkdownToBlocks(markdown) {
     }
 
     // Horizontal Rule
-    if (trimmed === '---' || trimmed === '***') {
+    if (/^([-*_])\s*(\1\s*){2,}$/.test(trimmed)) {
       flush()
       blocks.push({
         id: `block-${currentId++}`,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "md-annotator",
-  "version": "0.5.8",
+  "version": "0.7.0",
   "description": "Browser-based Markdown annotator for AI-assisted review",
   "type": "module",
   "bin": {
@@ -41,6 +41,8 @@
     "highlight.js": "^11.11.0",
     "mermaid": "^11.12.3",
     "open": "^10.1.0",
+    "pako": "^2.1.0",
+    "plantuml-encoder": "^1.4.0",
     "portfinder": "^1.0.32",
     "react": "^19.0.0",
     "react-dom": "^19.0.0",

package/server/config.js

@@ -27,10 +27,28 @@ function getHeartbeatTimeoutMs() {
   return DEFAULT_HEARTBEAT_TIMEOUT_MS
 }
 
+function getPlantumlServerUrl() {
+  const envUrl = process.env.PLANTUML_SERVER_URL
+  if (envUrl) {
+    return envUrl.replace(/\/+$/, '')
+  }
+  return 'https://www.plantuml.com/plantuml'
+}
+
+function getKrokiServerUrl() {
+  const envUrl = process.env.KROKI_SERVER_URL
+  if (envUrl) {
+    return envUrl.replace(/\/+$/, '')
+  }
+  return 'https://kroki.io'
+}
+
 export const config = {
   port: getServerPort(),
   browser: process.env.MD_ANNOTATOR_BROWSER || null,
   heartbeatTimeoutMs: getHeartbeatTimeoutMs(),
   forceExitTimeoutMs: 5000,
   jsonLimit: '10mb',
+  plantumlServerUrl: getPlantumlServerUrl(),
+  krokiServerUrl: getKrokiServerUrl(),
 }

package/server/feedback.js

@@ -20,11 +20,27 @@ function formatAnnotation(ann, block, heading) {
     return output + '\n'
   }
 
+  if (ann.targetType === 'pinpoint') {
+    const isDeletion = ann.type === 'DELETION'
+    const label = isDeletion ? 'Remove block' : 'Comment on block'
+    let output = `${heading} ${label} (Line ${blockStartLine})\n`
+    const preview = (block?.content || ann.originalText || '').slice(0, 200)
+    output += `\`\`\`\n${preview}\n\`\`\`\n`
+    if (isDeletion) {
+      output += `> User wants this block removed from the document.\n`
+    } else {
+      output += `> ${(ann.text ?? '').replace(/\n/g, '\n> ')}\n`
+    }
+    return output + '\n'
+  }
+
   if (ann.targetType === 'diagram') {
     const isDeletion = ann.type === 'DELETION'
-    const label = isDeletion ? 'Remove Mermaid diagram' : 'Comment on Mermaid diagram'
+    const diagramLang = block?.language === 'plantuml' ? 'PlantUML' : 'Mermaid'
+    const fence = block?.language === 'plantuml' ? 'plantuml' : 'mermaid'
+    const label = isDeletion ? `Remove ${diagramLang} diagram` : `Comment on ${diagramLang} diagram`
     let output = `${heading} ${label} (Line ${blockStartLine})\n`
-    output += `\`\`\`mermaid\n${block?.content || ann.originalText}\n\`\`\`\n`
+    output += `\`\`\`${fence}\n${block?.content || ann.originalText}\n\`\`\`\n`
     if (isDeletion) {
       output += `> User wants this diagram removed from the document.\n`
     } else {
@@ -50,13 +66,13 @@ function formatAnnotation(ann, block, heading) {
   } else if (ann.type === 'COMMENT') {
     output += `Comment on (${lineRef})\n`
     output += `\`\`\`\n${ann.originalText}\n\`\`\`\n`
-    output += `> ${ann.text.replace(/\n/g, '\n> ')}\n`
+    output += `> ${(ann.text ?? '').replace(/\n/g, '\n> ')}\n`
   } else if (ann.type === 'INSERTION') {
     output += `Insert text (${lineRef})\n`
     if (ann.afterContext) {
       output += `After: \`${ann.afterContext}\`\n`
     }
-    output += `\`\`\`\n${ann.text}\n\`\`\`\n`
+    output += `\`\`\`\n${ann.text ?? ''}\n\`\`\`\n`
     output += `> User wants this text inserted at this point in the document.\n`
   }
 

package/server/routes.js

@@ -3,6 +3,8 @@ import { relative, resolve, dirname, isAbsolute } from 'node:path'
 import { createHash } from 'node:crypto'
 import { readMarkdownFile, isMarkdownFile } from './file.js'
 import { exportFeedback, exportMultiFileFeedback } from './feedback.js'
+import { listWorkspaceFiles } from './workspace.js'
+import { config } from './config.js'
 
 function success(data) {
   return { success: true, data }
@@ -15,6 +17,16 @@ function failure(error) {
 export function createApiRouter(filePaths, resolveDecision, origin = 'cli', stores = []) {
   const router = Router()
 
+  // Workspace file listing for @-reference autocomplete
+  router.get('/api/workspace/files', async (_req, res) => {
+    try {
+      const files = await listWorkspaceFiles()
+      res.json(success({ files }))
+    } catch (error) {
+      res.status(500).json(failure(error.message))
+    }
+  })
+
   // Multi-file endpoint — returns all files
   router.get('/api/files', async (_req, res) => {
     try {
@@ -32,7 +44,7 @@ export function createApiRouter(filePaths, resolveDecision, origin = 'cli', stor
           }
         })
       )
-      res.json(success({ files, origin }))
+      res.json(success({ files, origin, config: { plantumlServerUrl: config.plantumlServerUrl, krokiServerUrl: config.krokiServerUrl } }))
     } catch (error) {
       res.status(500).json(failure(error.message))
     }

package/server/workspace.js

@@ -0,0 +1,43 @@
+import { execFile } from 'node:child_process'
+import { readdir } from 'node:fs/promises'
+import { relative, join } from 'node:path'
+
+const MAX_FILES = 5000
+const EXCLUDE_DIRS = new Set([
+  'node_modules', '.git', 'dist', 'build', 'coverage',
+  '__pycache__', '.next', '.cache', '.turbo', '.output'
+])
+
+function gitListFiles(cwd) {
+  return new Promise((resolve, reject) => {
+    execFile('git', ['ls-files'], { cwd, maxBuffer: 10 * 1024 * 1024, timeout: 10_000 }, (err, stdout) => {
+      if (err) { return reject(err) }
+      const files = stdout.split('\n').filter(Boolean)
+      resolve(files.slice(0, MAX_FILES))
+    })
+  })
+}
+
+async function fallbackListFiles(cwd) {
+  const entries = await readdir(cwd, { recursive: true, withFileTypes: true })
+  const files = []
+  for (const entry of entries) {
+    if (!entry.isFile()) { continue }
+    const parentPath = entry.parentPath || entry.path
+    const rel = relative(cwd, join(parentPath, entry.name))
+    const parts = rel.split('/')
+    if (parts.some(p => EXCLUDE_DIRS.has(p))) { continue }
+    files.push(rel)
+    if (files.length >= MAX_FILES) { break }
+  }
+  return files.sort()
+}
+
+export async function listWorkspaceFiles() {
+  const cwd = process.cwd()
+  try {
+    return await gitListFiles(cwd)
+  } catch {
+    return await fallbackListFiles(cwd)
+  }
+}