adba
Version:
Any DataBase to API
141 lines (140 loc) • 5.16 kB
JavaScript
import moment from 'moment';
/**
* Stores additional custom rules that can be added dynamically
*/
const extraRules = {};
/**
* Adds more rule actions to the `extraRules` object.
* @param moreRules - An object containing rules to be added.
*/
export function addRuleActions(moreRules) {
// Merge existing rules with new ones
Object.assign(extraRules, moreRules);
}
/**
* Applies given rule to the provided value using custom logic.
* @param ruleIn - Determines what rule should be applied.
* @param value - The value to transform.
* @param defaultFunc - Function to apply as default.
* @returns The value after transformation.
*/
function ruleActionReplace(ruleIn, value, defaultFunc = (v) => v) {
switch (ruleIn) {
case ':jsonStr':
return JSON.stringify(value);
case ':jsonObj':
return JSON.parse(value);
case ':datetime':
return moment(value).format('YYYY-MM-DD HH:mm:ss');
case ':date':
return moment(value).format('YYYY-MM-DD');
case ':time':
return moment(value).format('HH:mm:ss');
case ':booleanStr':
return !value ? 'FALSE' : 'TRUE';
case ':boolean':
return !!value;
case ':join':
return value.join(',');
case ':join:tab':
return value.join('\t');
case ':join:nl':
return value.join('\n');
case ':join:|':
return value.join('|');
case ':sortAsc':
return value.sort((a, b) => a - b);
case ':sortDesc':
return value.sort((a, b) => b - a);
default:
if (extraRules[ruleIn]) {
return extraRules[ruleIn](value);
}
return defaultFunc(value);
}
}
/**
* Formats data according to the provided rules.
* @param data - The object where keys are field names and values are the data to format.
* @param rules - The object containing rules to use for formatting.
* @returns Returns a new object with formatted values.
*/
export default function formatData(data, rules) {
const toReturn = Object.entries(data).reduce((r, [key, value]) => {
const colName = key;
const [ruleType, arg2, replaceFunc] = [rules[colName]].flat();
const ruleIn = typeof arg2 === 'string' && arg2;
const subrules = typeof arg2 === 'object' && arg2;
const willRemove = ruleIn === ':remove';
switch (ruleType) {
case ':replace': {
if (typeof replaceFunc !== 'function')
throw new Error("Missing replaceFunc");
const replace = replaceFunc(colName, value, data);
if (replace !== undefined)
r[colName] = replace;
return r;
}
case ':remove':
case false:
return r;
case true:
if (value !== undefined)
r[colName] = value;
break;
case 'string':
if (willRemove && typeof value !== 'string')
return r;
if (value === null)
r[colName] = value;
else if (value !== undefined) {
r[colName] = ruleActionReplace(ruleIn, value, (v) => v.toString());
}
break;
case 'boolean':
if (willRemove && typeof value !== 'boolean')
return r;
if (value === null)
r[colName] = value;
else if (value !== undefined) {
r[colName] = ruleActionReplace(ruleIn, value, (v) => (!v ? 0 : 1));
}
break;
case 'number':
if (willRemove && typeof value !== 'number')
return r;
if (value === null)
r[colName] = value;
else if (value !== undefined) {
r[colName] = ruleActionReplace(ruleIn, value, (v) => (Number(v) || 0));
}
break;
case 'array':
if (willRemove && !Array.isArray(value))
return r;
if (value === null)
r[colName] = [];
else if (value !== undefined) {
r[colName] = ruleActionReplace(ruleIn, [value].flat());
if (subrules)
r[colName] = formatData(r[colName], subrules);
}
break;
case 'object':
if (willRemove && typeof value !== 'object')
return r;
if (value === null)
r[colName] = value;
else if (value !== undefined) {
r[colName] = ruleActionReplace(ruleIn, value);
if (subrules)
r[colName] = formatData(r[colName], subrules);
}
break;
default:
break;
}
return r;
}, {});
return toReturn;
}