From 1b43a5f01bf231d7f4935d62b0d7581b080f94d1 Mon Sep 17 00:00:00 2001
From: allohamora <herman30698@gmail.com>
Date: Mon, 10 Apr 2023 18:12:05 +0300
Subject: [PATCH] feat: add atLeastOne to for methods

---
 src/env/env.manager.ts | 12 ++++++------
 src/env/env.picker.ts  |  5 ++++-
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/env/env.manager.ts b/src/env/env.manager.ts
index 66572ca..61ed0a7 100644
--- a/src/env/env.manager.ts
+++ b/src/env/env.manager.ts
@@ -1,5 +1,5 @@
 import { Manager } from 'src/utils/manager.utils.js';
-import { BaseConfig, EnvPicker } from './env.picker.js';
+import { BaseConfig, AtLeastOne, EnvPicker } from './env.picker.js';
 
 interface Options<E, C> {
   load?: () => C;
@@ -29,7 +29,7 @@ export class EnvManager<E extends string, C extends BaseConfig = BaseConfig> ext
     return value;
   }
 
-  private getKeyOrThrow<K extends keyof C>(envRecord: Partial<Record<E, K>>): K {
+  private getKeyOrThrow<K extends keyof C>(envRecord: AtLeastOne<Record<E, K>>): K {
     const key = envRecord[this.nodeEnv as E] as string | undefined;
 
     if (!key) {
@@ -57,7 +57,7 @@ export class EnvManager<E extends string, C extends BaseConfig = BaseConfig> ext
     return this.pick(key) as EnvPicker<E, C[K]>;
   }
 
-  public pickFor<K extends keyof C>(envRecord: Partial<Record<E, K>>): undefined | EnvPicker<E, C[K] | undefined> {
+  public pickFor<K extends keyof C>(envRecord: AtLeastOne<Record<E, K>>): undefined | EnvPicker<E, C[K] | undefined> {
     const key = envRecord[this.nodeEnv as E] as string | undefined;
 
     if (!key) {
@@ -67,13 +67,13 @@ export class EnvManager<E extends string, C extends BaseConfig = BaseConfig> ext
     return this.pick(key) as EnvPicker<E, C[K] | undefined>;
   }
 
-  public pickForOrThrow<K extends keyof C>(envRecord: Partial<Record<E, K>>): EnvPicker<E, C[K]> {
+  public pickForOrThrow<K extends keyof C>(envRecord: AtLeastOne<Record<E, K>>): EnvPicker<E, C[K]> {
     const key = this.getKeyOrThrow(envRecord);
 
     return this.pickOrThrow(key as K);
   }
 
-  public getFor<K extends keyof C>(envRecord: Partial<Record<E, K>>) {
+  public getFor<K extends keyof C>(envRecord: AtLeastOne<Record<E, K>>) {
     const key = envRecord[this.nodeEnv as E] as string | undefined;
 
     if (!key) {
@@ -83,7 +83,7 @@ export class EnvManager<E extends string, C extends BaseConfig = BaseConfig> ext
     return this.get(key as K);
   }
 
-  public getForOrThrow<K extends keyof C>(envRecord: Partial<Record<E, K>>) {
+  public getForOrThrow<K extends keyof C>(envRecord: AtLeastOne<Record<E, K>>) {
     const key = this.getKeyOrThrow(envRecord);
 
     return this.getOrThrow(key as K);
diff --git a/src/env/env.picker.ts b/src/env/env.picker.ts
index c2cd8dc..e2b0b15 100644
--- a/src/env/env.picker.ts
+++ b/src/env/env.picker.ts
@@ -12,6 +12,9 @@ type MoveOptional<T, R> = T extends null
 
 export type BaseConfig = Record<string, string | undefined>;
 
+// original https://stackoverflow.com/a/59987826/15681288
+export type AtLeastOne<T> = { [K in keyof T]: Pick<T, K> }[keyof T];
+
 export const wrapInEnvPickers = <E extends string, C extends BaseConfig>(config: C, nodeEnv: E) => {
   return Object.keys(config).reduce((result, key) => {
     return { ...result, [key]: new EnvPicker(config[key], nodeEnv) };
@@ -27,7 +30,7 @@ export class EnvPicker<E extends string, S> {
     return this as unknown as EnvPicker<E, NS | NonNullable<S>>;
   }
 
-  public defaultFor(envRecord: Partial<Record<E, S>>) {
+  public defaultFor(envRecord: AtLeastOne<Record<E, S>>) {
     this.state ??= envRecord[this.nodeEnv] as S;
 
     return this;