add option to write back to original environment object (#26)

arundo · Apr 10, 2024 · 6d2d2ab · 6d2d2ab
1 parent 1ac29b9
commit 6d2d2ab
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/.changeset/lemon-actors-sneeze.md b/.changeset/lemon-actors-sneeze.md
@@ -0,0 +1,5 @@
+---
+"@arundo/typed-env": minor
+---
+
+add option to write back to original environment object
diff --git a/src/contracts.ts b/src/contracts.ts
@@ -7,6 +7,7 @@ export type Options<TTransform, TPrefixRemoval> = {
   transform?: TTransform;
   constructErrorFn?: (issues: ZodIssue[]) => Error;
   excludePrefix?: TPrefixRemoval;
+  writeBackToEnv?: boolean;
 };
 
 export type ConditionalType<TTransform extends NamingConvention | undefined, TSchema> = 'default' extends TTransform

diff --git a/src/index.ts b/src/index.ts
@@ -80,10 +80,16 @@ export const typeEnvironment = <
     transform = 'default',
     constructErrorFn: constructErrorFn = constructError,
     excludePrefix = '' as TPrefixRemoval,
+    writeBackToEnv = false,
   } = options;
 
   try {
     const parsed = schema.parse(overrideEnv);
+    if (writeBackToEnv) {
+      for (const key of Object.keys(parsed)) {
+        overrideEnv[key] = typeof parsed[key] === 'string' ? parsed[key] : JSON.stringify(parsed[key]);
+      }
+    }
     type TSchemaOutput = TSchema['_output'];
     const prefixRemoved = removePrefix(parsed, excludePrefix) as PrefixRemoved<TSchemaOutput, TPrefixRemoval>;
     return changeCase(transform, prefixRemoved) as ConditionalType<