jan/electron/utils/migration.ts

import { app } from 'electron'

import { join } from 'path'
import {
  rmdirSync,
  readFileSync,
  existsSync,
  mkdirSync,
  readdirSync,
  cpSync,
  lstatSync,
} from 'fs'
import Store from 'electron-store'
import { getJanDataFolderPath, appResourcePath } from './../utils/path'

/**
 * Migrates the extensions & themes.
 * If the `migrated_version` key in the `Store` object does not match the current app version,
 * the function deletes the `extensions` directory and sets the `migrated_version` key to the current app version.
 * @returns A Promise that resolves when the migration is complete.
 */
export async function migrate() {
  const store = new Store()
  if (store.get('migrated_version') !== app.getVersion()) {
    console.debug('start migration:', store.get('migrated_version'))

    await migrateThemes()

    store.set('migrated_version', app.getVersion())
    console.debug('migrate extensions done')
  } else if (!existsSync(join(getJanDataFolderPath(), 'themes'))) {
    await migrateThemes()
  }
}

async function migrateThemes() {
  if (!existsSync(join(getJanDataFolderPath(), 'themes')))
    mkdirSync(join(getJanDataFolderPath(), 'themes'), { recursive: true })

  const themes = readdirSync(join(await appResourcePath(), 'themes'))
  for (const theme of themes) {
    const themePath = join(await appResourcePath(), 'themes', theme)
    if (existsSync(themePath) && !lstatSync(themePath).isDirectory()) {
      continue
    }
    await checkAndMigrateTheme(theme, themePath)
  }
}

async function checkAndMigrateTheme(
  sourceThemeName: string,
  sourceThemePath: string
) {
  const janDataThemesFolder = join(getJanDataFolderPath(), 'themes')
  const existingTheme = readdirSync(janDataThemesFolder).find(
    (theme) => theme === sourceThemeName
  )
  if (existingTheme) {
    const desTheme = join(janDataThemesFolder, existingTheme)
    if (!existsSync(desTheme) || !lstatSync(desTheme).isDirectory()) return

    const desThemeData = JSON.parse(
      readFileSync(join(desTheme, 'theme.json'), 'utf-8')
    )
    const sourceThemeData = JSON.parse(
      readFileSync(join(sourceThemePath, 'theme.json'), 'utf-8')
    )
    if (desThemeData.version !== sourceThemeData.version) {
      console.debug('Updating theme', existingTheme)
      rmdirSync(desTheme, { recursive: true })
      cpSync(sourceThemePath, join(janDataThemesFolder, sourceThemeName), {
        recursive: true,
      })
    }
  } else {
    console.debug('Adding new theme', sourceThemeName)
    cpSync(sourceThemePath, join(janDataThemesFolder, sourceThemeName), {
      recursive: true,
    })
  }
}