@gooddata/react-components/src/helpers/metadata.ts

GoodData React Components

import get = require('lodash/get');
import cloneDeep = require('lodash/cloneDeep');
import { VisualizationObject, Transformation } from '@gooddata/data-layer';

import { ISortingChange } from './sorting';

interface IMeasureInfo {
    generatedId: string;
    isPoP: boolean;
}

function getMeasureInfo(id: string): IMeasureInfo {
    if (id.endsWith('_pop')) {
        return {
            generatedId: id.split('_')[0],
            isPoP: true
        };
    }

    return {
        generatedId: id,
        isPoP: false
    };
}

export interface ISorting {
    sorting: Transformation.ISort;
    change: ISortingChange;
}

export interface IUpdateSortingResult {
    updatedMetadata: VisualizationObject.IVisualizationObject;
    updatedSorting: ISorting;
}

export function updateSorting(
    metadata: VisualizationObject.IVisualizationObject,
    sortingInfo: ISorting
): IUpdateSortingResult {
    const { sorting, change } = sortingInfo;
    const updatedMetadata = cloneDeep(metadata);
    const buckets = updatedMetadata.content.buckets;
    const { column, direction } = sorting;
    const { type } = change;

    let bucketItem;
    let sort: VisualizationObject.IMeasureSort | string;

    if (type === 'metric') {
        const { generatedId, isPoP } = getMeasureInfo(column);
        const measures = get(buckets, 'measures', []);
        const measure: VisualizationObject.IMeasure = measures
            .find(item => item.measure.generatedId === generatedId);

        bucketItem = get(measure, 'measure');
        // Sort direction needs to be fixed in data-layer
        // tslint:disable-next-line:no-object-literal-type-assertion
        sort = { direction } as VisualizationObject.IMeasureSort;
        if (isPoP) {
            sort.sortByPoP = true;
        }
    } else {
        const categories = get(buckets, 'categories', []);
        const category: VisualizationObject.ICategory = categories
            .find(item => item.category.displayForm === column);

        bucketItem = get(category, 'category') as VisualizationObject.ICategory;
        sort = direction; // string instead of object for categories :(
    }
    // handle column deletion
    if (!bucketItem) {
        return {
            updatedMetadata: metadata,
            updatedSorting: null
        };
    }

    buckets.categories = buckets.categories.map((category) => {
        category.category.sort = null;
        return category;
    });
    buckets.measures = buckets.measures.map((measure) => {
        measure.measure.sort = null;
        return measure;
    });
    bucketItem.sort = sort;

    return {
        updatedMetadata,
        updatedSorting: sortingInfo
    };
}