feat: add option to force infisical select env

anduintransaction · Nov 22, 2023 · 0206a7d · 0206a7d
1 parent b3a9dfe
commit 0206a7d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/js/package.json b/js/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@anduintransaction/rivendell",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/anduintransaction/rivendell.git"

diff --git a/js/src/secrets/infisical.ts b/js/src/secrets/infisical.ts
@@ -3,20 +3,31 @@ import { SecretProvider } from "../context";
 
 export class InfisicalSecretProvider implements SecretProvider {
   client: InfisicalClient;
+  forceEnv: string | undefined;
 
-  constructor(siteURL: string, token: string, debug: boolean = false) {
+  constructor(
+    siteURL: string,
+    token: string,
+    debug: boolean = false,
+    forceEnv?: string,
+  ) {
+    this.forceEnv = forceEnv;
     this.client = new InfisicalClient({
       token: token,
       siteURL: siteURL,
       debug: debug,
     });
   }
 
+  getTargetEnv(env: string): string {
+    return this.forceEnv || env;
+  }
+
   async get(env: string, name: string) {
     const parts = name.split("/");
     const [secretName, ...paths] = parts.reverse();
     const secret = await this.client.getSecret(secretName, {
-      environment: env,
+      environment: this.getTargetEnv(env),
       path: `/${paths.reverse().join("/")}`,
       type: "shared",
     });
@@ -28,7 +39,7 @@ export class InfisicalSecretProvider implements SecretProvider {
 
   async getPrefix(env: string, prefix: string) {
     const secrets = await this.client.getAllSecrets({
-      environment: env,
+      environment: this.getTargetEnv(env),
       path: `/${prefix}`,
       attachToProcessEnv: false,
       includeImports: false,