quality: move ai functions in a separated file

Author: aeltorio

src/components/viewers/MonacoViewer.tsx

@@ -27,14 +27,23 @@ import Editor, { type Monaco, type OnMount } from '@monaco-editor/react'
 import { toast } from '@heroui/react'
 import type { FileNode } from '../../types'
 import { useAuthContext } from '../../context/AuthContext'
-import { decodeUtf8, fetchJson, parseJsonSafe, postJson } from '../../utils/http'
+import { decodeUtf8, fetchJson, parseJsonSafe } from '../../utils/http'
 import {
   loadAiConfig,
-  resolveAiEndpoint,
   hasAiAuth,
   is403Error,
-  streamChatCompletion,
 } from '../../services/aiSettings'
+import {
+  collectExtraContext,
+  fetchAiCompletion,
+  fetchAiSanitizedCode,
+  inlineCommentTargetLanguage,
+  MAX_CONTEXT_CHARS,
+  MAX_TRANSLATE_CHUNK_CHARS,
+  MAX_TRANSLATE_CHUNKS,
+  normalizedInterfaceLanguage,
+  splitTextIntoChunks,
+} from './aiHelpers'
 
 // ── Navigation history (module-level: persists across component remounts on file change) ──
 interface NavEntry { path: string; line: number; column: number }
@@ -141,81 +150,6 @@ function hashFromEncryptedPath(path?: string): string | null {
 const IMPL_KINDS = new Set(['function', 'method', 'class', 'struct', 'trait', 'interface'])
 const AI_CACHE_PREFIX = 'INDUSTRIAL-ANALYZER_monaco_ai_v1'
 const MAX_PROXY_REQ_PER_MIN = 10
-const MAX_CONTEXT_CHARS = 80_000
-const MAX_TRANSLATE_CHUNK_CHARS = 9_000
-const MAX_TRANSLATE_CHUNKS = 6
-
-function splitTextIntoChunks(input: string, chunkSize: number): string[] {
-  if (!input) return ['']
-  if (input.length <= chunkSize) return [input]
-
-  const lines = input.split('\n')
-  const chunks: string[] = []
-  let current = ''
-
-  for (const line of lines) {
-    const lineWithBreak = current.length === 0 ? line : `\n${line}`
-    if (current.length + lineWithBreak.length <= chunkSize) {
-      current += lineWithBreak
-      continue
-    }
-    if (current.length > 0) chunks.push(current)
-    if (line.length <= chunkSize) {
-      current = line
-      continue
-    }
-
-    let start = 0
-    while (start < line.length) {
-      const part = line.slice(start, start + chunkSize)
-      if (part.length === chunkSize) chunks.push(part)
-      else current = part
-      start += chunkSize
-    }
-    if (start >= line.length && line.length % chunkSize === 0) current = ''
-  }
-
-  if (current.length > 0) chunks.push(current)
-  return chunks
-}
-
-function stripAiCodeWrappers(raw: string): string {
-  let cleaned = raw.trim()
-
-  const fullFence = cleaned.match(/^```[a-zA-Z0-9_-]*\s*\n([\s\S]*?)\n```\s*$/)
-  if (fullFence?.[1]) return fullFence[1]
-
-  const firstFenceStart = cleaned.indexOf('```')
-  if (firstFenceStart >= 0) {
-    const rest = cleaned.slice(firstFenceStart)
-    const fenced = rest.match(/^```[a-zA-Z0-9_-]*\s*\n([\s\S]*?)\n```/)
-    if (fenced?.[1]) return fenced[1]
-  }
-
-  const lines = cleaned.split('\n')
-  if (lines.length > 1) {
-    const first = lines[0].toLowerCase()
-    const isPreamble = first.includes('translated') || first.includes('source code') || first.includes('here is')
-    if (isPreamble) {
-      cleaned = lines.slice(1).join('\n').trim()
-    }
-  }
-
-  return cleaned
-}
-
-function normalizedInterfaceLanguage(lang: string | undefined): 'fr' | 'en' | 'zh' {
-  const normalized = (lang ?? '').toLowerCase()
-  if (normalized.startsWith('fr')) return 'fr'
-  if (normalized.startsWith('zh')) return 'zh'
-  return 'en'
-}
-
-function inlineCommentTargetLanguage(lang: 'fr' | 'en' | 'zh'): string {
-  if (lang === 'fr') return 'French'
-  if (lang === 'zh') return 'Chinese'
-  return 'English'
-}
 
 export default function MonacoViewer({
   node,
@@ -379,39 +313,6 @@ export default function MonacoViewer({
    * Throws an `Error` on non-2xx HTTP responses; callers can use
    * {@link is403Error} to distinguish authentication failures.
    */
-  const fetchAiCompletion = useCallback(async (
-    messages: Array<{ role: 'system' | 'user'; content: string }>,
-    onChunk?: (chunk: string) => void,
-  ): Promise<string> => {
-    const config = loadAiConfig()
-    const { url, model, apiKey } = resolveAiEndpoint(config, privateKey)
-    const requestBody = {
-      model,
-      temperature: 0.2,
-      top_p: 1,
-      max_tokens: 8192,
-      messages,
-    }
-
-    if (config.streaming) {
-      return streamChatCompletion(url, requestBody, apiKey, onChunk)
-    }
-
-    const body = await postJson<{ choices?: Array<{ message?: { content?: string } }> }>(
-      url,
-      { ...requestBody, stream: false },
-      { headers: { Authorization: `Bearer ${apiKey}` } },
-    )
-    const content = body.choices?.[0]?.message?.content?.trim()
-    if (!content) throw new Error('Empty model response')
-    return content
-  }, [privateKey])
-
-  const fetchAiSanitizedCode = useCallback(async (messages: Array<{ role: 'system' | 'user'; content: string }>): Promise<string> => {
-    const content = await fetchAiCompletion(messages)
-    return stripAiCodeWrappers(content)
-  }, [fetchAiCompletion])
-
   const openAiLoadingToast = useCallback((message: string) => {
     if (aiLoadingToastIdRef.current !== null) toast.close(aiLoadingToastIdRef.current)
     aiLoadingToastIdRef.current = toast(message, {
@@ -426,30 +327,6 @@ export default function MonacoViewer({
     aiLoadingToastIdRef.current = null
   }, [])
 
-  const collectExtraContext = useCallback(async (): Promise<string> => {
-    const includeMatches = text.match(/#include\s+[<"]([^>"]+)[>"]/g) ?? []
-    const includeNames = includeMatches
-      .map((line) => line.match(/#include\s+[<"]([^>"]+)[>"]/)?.[1] ?? '')
-      .filter(Boolean)
-    const targets = allNodes
-      .filter((candidate) => candidate.path && candidate.path !== node.path)
-      .filter((candidate) => includeNames.some((inc) => candidate.name.endsWith(inc)))
-      .slice(0, 4)
-
-    let remainingChars = Math.max(0, MAX_CONTEXT_CHARS - text.length)
-    const chunks: string[] = []
-    for (const target of targets) {
-      if (remainingChars <= 500 || !target.path) break
-      const decrypted = await decryptFile(target.path)
-      if (!decrypted) continue
-      const decoded = new TextDecoder('utf-8', { fatal: false }).decode(decrypted)
-      const clipped = decoded.slice(0, remainingChars)
-      remainingChars -= clipped.length
-      chunks.push(`\n\n### Related file: ${target.path}\n${clipped}`)
-    }
-    return chunks.join('')
-  }, [allNodes, decryptFile, node.path, text])
-
   const downloadText = useCallback((filename: string, content: string) => {
     const blob = new Blob([content], { type: 'text/plain;charset=utf-8' })
     const url = URL.createObjectURL(blob)
@@ -486,7 +363,7 @@ export default function MonacoViewer({
           break
         }
 
-        const translatedChunk = await fetchAiSanitizedCode([
+        const translatedChunk = await fetchAiSanitizedCode(privateKey, [
           {
             role: 'system',
             content: 'Translate comments in the provided source code to English. Keep code tokens, identifiers, literals, spacing, and line structure unchanged. Do not wrap with markdown code fences. Do not add introductory or explanatory text. Return only source code.',
@@ -538,11 +415,12 @@ export default function MonacoViewer({
       setIsExplaining(true)
       openAiLoadingToast(t('viewer.monaco.ai_toast_explaining'))
       setStatusMessage(t('viewer.monaco.ai_explaining'))
-      const extraContext = await collectExtraContext()
+      const extraContext = await collectExtraContext({ text, allNodes, nodePath: node.path, decryptFile })
       // Open overlay immediately so streaming content appears progressively.
       setAiOverlay({ mode: 'file-explanation', title: t('viewer.monaco.ai_explain_file'), content: '' })
       let streamed = ''
       const explanation = await fetchAiCompletion(
+        privateKey,
         [
           {
             role: 'system',
@@ -553,7 +431,7 @@ export default function MonacoViewer({
             content: `### Main file: ${node.path ?? node.name}\n${text.slice(0, MAX_CONTEXT_CHARS)}${extraContext}`,
           },
         ],
-        (chunk) => {
+        (chunk: string) => {
           streamed += chunk
           setAiOverlay((prev) => prev ? { ...prev, content: streamed } : null)
         },
@@ -600,7 +478,7 @@ export default function MonacoViewer({
           break
         }
 
-        const annotatedChunk = await fetchAiSanitizedCode([
+        const annotatedChunk = await fetchAiSanitizedCode(privateKey, [
           {
             role: 'system',
             content: `You are mentoring a junior developer. Add concise inline comments in ${targetLang}. Keep the original code behavior exactly unchanged. You may add comments and can translate existing comments into ${targetLang} if needed. Do not rename identifiers. Do not remove logic. Preserve formatting and line structure as much as possible. Return only source code without markdown fences or introductory text.`,
@@ -681,6 +559,7 @@ export default function MonacoViewer({
       setAiOverlay({ mode: 'selection-explanation', title: overlayTitle, content: '' })
       let streamed = ''
       const explanation = await fetchAiCompletion(
+        privateKey,
         [
           {
             role: 'system',
@@ -691,7 +570,7 @@ export default function MonacoViewer({
             content: `File: ${node.path ?? node.name}\n\nSelected code:\n${selectedText.slice(0, 12_000)}\n\nNearby context:\n${surrounding.join('\n').slice(0, 12_000)}`,
           },
         ],
-        (chunk) => {
+        (chunk: string) => {
           streamed += chunk
           setAiOverlay((prev) => prev ? { ...prev, content: streamed } : null)
         },