probability-picker 1.2.0 → 2.0.0

package/README.md

@@ -1,5 +1,6 @@
-# Probability Picker
-A lightweight library for selecting random values based on weighted probabilities.
+# probability-picker
+
+A small, type-safe utility for random selection based on weighted probabilities. It handles simple maps, nested objects, and selection without replacement.
 
 ## Installation
 
@@ -7,20 +8,65 @@ A lightweight library for selecting random values based on weighted probabilitie
 npm i probability-picker
 ```
 
-## Usage
+## Basic Usage
+
+The `picker` function takes an object where keys are your options and values are their weights.
 
-The `probabilityPicker` function will return `a`, `b`, or `c`. The probability of each one is defined as a value in the object.
-    
-```javascript
-import picker from 'probability-picker';
+```typescript
+import { picker } from 'probability-picker';
 
-const value = picker({
-    a: 10,
-    b: 70,
-    c: 20,
-}); 
+const drop = picker({
+  common: 80,
+  rare: 15,
+  legendary: 5
+}).one();
 ```
 
-## License
+## Advanced Features
+
+### Picking multiple items
+Use `.take(n)` to get an array of results.
+
+```typescript
+const items = picker({ grass: 95, clover: 5 }).take(5);
+// returns something like ['grass', 'grass', 'clover', 'grass', 'grass']
+```
+
+### Luck Modifier
+The `luck()` method adjusts probabilities using an exponential curve. A luck value > 1 increases the weight of all items (favoring rarer ones relatively), while < 1 does the opposite.
+
+```typescript
+const loot = picker({ sword: 1, gold: 99 })
+  .luck(5)
+  .one();
+```
+
+### Nested Maps
+If a value is another object, the picker will recursively select until it hits a leaf node (a string key).
 
-[MIT](LICENSE)
+```typescript
+const map = {
+  weapons: {
+    sword: 10,
+    axe: 5
+  },
+  consumables: 85
+};
+
+const result = picker(map).one(); // returns 'sword', 'axe', or 'consumables'
+```
+
+### The "Bag" System
+If you need to pick items without repeating them until the set is exhausted, use `bag`.
+
+```typescript
+import { bag } from 'probability-picker';
+
+const deck = bag({ Ace: 1, King: 1, Queen: 1 });
+
+console.log(deck.next()); // 'King'
+console.log(deck.next()); // 'Ace' (won't repeat until the deck is empty)
+```
+
+## License
+MIT

package/dist/index.cjs

@@ -1,59 +1,83 @@
 'use strict';
 
-// src/index.ts
-function filterNumbers(entries) {
-  return entries.filter((e) => typeof e[1] === "number" && !Number.isNaN(e[1]));
-}
-function sortEntries(entries) {
-  return entries.sort((a, b) => a[1] - b[1]);
-}
-function sumProbabilities(entries) {
-  return entries.reduce((acc, e) => acc + e[1], 0);
-}
-function normalizeProbabilities(entries, total) {
-  return entries.map(([k, v]) => [k, Math.round(v / total * 100)]);
-}
-function prepareEntries(entries) {
-  const filtered = filterNumbers(entries);
-  if (filtered.length === 0) return;
-  const sorted = sortEntries(filtered);
-  const sum = sumProbabilities(sorted);
-  if (sum === 0) return;
-  if (sum === 100) return sorted;
-  return normalizeProbabilities(sorted, sum);
-}
-function secureRandom() {
-  if (typeof globalThis.crypto === "undefined" || typeof globalThis.crypto.getRandomValues !== "function") {
-    return Math.random();
-  }
+// src/utils.ts
+var secureRandom = () => {
   const array = new Uint32Array(1);
   globalThis.crypto.getRandomValues(array);
   return array[0] / 4294967296;
-}
-function random(min, max) {
-  return secureRandom() * (max - min) + min;
-}
-function chooseOne(entries) {
-  const num = random(1, 100);
-  let count = 0;
-  for (const entry of entries) {
-    count += entry[1];
-    if (num <= count) return entry[0];
+};
+var normalize = (map, luck) => {
+  const valid = [];
+  for (const [key, value] of Object.entries(map)) {
+    const weight = typeof value === "number" ? value : 1;
+    if (!Number.isNaN(weight) && weight > 0) {
+      valid.push([key, Math.pow(weight, 1 / luck)]);
+    }
   }
-  return entries[entries.length - 1][0];
+  return valid;
+};
+var select = (items) => {
+  const total = items.reduce((acc, [, v]) => acc + v, 0);
+  let target = secureRandom() * total;
+  for (const [k, v] of items) {
+    target -= v;
+    if (target <= 0) return k;
+  }
+  return items[items.length - 1][0];
+};
+
+// src/picker.ts
+var ProbabilityEngine = class _ProbabilityEngine {
+  _luck = 1;
+  _map;
+  constructor(map) {
+    this._map = map;
+  }
+  luck(value) {
+    this._luck = value;
+    return this;
+  }
+  one() {
+    const entries = normalize(this._map, this._luck);
+    if (entries.length === 0) return;
+    const key = select(entries);
+    const value = this._map[key];
+    if (value && typeof value === "object" && !Array.isArray(value)) {
+      return new _ProbabilityEngine(value).luck(this._luck).one();
+    }
+    return key;
+  }
+  take(n) {
+    const results = [];
+    for (let i = 0; i < n; i++) {
+      const picked = this.one();
+      if (picked) results.push(picked);
+    }
+    return results;
+  }
+};
+function picker(map) {
+  if (!map || typeof map !== "object") {
+    return new ProbabilityEngine({});
+  }
+  return new ProbabilityEngine(map);
 }
-function probabilityPicker(map) {
-  if (!(map instanceof Object)) return null;
-  const entries = Object.entries(map);
-  if (entries.length === 0) return void 0;
-  const first = entries[0];
-  if (first && entries.length === 1) return first[0];
-  const prepared = prepareEntries(entries);
-  if (!prepared) return void 0;
-  return chooseOne(prepared);
+
+// src/bag.ts
+function bag(map) {
+  let remaining = { ...map };
+  return {
+    next() {
+      const keys = Object.keys(remaining);
+      if (keys.length === 0) remaining = { ...map };
+      const picked = picker(remaining).one();
+      if (picked) delete remaining[picked];
+      return picked;
+    }
+  };
 }
-var index_default = probabilityPicker;
 
-module.exports = index_default;
+exports.bag = bag;
+exports.picker = picker;
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts"],"names":[],"mappings":";;;AAGA,SAAS,cAAc,OAAA,EAAkB;AACrC,EAAA,OAAO,OAAA,CAAQ,MAAA,CAAO,CAAA,CAAA,KAAK,OAAO,EAAE,CAAC,CAAA,KAAM,QAAA,IAAY,CAAC,MAAA,CAAO,KAAA,CAAM,CAAA,CAAE,CAAC,CAAC,CAAC,CAAA;AAC9E;AAEA,SAAS,YAAY,OAAA,EAAkB;AACnC,EAAA,OAAO,OAAA,CAAQ,IAAA,CAAK,CAAC,CAAA,EAAG,CAAA,KAAM,EAAE,CAAC,CAAA,GAAI,CAAA,CAAE,CAAC,CAAC,CAAA;AAC7C;AAEA,SAAS,iBAAiB,OAAA,EAAkB;AACxC,EAAA,OAAO,OAAA,CAAQ,OAAO,CAAC,GAAA,EAAK,MAAM,GAAA,GAAM,CAAA,CAAE,CAAC,CAAA,EAAG,CAAC,CAAA;AACnD;AAEA,SAAS,sBAAA,CAAuB,SAAkB,KAAA,EAAe;AAC7D,EAAA,OAAO,OAAA,CAAQ,GAAA,CAAI,CAAC,CAAC,GAAG,CAAC,CAAA,KAAM,CAAC,CAAA,EAAG,KAAK,KAAA,CAAO,CAAA,GAAI,KAAA,GAAS,GAAG,CAAC,CAAU,CAAA;AAC9E;AAEA,SAAS,eAAe,OAAA,EAAkB;AACtC,EAAA,MAAM,QAAA,GAAW,cAAc,OAAO,CAAA;AACtC,EAAA,IAAI,QAAA,CAAS,WAAW,CAAA,EAAG;AAC3B,EAAA,MAAM,MAAA,GAAS,YAAY,QAAQ,CAAA;AACnC,EAAA,MAAM,GAAA,GAAM,iBAAiB,MAAM,CAAA;AACnC,EAAA,IAAI,QAAQ,CAAA,EAAG;AACf,EAAA,IAAI,GAAA,KAAQ,KAAK,OAAO,MAAA;AACxB,EAAA,OAAO,sBAAA,CAAuB,QAAQ,GAAG,CAAA;AAC7C;AAEA,SAAS,YAAA,GAAuB;AAC5B,EAAA,IAAI,OAAO,WAAW,MAAA,KAAW,WAAA,IAAe,OAAO,UAAA,CAAW,MAAA,CAAO,oBAAoB,UAAA,EAAY;AACrG,IAAA,OAAO,KAAK,MAAA,EAAO;AAAA,EACvB;AACA,EAAA,MAAM,KAAA,GAAQ,IAAI,WAAA,CAAY,CAAC,CAAA;AAC/B,EAAA,UAAA,CAAW,MAAA,CAAO,gBAAgB,KAAK,CAAA;AACvC,EAAA,OAAO,KAAA,CAAM,CAAC,CAAA,GAAK,UAAA;AACvB;AAEA,SAAS,MAAA,CAAO,KAAa,GAAA,EAAa;AACtC,EAAA,OAAO,YAAA,EAAa,IAAK,GAAA,GAAM,GAAA,CAAA,GAAO,GAAA;AAC1C;AAEA,SAAS,UAAU,OAAA,EAAkB;AACjC,EAAA,MAAM,GAAA,GAAM,MAAA,CAAO,CAAA,EAAG,GAAG,CAAA;AACzB,EAAA,IAAI,KAAA,GAAQ,CAAA;AACZ,EAAA,KAAA,MAAW,SAAS,OAAA,EAAS;AACzB,IAAA,KAAA,IAAS,MAAM,CAAC,CAAA;AAChB,IAAA,IAAI,GAAA,IAAO,KAAA,EAAO,OAAO,KAAA,CAAM,CAAC,CAAA;AAAA,EACpC;AACA,EAAA,OAAO,OAAA,CAAQ,OAAA,CAAQ,MAAA,GAAS,CAAC,EAAG,CAAC,CAAA;AACzC;AAEA,SAAS,kBAAkB,GAAA,EAAgD;AACvE,EAAA,IAAI,EAAE,GAAA,YAAe,MAAA,CAAA,EAAS,OAAO,IAAA;AACrC,EAAA,MAAM,OAAA,GAAU,MAAA,CAAO,OAAA,CAAQ,GAAG,CAAA;AAClC,EAAA,IAAI,OAAA,CAAQ,MAAA,KAAW,CAAA,EAAG,OAAO,MAAA;AACjC,EAAA,MAAM,KAAA,GAAQ,QAAQ,CAAC,CAAA;AACvB,EAAA,IAAI,SAAS,OAAA,CAAQ,MAAA,KAAW,CAAA,EAAG,OAAO,MAAM,CAAC,CAAA;AACjD,EAAA,MAAM,QAAA,GAAW,eAAe,OAAO,CAAA;AACvC,EAAA,IAAI,CAAC,UAAU,OAAO,MAAA;AACtB,EAAA,OAAO,UAAU,QAAQ,CAAA;AAC7B;AAEA,IAAO,aAAA,GAAQ","file":"index.cjs","sourcesContent":["type ProbabilityMap = Record<string, number>\r\ntype Entry = [string, number]\r\n\r\nfunction filterNumbers(entries: Entry[]) {\r\n    return entries.filter(e => typeof e[1] === 'number' && !Number.isNaN(e[1]))\r\n}\r\n\r\nfunction sortEntries(entries: Entry[]) {\r\n    return entries.sort((a, b) => a[1] - b[1])\r\n}\r\n\r\nfunction sumProbabilities(entries: Entry[]) {\r\n    return entries.reduce((acc, e) => acc + e[1], 0)\r\n}\r\n\r\nfunction normalizeProbabilities(entries: Entry[], total: number) {\r\n    return entries.map(([k, v]) => [k, Math.round((v / total) * 100)] as Entry)\r\n}\r\n\r\nfunction prepareEntries(entries: Entry[]) {\r\n    const filtered = filterNumbers(entries)\r\n    if (filtered.length === 0) return\r\n    const sorted = sortEntries(filtered)\r\n    const sum = sumProbabilities(sorted)\r\n    if (sum === 0) return\r\n    if (sum === 100) return sorted\r\n    return normalizeProbabilities(sorted, sum)\r\n}\r\n\r\nfunction secureRandom(): number {\r\n    if (typeof globalThis.crypto === 'undefined' || typeof globalThis.crypto.getRandomValues !== 'function') {\r\n        return Math.random()\r\n    }\r\n    const array = new Uint32Array(1)\r\n    globalThis.crypto.getRandomValues(array)\r\n    return array[0]! / 0x100000000\r\n}\r\n\r\nfunction random(min: number, max: number) {\r\n    return secureRandom() * (max - min) + min\r\n}\r\n\r\nfunction chooseOne(entries: Entry[]) {\r\n    const num = random(1, 100)\r\n    let count = 0\r\n    for (const entry of entries) {\r\n        count += entry[1]\r\n        if (num <= count) return entry[0]\r\n    }\r\n    return entries[entries.length - 1]![0]\r\n}\r\n\r\nfunction probabilityPicker(map: ProbabilityMap): string | null | undefined {\r\n    if (!(map instanceof Object)) return null\r\n    const entries = Object.entries(map) as Entry[]\r\n    if (entries.length === 0) return undefined\r\n    const first = entries[0]\r\n    if (first && entries.length === 1) return first[0]\r\n    const prepared = prepareEntries(entries)\r\n    if (!prepared) return undefined\r\n    return chooseOne(prepared)\r\n}\r\n\r\nexport default probabilityPicker;"]}
+{"version":3,"sources":["../src/utils.ts","../src/picker.ts","../src/bag.ts"],"names":[],"mappings":";;;AAEO,IAAM,eAAe,MAAc;AACtC,EAAA,MAAM,KAAA,GAAQ,IAAI,WAAA,CAAY,CAAC,CAAA;AAC/B,EAAA,UAAA,CAAW,MAAA,CAAO,gBAAgB,KAAK,CAAA;AACvC,EAAA,OAAO,KAAA,CAAM,CAAC,CAAA,GAAK,UAAA;AACvB,CAAA;AAEO,IAAM,SAAA,GAAY,CAAC,GAAA,EAAgB,IAAA,KAAqC;AAC3E,EAAA,MAAM,QAA4B,EAAC;AAEnC,EAAA,KAAA,MAAW,CAAC,GAAA,EAAK,KAAK,KAAK,MAAA,CAAO,OAAA,CAAQ,GAAG,CAAA,EAAG;AAC5C,IAAA,MAAM,MAAA,GAAS,OAAO,KAAA,KAAU,QAAA,GAAW,KAAA,GAAQ,CAAA;AACnD,IAAA,IAAI,CAAC,MAAA,CAAO,KAAA,CAAM,MAAM,CAAA,IAAK,SAAS,CAAA,EAAG;AACrC,MAAA,KAAA,CAAM,IAAA,CAAK,CAAC,GAAA,EAAK,IAAA,CAAK,IAAI,MAAA,EAAQ,CAAA,GAAI,IAAI,CAAC,CAAC,CAAA;AAAA,IAChD;AAAA,EACJ;AAEA,EAAA,OAAO,KAAA;AACX,CAAA;AAEO,IAAM,MAAA,GAAS,CAAC,KAAA,KAAsC;AACzD,EAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,MAAA,CAAO,CAAC,GAAA,EAAK,GAAG,CAAC,CAAA,KAAM,GAAA,GAAM,CAAA,EAAG,CAAC,CAAA;AACrD,EAAA,IAAI,MAAA,GAAS,cAAa,GAAI,KAAA;AAE9B,EAAA,KAAA,MAAW,CAAC,CAAA,EAAG,CAAC,CAAA,IAAK,KAAA,EAAO;AACxB,IAAA,MAAA,IAAU,CAAA;AACV,IAAA,IAAI,MAAA,IAAU,GAAG,OAAO,CAAA;AAAA,EAC5B;AAEA,EAAA,OAAO,KAAA,CAAM,KAAA,CAAM,MAAA,GAAS,CAAC,EAAG,CAAC,CAAA;AACrC,CAAA;;;AC5BA,IAAM,iBAAA,GAAN,MAAM,kBAAA,CAAiE;AAAA,EAC3D,KAAA,GAAQ,CAAA;AAAA,EACR,IAAA;AAAA,EAER,YAAY,GAAA,EAAgB;AACxB,IAAA,IAAA,CAAK,IAAA,GAAO,GAAA;AAAA,EAChB;AAAA,EAEA,KAAK,KAAA,EAAe;AAChB,IAAA,IAAA,CAAK,KAAA,GAAQ,KAAA;AACb,IAAA,OAAO,IAAA;AAAA,EACX;AAAA,EAEA,GAAA,GAAqB;AACjB,IAAA,MAAM,OAAA,GAAU,SAAA,CAAU,IAAA,CAAK,IAAA,EAAM,KAAK,KAAK,CAAA;AAC/C,IAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AAE1B,IAAA,MAAM,GAAA,GAAM,OAAO,OAAO,CAAA;AAC1B,IAAA,MAAM,KAAA,GAAQ,IAAA,CAAK,IAAA,CAAK,GAAG,CAAA;AAE3B,IAAA,IAAI,KAAA,IAAS,OAAO,KAAA,KAAU,QAAA,IAAY,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AAC7D,MAAA,OAAO,IAAI,mBAAkB,KAAK,CAAA,CAAE,KAAK,IAAA,CAAK,KAAK,EAAE,GAAA,EAAI;AAAA,IAC7D;AAEA,IAAA,OAAO,GAAA;AAAA,EACX;AAAA,EAEA,KAAK,CAAA,EAAgB;AACjB,IAAA,MAAM,UAAe,EAAC;AACtB,IAAA,KAAA,IAAS,CAAA,GAAI,CAAA,EAAG,CAAA,GAAI,CAAA,EAAG,CAAA,EAAA,EAAK;AACxB,MAAA,MAAM,MAAA,GAAS,KAAK,GAAA,EAAI;AACxB,MAAA,IAAI,MAAA,EAAQ,OAAA,CAAQ,IAAA,CAAK,MAAM,CAAA;AAAA,IACnC;AACA,IAAA,OAAO,OAAA;AAAA,EACX;AACJ,CAAA;AAEO,SAAS,OAAyB,GAAA,EAAuD;AAC5F,EAAA,IAAI,CAAC,GAAA,IAAO,OAAO,GAAA,KAAQ,QAAA,EAAU;AACjC,IAAA,OAAO,IAAI,iBAAA,CAAqB,EAAe,CAAA;AAAA,EACnD;AACA,EAAA,OAAO,IAAI,kBAAqB,GAAgB,CAAA;AACpD;;;AC3CO,SAAS,IAAsB,GAAA,EAAwB;AAC1D,EAAA,IAAI,SAAA,GAAY,EAAE,GAAG,GAAA,EAAI;AAEzB,EAAA,OAAO;AAAA,IACH,IAAA,GAAsB;AAClB,MAAA,MAAM,IAAA,GAAO,MAAA,CAAO,IAAA,CAAK,SAAS,CAAA;AAClC,MAAA,IAAI,KAAK,MAAA,KAAW,CAAA,EAAG,SAAA,GAAY,EAAE,GAAG,GAAA,EAAI;AAC5C,MAAA,MAAM,MAAA,GAAS,MAAA,CAAO,SAAS,CAAA,CAAE,GAAA,EAAI;AACrC,MAAA,IAAI,MAAA,EAAQ,OAAO,SAAA,CAAU,MAAW,CAAA;AACxC,MAAA,OAAO,MAAA;AAAA,IACX;AAAA,GACJ;AACJ","file":"index.cjs","sourcesContent":["import type { NestedMap } from './types'\r\n\r\nexport const secureRandom = (): number => {\r\n    const array = new Uint32Array(1)\r\n    globalThis.crypto.getRandomValues(array)\r\n    return array[0]! / 0x100000000\r\n}\r\n\r\nexport const normalize = (map: NestedMap, luck: number): [string, number][] => {\r\n    const valid: [string, number][] = []\r\n\r\n    for (const [key, value] of Object.entries(map)) {\r\n        const weight = typeof value === 'number' ? value : 1\r\n        if (!Number.isNaN(weight) && weight > 0) {\r\n            valid.push([key, Math.pow(weight, 1 / luck)])\r\n        }\r\n    }\r\n\r\n    return valid\r\n}\r\n\r\nexport const select = (items: [string, number][]): string => {\r\n    const total = items.reduce((acc, [, v]) => acc + v, 0)\r\n    let target = secureRandom() * total\r\n\r\n    for (const [k, v] of items) {\r\n        target -= v\r\n        if (target <= 0) return k\r\n    }\r\n\r\n    return items[items.length - 1]![0]\r\n}\r\n","import type { NestedMap, PickerInstance } from './types'\r\nimport { normalize, select } from './utils'\r\n\r\nclass ProbabilityEngine<T extends string> implements PickerInstance<T> {\r\n    private _luck = 1\r\n    private _map: NestedMap\r\n\r\n    constructor(map: NestedMap) {\r\n        this._map = map\r\n    }\r\n\r\n    luck(value: number) {\r\n        this._luck = value\r\n        return this\r\n    }\r\n\r\n    one(): T | undefined {\r\n        const entries = normalize(this._map, this._luck)\r\n        if (entries.length === 0) return\r\n\r\n        const key = select(entries)\r\n        const value = this._map[key]\r\n\r\n        if (value && typeof value === 'object' && !Array.isArray(value)) {\r\n            return new ProbabilityEngine(value).luck(this._luck).one() as T\r\n        }\r\n\r\n        return key as T\r\n    }\r\n\r\n    take(n: number): T[] {\r\n        const results: T[] = []\r\n        for (let i = 0; i < n; i++) {\r\n            const picked = this.one()\r\n            if (picked) results.push(picked)\r\n        }\r\n        return results\r\n    }\r\n}\r\n\r\nexport function picker<T extends string>(map: Record<T, number | NestedMap>): PickerInstance<T> {\r\n    if (!map || typeof map !== 'object') {\r\n        return new ProbabilityEngine<T>({} as NestedMap)\r\n    }\r\n    return new ProbabilityEngine<T>(map as NestedMap)\r\n}\r\n","import { picker } from './picker'\r\n\r\nexport function bag<T extends string>(map: Record<T, number>) {\r\n    let remaining = { ...map }\r\n\r\n    return {\r\n        next(): T | undefined {\r\n            const keys = Object.keys(remaining)\r\n            if (keys.length === 0) remaining = { ...map }\r\n            const picked = picker(remaining).one()\r\n            if (picked) delete remaining[picked as T]\r\n            return picked as T\r\n        }\r\n    }\r\n}\r\n"]}

package/dist/index.d.cts

@@ -1,4 +1,16 @@
-type ProbabilityMap = Record<string, number>;
-declare function probabilityPicker(map: ProbabilityMap): string | null | undefined;
+type NestedMap = {
+    [key: string]: number | NestedMap;
+};
+interface PickerInstance<T extends string> {
+    luck(value: number): this;
+    one(): T | undefined;
+    take(n: number): T[];
+}
 
-export { probabilityPicker as default };
+declare function picker<T extends string>(map: Record<T, number | NestedMap>): PickerInstance<T>;
+
+declare function bag<T extends string>(map: Record<T, number>): {
+    next(): T | undefined;
+};
+
+export { type NestedMap, type PickerInstance, bag, picker };

package/dist/index.d.ts

@@ -1,4 +1,16 @@
-type ProbabilityMap = Record<string, number>;
-declare function probabilityPicker(map: ProbabilityMap): string | null | undefined;
+type NestedMap = {
+    [key: string]: number | NestedMap;
+};
+interface PickerInstance<T extends string> {
+    luck(value: number): this;
+    one(): T | undefined;
+    take(n: number): T[];
+}
 
-export { probabilityPicker as default };
+declare function picker<T extends string>(map: Record<T, number | NestedMap>): PickerInstance<T>;
+
+declare function bag<T extends string>(map: Record<T, number>): {
+    next(): T | undefined;
+};
+
+export { type NestedMap, type PickerInstance, bag, picker };

package/dist/index.js

@@ -1,57 +1,80 @@
-// src/index.ts
-function filterNumbers(entries) {
-  return entries.filter((e) => typeof e[1] === "number" && !Number.isNaN(e[1]));
-}
-function sortEntries(entries) {
-  return entries.sort((a, b) => a[1] - b[1]);
-}
-function sumProbabilities(entries) {
-  return entries.reduce((acc, e) => acc + e[1], 0);
-}
-function normalizeProbabilities(entries, total) {
-  return entries.map(([k, v]) => [k, Math.round(v / total * 100)]);
-}
-function prepareEntries(entries) {
-  const filtered = filterNumbers(entries);
-  if (filtered.length === 0) return;
-  const sorted = sortEntries(filtered);
-  const sum = sumProbabilities(sorted);
-  if (sum === 0) return;
-  if (sum === 100) return sorted;
-  return normalizeProbabilities(sorted, sum);
-}
-function secureRandom() {
-  if (typeof globalThis.crypto === "undefined" || typeof globalThis.crypto.getRandomValues !== "function") {
-    return Math.random();
-  }
+// src/utils.ts
+var secureRandom = () => {
   const array = new Uint32Array(1);
   globalThis.crypto.getRandomValues(array);
   return array[0] / 4294967296;
-}
-function random(min, max) {
-  return secureRandom() * (max - min) + min;
-}
-function chooseOne(entries) {
-  const num = random(1, 100);
-  let count = 0;
-  for (const entry of entries) {
-    count += entry[1];
-    if (num <= count) return entry[0];
+};
+var normalize = (map, luck) => {
+  const valid = [];
+  for (const [key, value] of Object.entries(map)) {
+    const weight = typeof value === "number" ? value : 1;
+    if (!Number.isNaN(weight) && weight > 0) {
+      valid.push([key, Math.pow(weight, 1 / luck)]);
+    }
   }
-  return entries[entries.length - 1][0];
+  return valid;
+};
+var select = (items) => {
+  const total = items.reduce((acc, [, v]) => acc + v, 0);
+  let target = secureRandom() * total;
+  for (const [k, v] of items) {
+    target -= v;
+    if (target <= 0) return k;
+  }
+  return items[items.length - 1][0];
+};
+
+// src/picker.ts
+var ProbabilityEngine = class _ProbabilityEngine {
+  _luck = 1;
+  _map;
+  constructor(map) {
+    this._map = map;
+  }
+  luck(value) {
+    this._luck = value;
+    return this;
+  }
+  one() {
+    const entries = normalize(this._map, this._luck);
+    if (entries.length === 0) return;
+    const key = select(entries);
+    const value = this._map[key];
+    if (value && typeof value === "object" && !Array.isArray(value)) {
+      return new _ProbabilityEngine(value).luck(this._luck).one();
+    }
+    return key;
+  }
+  take(n) {
+    const results = [];
+    for (let i = 0; i < n; i++) {
+      const picked = this.one();
+      if (picked) results.push(picked);
+    }
+    return results;
+  }
+};
+function picker(map) {
+  if (!map || typeof map !== "object") {
+    return new ProbabilityEngine({});
+  }
+  return new ProbabilityEngine(map);
 }
-function probabilityPicker(map) {
-  if (!(map instanceof Object)) return null;
-  const entries = Object.entries(map);
-  if (entries.length === 0) return void 0;
-  const first = entries[0];
-  if (first && entries.length === 1) return first[0];
-  const prepared = prepareEntries(entries);
-  if (!prepared) return void 0;
-  return chooseOne(prepared);
+
+// src/bag.ts
+function bag(map) {
+  let remaining = { ...map };
+  return {
+    next() {
+      const keys = Object.keys(remaining);
+      if (keys.length === 0) remaining = { ...map };
+      const picked = picker(remaining).one();
+      if (picked) delete remaining[picked];
+      return picked;
+    }
+  };
 }
-var index_default = probabilityPicker;
 
-export { index_default as default };
+export { bag, picker };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts"],"names":[],"mappings":";AAGA,SAAS,cAAc,OAAA,EAAkB;AACrC,EAAA,OAAO,OAAA,CAAQ,MAAA,CAAO,CAAA,CAAA,KAAK,OAAO,EAAE,CAAC,CAAA,KAAM,QAAA,IAAY,CAAC,MAAA,CAAO,KAAA,CAAM,CAAA,CAAE,CAAC,CAAC,CAAC,CAAA;AAC9E;AAEA,SAAS,YAAY,OAAA,EAAkB;AACnC,EAAA,OAAO,OAAA,CAAQ,IAAA,CAAK,CAAC,CAAA,EAAG,CAAA,KAAM,EAAE,CAAC,CAAA,GAAI,CAAA,CAAE,CAAC,CAAC,CAAA;AAC7C;AAEA,SAAS,iBAAiB,OAAA,EAAkB;AACxC,EAAA,OAAO,OAAA,CAAQ,OAAO,CAAC,GAAA,EAAK,MAAM,GAAA,GAAM,CAAA,CAAE,CAAC,CAAA,EAAG,CAAC,CAAA;AACnD;AAEA,SAAS,sBAAA,CAAuB,SAAkB,KAAA,EAAe;AAC7D,EAAA,OAAO,OAAA,CAAQ,GAAA,CAAI,CAAC,CAAC,GAAG,CAAC,CAAA,KAAM,CAAC,CAAA,EAAG,KAAK,KAAA,CAAO,CAAA,GAAI,KAAA,GAAS,GAAG,CAAC,CAAU,CAAA;AAC9E;AAEA,SAAS,eAAe,OAAA,EAAkB;AACtC,EAAA,MAAM,QAAA,GAAW,cAAc,OAAO,CAAA;AACtC,EAAA,IAAI,QAAA,CAAS,WAAW,CAAA,EAAG;AAC3B,EAAA,MAAM,MAAA,GAAS,YAAY,QAAQ,CAAA;AACnC,EAAA,MAAM,GAAA,GAAM,iBAAiB,MAAM,CAAA;AACnC,EAAA,IAAI,QAAQ,CAAA,EAAG;AACf,EAAA,IAAI,GAAA,KAAQ,KAAK,OAAO,MAAA;AACxB,EAAA,OAAO,sBAAA,CAAuB,QAAQ,GAAG,CAAA;AAC7C;AAEA,SAAS,YAAA,GAAuB;AAC5B,EAAA,IAAI,OAAO,WAAW,MAAA,KAAW,WAAA,IAAe,OAAO,UAAA,CAAW,MAAA,CAAO,oBAAoB,UAAA,EAAY;AACrG,IAAA,OAAO,KAAK,MAAA,EAAO;AAAA,EACvB;AACA,EAAA,MAAM,KAAA,GAAQ,IAAI,WAAA,CAAY,CAAC,CAAA;AAC/B,EAAA,UAAA,CAAW,MAAA,CAAO,gBAAgB,KAAK,CAAA;AACvC,EAAA,OAAO,KAAA,CAAM,CAAC,CAAA,GAAK,UAAA;AACvB;AAEA,SAAS,MAAA,CAAO,KAAa,GAAA,EAAa;AACtC,EAAA,OAAO,YAAA,EAAa,IAAK,GAAA,GAAM,GAAA,CAAA,GAAO,GAAA;AAC1C;AAEA,SAAS,UAAU,OAAA,EAAkB;AACjC,EAAA,MAAM,GAAA,GAAM,MAAA,CAAO,CAAA,EAAG,GAAG,CAAA;AACzB,EAAA,IAAI,KAAA,GAAQ,CAAA;AACZ,EAAA,KAAA,MAAW,SAAS,OAAA,EAAS;AACzB,IAAA,KAAA,IAAS,MAAM,CAAC,CAAA;AAChB,IAAA,IAAI,GAAA,IAAO,KAAA,EAAO,OAAO,KAAA,CAAM,CAAC,CAAA;AAAA,EACpC;AACA,EAAA,OAAO,OAAA,CAAQ,OAAA,CAAQ,MAAA,GAAS,CAAC,EAAG,CAAC,CAAA;AACzC;AAEA,SAAS,kBAAkB,GAAA,EAAgD;AACvE,EAAA,IAAI,EAAE,GAAA,YAAe,MAAA,CAAA,EAAS,OAAO,IAAA;AACrC,EAAA,MAAM,OAAA,GAAU,MAAA,CAAO,OAAA,CAAQ,GAAG,CAAA;AAClC,EAAA,IAAI,OAAA,CAAQ,MAAA,KAAW,CAAA,EAAG,OAAO,MAAA;AACjC,EAAA,MAAM,KAAA,GAAQ,QAAQ,CAAC,CAAA;AACvB,EAAA,IAAI,SAAS,OAAA,CAAQ,MAAA,KAAW,CAAA,EAAG,OAAO,MAAM,CAAC,CAAA;AACjD,EAAA,MAAM,QAAA,GAAW,eAAe,OAAO,CAAA;AACvC,EAAA,IAAI,CAAC,UAAU,OAAO,MAAA;AACtB,EAAA,OAAO,UAAU,QAAQ,CAAA;AAC7B;AAEA,IAAO,aAAA,GAAQ","file":"index.js","sourcesContent":["type ProbabilityMap = Record<string, number>\r\ntype Entry = [string, number]\r\n\r\nfunction filterNumbers(entries: Entry[]) {\r\n    return entries.filter(e => typeof e[1] === 'number' && !Number.isNaN(e[1]))\r\n}\r\n\r\nfunction sortEntries(entries: Entry[]) {\r\n    return entries.sort((a, b) => a[1] - b[1])\r\n}\r\n\r\nfunction sumProbabilities(entries: Entry[]) {\r\n    return entries.reduce((acc, e) => acc + e[1], 0)\r\n}\r\n\r\nfunction normalizeProbabilities(entries: Entry[], total: number) {\r\n    return entries.map(([k, v]) => [k, Math.round((v / total) * 100)] as Entry)\r\n}\r\n\r\nfunction prepareEntries(entries: Entry[]) {\r\n    const filtered = filterNumbers(entries)\r\n    if (filtered.length === 0) return\r\n    const sorted = sortEntries(filtered)\r\n    const sum = sumProbabilities(sorted)\r\n    if (sum === 0) return\r\n    if (sum === 100) return sorted\r\n    return normalizeProbabilities(sorted, sum)\r\n}\r\n\r\nfunction secureRandom(): number {\r\n    if (typeof globalThis.crypto === 'undefined' || typeof globalThis.crypto.getRandomValues !== 'function') {\r\n        return Math.random()\r\n    }\r\n    const array = new Uint32Array(1)\r\n    globalThis.crypto.getRandomValues(array)\r\n    return array[0]! / 0x100000000\r\n}\r\n\r\nfunction random(min: number, max: number) {\r\n    return secureRandom() * (max - min) + min\r\n}\r\n\r\nfunction chooseOne(entries: Entry[]) {\r\n    const num = random(1, 100)\r\n    let count = 0\r\n    for (const entry of entries) {\r\n        count += entry[1]\r\n        if (num <= count) return entry[0]\r\n    }\r\n    return entries[entries.length - 1]![0]\r\n}\r\n\r\nfunction probabilityPicker(map: ProbabilityMap): string | null | undefined {\r\n    if (!(map instanceof Object)) return null\r\n    const entries = Object.entries(map) as Entry[]\r\n    if (entries.length === 0) return undefined\r\n    const first = entries[0]\r\n    if (first && entries.length === 1) return first[0]\r\n    const prepared = prepareEntries(entries)\r\n    if (!prepared) return undefined\r\n    return chooseOne(prepared)\r\n}\r\n\r\nexport default probabilityPicker;"]}
+{"version":3,"sources":["../src/utils.ts","../src/picker.ts","../src/bag.ts"],"names":[],"mappings":";AAEO,IAAM,eAAe,MAAc;AACtC,EAAA,MAAM,KAAA,GAAQ,IAAI,WAAA,CAAY,CAAC,CAAA;AAC/B,EAAA,UAAA,CAAW,MAAA,CAAO,gBAAgB,KAAK,CAAA;AACvC,EAAA,OAAO,KAAA,CAAM,CAAC,CAAA,GAAK,UAAA;AACvB,CAAA;AAEO,IAAM,SAAA,GAAY,CAAC,GAAA,EAAgB,IAAA,KAAqC;AAC3E,EAAA,MAAM,QAA4B,EAAC;AAEnC,EAAA,KAAA,MAAW,CAAC,GAAA,EAAK,KAAK,KAAK,MAAA,CAAO,OAAA,CAAQ,GAAG,CAAA,EAAG;AAC5C,IAAA,MAAM,MAAA,GAAS,OAAO,KAAA,KAAU,QAAA,GAAW,KAAA,GAAQ,CAAA;AACnD,IAAA,IAAI,CAAC,MAAA,CAAO,KAAA,CAAM,MAAM,CAAA,IAAK,SAAS,CAAA,EAAG;AACrC,MAAA,KAAA,CAAM,IAAA,CAAK,CAAC,GAAA,EAAK,IAAA,CAAK,IAAI,MAAA,EAAQ,CAAA,GAAI,IAAI,CAAC,CAAC,CAAA;AAAA,IAChD;AAAA,EACJ;AAEA,EAAA,OAAO,KAAA;AACX,CAAA;AAEO,IAAM,MAAA,GAAS,CAAC,KAAA,KAAsC;AACzD,EAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,MAAA,CAAO,CAAC,GAAA,EAAK,GAAG,CAAC,CAAA,KAAM,GAAA,GAAM,CAAA,EAAG,CAAC,CAAA;AACrD,EAAA,IAAI,MAAA,GAAS,cAAa,GAAI,KAAA;AAE9B,EAAA,KAAA,MAAW,CAAC,CAAA,EAAG,CAAC,CAAA,IAAK,KAAA,EAAO;AACxB,IAAA,MAAA,IAAU,CAAA;AACV,IAAA,IAAI,MAAA,IAAU,GAAG,OAAO,CAAA;AAAA,EAC5B;AAEA,EAAA,OAAO,KAAA,CAAM,KAAA,CAAM,MAAA,GAAS,CAAC,EAAG,CAAC,CAAA;AACrC,CAAA;;;AC5BA,IAAM,iBAAA,GAAN,MAAM,kBAAA,CAAiE;AAAA,EAC3D,KAAA,GAAQ,CAAA;AAAA,EACR,IAAA;AAAA,EAER,YAAY,GAAA,EAAgB;AACxB,IAAA,IAAA,CAAK,IAAA,GAAO,GAAA;AAAA,EAChB;AAAA,EAEA,KAAK,KAAA,EAAe;AAChB,IAAA,IAAA,CAAK,KAAA,GAAQ,KAAA;AACb,IAAA,OAAO,IAAA;AAAA,EACX;AAAA,EAEA,GAAA,GAAqB;AACjB,IAAA,MAAM,OAAA,GAAU,SAAA,CAAU,IAAA,CAAK,IAAA,EAAM,KAAK,KAAK,CAAA;AAC/C,IAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,EAAG;AAE1B,IAAA,MAAM,GAAA,GAAM,OAAO,OAAO,CAAA;AAC1B,IAAA,MAAM,KAAA,GAAQ,IAAA,CAAK,IAAA,CAAK,GAAG,CAAA;AAE3B,IAAA,IAAI,KAAA,IAAS,OAAO,KAAA,KAAU,QAAA,IAAY,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AAC7D,MAAA,OAAO,IAAI,mBAAkB,KAAK,CAAA,CAAE,KAAK,IAAA,CAAK,KAAK,EAAE,GAAA,EAAI;AAAA,IAC7D;AAEA,IAAA,OAAO,GAAA;AAAA,EACX;AAAA,EAEA,KAAK,CAAA,EAAgB;AACjB,IAAA,MAAM,UAAe,EAAC;AACtB,IAAA,KAAA,IAAS,CAAA,GAAI,CAAA,EAAG,CAAA,GAAI,CAAA,EAAG,CAAA,EAAA,EAAK;AACxB,MAAA,MAAM,MAAA,GAAS,KAAK,GAAA,EAAI;AACxB,MAAA,IAAI,MAAA,EAAQ,OAAA,CAAQ,IAAA,CAAK,MAAM,CAAA;AAAA,IACnC;AACA,IAAA,OAAO,OAAA;AAAA,EACX;AACJ,CAAA;AAEO,SAAS,OAAyB,GAAA,EAAuD;AAC5F,EAAA,IAAI,CAAC,GAAA,IAAO,OAAO,GAAA,KAAQ,QAAA,EAAU;AACjC,IAAA,OAAO,IAAI,iBAAA,CAAqB,EAAe,CAAA;AAAA,EACnD;AACA,EAAA,OAAO,IAAI,kBAAqB,GAAgB,CAAA;AACpD;;;AC3CO,SAAS,IAAsB,GAAA,EAAwB;AAC1D,EAAA,IAAI,SAAA,GAAY,EAAE,GAAG,GAAA,EAAI;AAEzB,EAAA,OAAO;AAAA,IACH,IAAA,GAAsB;AAClB,MAAA,MAAM,IAAA,GAAO,MAAA,CAAO,IAAA,CAAK,SAAS,CAAA;AAClC,MAAA,IAAI,KAAK,MAAA,KAAW,CAAA,EAAG,SAAA,GAAY,EAAE,GAAG,GAAA,EAAI;AAC5C,MAAA,MAAM,MAAA,GAAS,MAAA,CAAO,SAAS,CAAA,CAAE,GAAA,EAAI;AACrC,MAAA,IAAI,MAAA,EAAQ,OAAO,SAAA,CAAU,MAAW,CAAA;AACxC,MAAA,OAAO,MAAA;AAAA,IACX;AAAA,GACJ;AACJ","file":"index.js","sourcesContent":["import type { NestedMap } from './types'\r\n\r\nexport const secureRandom = (): number => {\r\n    const array = new Uint32Array(1)\r\n    globalThis.crypto.getRandomValues(array)\r\n    return array[0]! / 0x100000000\r\n}\r\n\r\nexport const normalize = (map: NestedMap, luck: number): [string, number][] => {\r\n    const valid: [string, number][] = []\r\n\r\n    for (const [key, value] of Object.entries(map)) {\r\n        const weight = typeof value === 'number' ? value : 1\r\n        if (!Number.isNaN(weight) && weight > 0) {\r\n            valid.push([key, Math.pow(weight, 1 / luck)])\r\n        }\r\n    }\r\n\r\n    return valid\r\n}\r\n\r\nexport const select = (items: [string, number][]): string => {\r\n    const total = items.reduce((acc, [, v]) => acc + v, 0)\r\n    let target = secureRandom() * total\r\n\r\n    for (const [k, v] of items) {\r\n        target -= v\r\n        if (target <= 0) return k\r\n    }\r\n\r\n    return items[items.length - 1]![0]\r\n}\r\n","import type { NestedMap, PickerInstance } from './types'\r\nimport { normalize, select } from './utils'\r\n\r\nclass ProbabilityEngine<T extends string> implements PickerInstance<T> {\r\n    private _luck = 1\r\n    private _map: NestedMap\r\n\r\n    constructor(map: NestedMap) {\r\n        this._map = map\r\n    }\r\n\r\n    luck(value: number) {\r\n        this._luck = value\r\n        return this\r\n    }\r\n\r\n    one(): T | undefined {\r\n        const entries = normalize(this._map, this._luck)\r\n        if (entries.length === 0) return\r\n\r\n        const key = select(entries)\r\n        const value = this._map[key]\r\n\r\n        if (value && typeof value === 'object' && !Array.isArray(value)) {\r\n            return new ProbabilityEngine(value).luck(this._luck).one() as T\r\n        }\r\n\r\n        return key as T\r\n    }\r\n\r\n    take(n: number): T[] {\r\n        const results: T[] = []\r\n        for (let i = 0; i < n; i++) {\r\n            const picked = this.one()\r\n            if (picked) results.push(picked)\r\n        }\r\n        return results\r\n    }\r\n}\r\n\r\nexport function picker<T extends string>(map: Record<T, number | NestedMap>): PickerInstance<T> {\r\n    if (!map || typeof map !== 'object') {\r\n        return new ProbabilityEngine<T>({} as NestedMap)\r\n    }\r\n    return new ProbabilityEngine<T>(map as NestedMap)\r\n}\r\n","import { picker } from './picker'\r\n\r\nexport function bag<T extends string>(map: Record<T, number>) {\r\n    let remaining = { ...map }\r\n\r\n    return {\r\n        next(): T | undefined {\r\n            const keys = Object.keys(remaining)\r\n            if (keys.length === 0) remaining = { ...map }\r\n            const picked = picker(remaining).one()\r\n            if (picked) delete remaining[picked as T]\r\n            return picked as T\r\n        }\r\n    }\r\n}\r\n"]}

package/package.json

@@ -1,42 +1,53 @@
-{
-    "name": "probability-picker",
-    "version": "1.2.0",
-    "description": "A lightweight library for selecting random values based on weighted probabilities.",
-    "main": "./dist/index.cjs",
-    "module": "./dist/index.js",
-    "types": "./dist/index.d.ts",
-    "exports": {
-        ".": {
-            "types": "./dist/index.d.ts",
-            "import": "./dist/index.js",
-            "require": "./dist/index.cjs"
-        }
-    },
-    "files": [
-        "dist",
-        "README.md",
-        "LICENSE"
-    ],
-    "scripts": {
-        "build": "tsup",
-        "dev": "tsup --watch",
-        "lint": "tsc --noEmit"
-    },
-    "keywords": [
-        "probability",
-        "weighted",
-        "random"
-    ],
-    "author": "qgave (https://github.com/qgave)",
-    "license": "MIT",
-    "repository": {
-        "type": "git",
-        "url": "https://github.com/qgave/probability-picker"
-    },
-    "type": "module",
-    "sideEffects": false,
-    "devDependencies": {
-        "tsup": "^8.5.1",
-        "typescript": "^5.9.3"
-    }
+{
+  "name": "probability-picker",
+  "version": "2.0.0",
+  "description": "A lightweight library for selecting random values based on weighted probabilities.",
+  "type": "module",
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "lint": "tsc --noEmit",
+    "test": "vitest run"
+  },
+  "keywords": [
+    "probability",
+    "weighted",
+    "random"
+  ],
+  "author": {
+    "name": "qgave",
+    "url": "https://github.com/qgave"
+  },
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/qgave/probability-picker"
+  },
+  "sideEffects": false,
+  "engines": {
+    "node": ">=18"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "devDependencies": {
+    "typescript": "^5.9.3",
+    "tsup": "^8.5.1",
+    "vitest": "^4.0.17"
+  }
 }