herokutools/src/utils/db.ts

Heroku CLI wrapper - helpful commands you use often - made easier!

import fs from "fs"
// import path from "path"
import ospath from "ospath"

const db = ospath.data()
const dirpath = db + "/Herokutools"
const filepath = dirpath + "/env.txt"

export default class DB {
  constructor(options: any) {
    this.checkIfDBFileExists()
    this.handleEnvParameters(options.app)
  }

  /**
   * * Get Default App Name (HEROKU_TOOL_APP_1)
   * @returns {string | undefined}
   */
  public getDefaultAppName = (): string | undefined => {
    const envFile = this.readDBFile()
    return envFile.HEROKU_TOOL_APP_1 || process.env.HEROKU_TOOL_APP_1
  }

  /**
   * * Check if env file exists
   *   If not, create it
   */
  private checkIfDBFileExists = () => {
    if (!fs.existsSync(filepath)) {
      const dir = fs.mkdirSync(dirpath, {
        recursive: true,
      })
      if (dir) {
        fs.writeFileSync(filepath, "")
      }
    }
  }

  /**
   * * Persist environment variables to a file
   * @param env
   */
  private saveToDBFile = (env: { [key: string]: string }) => {
    const envFile = this.readDBFile()
    const newEnv = { ...envFile, ...env }
    const writeStream = fs.createWriteStream(filepath, {
      flags: "w",
    })
    Object.entries(newEnv).map(([key, value]) => {
      console.log(`Setting ${key} to: ${value}`)
      writeStream.write(`${key}=${value}\n`)
    })
    writeStream.end()
  }

  /**
   * * Read environment variables from a file
   * @returns
   */
  private readDBFile = () => {
    const envFile = fs.readFileSync(filepath, "utf8")
    const env = envFile
      .split("\n")
      .filter((line) => line)
      .reduce((acc: { [key: string]: string }, curr) => {
        const [key, value] = curr.split("=")
        acc[key] = value
        return acc
      }, {})
    return env
  }

  /**
   * * Handle environment variables
   * @param appName
   */
  private handleEnvParameters = (appName?: string) => {
    let app1 = process.env.HEROKU_TOOL_APP_1
    if (appName) {
      app1 = appName
    } else {
      app1 = this.readDBFile().HEROKU_TOOL_APP_1
    }
    let app2 = process.env.HEROKU_TOOL_APP_2
    let app3 = process.env.HEROKU_TOOL_APP_3

    if (app1) {
      process.env.HEROKU_TOOL_APP_1 = app1
      this.saveToDBFile({ HEROKU_TOOL_APP_1: app1 })
    }
    if (app2) {
      process.env.HEROKU_TOOL_APP_2 = app2
      this.saveToDBFile({ HEROKU_TOOL_APP_2: app2 })
    }
    if (app3) {
      process.env.HEROKU_TOOL_APP_3 = app3
      this.saveToDBFile({ HEROKU_TOOL_APP_3: app3 })
    }
  }
}