🎵

src/index.ts

@@ -2,5 +2,5 @@ export * from './generators'
 export * from './random'
 export { default } from './random'
 export * from './rng'
-export * from './rng-factory'
 export type * from './types'
+export * from './utils'

src/random.ts

@@ -14,7 +14,7 @@ import { uniform } from './distributions/uniform'
 import { uniformBoolean } from './distributions/uniform-boolean'
 import { uniformInt } from './distributions/uniform-int'
 import { MathRandomRNG } from './generators/math-random'
-import { RNGFactory } from './rng-factory'
+import { createRNG } from './utils'
 
 /**
  * Distribution function
@@ -47,8 +47,9 @@ export class Random {
   protected readonly _cache: {
     [k: string]: ICacheEntry<any>
   } = {}
+
   constructor(seedOrRNG: SeedOrRNG = new MathRandomRNG()) {
-    this._rng = RNGFactory(seedOrRNG)
+    this._rng = createRNG(seedOrRNG)
   }
 
   /**
@@ -81,7 +82,7 @@ export class Random {
    * ```
    */
   use(seedOrRNG: SeedOrRNG) {
-    this._rng = RNGFactory(seedOrRNG)
+    this._rng = createRNG(seedOrRNG)
   }
 
   // --------------------------------------------------------------------------

src/utils.ts

@@ -1,4 +1,25 @@
-import type { Seed } from './types'
+import type { Seed, SeedOrRNG } from './types'
+import { ARC4RNG } from './generators/arc4'
+import { FunctionRNG } from './generators/function'
+import { RNG } from './rng'
+
+export function createRNG(seedOrRNG?: SeedOrRNG) {
+  switch (typeof seedOrRNG) {
+    case 'object':
+      if (seedOrRNG instanceof RNG) {
+        return seedOrRNG
+      }
+      break
+
+    case 'function':
+      return new FunctionRNG(seedOrRNG)
+
+    default:
+      return new ARC4RNG(seedOrRNG)
+  }
+
+  throw new Error(`invalid RNG seed or instance "${seedOrRNG}"`)
+}
 
 export function processSeed(seed?: Seed): number {
   if (seed === undefined) {