feat: add secret command

zxypro1 · zxypro1 · commit f1bc8d20b0ba · 2024-07-29T17:24:42.000+08:00
Signed-off-by: zxypro1 &lt;1018995004@qq.com&gt;
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -66,12 +66,13 @@
   "devDependencies": {
     "@serverless-devs/core": "^0.1.67-beta.1",
     "@serverless-devs/credential": "^0.0.9-beta.1",
-    "@serverless-devs/engine": "^0.1.4-beta.8",
-    "@serverless-devs/load-application": "^0.0.14-beta.5",
+    "@serverless-devs/engine": "^0.1.4-beta.10",
+    "@serverless-devs/load-application": "^0.0.14-beta.6",
     "@serverless-devs/load-component": "^0.0.8-beta.1",
     "@serverless-devs/logger": "^0.0.5",
-    "@serverless-devs/parse-spec": "^0.0.28-beta.7",
+    "@serverless-devs/parse-spec": "^0.0.28-beta.9",
     "@serverless-devs/registry": "^0.0.10",
+    "@serverless-devs/secret": "^0.0.2-beta.4",
     "@serverless-devs/utils": "^0.0.16-beta.1",
     "@types/fs-extra": "^11.0.4",
     "@types/inquirer": "^9.0.7",
diff --git a/src/command/index.ts b/src/command/index.ts
@@ -15,7 +15,8 @@ const commandDict = {
   preview: 'preview/index.ts',
   component: 'component/index.ts',
   verify: 'verify/index.ts',
-  cli: 'cli/index.ts'
+  cli: 'cli/index.ts',
+  secret: 'secret/index.ts',
 };
 
 const root = async (program: Command) => {
diff --git a/src/command/secret/command/add.ts b/src/command/secret/command/add.ts
@@ -0,0 +1,36 @@
+import { Command } from 'commander';
+import chalk from 'chalk';
+import { emoji } from '@/utils';
+import { getSecretManager } from '../utils';
+import logger from '@/logger';
+
+const description = `Add a secret or overwrite a secret.
+
+    Example:
+        $ s secret add --key demo --value demo
+        $ s secret add --key demo --value demo -f
+
+${emoji('📖')} Document: Temporarily unavailable`;
+
+export default (program: Command) => {
+  const command = program.command('add');
+
+  command
+    .usage('[options]')
+    .description(description)
+    .summary(`Add a secret`)
+    .requiredOption('--key <key>', 'Secret key')
+    .requiredOption('--value <value>', 'Secret value')
+    .option('-f, --force', 'Mandatory overwrite key information')
+    .helpOption('-h, --help', 'Display help for command')
+    .action(async options => {
+      const secretManager = getSecretManager();
+      const value = secretManager.getSecret(options.key);
+      if (value && !options.force) {
+        logger.warn(`The secret ${options.key} already exists, please use --force to overwrite`);
+        return;
+      }
+      secretManager.addSecret(options.key, options.value);
+      logger.write(chalk.green(`Secret [${options.key}] has been successfully added.`));
+    });
+};
diff --git a/src/command/secret/command/get.ts b/src/command/secret/command/get.ts
@@ -0,0 +1,41 @@
+import { Command } from 'commander';
+import chalk from 'chalk';
+import { emoji } from '@/utils';
+import { getSecretManager } from '../utils';
+import { handleSecret } from '@/command/config/utils';
+import logger from '@/logger';
+import { isEmpty } from 'lodash';
+import jsYaml from 'js-yaml';
+
+const description = `Preview Secrets.
+ 
+  Example:
+    $ s secret get
+    $ s secret get --key demo
+    
+${emoji('📖')} Document: Temporarily unavailable`;
+
+export default (program: Command) => {
+  const command = program.command('get');
+
+  command
+    .usage('[options]')
+    .description(description)
+    .summary(`Preview secrets`)
+    .option('--key <key>', 'Secret key')
+    .helpOption('-h, --help', 'Display help for command')
+    .action(async options => {
+      const { key } = options;
+      const secretManager = getSecretManager();
+      const secrets = handleSecret(secretManager.getAllSecrets());
+      if (isEmpty(key)) {
+        logger.write(jsYaml.dump(secrets));
+      } else {
+        if (!secrets[key]) {
+          logger.error(`Secret [${key}] does not exist.`);
+          return;
+        }
+        logger.write(jsYaml.dump(secrets[key]));
+      }
+    });
+};
diff --git a/src/command/secret/command/remove.ts b/src/command/secret/command/remove.ts
@@ -0,0 +1,33 @@
+import { Command } from 'commander';
+import chalk from 'chalk';
+import { emoji } from '@/utils';
+import logger from '@/logger';
+import { getSecretManager } from '../utils';
+
+const description = `Delete a secret.
+  
+  Example:
+    $ s secret delete --key demo
+    
+${emoji('📖')} Document: Temporarily unavailable`;
+
+export default (program: Command) => {
+  const command = program.command('delete');
+
+  command
+    .usage('[options]')
+    .description(description)
+    .summary(`Delete a secret`)
+    .requiredOption('--key <key>', 'Secret key')
+    .helpOption('-h, --help', 'Display help for command')
+    .action(async options => {
+      const { key } = options;
+      const secretManager = getSecretManager();
+      try {
+        secretManager.deleteSecret(key);
+        logger.write(chalk.green(`Secret [${key}] has been successfully deleted.`));
+      } catch (error) {
+        logger.error(error);
+      }
+    });
+};
diff --git a/src/command/secret/index.ts b/src/command/secret/index.ts
@@ -0,0 +1,22 @@
+import { Command } from 'commander';
+import { emoji, mount, suggestCommand } from '@/utils';
+import chalk from 'chalk';
+
+const description = `Secret Manager is used to manage the secret of cloud vendors.
+
+${emoji('📖')} Document: Temporarily unavailable`;
+
+export default (program: Command) => {
+  const configProgram = program.command('secret');
+  suggestCommand(configProgram);
+  configProgram
+    .description(description)
+    .summary(`Manage the secret of cloud vendors`)
+    .usage('[commands] [options]')
+    .addHelpCommand(false)
+    .helpOption('-h, --help', 'Display help for command');
+
+  mount('secret/command/add.ts', configProgram);
+  mount('secret/command/get.ts', configProgram);
+  mount('secret/command/remove.ts', configProgram);
+};
diff --git a/src/command/secret/utils/index.ts b/src/command/secret/utils/index.ts
@@ -0,0 +1,3 @@
+import SecretManager from '@serverless-devs/secret';
+
+export const getSecretManager = () => SecretManager.getInstance();

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+import SecretManager from '@serverless-devs/secret';`
	`2`	`+`
	`3`	`+export const getSecretManager = () => SecretManager.getInstance();`