@monstermann/fn/dist/map/mapOr.js

A utility library for TypeScript.

import { is } from "../function/is.js";
import { dfdlT } from "@monstermann/dfdl";
import { cloneMap } from "@monstermann/remmi";

//#region src/map/mapOr.ts
/**
* `mapOr(map, key, transform, or)`
*
* Transforms the value at `key` in `map` using `transform`, returning `or` if the key doesn't exist.
*
* ```ts
* mapOr(
*     new Map([
*         ["a", 1],
*         ["b", 2],
*     ]),
*     "a",
*     (value) => value * 2,
*     null,
* ); // Map(2) { "a" => 2, "b" => 2 }
*
* mapOr(
*     new Map([
*         ["a", 1],
*         ["b", 2],
*     ]),
*     "c",
*     (value) => value * 2,
*     null,
* ); // null
* ```
*
* ```ts
* pipe(
*     new Map([
*         ["a", 1],
*         ["b", 2],
*     ]),
*     mapOr("a", (value) => value * 2, null),
* ); // Map(2) { "a" => 2, "b" => 2 }
*
* pipe(
*     new Map([
*         ["a", 1],
*         ["b", 2],
*     ]),
*     mapOr("c", (value) => value * 2, null),
* ); // null
* ```
*/
const mapOr = dfdlT((target, key, transform, or) => {
	if (!target.has(key)) return or;
	const prev = target.get(key);
	const next = transform(prev, key, target);
	if (is(prev, next)) return target;
	const result = cloneMap(target);
	result.set(key, next);
	return result;
}, 4);

//#endregion
export { mapOr };