feat: 'secrets edit' now makes a new secret if it doesnt exist already

MatrixAI · Aug 26, 2024 · 064b69a · 064b69a
1 parent 755e97f
commit 064b69a
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 49 deletions.
diff --git a/flake.lock b/flake.lock
diff --git a/src/secrets/CommandEdit.ts b/src/secrets/CommandEdit.ts
@@ -1,8 +1,9 @@
 import type PolykeyClient from 'polykey/dist/PolykeyClient';
 import fs from 'fs';
 import path from 'path';
-import * as errors from '../errors';
 import CommandPolykey from '../CommandPolykey';
+import * as vaultsErrors from 'polykey/dist/vaults/errors';
+import * as errors from '../errors';
 import * as binUtils from '../utils';
 import * as binOptions from '../utils/options';
 import * as binParsers from '../utils/parsers';
@@ -54,15 +55,32 @@ class CommandEdit extends CommandPolykey {
           },
           logger: this.logger.getChild(PolykeyClient.name),
         });
-        const response = await binUtils.retryAuthentication(
-          (auth) =>
-            pkClient.rpcClient.methods.vaultsSecretsGet({
+        const response = await binUtils.retryAuthentication((auth) => {
+          try {
+            pkClient.rpcClient.methods.vaultsSecretsStat({
               metadata: auth,
               nameOrId: secretPath[0],
               secretName: secretPath[1],
-            }),
-          meta,
-        );
+            });
+          } catch (e) {
+            console.log(e.cause);
+            if (e.cause === vaultsErrors.ErrorSecretsSecretUndefined) {
+              pkClient.rpcClient.methods.vaultsSecretsNew({
+                metadata: auth,
+                nameOrId: secretPath[0],
+                secretName: secretPath[1],
+                secretContent: '',
+              });
+            }
+            else throw e;
+          }
+
+          return pkClient.rpcClient.methods.vaultsSecretsGet({
+            metadata: auth,
+            nameOrId: secretPath[0],
+            secretName: secretPath[1],
+          });
+        }, meta);
         const secretContent = response.secretContent;
         const tmpDir = await fs.promises.mkdtemp(
           path.join(os.tmpdir(), 'polykey-'),