tolaria/src/hooks/useThemeManager.ts

import { useCallback, useEffect, useMemo, useRef, useState } from 'react'
import { invoke } from '@tauri-apps/api/core'
import { isTauri, mockInvoke } from '../mock-tauri'
import { parseFrontmatter } from '../utils/frontmatter'
import type { ThemeFile, VaultEntry, VaultSettings } from '../types'

function tauriCall<T>(command: string, args: Record<string, unknown>): Promise<T> {
  return isTauri() ? invoke<T>(command, args) : mockInvoke<T>(command, args)
}

/** Frontmatter keys that are metadata — not CSS custom properties. */
const NON_THEME_KEYS = new Set([
  'Is A', 'type', 'is_a', 'is a',
  'Name', 'name', 'title', 'Title',
  'Description', 'description',
  'Archived', 'archived',
  'Trashed', 'trashed',
  'Trashed at', 'trashed at', 'trashed_at',
  'Created at', 'created at', 'created_at',
  'Created time', 'created_time',
  'Owner', 'owner',
  'Status', 'status',
  'Cadence', 'cadence',
  'aliases',
  'Belongs to', 'belongs_to', 'belongs to',
  'Related to', 'related_to', 'related to',
])

/** Extract CSS custom properties from a theme note's frontmatter content. */
export function extractCssVars(content: string): Record<string, string> {
  const fm = parseFrontmatter(content)
  const vars: Record<string, string> = {}
  for (const [key, value] of Object.entries(fm)) {
    if (NON_THEME_KEYS.has(key)) continue
    if (typeof value === 'string' && value) {
      vars[`--${key}`] = value
    } else if (typeof value === 'number') {
      vars[`--${key}`] = String(value)
    }
  }
  return vars
}

/** Extract bare colors (without -- prefix) for ThemeFile.colors from content. */
function extractColorsFromContent(content: string): Record<string, string> {
  const fm = parseFrontmatter(content)
  const colors: Record<string, string> = {}
  for (const [key, value] of Object.entries(fm)) {
    if (NON_THEME_KEYS.has(key)) continue
    if (typeof value === 'string' && value.startsWith('#')) {
      colors[key] = value
    }
  }
  return colors
}

/** Check if a hex color is perceptually dark (luminance < 0.5). */
export function isColorDark(hex: string): boolean {
  if (!hex.startsWith('#') || hex.length < 7) return false
  const r = parseInt(hex.slice(1, 3), 16)
  const g = parseInt(hex.slice(3, 5), 16)
  const b = parseInt(hex.slice(5, 7), 16)
  return (0.299 * r + 0.587 * g + 0.114 * b) / 255 < 0.5
}

/** Update color-scheme and data-theme-mode on document root based on --background. */
function updateColorScheme(vars: Record<string, string>): void {
  const bg = vars['--background']
  if (!bg) return
  const dark = isColorDark(bg)
  const root = document.documentElement
  root.style.setProperty('color-scheme', dark ? 'dark' : 'light')
  root.dataset.themeMode = dark ? 'dark' : 'light'
}

function clearColorScheme(): void {
  const root = document.documentElement
  root.style.removeProperty('color-scheme')
  delete root.dataset.themeMode
}

function applyVarsToDom(vars: Record<string, string>): void {
  const root = document.documentElement
  for (const [key, value] of Object.entries(vars)) {
    root.style.setProperty(key, value)
  }
  updateColorScheme(vars)
}

function clearVarsFromDom(vars: Record<string, string>): void {
  const root = document.documentElement
  for (const key of Object.keys(vars)) {
    root.style.removeProperty(key)
  }
  clearColorScheme()
}

/** Build a ThemeFile descriptor from a vault entry, enriched with content colors. */
function entryToThemeFile(entry: VaultEntry, content: string | undefined): ThemeFile {
  return {
    id: entry.path,
    name: entry.title,
    description: '',
    path: entry.path,
    colors: content ? extractColorsFromContent(content) : {},
    typography: {},
    spacing: {},
  }
}

/** True when a theme entry should no longer be applied (trashed or archived). */
function isEntryRemoved(entry: VaultEntry): boolean {
  return entry.trashed || entry.archived
}

export interface ThemeManager {
  themes: ThemeFile[]
  activeThemeId: string | null
  activeTheme: ThemeFile | null
  activeThemeContent: string | undefined
  isDark: boolean
  switchTheme: (themeId: string) => Promise<void>
  createTheme: (name?: string) => Promise<string>
  reloadThemes: () => Promise<void>
  /** Update a single frontmatter property on the active theme note. */
  updateThemeProperty: (key: string, value: string) => Promise<void>
}

/** Manages loading and persisting the active theme path from vault settings. */
function useThemeSetting(vaultPath: string | null) {
  const [activeThemeId, setActiveThemeId] = useState<string | null>(null)

  const load = useCallback(async () => {
    if (!vaultPath) return
    try {
      const s = await tauriCall<VaultSettings>('get_vault_settings', { vaultPath })
      setActiveThemeId(s.theme)
    } catch { /* no settings file — fine, no active theme */ }
  }, [vaultPath])

  // eslint-disable-next-line react-hooks/set-state-in-effect -- async fn; setState runs after await
  useEffect(() => { load() }, [load])

  useEffect(() => {
    window.addEventListener('focus', load)
    return () => window.removeEventListener('focus', load)
  }, [load])

  return { activeThemeId, setActiveThemeId, reload: load }
}

/** Applies CSS custom properties to the document root from the active theme. */
function useThemeApplier(
  activeThemeId: string | null,
  cachedContent: string | undefined,
) {
  const appliedVarsRef = useRef<Record<string, string>>({})
  const [isDark, setIsDark] = useState(false)

  const applyDom = useCallback((content: string) => {
    const newVars = extractCssVars(content)
    clearVarsFromDom(appliedVarsRef.current)
    applyVarsToDom(newVars)
    appliedVarsRef.current = newVars
    return newVars
  }, [])

  const clearDom = useCallback(() => {
    clearVarsFromDom(appliedVarsRef.current)
    appliedVarsRef.current = {}
  }, [])

  // Apply theme when activeThemeId or cached content changes.
  // Also serves as live-preview: re-applies when the user saves the theme note.
  useEffect(() => {
    if (!activeThemeId) {
      clearDom()
      setIsDark(false) // eslint-disable-line react-hooks/set-state-in-effect -- sync dark mode with cleared theme
      return
    }
    if (cachedContent) {
      const vars = applyDom(cachedContent)
      setIsDark(isColorDark(vars['--background'] ?? ''))
      return
    }
    tauriCall<string>('get_note_content', { path: activeThemeId })
      .then(content => {
        const vars = applyDom(content)
        setIsDark(isColorDark(vars['--background'] ?? ''))
      })
      .catch(() => { clearDom(); setIsDark(false) })
  }, [activeThemeId, cachedContent, applyDom, clearDom])

  return { clearDom, isDark }
}

export function useThemeManager(
  vaultPath: string | null,
  entries: VaultEntry[],
  allContent: Record<string, string>,
  updateContent?: (path: string, content: string) => void,
): ThemeManager {
  // Ensure default theme files exist on vault open (creates theme/ dir + defaults if missing)
  useEffect(() => {
    if (vaultPath) tauriCall('ensure_vault_themes', { vaultPath }).catch(() => {})
  }, [vaultPath])

  const { activeThemeId, setActiveThemeId, reload } = useThemeSetting(vaultPath)
  const cachedThemeContent = activeThemeId ? allContent[activeThemeId] : undefined
  const { clearDom: clearTheme, isDark } = useThemeApplier(activeThemeId, cachedThemeContent)

  // Track IDs set by user actions (switchTheme/createTheme) so the stale-ID
  // cleanup doesn't clear a newly-created theme that isn't in entries yet.
  const userSetIdRef = useRef<string | null>(null)

  const themes = useMemo(
    () => entries
      .filter(e => e.isA === 'Theme' && !e.trashed && !e.archived)
      .map(e => entryToThemeFile(e, allContent[e.path])),
    [entries, allContent],
  )

  const activeTheme = useMemo(
    () => themes.find(t => t.id === activeThemeId) ?? null,
    [themes, activeThemeId],
  )

  // If active theme ID doesn't match any known theme (e.g. stale ID from old
  // JSON-based theme system), clear it so the app doesn't try to load a
  // non-existent path. Skip IDs just set by switchTheme/createTheme — the
  // entry may not have appeared in `entries` yet.
  useEffect(() => {
    if (!activeThemeId || themes.length === 0) return
    if (activeThemeId === userSetIdRef.current) return
    const isKnown = themes.some(t => t.id === activeThemeId)
    if (!isKnown) {
      clearTheme()
      setActiveThemeId(null)
      if (vaultPath) tauriCall('set_active_theme', { vaultPath, themeId: null }).catch(() => {})
    }
  }, [activeThemeId, themes, clearTheme, vaultPath, setActiveThemeId])

  // If active theme is trashed or archived: clear CSS vars and fall back to no theme
  useEffect(() => {
    if (!activeThemeId) return
    const entry = entries.find(e => e.path === activeThemeId)
    if (!entry || !isEntryRemoved(entry)) return
    clearTheme()
    setActiveThemeId(null)
    if (vaultPath) tauriCall('set_active_theme', { vaultPath, themeId: null }).catch(() => {})
  }, [entries, activeThemeId, clearTheme, vaultPath, setActiveThemeId])

  const switchTheme = useCallback(async (themeId: string) => {
    if (!vaultPath) return
    try {
      await tauriCall<null>('set_active_theme', { vaultPath, themeId })
      userSetIdRef.current = themeId
      setActiveThemeId(themeId)
    } catch (err) { console.error('Failed to switch theme:', err) }
  }, [vaultPath, setActiveThemeId])

  const createTheme = useCallback(async (name?: string) => {
    if (!vaultPath) return ''
    try {
      const path = await tauriCall<string>('create_vault_theme', { vaultPath, name: name ?? null })
      await tauriCall<null>('set_active_theme', { vaultPath, themeId: path })
      userSetIdRef.current = path
      setActiveThemeId(path)
      return path
    } catch (err) { console.error('Failed to create theme:', err); return '' }
  }, [vaultPath, setActiveThemeId])

  const reloadThemes = useCallback(async () => { await reload() }, [reload])

  const updateThemeProperty = useCallback(async (key: string, value: string) => {
    if (!activeThemeId) return
    try {
      const newContent = await tauriCall<string>('update_frontmatter', {
        path: activeThemeId,
        key,
        value,
      })
      updateContent?.(activeThemeId, newContent)
    } catch (err) { console.error('Failed to update theme property:', err) }
  }, [activeThemeId, updateContent])

  return {
    themes, activeThemeId, activeTheme,
    activeThemeContent: cachedThemeContent,
    isDark, switchTheme, createTheme, reloadThemes, updateThemeProperty,
  }
}