// (C) 2007-2020 GoodData Corporation
import { Execution } from "@gooddata/typings";
import { getTreemapAttributes } from "../components/visualizations/chart/chartOptionsBuilder";
import { IUnwrappedAttributeHeaderWithItems } from "../components/visualizations/typings/chart";
import { STACK_BY_DIMENSION_INDEX } from "../components/visualizations/chart/constants";
import { isTreemap } from "../components/visualizations/utils/common";
import { findAttributeInDimension } from "./executionResultHelper";
import { IChartConfig } from "../interfaces/Config";

export function getStackByAttribute(
    config: IChartConfig,
    dimensions: Execution.IResultDimension[],
    attributeHeaderItems: Execution.IResultHeaderItem[][][],
): IUnwrappedAttributeHeaderWithItems {
    const { type, mdObject } = config;

    if (isTreemap(type)) {
        const { stackByAttribute } = getTreemapAttributes(dimensions, attributeHeaderItems, mdObject);
        return stackByAttribute;
    }

    return findAttributeInDimension(
        dimensions[STACK_BY_DIMENSION_INDEX],
        attributeHeaderItems[STACK_BY_DIMENSION_INDEX],
    );
}