hohoff/src/renderer/components/Toolbar/AnalysisToolbar.tsx

import { useEffect } from 'react'
import { useEditorStore } from '../../store/editorStore'
import { detectPassiveVoice } from '../../utils/passiveVoice'
import { parseAnnotationsFromAIResponse } from '../../utils/annotationParser'
import './Toolbar.css'

function countWords(text: string): number {
  return text.trim() === '' ? 0 : text.trim().split(/\s+/).length
}

function formatWordCount(n: number): string {
  if (n >= 1000) return `${(n / 1000).toFixed(1)}k`
  return String(n)
}

export function AnalysisToolbar(): JSX.Element {
  const {
    activeFilePath,
    activeFileContent,
    isDirty,
    analysisMode,
    annotations,
    isAILoading,
    setAnnotations,
    clearAnnotations,
    setAnalysisMode,
    addUserMessage,
    startAssistantMessage,
    appendToLastAssistantMessage,
    setAILoading,
    setAIError,
    chatHistory,
    projectWordCount,
    setProjectWordCount,
    fontSize,
    setFontSize
  } = useEditorStore()

  useEffect(() => {
    window.api.getProjectWordCount().then(setProjectWordCount).catch(() => {})
  }, [])

  // Refresh project count after a save (isDirty transitions from true → false)
  useEffect(() => {
    if (!isDirty) {
      window.api.getProjectWordCount().then(setProjectWordCount).catch(() => {})
    }
  }, [isDirty])

  const hasFile = Boolean(activeFilePath)

  const runPassiveVoice = (): void => {
    if (!hasFile) return
    const found = detectPassiveVoice(activeFileContent)
    const existing = useEditorStore.getState().annotations.filter((a) => a.type !== 'passive_voice')
    setAnnotations([...existing, ...found])
    setAnalysisMode('passive_voice')
  }

  const runAIAnalysis = async (mode: 'consistency' | 'style' | 'critique'): Promise<void> => {
    if (!activeFilePath || isAILoading) return

    setAnalysisMode(mode)
    setAIError(null)

    const prompt =
      mode === 'consistency'
        ? 'Please check this chapter for consistency issues (character names, timeline, repeated phrases).'
        : mode === 'style'
          ? 'Please analyze the style and pacing of this chapter and suggest improvements.'
          : 'Please give me an honest critique of this chapter.'

    addUserMessage(prompt)
    startAssistantMessage()
    setAILoading(true)

    try {
      await window.api.streamAIMessage(
        {
          mode,
          documentContent: activeFileContent,
          documentPath: activeFilePath,
          conversationHistory: chatHistory
            .slice(-10)
            .map((m) => ({ role: m.role, content: m.content })),
          userMessage: prompt
        },
        (chunk: string) => {
          appendToLastAssistantMessage(chunk)
        }
      )

      // Parse annotations from response
      const currentHistory = useEditorStore.getState().chatHistory
      const lastMsg = currentHistory[currentHistory.length - 1]
      if (lastMsg?.role === 'assistant' && lastMsg.content.length > 0) {
        const newAnnotations = parseAnnotationsFromAIResponse(lastMsg.content, activeFileContent)
        if (newAnnotations.length > 0) {
          const existing = useEditorStore.getState().annotations.filter((a) => a.type !== mode)
          setAnnotations([...existing, ...newAnnotations])
        }
      }
    } catch (err) {
      setAIError(err instanceof Error ? err.message : 'Analysis failed')
    } finally {
      setAILoading(false)
    }
  }

  const passiveCount = annotations.filter((a) => a.type === 'passive_voice').length
  const consistencyCount = annotations.filter((a) => a.type === 'consistency').length
  const styleCount = annotations.filter((a) => a.type === 'style').length
  const critiqueCount = annotations.filter((a) => a.type === 'critique').length
  const docWordCount = countWords(activeFileContent)

  return (
    <div className="toolbar">
      <div className="toolbar-left">
        <button
          className={`toolbar-btn${analysisMode === 'passive_voice' ? ' active' : ''}`}
          onClick={runPassiveVoice}
          disabled={!hasFile}
          title="Highlight passive voice sentences instantly (no AI required)"
        >
          Passive Voice
          {passiveCount > 0 && (
            <span className="toolbar-badge">{passiveCount}</span>
          )}
        </button>

        <button
          className={`toolbar-btn${analysisMode === 'consistency' ? ' active' : ''}`}
          onClick={() => runAIAnalysis('consistency')}
          disabled={!hasFile || isAILoading}
          title="Check character names, timeline, and repeated phrases via AI"
        >
          {isAILoading && analysisMode === 'consistency' ? 'Checking…' : 'Consistency'}
          {consistencyCount > 0 && (
            <span className="toolbar-badge">{consistencyCount}</span>
          )}
        </button>

        <button
          className={`toolbar-btn${analysisMode === 'style' ? ' active' : ''}`}
          onClick={() => runAIAnalysis('style')}
          disabled={!hasFile || isAILoading}
          title="Pacing, sentence variety, show-don't-tell feedback via AI"
        >
          {isAILoading && analysisMode === 'style' ? 'Analyzing…' : 'Style'}
          {styleCount > 0 && (
            <span className="toolbar-badge">{styleCount}</span>
          )}
        </button>

        <button
          className={`toolbar-btn${analysisMode === 'critique' ? ' active' : ''}`}
          onClick={() => runAIAnalysis('critique')}
          disabled={!hasFile || isAILoading}
          title="Honest overall critique of this chapter via AI"
        >
          {isAILoading && analysisMode === 'critique' ? 'Reading…' : 'Critique'}
          {critiqueCount > 0 && (
            <span className="toolbar-badge">{critiqueCount}</span>
          )}
        </button>

        {annotations.length > 0 && (
          <button
            className="toolbar-btn toolbar-btn-clear"
            onClick={clearAnnotations}
            title="Remove all highlights"
          >
            Clear
          </button>
        )}
      </div>

      <div className="toolbar-right">
        <div className="toolbar-fontsize">
          <button
            className="toolbar-fontsize-btn"
            onClick={() => setFontSize(fontSize - 1)}
            disabled={fontSize <= 11}
            title="Decrease font size"
          >
            A−
          </button>
          <button
            className="toolbar-fontsize-btn"
            onClick={() => setFontSize(fontSize + 1)}
            disabled={fontSize >= 24}
            title="Increase font size"
          >
            A+
          </button>
        </div>
        {activeFilePath && (
          <span
            className="toolbar-wordcount"
            title={`This document: ${docWordCount.toLocaleString()} words · Entire project: ${projectWordCount.toLocaleString()} words`}
          >
            {formatWordCount(docWordCount)}
            <span className="toolbar-wordcount-sep">/</span>
            {formatWordCount(projectWordCount)}
          </span>
        )}
        {isDirty && (
          <span className="toolbar-dirty" title="Unsaved changes — press Cmd+S to save">
            ●
          </span>
        )}
        {activeFilePath && (
          <span className="toolbar-filename">
            {activeFilePath.split('/').pop()?.replace(/\.md$/, '')}
          </span>
        )}
      </div>
    </div>
  )
}