wip: working set secrets and vars

Author: Giorgiosaud

src/commands/rm-secret/index.ts src/commands/rm-secrets/index.tssrc/commands/rm-secret/index.ts renamed to src/commands/rm-secrets/index.ts

@@ -1,14 +1,11 @@
 /* eslint-disable no-await-in-loop */
 import {Command} from '@oclif/core'
 import {info} from '../../helpers/logger'
-import {validateRepoNames, validateSecrets} from '../../helpers/validations'
-import secretVarsFlags from '../../helpers/set-vars-helpers/secret-vars-flags'
+import {validateRepoNames} from '../../helpers/validations'
 import repositoryFactory from '../../repositories/repository-factory'
-import encryptSecret from '../../set-secret-helpers/encrypt-secret'
-import {getPublicKey} from '../../set-secret-helpers/get-public-key'
 import RmSecretFlags from '../../helpers/rm-secret-helpers/rm-secret-flags'
 
-export default class SetSecret extends Command {
+export default class RmSecret extends Command {
   static description = 'describe the command here'
 
   static examples = [
@@ -26,7 +23,7 @@ export default class SetSecret extends Command {
   static flags = RmSecretFlags
 
   async run(): Promise<void> {
-    const {flags: {organization, repositories, secrets, environment}} = await this.parse(SetSecret)
+    const {flags: {organization, repositories, secrets, environment}} = await this.parse(RmSecret)
     validateRepoNames(repositories)
     const octoFactory = repositoryFactory.get('octokit')
     for (const repo of repositories) {

src/commands/set-secret/index.ts src/commands/set-secrets/index.tssrc/commands/set-secret/index.ts renamed to src/commands/set-secrets/index.ts

@@ -1,43 +1,41 @@
+/* eslint-disable no-await-in-loop */
 import {Command} from '@oclif/core'
 import {info} from '../../helpers/logger'
-import {validateEqualLengths, validateRepoNames} from '../../helpers/validations'
-import updateVars from '../../helpers/set-vars-helpers/update-vars'
+import {validateRepoNames, validateSecrets} from '../../helpers/validations'
 import secretVarsFlags from '../../helpers/set-vars-helpers/secret-vars-flags'
+import repositoryFactory from '../../repositories/repository-factory'
+import encryptSecret from '../../set-secret-helpers/encrypt-secret'
+import {getPublicKey} from '../../set-secret-helpers/get-public-key'
 
 export default class SetVars extends Command {
-  static description = 'Set Variables for a list of repositories'
+  static description = 'describe the command here'
 
   static examples = [
     `
-    $ github-automation set-vars -r OWNER/NAME1 OWNER/NAME2 ... OWNER/NAMEn --secret-name SECRET_NAME1 SECRET_NAME2 ... SECRET_NAMEN --secret-value SECRETVALUE1 SECRETVALUE2 ... SECRETVALUEN
-    $ github-automation set-vars -r OWNER/NAME1 OWNER/NAME2 ... OWNER/NAMEn -n SECRET_NAME1 SECRET_NAME2 ... SECRET_NAMEN -x SECRETVALUE1 SECRETVALUE2 ... SECRETVALUEN
+    you must have a personal github token to set the first time that uses this tool
+    $ github-automation set-secret -r OWNER/NAME1 OWNER/NAME2 ... OWNER/NAMEn --secrets SECRET_NAME1:SECRET_VALUE_1 SECRET_NAME2:SECRET_VALUE_2 ... SECRET_NAMEN:SECRET_VALUE_N
+    $ github-automation set-secret -r OWNER/NAME1 OWNER/NAME2 ... OWNER/NAMEn -s SECRET_NAME1:SECRET_VALUE_1 SECRET_NAME2 ... SECRET_NAMEN -x SECRETVALUE1 SECRETVALUE2:SECRET_VALUE_2 ... SECRETVALUEN:SECRET_VALUE_N
     `,
   ]
 
-  static usage='set-vars -r REPOS -n NAMES -x VALUES'
+  static usage='set-secret -r REPOS -n NAMES -x VALUES'
 
-  static strict = true
+  static strict = false
 
   static flags = secretVarsFlags
 
   async run(): Promise<void> {
-    const {flags} = await this.parse(SetVars)
-    validateEqualLengths(flags['secret-name'], flags['secret-value'])
-    validateRepoNames(flags.repositories)
-
-    // const token = await getGithubToken(flags.organization)
-
-    const varsToSet = []
-    for (const repo of flags.repositories) {
-      for (const [index, secret] of flags['secret-value'].entries()) {
-        varsToSet.push(updateVars({value: secret, owner: flags.organization, repo, name: flags['secret-name'][index], environment_name: flags.environment}))
-      }
-    }
-
-    await Promise.all(varsToSet)
-    for (const repo of flags.repositories) {
-      for (const [index, secret] of flags['secret-value'].entries()) {
-        this.log(info(`Updated var ${flags['secret-name'][index]} with value ${secret} in org: ${flags.organization} in repo: ${repo}`))
+    const {flags: {organization, repositories, secrets, environment}} = await this.parse(SetVars)
+    validateSecrets(secrets)
+    validateRepoNames(repositories)
+    const octoFactory = repositoryFactory.get('octokit')
+    for (const repo of repositories) {
+      this.log(info(`Updating secrets in org: ${organization} in repo: ${repo}`))
+      for (const secret of secrets) {
+        const [name, value] = secret.split(':')
+        this.log(info(`Updating variables ${name} with value ${value} in org: ${organization} in repo: ${repo} ${environment ? `in environment: ${environment}` : ''}`))
+        await octoFactory.updateVariables({owner: organization, repo, name, value, environment})
+        this.log(info(`Updated variable ${name} with value ${value} in org: ${organization} in repo: ${repo} ${environment ? `in environment: ${environment}` : ''}`))
       }
     }
   }

src/commands/set-vars/index.ts

@@ -285,5 +285,31 @@ export default {
       },
     })
   },
+  async updateVariables({owner, repo, name, value, environment}:{owner:string, repo:string, name:string, value:string, environment?:string}):Promise<postVariableResponse|patchVariableResponse> {
+    if (environment) {
+      const {data: {environments}} = await this.getEnvironments({organization: owner, repository: repo})
+      if (!environments?.find(env => env.name === environment)) {
+        const confirm = await ux.confirm('The environment does not exist. Would you like to create it? (yes/no)')
+        if (confirm) {
+          await this.defineEnvironment({owner, repo, environment_name: environment})
+        } else {
+          throw new Error(`Environment ${environment} does not exist`)
+        }
+      }
 
+      try {
+        await this.getEnvironmentVariable({owner, repo, name, environment_name: environment})
+        return this.patchEnvironmentVariable({owner, repo, name, value, environment_name: environment})
+      } catch  {
+        return this.setEnvironmentVariable({owner, repo, name, value, environment_name: environment})
+      }
+    }
+
+    try {
+      await this.getGlobalVariable({owner, repo, name})
+      return this.patchGlobalVariable({owner, repo, name, value})
+    } catch  {
+      return this.setGlobalVariable({owner, repo, name, value})
+    }
+  },
 }