52 lines
1.3 KiB
TypeScript
52 lines
1.3 KiB
TypeScript
|
import get from './get';
|
||
|
|
|
||
|
|
function internalSet<Entity = any, Output = Entity, Value = any>(
|
||
|
|
entity: Entity,
|
||
|
|
paths: (string | number)[],
|
||
|
|
value: Value,
|
||
|
|
removeIfUndefined: boolean,
|
||
|
|
): Output {
|
||
|
|
if (!paths.length) {
|
||
|
|
return value as unknown as Output;
|
||
|
|
}
|
||
|
|
|
||
|
|
const [path, ...restPath] = paths;
|
||
|
|
|
||
|
|
let clone: Output;
|
||
|
|
if (!entity && typeof path === 'number') {
|
||
|
|
clone = [] as unknown as Output;
|
||
|
|
} else if (Array.isArray(entity)) {
|
||
|
|
clone = [...entity] as unknown as Output;
|
||
|
|
} else {
|
||
|
|
clone = { ...entity } as unknown as Output;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Delete prop if `removeIfUndefined` and value is undefined
|
||
|
|
if (removeIfUndefined && value === undefined && restPath.length === 1) {
|
||
|
|
delete clone[path][restPath[0]];
|
||
|
|
} else {
|
||
|
|
clone[path] = internalSet(clone[path], restPath, value, removeIfUndefined);
|
||
|
|
}
|
||
|
|
|
||
|
|
return clone;
|
||
|
|
}
|
||
|
|
|
||
|
|
export default function set<Entity = any, Output = Entity, Value = any>(
|
||
|
|
entity: Entity,
|
||
|
|
paths: (string | number)[],
|
||
|
|
value: Value,
|
||
|
|
removeIfUndefined = false,
|
||
|
|
): Output {
|
||
|
|
// Do nothing if `removeIfUndefined` and parent object not exist
|
||
|
|
if (
|
||
|
|
paths.length &&
|
||
|
|
removeIfUndefined &&
|
||
|
|
value === undefined &&
|
||
|
|
!get(entity, paths.slice(0, -1))
|
||
|
|
) {
|
||
|
|
return entity as unknown as Output;
|
||
|
|
}
|
||
|
|
|
||
|
|
return internalSet(entity, paths, value, removeIfUndefined);
|
||
|
|
}
|