prism/src/main/themeLibrary.ts

import { access, mkdir, readFile, readdir, unlink, writeFile } from 'node:fs/promises'
import { basename, dirname, extname, isAbsolute, join, relative, resolve } from 'node:path'
import { randomUUID } from 'node:crypto'
import {
  DEFAULT_THEME_ID,
  DEFAULT_THEME_NAME,
  LEGACY_THEME_MIGRATION_VERSION,
  type LegacyThemeMigrationPayload,
  type LegacyThemeMigrationResult,
  type PrismTheme,
  type PrismThemeLocalStateV1,
  type ThemeLibrarySnapshot,
} from '../types/theme'
import {
  createBundledThemes,
  createDefaultTheme,
  createEmptyThemeLocalState,
  createMigratedAccentTheme,
  createTemplateThemeFile,
  getDefaultThemeIdForLocalState,
  normalizeLegacyThemePayload,
  normalizeThemeLocalState,
  parseThemeFileContent,
  resolveLegacyThemeToPresetId,
  serializeThemeFile,
} from '../shared/themeState'

const THEME_EXTENSION = '.iro'
const TEMPLATE_THEME_FILE_NAME = '_TEMPLATE.iro'

interface ManagedThemeEntry {
  id: string
  path: string
  theme: PrismTheme
}

export class FileBackedThemeLibrary {
  constructor(
    private readonly themesDir: string,
    private readonly localStatePath: string,
  ) {}

  getThemesDirectory(): string {
    return this.themesDir
  }

  async getActiveThemeId(): Promise<string | null> {
    const { entries, localState } = await this.loadLibrary()
    return localState.activeThemeId && entries.some((entry) => entry.id === localState.activeThemeId)
      ? localState.activeThemeId
      : (entries[0]?.id ?? null)
  }

  async getSnapshot(): Promise<ThemeLibrarySnapshot> {
    const { entries, localState } = await this.loadLibrary()
    return this.buildSnapshot(entries, localState)
  }

  async loadTheme(id: string): Promise<ThemeLibrarySnapshot> {
    const { entries, localState } = await this.loadLibrary()
    this.findEntry(entries, id)
    localState.activeThemeId = id
    await this.writeLocalState(localState)
    return this.buildSnapshot(entries, localState)
  }

  async importThemeFromPath(sourcePath: string): Promise<ThemeLibrarySnapshot> {
    const { entries, localState } = await this.loadLibrary()
    const resolvedSourcePath = resolve(sourcePath)
    const theme = await this.readThemeFile(resolvedSourcePath)
    const existingEntry = entries.find((entry) => entry.id === theme.id) ?? null
    const insideManagedDirectory = this.isPathInsideDirectory(resolvedSourcePath, this.themesDir)
    const currentPath = existingEntry?.path ?? (insideManagedDirectory ? resolvedSourcePath : undefined)
    const targetPath = await this.writeManagedTheme(entries, theme.id, theme, currentPath)

    if (insideManagedDirectory && resolvedSourcePath !== targetPath) {
      await this.unlinkIfExists(resolvedSourcePath)
    }

    localState.activeThemeId = theme.id
    await this.writeLocalState(localState)
    return this.getSnapshot()
  }

  async renameTheme(id: string, name: string): Promise<ThemeLibrarySnapshot> {
    if (id === DEFAULT_THEME_ID) {
      throw new Error('The default theme cannot be renamed.')
    }

    const { entries, localState } = await this.loadLibrary()
    const entry = this.findEntry(entries, id)
    const normalized = {
      ...entry.theme,
      name: name.trim() || entry.theme.name,
    }
    await this.writeManagedTheme(entries, id, normalized, entry.path)
    return this.buildSnapshot(await this.readManagedEntries(), localState)
  }

  async deleteTheme(id: string): Promise<ThemeLibrarySnapshot> {
    if (id === DEFAULT_THEME_ID) {
      throw new Error('The default theme cannot be deleted.')
    }

    const { entries, localState } = await this.loadLibrary()
    const entry = this.findEntry(entries, id)
    await unlink(entry.path)
    if (localState.activeThemeId === id) {
      localState.activeThemeId = DEFAULT_THEME_ID
    }
    await this.writeLocalState(localState)
    return this.getSnapshot()
  }

  async reloadThemes(): Promise<ThemeLibrarySnapshot> {
    return this.getSnapshot()
  }

  async migrateLegacyTheme(payload: LegacyThemeMigrationPayload): Promise<LegacyThemeMigrationResult> {
    const { entries, localState } = await this.loadLibrary()
    if (localState.migrationVersion >= LEGACY_THEME_MIGRATION_VERSION) {
      return {
        didMigrate: false,
        snapshot: this.buildSnapshot(entries, localState),
      }
    }

    const normalizedPayload = normalizeLegacyThemePayload(payload)
    let nextActiveThemeId = resolveLegacyThemeToPresetId(normalizedPayload)
    let didMigrate = false

    if (normalizedPayload.customAccent) {
      const migratedTheme = createMigratedAccentTheme(normalizedPayload.customAccent)
      if (migratedTheme) {
        await this.writeManagedTheme(entries, migratedTheme.id, migratedTheme)
        nextActiveThemeId = migratedTheme.id
        didMigrate = true
      }
    } else if (nextActiveThemeId) {
      didMigrate = true
    }

    localState.migrationVersion = LEGACY_THEME_MIGRATION_VERSION
    localState.activeThemeId = nextActiveThemeId && (await this.themeExists(nextActiveThemeId))
      ? nextActiveThemeId
      : getDefaultThemeIdForLocalState()
    await this.writeLocalState(localState)

    return {
      didMigrate,
      snapshot: await this.getSnapshot(),
    }
  }

  private async loadLibrary(): Promise<{
    entries: ManagedThemeEntry[]
    localState: PrismThemeLocalStateV1
  }> {
    await mkdir(this.themesDir, { recursive: true })
    let localState = await this.readLocalState()
    let entries = await this.readManagedEntries()

    if (entries.length === 0) {
      for (const theme of createBundledThemes()) {
        await this.writeManagedTheme(entries, theme.id, theme)
      }
      entries = await this.readManagedEntries()
    }

    entries = await this.syncBundledThemes(entries)

    await this.ensureTemplateFile()

    if (!localState.activeThemeId || !entries.some((entry) => entry.id === localState.activeThemeId)) {
      localState = {
        ...localState,
        activeThemeId: entries.find((entry) => entry.id === DEFAULT_THEME_ID)?.id ?? entries[0]?.id ?? null,
      }
      await this.writeLocalState(localState)
    }

    return {
      entries: this.sortEntries(entries),
      localState,
    }
  }

  private async syncBundledThemes(entries: ManagedThemeEntry[]): Promise<ManagedThemeEntry[]> {
    let nextEntries = entries

    for (const theme of createBundledThemes()) {
      const existingEntry = nextEntries.find((entry) => entry.id === theme.id) ?? null
      const shouldWrite = !existingEntry || serializeThemeFile(existingEntry.theme) !== serializeThemeFile(theme)
      if (!shouldWrite) continue

      await this.writeManagedTheme(nextEntries, theme.id, theme, existingEntry?.path)
      nextEntries = await this.readManagedEntries()
    }

    if (!nextEntries.some((entry) => entry.id === DEFAULT_THEME_ID)) {
      await this.writeManagedTheme(nextEntries, DEFAULT_THEME_ID, createDefaultTheme())
      nextEntries = await this.readManagedEntries()
    }

    return nextEntries
  }

  private async ensureTemplateFile(): Promise<void> {
    const targetPath = resolve(join(this.themesDir, TEMPLATE_THEME_FILE_NAME))
    const nextContent = createTemplateThemeFile()

    if (await this.pathExists(targetPath)) {
      const currentContent = await readFile(targetPath, 'utf8')
      if (currentContent === nextContent) {
        return
      }
    }

    await writeFile(targetPath, nextContent, 'utf8')
  }

  private async readManagedEntries(): Promise<ManagedThemeEntry[]> {
    const dirEntries = await readdir(this.themesDir, { withFileTypes: true })
    const themePaths = dirEntries
      .filter((entry) => {
        if (!entry.isFile()) return false
        if (entry.name.startsWith('_')) return false
        return extname(entry.name).toLowerCase() === THEME_EXTENSION
      })
      .map((entry) => resolve(join(this.themesDir, entry.name)))
      .sort((left, right) => left.localeCompare(right))

    const entries: ManagedThemeEntry[] = []
    const seenIds = new Set<string>()

    for (const filePath of themePaths) {
      try {
        const theme = await this.readThemeFile(filePath)
        if (seenIds.has(theme.id)) {
          console.warn(`Skipping duplicate theme id "${theme.id}" in ${basename(filePath)}.`)
          continue
        }
        seenIds.add(theme.id)
        entries.push({
          id: theme.id,
          path: filePath,
          theme,
        })
      } catch (error) {
        console.warn(`Skipping invalid theme file at ${filePath}:`, error)
      }
    }

    return entries
  }

  private async readThemeFile(filePath: string): Promise<PrismTheme> {
    const content = await readFile(filePath, 'utf8')
    return parseThemeFileContent(
      content,
      this.buildFallbackThemeId(filePath),
      basename(filePath, THEME_EXTENSION),
    )
  }

  private async readLocalState(): Promise<PrismThemeLocalStateV1> {
    try {
      const raw = await readFile(this.localStatePath, 'utf8')
      return normalizeThemeLocalState(JSON.parse(raw) as unknown)
    } catch {
      return createEmptyThemeLocalState()
    }
  }

  private async writeLocalState(state: PrismThemeLocalStateV1): Promise<void> {
    await mkdir(dirname(this.localStatePath), { recursive: true })
    await writeFile(this.localStatePath, `${JSON.stringify(state, null, 2)}\n`, 'utf8')
  }

  private async writeManagedTheme(
    entries: ManagedThemeEntry[],
    id: string,
    theme: PrismTheme,
    currentPath?: string,
  ): Promise<string> {
    const nextPath = await this.getManagedThemePath(entries, id, theme.name, currentPath)
    const existingPath = currentPath ? resolve(currentPath) : null

    await mkdir(dirname(nextPath), { recursive: true })
    await writeFile(nextPath, serializeThemeFile(theme), 'utf8')

    if (existingPath && existingPath !== nextPath) {
      await this.unlinkIfExists(existingPath)
    }

    return nextPath
  }

  private async getManagedThemePath(
    entries: ManagedThemeEntry[],
    id: string,
    name: string,
    currentPath?: string,
  ): Promise<string> {
    if (id === DEFAULT_THEME_ID) {
      const defaultPath = resolve(join(this.themesDir, `${DEFAULT_THEME_NAME}${THEME_EXTENSION}`))
      if (!currentPath || resolve(currentPath) === defaultPath) {
        return defaultPath
      }
      const occupiedByOtherEntry = entries.some((entry) => entry.id !== id && entry.path === defaultPath)
      return occupiedByOtherEntry ? resolve(currentPath) : defaultPath
    }

    const preferredCurrentPath = currentPath ? resolve(currentPath) : null
    const occupiedPaths = new Set(entries.filter((entry) => entry.id !== id).map((entry) => entry.path))
    const baseStem = this.sanitizeFileStem(name)

    let attempt = 0
    while (true) {
      const suffix = attempt === 0 ? '' : ` (${attempt + 1})`
      const candidatePath = resolve(join(this.themesDir, `${baseStem}${suffix}${THEME_EXTENSION}`))
      if (preferredCurrentPath === candidatePath) {
        return candidatePath
      }
      if (occupiedPaths.has(candidatePath) || await this.pathExists(candidatePath)) {
        attempt += 1
        continue
      }
      return candidatePath
    }
  }

  private sortEntries(entries: ManagedThemeEntry[]): ManagedThemeEntry[] {
    return [...entries].sort((left, right) => {
      if (left.id === DEFAULT_THEME_ID) return -1
      if (right.id === DEFAULT_THEME_ID) return 1
      return left.theme.name.localeCompare(right.theme.name)
    })
  }

  private buildSnapshot(entries: ManagedThemeEntry[], localState: PrismThemeLocalStateV1): ThemeLibrarySnapshot {
    const themes = this.sortEntries(entries).reduce((acc, entry) => {
      acc[entry.id] = entry.theme
      return acc
    }, {} as Record<string, PrismTheme>)

    return {
      themes,
      activeThemeId: localState.activeThemeId && themes[localState.activeThemeId]
        ? localState.activeThemeId
        : (themes[DEFAULT_THEME_ID] ? DEFAULT_THEME_ID : Object.keys(themes)[0] ?? null),
    }
  }

  private findEntry(entries: ManagedThemeEntry[], id: string): ManagedThemeEntry {
    const entry = entries.find((candidate) => candidate.id === id)
    if (!entry) {
      throw new Error(`Theme "${id}" was not found.`)
    }
    return entry
  }

  private sanitizeFileStem(name: string): string {
    const sanitized = name
      .trim()
      .replace(/[<>:"/\\|?*\u0000-\u001f]/g, ' ')
      .replace(/\s+/g, ' ')
      .trim()

    return sanitized || 'Theme'
  }

  private buildFallbackThemeId(filePath: string): string {
    const stem = basename(filePath, THEME_EXTENSION)
      .toLowerCase()
      .replace(/[^a-z0-9]+/g, '_')
      .replace(/^_+|_+$/g, '')
    return stem ? `theme_${stem}` : `theme_${randomUUID().replace(/-/g, '')}`
  }

  private isPathInsideDirectory(candidatePath: string, directoryPath: string): boolean {
    const relativePath = relative(resolve(directoryPath), resolve(candidatePath))
    return relativePath === '' || (!relativePath.startsWith('..') && !isAbsolute(relativePath))
  }

  private async themeExists(id: string): Promise<boolean> {
    const entries = await this.readManagedEntries()
    return entries.some((entry) => entry.id === id)
  }

  private async pathExists(targetPath: string): Promise<boolean> {
    try {
      await access(targetPath)
      return true
    } catch {
      return false
    }
  }

  private async unlinkIfExists(targetPath: string): Promise<void> {
    try {
      await unlink(targetPath)
    } catch {
      // Ignore missing files.
    }
  }
}