herokutools/src/commands/env.ts

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

import { spawnSync } from "child_process"
import DB from "../utils/db"
import processIndex from "../utils/process.index"
import verifyApp from "../utils/verify.app"
import readEnvVariablePrams from "../utils/read.env.var"

/**
 * This will allow you to set environment variables for a Heroku app
 */

export default class Env {
  constructor(options: any, args: any) {
    new DB(options)
    if (options.index) {
      options.app = processIndex(options.index)
    }
    if (options.get) {
      // GET
      this.get(options.app)
      return
    }
    if (options.remove) {
      // REMOVE
      this.removeMulti(readEnvVariablePrams(args), options.app)
      return
    }
    // SET Multi - Default
    this.setMulti(readEnvVariablePrams(args), options.app)
  }

  /**
   * * GET env vars from Heroku
   * @param app
   * @returns
   */
  get = async (app?: string | null) => {
    app = verifyApp(app)
    if (!app) {
      return
    }
    try {
      const command = `heroku config --app ${app}`
      const child = spawnSync(command, {
        shell: true,
        stdio: "inherit",
      })
      if (child.error) {
        console.log(child.error)
      } else {
        console.log(child.stdout) // eslint-disable-line
      }
    } catch (error) {
      console.error(error)
    }
  }

  /**
   * * SET env var on Heroku
   * @param key
   * @param value
   * @param app
   * @returns
   */
  setSingle = async (key: string, value: string, app?: string | null) => {
    app = verifyApp(app)
    if (!app) {
      return
    }
    if (!key || !value) {
      console.error("Please provide a key and value")
      return
    }
    try {
      const command = `heroku config:set ${key}=${value} --app ${app}`
      const child = spawnSync(command, {
        shell: true,
        stdio: "inherit",
      })
      if (child.error) {
        console.log(child.error)
      } else {
        console.log(child.stdout) // eslint-disable-line
      }
      console.log(`Successfully set ${key}=${value}`)
    } catch (error) {
      console.error(error)
    }
  }

  /**
   * * SET multiple env vars on Heroku
   * @param envVars
   * @param app
   * @returns
   */
  setMulti = async (
    envVars: {
      [key: string]: string
    },
    app?: string | null
  ) => {
    app = verifyApp(app)
    if (!app) {
      return
    }
    try {
      // set all the env vars
      Object.entries(envVars).forEach(([key, value]) => {
        this.setSingle(key, value, app)
      })
      console.log(`Successfully set all environment variables`)
    } catch (error) {
      console.error(error)
    }
  }

  /**
   * * UNSET env var on Heroku
   * @param key
   * @param value
   * @param app
   * @returns
   */
  unsetSingle = async (key: string, value: string, app?: string | null) => {
    app = verifyApp(app)
    if (!app) {
      return
    }
    if (!key || !value) {
      console.error("Please provide a key and value")
      return
    }
    try {
      const command = `heroku config:unset ${key}=${value} --app ${app}`
      // execSync(`heroku config:set ${key}=${value} --app ${app}`)
      const child = spawnSync(command, {
        shell: true,
        stdio: "inherit",
      })
      if (child.error) {
        console.log(child.error)
      } else {
        console.log(child.stdout) // eslint-disable-line
      }
      console.log(`Successfully unset ${key}=${value}`)
    } catch (error) {
      console.error(error)
    }
  }

  /**
   * * REMOVE multiple env vars on Heroku
   * @param envVars
   * @param app
   * @returns
   */
  removeMulti = async (
    envVars: {
      [key: string]: string
    },
    app?: string | null
  ) => {
    app = verifyApp(app)
    if (!app) {
      return
    }
    try {
      // set all the env vars
      Object.entries(envVars).forEach(([key, value]) => {
        this.unsetSingle(key, value, app)
      })
      console.log(`Successfully unset environment variables`)
    } catch (error) {
      console.error(error)
    }
  }
}