@qrvey/formula-lang
Version:
QFormula support for qrvey projects
127 lines • 4.12 kB
JavaScript
import { AST_PRIMITIVES, OPERATION_SCOPE, EXPECTED_AGGREGATE_RESULT, } from '../constants';
import { ERROR_DICTIONARY, ERROR_LIST } from '../errors/dictionary';
import { isAggregateScope } from '../utils/isAggregate';
import { isBooleanExpression } from '../utils/isBooleanExpression';
import { inferPrimitive, isSinglePrimitive } from '../utils/primitiveFunctions';
/**
* `IFS` Returns one value if a logical expression is TRUE
* when is False another logical expression have to be set with their returned value
* and so (recursive operation)
*/
export const IFS = {
identifier: 'IFS',
operationScope: OPERATION_SCOPE.RAW,
functionScope: [OPERATION_SCOPE.RAW, OPERATION_SCOPE.AGGREGATE],
recursiveStartIn: 0,
parameters: [
{
identifier: 'LOGICAL_EXPRESSION',
optional: false,
expectedPrimitive: AST_PRIMITIVES.BOOLEAN,
validator: [isBooleanExpression],
},
{
identifier: 'VALUE_IF_TRUE',
optional: false,
generic: true,
validator: [sameRecursivePrimitiveType],
},
],
transpiler: {
elasticsearch,
snowflake,
redshift,
postgresql,
databricks,
},
primitiveResult(args) {
return calculateIfsPrimitive(args);
},
};
function sameRecursivePrimitiveType(_param, _dataType, context) {
const args = context.fnNode.arguments;
const firstArgsPrimitive = calculateFirstIFSPrimitive(args);
const ifsPrimitive = calculateIfsPrimitive(args);
const isAggregate = isAggregateScope(context.globalContext);
const valid = isAggregate
? EXPECTED_AGGREGATE_RESULT.includes(firstArgsPrimitive)
: isSinglePrimitive(ifsPrimitive);
if (!valid)
return Object.assign(Object.assign({ valid }, ERROR_DICTIONARY[ERROR_LIST.inferredPrimitive]), { mismatchData: {
primitive: isAggregate
? EXPECTED_AGGREGATE_RESULT[0]
: firstArgsPrimitive,
} });
return {
valid,
mismatchData: {
primitive: ifsPrimitive,
},
};
}
function calculateIfsPrimitive(args) {
const primitiveArgs = [];
args.forEach((value, index) => {
if (index % 2 === 1) {
primitiveArgs.push(value.primitive);
}
});
return inferPrimitive(...primitiveArgs);
}
function calculateFirstIFSPrimitive(args) {
var _a, _b;
return ((_b = (_a = args.find((arg, index) => index % 2 === 1 &&
isSinglePrimitive(arg.primitive) &&
arg.primitive !== AST_PRIMITIVES.UNKNOWN)) === null || _a === void 0 ? void 0 : _a.primitive) !== null && _b !== void 0 ? _b : AST_PRIMITIVES.UNKNOWN);
}
function elasticsearch(...args) {
const _args = [...args];
const logicalExpression = _args.shift();
const valueIfTrue = _args.shift();
if (_args.length === 0) {
return `(${logicalExpression} ? ${valueIfTrue} : null)`;
}
else {
return `(${logicalExpression} ? ${valueIfTrue} : ${elasticsearch(..._args)})`;
}
}
function snowflake(...args) {
const _args = [...args];
const logicalExpression = _args.shift();
const valueIfTrue = _args.shift();
if (_args.length === 0) {
return `IFF(${logicalExpression}, ${valueIfTrue}, NULL)`;
}
else {
return `IFF(${logicalExpression}, ${valueIfTrue}, ${snowflake(..._args)})`;
}
}
function sql(...args) {
const _args = [...args];
const logicalExpression = _args.shift();
const valueIfTrue = _args.shift();
if (_args.length === 0) {
return `
(CASE
WHEN ${logicalExpression} THEN ${valueIfTrue}
ELSE NULL
END)`.trim();
}
else {
return `
(CASE
WHEN ${logicalExpression} THEN ${valueIfTrue}
ELSE ${sql(..._args)}
END)`.trim();
}
}
function redshift(...args) {
return sql(...args);
}
function postgresql(...args) {
return sql(...args);
}
function databricks(...args) {
return sql(...args);
}
//# sourceMappingURL=ifs.js.map