@wener/utils 1.1.54 → 1.1.58

package/src/maths/random.ts

@@ -1,52 +1,209 @@
-// LCG pseudo random
-export function createRandom(_seed: number | string = Date.now()): {
-	random(n?: number): number;
-	randomBytes(length: number): Uint8Array;
-	[Symbol.iterator](): Generator<number, void, unknown>;
-} {
-	let seed = 0;
+export type RNG = {
+	/** 返回 [0,1) 或 [0,a) 或 [a,b) 范围的随机浮点数 */
+	random(a?: number, b?: number): number;
+	/** 返回 [0, max] 或 [min, max] 范围的随机整数（包含两端） */
+	randomInt(max: number): number;
+	randomInt(min: number, max: number): number;
+	/** 生成随机字节，填充到提供的数组或返回指定长度的新数组 */
+	randomBytes(n: number): Uint8Array;
+	randomBytes(buf: Uint8Array): Uint8Array;
+	/** Fisher-Yates 洗牌算法，原地打乱数组 */
+	shuffle<T>(arr: T[]): T[];
+	/** 从数组中随机采样 n 个元素（不重复） */
+	sample<T>(arr: T[], n: number): T[];
+	/** 从数组中随机选择一个元素 */
+	pick<T>(arr: T[]): T;
+	/** 重置随机数生成器到初始种子或指定种子 */
+	reset(seed?: SeedSource): void;
+	/** 当前种子值 */
+	readonly seed: number;
+};
+
+type SeedSource = string | number;
+
+function resolveSeed(seed: SeedSource = Date.now()) {
+	let v = 0;
 
-	if (typeof _seed === 'string') {
-		for (let i = 0; i < _seed.length; i++) {
-			seed = (Math.imul(31, seed) + _seed.charCodeAt(i)) | 0;
+	if (typeof seed === 'number') {
+		v = seed;
+	} else {
+		seed = String(seed);
+		for (let i = 0; i < seed.length; i++) {
+			v = (Math.imul(31, v) + seed.charCodeAt(i)) | 0;
 		}
-	} else if (typeof _seed === 'number') {
-		seed = _seed;
 	}
 
-	// int32
-	seed = seed >>> 0;
+	// uint32
+	v = v >>> 0;
 
-	// alternative 梅森旋转算法 (Mersenne Twister), Xorshift
-	// LCG
-	const m = 0x80000000; // 2^31
-	const a = 1664525;
-	const c = 1013904223;
+	return v;
+}
+
+export function createRandom(s: SeedSource): RNG {
+	// seealso https://github.com/skeeto/rng-js/blob/master/rng.js
+	const initialSeed = resolveSeed(s);
+	const rng = createXorshift128plus(initialSeed);
 
-	const random = (n?: number): number => {
-		seed = (a * seed + c) % m;
-		let r = seed / m;
-		if (n) {
-			r = Math.floor(r * n);
+	const random = (a?: number, b?: number) => {
+		const r = rng.next();
+		if (a !== undefined && b !== undefined) {
+			return r * (b - a) + a;
+		}
+		if (a !== undefined) {
+			return r * a;
 		}
 		return r;
 	};
 
-	const randomBytes = (length: number): Uint8Array => {
-		const bytes = new Uint8Array(length);
-		for (let i = 0; i < length; i++) {
-			bytes[i] = Math.floor(random() * 256);
+	const randomInt = (min: number, max?: number) => {
+		if (max === undefined) {
+			max = min;
+			min = 0;
+		}
+		return Math.floor(random(min, max + 1));
+	};
+
+	const shuffle = <T>(arr: T[]): T[] => {
+		// Fisher-Yates shuffle
+		for (let i = arr.length - 1; i > 0; i--) {
+			const j = randomInt(0, i);
+			const tmp = arr[i]!;
+			arr[i] = arr[j]!;
+			arr[j] = tmp;
+		}
+		return arr;
+	};
+
+	const sample = <T>(arr: T[], n: number): T[] => {
+		if (n >= arr.length) {
+			return shuffle([...arr]);
+		}
+		// 使用 partial Fisher-Yates
+		const copy = [...arr];
+		const result: T[] = [];
+		for (let i = 0; i < n; i++) {
+			const j = randomInt(i, copy.length - 1);
+			const tmp = copy[i]!;
+			copy[i] = copy[j]!;
+			copy[j] = tmp;
+			result.push(copy[i]!);
 		}
-		return bytes;
+		return result;
+	};
+
+	const pick = <T>(arr: T[]): T => {
+		return arr[randomInt(0, arr.length - 1)]!;
+	};
+
+	const randomBytes = (input: number | Uint8Array): Uint8Array => {
+		const buf = typeof input === 'number' ? new Uint8Array(input) : input;
+		const len = buf.length;
+		// 每次 nextInt32 产生 4 字节
+		let i = 0;
+		while (i + 4 <= len) {
+			const v = rng.nextInt32();
+			buf[i] = v & 0xff;
+			buf[i + 1] = (v >>> 8) & 0xff;
+			buf[i + 2] = (v >>> 16) & 0xff;
+			buf[i + 3] = (v >>> 24) & 0xff;
+			i += 4;
+		}
+		// 处理剩余字节
+		if (i < len) {
+			const v = rng.nextInt32();
+			for (let j = 0; i < len; i++, j++) {
+				buf[i] = (v >>> (j * 8)) & 0xff;
+			}
+		}
+		return buf;
 	};
 
 	return {
 		random,
+		randomInt,
 		randomBytes,
-		[Symbol.iterator]: function* () {
-			while (true) {
-				yield random();
-			}
+		shuffle,
+		sample,
+		pick,
+		reset: (seed?: SeedSource) => rng.reset(seed !== undefined ? resolveSeed(seed) : undefined),
+		get seed() {
+			return rng.seed;
+		},
+	};
+}
+
+export function resolveRandom(r: RNG | SeedSource = Date.now()): RNG {
+	if (typeof r !== 'string' && typeof r !== 'number') {
+		return r;
+	}
+	return createRandom(r);
+}
+
+/**
+ * xorshift128+ 算法
+ * 周期: 2^128 - 1，统计特性优于 LCG
+ * 参考: https://prng.di.unimi.it/
+ */
+function createXorshift128plus(seed: number): {
+	next: () => number;
+	nextFloat: () => number;
+	nextDouble: () => number;
+	nextInt32: () => number;
+	reset: (seed?: number) => void;
+	readonly seed: number;
+} {
+	const initialSeed = seed;
+	// 使用 SplitMix64 风格初始化两个状态
+	let s0 = seed >>> 0;
+	let s1 = (seed * 1812433253 + 1) >>> 0;
+
+	// 确保状态非零
+	if (s0 === 0) s0 = 0xdeadbeef;
+	if (s1 === 0) s1 = 0xcafebabe;
+
+	const nextInt32 = () => {
+		// xorshift128+ 核心算法（32位简化版）
+		let x = s0;
+		const y = s1;
+
+		s0 = y;
+		x ^= x << 23;
+		x ^= x >>> 17;
+		x ^= y;
+		x ^= y >>> 26;
+		s1 = x >>> 0;
+
+		// 返回 uint32
+		return (s0 + s1) >>> 0;
+	};
+
+	// 32 位精度 [0, 1)
+	const nextFloat = () => nextInt32() / 0x100000000;
+
+	// 53 位精度 [0, 1)，两次迭代组合
+	const nextDouble = () => {
+		const hi = nextInt32() >>> 11; // 21 位
+		const lo = nextInt32(); // 32 位
+		return hi / (1 << 21) + lo / ((1 << 21) * 0x100000000);
+	};
+
+	const next = nextFloat;
+
+	const reset = (newSeed: number = initialSeed) => {
+		s0 = newSeed >>> 0;
+		s1 = (newSeed * 1812433253 + 1) >>> 0;
+		if (s0 === 0) s0 = 0xdeadbeef;
+		if (s1 === 0) s1 = 0xcafebabe;
+	};
+
+	return {
+		next,
+		nextFloat,
+		nextDouble,
+		nextInt32,
+		reset,
+		get seed() {
+			return initialSeed;
 		},
 	};
 }

package/src/objects/set.ts

@@ -13,7 +13,9 @@ export function set<T extends object, V>(obj: T, key: ObjectKey | ObjectPath, va
 	let x, k;
 	while (i < len) {
 		k = path[i++];
+		// Security: Prevent prototype pollution
 		if (k === '__proto__' || k === 'constructor' || k === 'prototype') break;
+
 		// noinspection PointlessArithmeticExpressionJS
 		current = current[k] =
 			i === len
@@ -22,7 +24,14 @@ export function set<T extends object, V>(obj: T, key: ObjectKey | ObjectPath, va
 					: val
 				: typeof (x = current[k]) === typeof path
 					? x
-					: // @ts-expect-error hacky type check
+					: // Determine if we should create an Object or an Array for the next level
+						// If the next key is NOT an integer-like index, or contains a dot, create an Object.
+						// Otherwise, create an Array.
+						//
+						// path[i] * 0 !== 0 checks if it is NOT a number (NaN * 0 is NaN).
+						// !!~('' + path[i]).indexOf('.') checks if it contains a dot.
+						//
+						// @ts-expect-error hacky type check from dset
 						path[i] * 0 !== 0 || !!~('' + path[i]).indexOf('.') // eslint-disable-line
 						? {}
 						: [];

package/src/types.d.ts

@@ -1,6 +1,6 @@
 declare var __DEV__: boolean;
 
-namespace NodeJS {
+declare namespace NodeJS {
 	interface Process {
 		// webpack check
 		readonly browser?: boolean;

package/src/web/getGlobalThis.ts

@@ -14,7 +14,6 @@ export const getGlobalThis = (): typeof globalThis => {
 	if (typeof globalThis !== 'undefined') return globalThis;
 	if (typeof self !== 'undefined') return self;
 	if (typeof window !== 'undefined') return window;
-	if (typeof global !== 'undefined') return global as any;
-	if (typeof this !== 'undefined') return this as any;
+	if (typeof global !== 'undefined') return global;
 	throw new Error('Unable to locate global `this`');
 };

package/src/web/getRandomValues.ts

@@ -1,35 +1,34 @@
-// eslint-disable-next-line @typescript-eslint/consistent-type-imports
 import { getNodeCrypto } from '../crypto/getNodeCrypto';
 import type { TypedArray } from '../io/types';
 import { getGlobalThis } from './getGlobalThis';
 
-const globalThis = getGlobalThis();
+type RandomValuesArray = Exclude<TypedArray, Float32Array | Float64Array>;
+
+const _globalThis = getGlobalThis();
 
 // chrome 11+, safari 5+, nodejs 17.4+
 // https://developer.mozilla.org/en-US/docs/Web/API/Crypto/getRandomValues
-export let getRandomValues: <T extends Exclude<TypedArray, Float32Array | Float64Array>>(typedArray: T) => T =
-	globalThis.crypto?.getRandomValues?.bind(globalThis.crypto)
-	|| (globalThis as any).msCrypto?.getRandomValues?.bind((globalThis as any).msCrypto)
-	|| (() => {
-		throw new Error('[getRandomValues]: No secure random number generator available.');
-	});
+export let getRandomValues: <T extends RandomValuesArray>(typedArray: T) => T =
+	_globalThis.crypto?.getRandomValues?.bind(_globalThis.crypto)
+	|| (_globalThis as any).msCrypto?.getRandomValues?.bind((_globalThis as any).msCrypto)
+	|| _getRandomValues;
 
-function _getRandomValues<T extends Exclude<TypedArray, Float32Array | Float64Array>>(buf: T) {
+function _getRandomValues<T extends RandomValuesArray>(buf: T): T {
 	const nodeCrypto = getNodeCrypto();
-	// avoid type error
-	let wc = nodeCrypto?.webcrypto as any;
+	const wc = nodeCrypto?.webcrypto as Crypto | undefined;
 	if (wc?.getRandomValues) {
-		getRandomValues = wc.getRandomValues?.bind(nodeCrypto?.webcrypto);
-		return wc.getRandomValues(buf);
+		getRandomValues = wc.getRandomValues.bind(wc);
+		return getRandomValues(buf);
 	}
 	if (nodeCrypto?.randomBytes) {
 		if (!(buf instanceof Uint8Array)) {
 			throw new TypeError('expected Uint8Array');
 		}
 		if (buf.length > 65536) {
-			const e: any = new Error();
+			const e = new Error(
+				`Failed to execute 'getRandomValues' on 'Crypto': The ArrayBufferView's byte length (${buf.length}) exceeds the number of bytes of entropy available via this API (65536).`,
+			) as Error & { code: number; name: string };
 			e.code = 22;
-			e.message = `Failed to execute 'getRandomValues' on 'Crypto': The ArrayBufferView's byte length (${buf.length}) exceeds the number of bytes of entropy available via this API (65536).`;
 			e.name = 'QuotaExceededError';
 			throw e;
 		}

package/tsconfig.json

@@ -1,16 +1,8 @@
 {
+	"extends": ["../../tsconfig.base.json"],
 	"compilerOptions": {
-		"moduleResolution": "bundler",
-		"target": "ESNext",
-		"module": "ESNext",
-		"lib": ["dom", "dom.iterable", "ESNext"],
-		"strict": true,
 		"sourceMap": true,
-		"skipLibCheck": true,
-		"esModuleInterop": true,
-		"verbatimModuleSyntax": true,
 		"erasableSyntaxOnly": false,
-		"allowSyntheticDefaultImports": true,
 		"noImplicitAny": true,
 		"noImplicitReturns": true,
 		"noUnusedLocals": false,
@@ -18,11 +10,13 @@
 		"pretty": true,
 		"removeComments": false,
 		"stripInternal": true,
-		"isolatedModules": true,
 		"outDir": "lib",
 		"baseUrl": ".",
-		"rootDir": "./src"
+		"rootDir": "./src",
+		"paths": {
+			"#/*": ["./src/*"]
+		}
 	},
-	"exclude": ["node_modules"],
+	"exclude": ["node_modules", "**/*.test.ts", "**/*.spec.ts"],
 	"include": ["src/types.d.ts", "src"]
 }

package/src/schema/README.md

@@ -1,2 +0,0 @@
-- 增加了 literal 处理
-- 关闭了内置 prettier