Function brushPreprocessor

src/component/brush/preprocessor.ts:30–69

(option: ECUnitOption, isNew: boolean)

const DEFAULT_TOOLBOX_BTNS: BrushToolboxIconType[] = ['rect', 'polygon', 'keep', 'clear'];

export default function brushPreprocessor(option: ECUnitOption, isNew: boolean): void {
    const brushComponents = normalizeToArray(option ? option.brush : []);

    if (!brushComponents.length) {
        return;
    }

    let brushComponentSpecifiedBtns = [] as string[];

    zrUtil.each(brushComponents, function (brushOpt: BrushOption) {
        const tbs = brushOpt.hasOwnProperty('toolbox')
            ? brushOpt.toolbox : [];

        if (tbs instanceof Array) {
            brushComponentSpecifiedBtns = brushComponentSpecifiedBtns.concat(tbs);
        }
    });

    let toolbox: ToolboxOption = option && option.toolbox;

    if (zrUtil.isArray(toolbox)) {
        toolbox = toolbox[0];
    }
    if (!toolbox) {
        toolbox = {feature: {}};
        option.toolbox = [toolbox];
    }

    const toolboxFeature = (toolbox.feature || (toolbox.feature = {}));
    const toolboxBrush = (toolboxFeature.brush || (toolboxFeature.brush = {})) as ToolboxBrushFeatureOption;
    const brushTypes = toolboxBrush.type || (toolboxBrush.type = []);

    brushTypes.push.apply(brushTypes, brushComponentSpecifiedBtns);

    removeDuplicates(brushTypes, item => item + '', null);

    if (isNew && !brushTypes.length) {
        brushTypes.push.apply(brushTypes, DEFAULT_TOOLBOX_BTNS);
    }
}