🎉 initial commit

src/renderer/utils/annotationParser.ts

@@ -0,0 +1,125 @@
+import type { TextAnnotation, AnnotationType } from '../types/editor'
+
+// Attempt to locate AI-quoted text in the document and create highlight annotations
+export function parseAnnotationsFromAIResponse(
+  aiResponse: string,
+  documentContent: string
+): TextAnnotation[] {
+  const annotations: TextAnnotation[] = []
+  let id = 0
+
+  // Match quoted strings — handles "straight", "curly", and 'single' quotes
+  // Minimum 10 chars to avoid matching short words
+  const quotePattern = /["""''](.{10,300}?)["""'']/g
+  let match: RegExpExecArray | null
+
+  while ((match = quotePattern.exec(aiResponse)) !== null) {
+    const quotedText = match[1].trim()
+
+    // Try exact match first
+    let docIndex = documentContent.indexOf(quotedText)
+
+    // If not found exactly, try a normalized version (collapse whitespace)
+    if (docIndex === -1) {
+      const normalized = quotedText.replace(/\s+/g, ' ')
+      docIndex = findNormalized(documentContent, normalized)
+    }
+
+    if (docIndex === -1) continue
+
+    // Don't annotate the same range twice
+    const alreadyAnnotated = annotations.some(
+      (a) => a.from === docIndex && a.to === docIndex + quotedText.length
+    )
+    if (alreadyAnnotated) continue
+
+    // Determine annotation type from context around the quote in the AI response
+    const contextStart = Math.max(0, match.index - 200)
+    const contextBefore = aiResponse.slice(contextStart, match.index).toLowerCase()
+    const type = classifyType(contextBefore)
+
+    // Try to extract a suggestion from text after the quote
+    const afterQuote = aiResponse.slice(match.index + match[0].length, match.index + match[0].length + 400)
+    const suggestion = extractSuggestion(afterQuote)
+
+    // Extract a short message label from the ISSUE/PROBLEM line before the quote
+    const message = extractMessage(aiResponse, match.index)
+
+    annotations.push({
+      id: `ai-${id++}`,
+      type,
+      from: docIndex,
+      to: docIndex + quotedText.length,
+      matchedText: quotedText,
+      message: message || `${type.replace('_', ' ')} — hover for details`,
+      suggestion
+    })
+  }
+
+  return annotations
+}
+
+function classifyType(contextBefore: string): AnnotationType {
+  if (contextBefore.includes('passive')) return 'passive_voice'
+  if (
+    contextBefore.includes('consistency') ||
+    contextBefore.includes('character') ||
+    contextBefore.includes('timeline') ||
+    contextBefore.includes('repeated') ||
+    contextBefore.includes('contradiction')
+  ) {
+    return 'consistency'
+  }
+  return 'style'
+}
+
+function extractSuggestion(text: string): string | undefined {
+  // Look for SUGGESTION: "..." pattern
+  const m = text.match(/SUGGESTION:\s*["""'](.{5,200}?)["""']/i)
+  return m?.[1]?.trim()
+}
+
+function extractMessage(response: string, quoteIndex: number): string {
+  // Look back for ISSUE: or PROBLEM: line
+  const before = response.slice(Math.max(0, quoteIndex - 300), quoteIndex)
+  const issueMatch = before.match(/(?:ISSUE|PROBLEM|WHY):\s*(.+?)(?:\n|$)/gi)
+  if (issueMatch) {
+    const last = issueMatch[issueMatch.length - 1]
+    return last.replace(/^(?:ISSUE|PROBLEM|WHY):\s*/i, '').trim().slice(0, 120)
+  }
+  // Fall back to the last sentence before the quote
+  const sentences = before.split(/[.!?]\s+/)
+  const last = sentences[sentences.length - 1]?.trim()
+  return last?.slice(0, 120) ?? ''
+}
+
+// Find a normalized string in a document (ignores whitespace differences)
+function findNormalized(document: string, normalized: string): number {
+  const words = normalized.split(' ')
+  if (words.length < 3) return -1
+
+  // Search for the first few words as an anchor
+  const anchor = words.slice(0, 4).join(' ')
+  let searchFrom = 0
+
+  while (searchFrom < document.length) {
+    const idx = document.indexOf(words[0], searchFrom)
+    if (idx === -1) break
+
+    // Extract a comparable slice from the document
+    const slice = document.slice(idx, idx + normalized.length * 2).replace(/\s+/g, ' ')
+    if (slice.startsWith(normalized)) {
+      return idx
+    }
+
+    // Check if the anchor matches
+    const docSlice = document.slice(idx, idx + anchor.length + 20).replace(/\s+/g, ' ')
+    if (docSlice.startsWith(anchor)) {
+      return idx
+    }
+
+    searchFrom = idx + 1
+  }
+
+  return -1
+}