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

import { typeAll } from '@antv/dw-analyzer';
import { parse } from './parse';
/**
 * @public
 */
export 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);
}
/**
 * @public
 */
export function autoSchema(data, renameOption, defaultAgg) {
    if (renameOption === void 0) { renameOption = 'brackets'; }
    if (defaultAgg === void 0) { defaultAgg = 'sum'; }
    var schemas = [];
    var schema = {
        actions: [],
    };
    var dataAnalyze = 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: defaultAgg,
                field: colName,
                as: rename(colName, defaultAgg, renameOption),
            });
        });
        schemas.push(schema);
    }
    return schemas;
}
/**
 * @public
 */
export function autoTransform(data, renameOption, defaultAgg) {
    if (renameOption === void 0) { renameOption = 'brackets'; }
    if (defaultAgg === void 0) { defaultAgg = 'sum'; }
    var result = {
        result: [],
        schemas: autoSchema(data, renameOption, defaultAgg),
    };
    result.result = parse(data, result.schemas);
    return result;
}