@antv/dw-transform/lib/autoTransform.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var dw_analyzer_1 = require("@antv/dw-analyzer");
var parse_1 = require("./parse");
function rename(originStr, aggType, option) {
    if (option === void 0) { option = 'brackets'; }
    if (option === false || option === 'origin') {
        return originStr;
    }
    if (option === true || option === 'brackets' || !option) {
        return aggType.toUpperCase() + "(" + originStr + ")";
    }
    if (option === 'underline') {
        return aggType.toUpperCase() + "_" + originStr;
    }
    if (typeof option === 'function') {
        return String(option(originStr, aggType));
    }
    throw new Error("Invalid rename option " + option);
}
/**
 * @beta
 */
function autoSchema(data, renameOption) {
    if (renameOption === void 0) { renameOption = 'brackets'; }
    var schemas = [];
    var schema = {
        actions: [],
    };
    var dataAnalyze = dw_analyzer_1.typeAll(data);
    var toGroupBy = [];
    var toNotGroupBy = [];
    dataAnalyze.forEach(function (col) {
        if (['string', 'date', 'boolean'].includes(col.recommendation)) {
            toGroupBy.push(col.name);
        }
        else {
            toNotGroupBy.push(col.name);
        }
    });
    if (toGroupBy.length) {
        schema.groupBy = toGroupBy;
        toNotGroupBy.forEach(function (colName) {
            schema.actions.push({
                type: 'sum',
                field: colName,
                as: rename(colName, 'sum', renameOption),
            });
        });
        schemas.push(schema);
    }
    return schemas;
}
exports.autoSchema = autoSchema;
/**
 * @beta
 */
function autoTransform(data, renameOption) {
    if (renameOption === void 0) { renameOption = 'brackets'; }
    var result = {
        result: [],
        schemas: autoSchema(data, renameOption),
    };
    result.result = parse_1.parse(data, result.schemas);
    return result;
}
exports.autoTransform = autoTransform;