FileConfig.ts

// @ts-ignore
const fromEntries = (entries) => Object.assign(...entries.map(([ k, v ]) => ({ [k]: v })))

import os from 'os'
import path from 'path'
import fs from 'fs-extra'
import isDocker from 'is-docker'
import yaml from 'js-yaml'
import Config from './Config'
import { YAMLConnections } from '../ConnectionsManager'
import diff from './diff'
import defaultConfig from './defaultConfig'
import { Session } from 'libfb'
import SetupServer from '../api/setup'

const log = logger.withScope('FileConfig')

export default class FileConfig extends Config {
  path: string
  connPath: string
  sessionPath: string

  constructor (dataPath = getConfigDir()) {
    super()
    dataPath = path.resolve(dataPath)
    this.path = path.join(dataPath, 'config.json')
    this.connPath = path.join(dataPath, 'connections.yml')
    this.sessionPath = path.join(dataPath, 'session.json')
  }

  async load () {
    if (!fs.existsSync(this.path)) {
      const server = new SetupServer()
      server.run()
      return new Promise(resolve => {
        server.once('config', async config => {
          await fs.ensureFile(this.path)
          await fs.writeJSON(this.path, config, { spaces: 2 })
          await fs.writeJSON(this.sessionPath, server.messengerSession, { spaces: 2 })
          this._load(config)
          await server.stop()
          resolve()
        })
      })
    }

    log.info(`Loading config from ${this.path}`)
    const config = await fs.readJSON(this.path)
    this._load(config)
  }

  set (key: string, value: string) {
    this._set(key, value)
    return this.save()
  }

  async save () {
    let config = JSON.parse(JSON.stringify(this.config))
    config = fromEntries(
      Object.entries(config)
        .map(([ key, value ]) => {
          if (typeof value === 'object' && defaultConfig[key]) {
            return [ key, diff(defaultConfig[key], value) ]
          }
          return [ key, value ]
        })
    )
    await fs.writeJSON(this.path, config, { spaces: 2 })
  }

  async loadConnections () {
    const file = await fs.readFile(this.connPath, 'utf8')
    return yaml.safeLoad(file) as YAMLConnections
  }
  saveConnections (connections: YAMLConnections) {
    return fs.writeFile(this.connPath, yaml.safeDump(connections), 'utf8')
  }

  loadSession () { return fs.readJSON(this.sessionPath) }
  saveSession (session: Session) { return fs.writeJSON(this.sessionPath, session) }
}
export function getConfigDir () {
  if (isDocker()) return '/config'
  switch (process.platform) {
    case 'win32':
      return path.join(process.env.APPDATA!!, 'Miscord')
    case 'linux':
      return path.join(os.homedir(), '.config', 'Miscord')
    case 'darwin':
      return path.join(os.homedir(), 'Library', 'Application Support', 'Miscord')
    default:
      return path.join(os.homedir(), '.miscord')
  }
}