facilmap/server/src/utils/utils.ts

import { AsyncAutoTasks, Dictionary, auto as asyncAuto } from "async";
import highland from "highland";

const LETTERS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";

export function generateRandomId(length: number): string {
	let randomPadId = "";
	for(let i=0; i<length; i++) {
		randomPadId += LETTERS[Math.floor(Math.random() * LETTERS.length)];
	}
	return randomPadId;
}

const FAILURE = Symbol('failure');

export function stripObject(obj: object, structure: object): boolean {
	return _stripObject(obj, structure) !== FAILURE;
}

function _stripObject(obj: any, type: any): any {
	if(obj === undefined)
		return obj;
	else if(obj === null)
		return obj;
	else if(type instanceof Array) {
		if(!(obj instanceof Array))
			return FAILURE;

		for(let i=0; i<obj.length; i++) {
			if((obj[i] = _stripObject(obj[i], type[0])) === FAILURE)
				return FAILURE;
		}
		return obj;
	}
	else if(typeof type == "function")
		return (obj instanceof type) ? obj : FAILURE;
	else if(type instanceof Object) {
		if(!(obj instanceof Object))
			return FAILURE;

		for(const i in obj) {
			if(type[i] == null || obj[i] === undefined)
				delete obj[i];
			else if((obj[i] = _stripObject(obj[i], type[i])) === FAILURE)
				return FAILURE;
		}
		return obj;
	}
	else if(type == "number" && typeof obj == "string")
		return obj == "" ? null : isNaN(obj = Number(obj)) ? FAILURE : obj;
	else if(type == "string" && typeof obj == "number")
		return ""+obj;
	else if(typeof type == "string")
		return (typeof obj == type) ? obj : FAILURE;
	else
		return FAILURE;
}

export function round(number: number, digits: number): number {
	const fac = Math.pow(10, digits);
	return Math.round(number*fac)/fac;
}

/* type PromisePropsResult<T extends Record<any, PromiseLike<any>>> = {
	[P in keyof T]: Parameters<T[P].then>
}; */

type PromiseMap<T extends object> = {
	[P in keyof T]: PromiseLike<T[P]> | T[P]
}

export async function promiseProps<T extends object>(obj: PromiseMap<T>): Promise<T> {
	const result = { } as T;
	await Promise.all((Object.keys(obj) as Array<keyof T>).map(async (key) => {
		result[key] = (await obj[key]) as any;
	}));
	return result;
}


export function auto<R extends Dictionary<any>>(tasks: AsyncAutoTasks<R, Error>): Promise<R> {
	// typing of async.auto is broken
	return asyncAuto(tasks) as any;
}


type PromiseCreatorMap<T extends object> = {
	[P in keyof T]: PromiseLike<T[P]> | ((...args: Array<any>) => Promise<T[P]>)
};

export function promiseAuto<T extends Record<string, any>>(obj: PromiseCreatorMap<T>): Promise<T> {
	const promises = { } as PromiseMap<T>;

	function _get(str: keyof T & string) {
		const dep = obj[str];
		if(!dep)
			throw new Error("Invalid dependency '" + str + "' in promiseAuto().");

		if(promises[str])
			return promises[str];

		if(dep instanceof Function) {
			const params = getFuncParams(dep) as Array<keyof T>;
			return promises[str] = _getDeps(params).then(function(res) {
				return (dep as any)(...params.map((param) => res[param]));
			});
		} else {
			return Promise.resolve(dep as any);
		}
	}

	function _getDeps(arr: Array<keyof T>) {
		const deps = { } as PromiseMap<T>;
		arr.forEach(function(it) {
			deps[it] = _get(it as string);
		});
		return promiseProps(deps);
	}

	return _getDeps(Object.keys(obj) as Array<keyof T>);
}

function getFuncParams(func: Function) {
	// Taken from angular injector code

	const ARROW_ARG = /^([^(]+?)\s*=>/;
	const FN_ARGS = /^[^(]*\(\s*([^)]*)\)/m;
	const FN_ARG_SPLIT = /\s*,\s*/;
	const STRIP_COMMENTS = /((\/\/.*$)|(\/\*[\s\S]*?\*\/))/mg;

	const fnText = (Function.prototype.toString.call(func) + ' ').replace(STRIP_COMMENTS, '');
	const match = (fnText.match(ARROW_ARG) || fnText.match(FN_ARGS));
	if (!match) {
		throw new Error("Could not parse function params.");
	}
	const params = match[1];
	return params == "" ? [ ] : params.split(FN_ARG_SPLIT);
}

export function clone<T>(obj: T): T {
	return obj == null ? obj : JSON.parse(JSON.stringify(obj));
}

export function throttle<A extends any[], R>(func: (...args: A) => Promise<R>, parallel = 1): (...args: A) => Promise<R> {
	const stream = highland<() => Promise<void>>();
	stream.map((func) => (highland(func()))).parallel(parallel).done(() => undefined);

	return (...args: A): Promise<R> => {
		return new Promise<R>((resolve, reject) => {
			stream.write(() => {
				return func(...args).then(resolve).catch(reject);
			});
		});
	};
}