autorestic/src/utils.ts

import axios from 'axios'
import { spawnSync, SpawnSyncOptions } from 'child_process'
import { randomBytes } from 'crypto'
import { createWriteStream } from 'fs'

export const exec = (
  command: string,
  args: string[],
  { env, ...rest }: SpawnSyncOptions = {}
) => {
  const cmd = spawnSync(command, args, {
    ...rest,
    env: {
      ...process.env,
      ...env,
    },
  })

  const out = cmd.stdout && cmd.stdout.toString().trim()
  const err = cmd.stderr && cmd.stderr.toString().trim()

  return { out, err }
}

export const checkIfResticIsAvailable = () =>
  checkIfCommandIsAvailable(
    'restic',
    'Restic is not installed'.red +
      ' https://restic.readthedocs.io/en/latest/020_installation.html#stable-releases'
  )

export const checkIfCommandIsAvailable = (cmd: string, errorMsg?: string) => {
  if (require('child_process').spawnSync(cmd).error)
    throw new Error(errorMsg ? errorMsg : `"${errorMsg}" is not installed`.red)
}

export const makeObjectKeysLowercase = (object: Object): any =>
  Object.fromEntries(
    Object.entries(object).map(([key, value]) => [key.toLowerCase(), value])
  )

export function rand(length = 32): string {
  return randomBytes(length / 2).toString('hex')
}

export const singleToArray = <T>(singleOrArray: T | T[]): T[] =>
  Array.isArray(singleOrArray) ? singleOrArray : [singleOrArray]

export const filterObject = <T>(
  obj: { [key: string]: T },
  filter: (item: [string, T]) => boolean
): { [key: string]: T } =>
  Object.fromEntries(Object.entries(obj).filter(filter))

export const filterObjectByKey = <T>(
  obj: { [key: string]: T },
  keys: string[]
) => filterObject(obj, ([key]) => keys.includes(key))

export const downloadFile = async (url: string, to: string) =>
  new Promise<void>(async res => {
    const { data: file } = await axios({
      method: 'get',
      url: url,
      responseType: 'stream',
    })

    const stream = createWriteStream(to)

    const writer = file.pipe(stream)
    writer.on('close', () => {
      stream.close()
      res()
    })
  })

export const ConfigError = new Error('Config file not found')