Cultivation/src/utils/themes.ts

import { invoke } from '@tauri-apps/api'
import { dataDir } from '@tauri-apps/api/path'
import { convertFileSrc } from '@tauri-apps/api/tauri'

interface Theme {
  name: string
  version: string
  description: string

  // Included custom CSS and JS files
  includes: {
    css: string[]
    js: string[]
  }

  customBackgroundURL?: string
  customBackgroundPath?: string
}

interface BackendThemeList {
  json: string
  path: string
}

interface ThemeList extends Theme {
  path: string
}

const defaultTheme = {
  name: 'default',
  version: '1.0.0',
  description: 'Default theme',
  includes: {
    css: [],
    js: []
  },
  path: 'default'
}
export async function getThemeList() {
  // Do some invoke to backend to get the theme list
  const themes = await invoke('get_theme_list', {
    dataDir: `${await dataDir()}/cultivation`
  }) as BackendThemeList[]
  const list: ThemeList[] = [
    // ALWAYS include default theme
    {
      name: 'default',
      version: '1.0.0',
      description: 'Default theme',
      includes: {
        css: [],
        js: []
      },
      path: 'default'
    }
  ]

  themes.forEach(t => {
    let obj

    try {
      obj = JSON.parse(t.json)
    } catch (e) {
      console.error(e)
    }

    list.push({ ...obj, path: t.path })
  })

  return list
}

export async function getTheme(name: string) {
  const themes = await getThemeList()

  return themes.find(t => t.name === name) || defaultTheme
}

export async function loadTheme(theme: ThemeList, document: Document) {
  // We are going to dynamically load stylesheets into the document
  const head = document.head

  // Get all CSS includes
  const cssIncludes = theme.includes.css
  const jsIncludes = theme.includes.js

  // Load CSS files
  cssIncludes.forEach(css => {
    if (!css) return

    const link = document.createElement('link')

    link.rel = 'stylesheet'
    link.href = convertFileSrc(theme.path + '/' + css)
    head.appendChild(link)
  })

  // Load JS files
  jsIncludes.forEach(js => {
    if (!js) return

    const script = document.createElement('script')

    script.src = convertFileSrc(theme.path + '/' + js)
    head.appendChild(script)
  })

  // Set custom background
  if (theme.customBackgroundURL) {
    document.body.style.backgroundImage = `url('${theme.customBackgroundURL}')`
  }

  // Set custom background
  if (theme.customBackgroundPath) {
    const bgPath = await dataDir() + 'cultivation/grasscutter/theme.png'

    // Save the background to our data dir
    await invoke('copy_file', {
      path: theme.path + '/' + theme.customBackgroundPath,
      new_path: bgPath
    })

    // Set the background
    document.body.style.backgroundImage = `url('${bgPath}')`
  }

  return
}