Source: ReactViews/Custom/registerCustomComponentTypes.js

'use strict';

/* global require */

const defined = require('terriajs-cesium/Source/Core/defined');
const DeveloperError = require('terriajs-cesium/Source/Core/DeveloperError');

const Chart = require('./Chart/Chart');
const ChartExpandAndDownloadButtons = require('./Chart/ChartExpandAndDownloadButtons');
const Collapsible = require('./Collapsible/Collapsible');
const CustomComponents = require('./CustomComponents');
const CustomComponentType = require('./CustomComponentType');
const TableStructure = require('../../Map/TableStructure');
const VarType = require('../../Map/VarType');

const React = require('react');

import ChartPreviewStyles from './Chart/chart-preview.scss';

const chartAttributes = [
    'src',
    'src-preview',
    'sources',
    'source-names',
    'downloads',
    'download-names',
    'preview-x-label',
    'data',
    'id',
    'x-column',
    'y-column',
    'y-columns',
    'colors',
    'column-names',
    'column-units',
    'styling',
    'highlight-x',
    'poll-seconds',
    'poll-sources',
    'poll-id-columns',
    'poll-replace',
    'title',
    'can-download',
    'hide-buttons'
];

/**
 * @private
 */
function splitStringIfDefined(string) {
    return defined(string) ? string.split(',') : undefined;
}

/**
 * Registers custom component types.
 *
 * Here we define the following:
 * - <chart>
 * - <collapsible>
 * You can define your own by replacing this file with your own version.
 *
 *
 * <collapsible> displays a collapsible section (see Collapsible.jsx) around its children components.
 * It has two allowed attributes:
 * - title:          The title of the section.
 * - [open]:         true or false (the default).
 *
 *
 * <chart> displays an interactive chart (see Chart.jsx), along with "expand" and "download" buttons (ChartExpandAndDownloadButtons.jsx).
 * This button enables a catalog item based on the data, for display in the Chart Panel (ChartPanel.jsx).
 * It also detects if it appears in the second column of a <table> and, if so, rearranges itself to span two columns.
 *
 * It can have the following attributes. Currently URLs must point to csv (not json) data; but inline json data is supported.
 * Note if you change any of these, also update the chartAttributes array above, or they won't make it here.
 *
 * - [title]:        The title of the chart.  If not supplied, defaults to the name of the context-supplied feature, if available, or else simply "Chart".
 * - [x-column]:     The x column name or number to show in the preview, if not the first appropriate column. NOT FULLY IMPLEMENTED YET.
 * - [y-column]:     The y column name or number to show in the preview, if not the first scalar column.
 * - [y-columns]:    Comma-separated list of y column names or numbers to show in the preview. Overrides "y-column" if provided.
 * - [colors]:       Comma-separated list of css colors to apply to data columns.
 * - [column-names]: Comma-separated list of column names to override those in the source data; empty strings retain the original column name.
 *                   Eg. column-names="Time,Height,Speed"
 * - [column-units]: Comma-separated list of the units for each column. Empty strings are ok.
 *                   Eg. column-units=",m,km/h"
 * - [preview-x-label]: The preview chart x-axis label. Defaults to empty string. Eg. long-names="Last 24 hours,Last 5 days,Time".
 * - [id]:           An id for the chart; give different charts from the same feature different ids. The actual catalogItem.id used for the expanded chart will
 *                   also incorporate the chart title and the catalog item name it came from.
 * - [styling]:      Defaults to 'feature-info'. Can also be 'histogram'. TODO: improve.
 * - [highlight-x]:  An x-coordinate to highlight.
 * - [poll-seconds]: If present, the chart is updated from [poll-sources] every [poll-seconds] seconds.
 *                   TODO: Returned data is merged into existing data and shown.
 * - [poll-sources]: Comma-separated list of URLs to poll every [poll-seconds] seconds. Defaults to sources.
 * - [poll-replace]: Either 'true' or 'false' (case sensitive). Pass 'true' to completely replace the data, 'false' to update
 *                   the existing data. Defaults to false (updating).
 * - [can-download]: 'false' to hide the Download button on the chart.  By default true and for any other value, the download button is shown.
 * - [hide-buttons]: 'true' to hide the Expand and Download buttons on the chart.  By default and for any other value, the buttons are shown when applicable.
 *                   Overrides can-download.
 *
 * Provide the data in one of these four ways:
 * - [sources]:      Comma-separated URLs for data at each available time range. The first in the list is shown in the feature info panel preview.
 *                   Eg. sources="http://example.com/series?offset=1d,http://example.com/series?offset=5d,http://example.com/series?all"
 * - [source-names]: Comma-separated display names for each available time range, used in the expand-chart dropdown button.
 *                   Eg. source-names="1d,5d,30d".
 * - [downloads]:    Same as sources, but for download only. Defaults to the same as sources.
 *                   Eg. sources="http://example.com/series?offset=1d,http://example.com/series?offset=5d,http://example.com/series?all"
 * - [download-names]: Same as source-names, but for download only. Defaults to the same as source-names.
 *                   Eg. source-names="1d,5d,30d,max".
 * Or:
 * - [src]:          The URL of the data to show in the chart panel, once "expand" is clicked. Eg. src="http://example.com/full_time_series.csv".
 * - [src-preview]:  The URL of the data to show in the feature info panel. Defaults to src. Eg. src-preview="http://example.com/preview_time_series.csv".
 * Or:
 * - [data]:         csv-formatted data, with \n for newlines. Eg. data="time,a,b\n2016-01-01,2,3\n2016-01-02,5,6".
 *                   or json-formatted string data, with \quot; for quotes, eg. data="[[\quot;a\quot;,\quot;b\quot;],[2,3],[5,6]]".
 * Or:
 * - None of the above, but supply csv or json-formatted data as the content of the chart data, with \n for newlines.
 *                   Eg. <chart>time,a,b\n2016-01-01,2,3\n2016-01-02,5,6</chart>.
 *                   or  <chart>[["x","y","z"],[1,10,3],[2,15,9],[3,8,12],[5,25,4]]</chart>.
 *
 *
 * See CustomComponentType for more details.
 */
const registerCustomComponentTypes = function(terria) {

    /**
     * @private
     */
    function processChartNode(context, node, children) {
        checkAllPropertyKeys(node.attribs, chartAttributes);
        const columnNames = splitStringIfDefined(node.attribs['column-names']);
        const columnUnits = splitStringIfDefined(node.attribs['column-units']);
        const styling = node.attribs['styling'] || 'feature-info';

        // Present src and src-preview as if they came from sources.
        let sources = splitStringIfDefined(node.attribs.sources);
        const sourceNames = splitStringIfDefined(node.attribs['source-names']);
        if (!defined(sources) && defined(node.attribs.src)) {
            // [src-preview, src], or [src] if src-preview is not defined.
            sources = [node.attribs.src];
            if (defined(node.attribs['src-preview'])) {
                sources.unshift(node.attribs['src-preview']);
            }
        }
        const downloads = splitStringIfDefined(node.attribs.downloads) || sources;
        const downloadNames = splitStringIfDefined(node.attribs['download-names']) || sourceNames;
        const pollSources = splitStringIfDefined(node.attribs['poll-sources']);

        const id = node.attribs.id;
        const xColumn = node.attribs['x-column'];
        let yColumns = splitStringIfDefined(node.attribs['y-columns']);
        if (!defined(yColumns) && defined(node.attribs['y-column'])) {
            yColumns = [node.attribs['y-column']];
        }
        const url = defined(sources) ? sources[0] : undefined;
        const tableStructure = tableStructureFromStringData(getSourceData(node, children));

        const colors = splitStringIfDefined(node.attribs['colors']);
        const title = node.attribs['title'];
        const updateCounterKeyProps = {
            url: url,
            xColumn: xColumn,
            yColumns: yColumns
        };
        const updateCounter = CustomComponents.getUpdateCounter(context.updateCounters, Chart, updateCounterKeyProps);

        // If any of these attributes change, change the key so that React knows to re-render the chart.
        const reactKeys = [
            title || '',
            id || '',
            (sources && sources.join('|')) || '',
            xColumn || '',
            (defined(yColumns) ? yColumns.join('|') : ''),
            colors || ''
        ];

        const chartElements = [];
        if (node.attribs['hide-buttons'] !== 'true') {
            chartElements.push(React.createElement(ChartExpandAndDownloadButtons, {
                key: 'button',
                terria: terria,
                catalogItem: context.catalogItem,
                title: title,
                colors: colors,  // The colors are used when the chart is expanded.
                feature: context.feature,
                sources: sources,
                sourceNames: sourceNames,
                downloads: downloads,
                downloadNames: downloadNames,
                tableStructure: tableStructure,
                columnNames: columnNames,
                columnUnits: columnUnits,
                xColumn: node.attribs['x-column'],
                yColumns: yColumns,
                id: id,
                canDownload: !(node.attribs['can-download'] === 'false'),
                raiseToTitle: !!getInsertedTitle(node),
                pollSources: pollSources,
                pollSeconds: node.attribs['poll-seconds'],
                pollReplace: node.attribs['poll-replace'] === 'true',
                updateCounter: updateCounter  // Change this to trigger an update.
            }));
        }

        chartElements.push(React.createElement(Chart, {
            key: 'chart',
            axisLabel: {
                x: node.attribs['preview-x-label'],
                y: undefined
            },
            catalogItem: context.catalogItem,
            url: url,
            tableStructure: tableStructure,
            xColumn: node.attribs['x-column'],
            yColumns: yColumns,
            styling: styling,
            highlightX: node.attribs['highlight-x'],
            // colors: colors,  // Note that the preview chart doesn't show the colors.
            pollUrl: defined(pollSources) ? pollSources[0] : undefined,
            pollSeconds: node.attribs['poll-seconds'],  // This is unorthodox: this prop is picked up not by Chart.jsx, but in selfUpdateSeconds below.
            // pollIdColumns: node.attribs['poll-id-columns'],  // TODO: implement.
            // pollReplace: (node.attribs['poll-replace'] === 'true'),
            updateCounter: updateCounter,
            transitionDuration: 300
        }));

        return React.createElement('div', {
            key: (reactKeys.join('-')) || 'chart-wrapper',
            className: ChartPreviewStyles.previewChartWrapper
        }, chartElements);
    }

    /**
     * Returns the 'data' attribute if available, otherwise the child of this node.
     * @private
     */
    function getSourceData(node, children) {
        const sourceData = node.attribs['data'];
        if (sourceData) {
            return sourceData;
        }
        if (Array.isArray(children) && children.length > 0) {
            return children[0];
        }
        return children;
    }

    /**
     * This function does not activate any columns in itself.
     * That should be done by TableCatalogItem when it is created around this.
     * @private
     */
    function tableStructureFromStringData(stringData) {
        // sourceData can be either json (starts with a '[') or csv format (contains a true line feed or '\n'; \n is replaced with a real linefeed).
        if (!defined(stringData) || stringData.length < 2) {
            return;
        }
        // We prevent ALT, LON and LAT from being chosen, since we know this is a non-geo csv already.
        const result = new TableStructure('chart', {unallowedTypes: [VarType.ALT, VarType.LAT, VarType.LON]});
        if (stringData[0] === '[') {
            // Treat as json.
            const json = JSON.parse(stringData.replace(/&quot;/g, '"'));
            return TableStructure.fromJson(json, result);
        }
        if (stringData.indexOf('\\n') >=0 || stringData.indexOf('\n') >= 0) {
            // Treat as csv.
            return TableStructure.fromCsv(stringData.replace(/\\n/g, '\n'), result);
        }
    }

    /**
     * @private
     */
    function getInsertedTitle(node) {
        // Check if there is a title in the position 'Title' relative to node <chart>:
        // <tr><td>Title</td><td><chart></chart></tr>
        if (defined(node.parent) && (node.parent.name === 'td')
                && defined(node.parent.parent) && (node.parent.parent.name === 'tr')
                && defined(node.parent.parent.children[0])
                && defined(node.parent.parent.children[0].children[0])) {
            return node.parent.parent.children[0].children[0].data;
        }
    }

    const chartComponentType = new CustomComponentType({
        name: 'chart',
        attributes: chartAttributes,
        processNode: processChartNode,
        furtherProcessing: [
            //
            // These replacements reformat <chart>s defined directly in a csv, so they take the full width of the 2-column table,
            // and present the column name as the title.
            // It replaces:
            // <tr><td>Title</td><td><chart></chart></tr>
            // with:
            // <tr><td colSpan:2><div class="chart-title">Title</div><chart></chart></tr>
            //
            {
                shouldProcessNode: node => (
                    // If this node is a <chart> in the second column of a 2-column table,
                    // then add a title taken from the first column, and give it colSpan 2.
                    node.name === 'td'
                    && node.children.length === 1
                    && node.children[0].name === 'chart'
                    && node.parent.name === 'tr'
                    && node.parent.children.length === 2
                ),
                processNode: (context, node, children) => { // eslint-disable-line react/display-name
                    const title = node.parent.children[0].children[0].data;
                    const revisedChildren = [
                        React.createElement('div', {key: 'title', className: ChartPreviewStyles.chartTitleFromTable}, title)
                    ].concat(children);
                    return React.createElement(node.name, {key: 'chart', colSpan: 2, className: ChartPreviewStyles.chartTd}, node.data, revisedChildren);
                }
            },
            {
                shouldProcessNode: node => (
                    // If this node is in the first column of a 2-column table, and the second column is a <chart>,
                    // then remove it.
                    node.name === 'td'
                    && node.children.length === 1
                    && node.parent.name === 'tr'
                    && node.parent.children.length === 2
                    && node.parent.children[1].name === 'td'
                    && node.parent.children[1].children.length === 1
                    && node.parent.children[1].children[0].name === 'chart'
                ),
                processNode: function() {
                    return;  // Do not return a node.
                }
            }
        ],
    /*
     * isCorresponding is a function which checks a ReactComponent and returns a Boolean,
     * indicating if that react component corresponds to this type.
     * "Correspondence" is whatever the component wants it to be, but must be consistent with selfUpdateSeconds.
     */
        isCorresponding: function(reactComponent) {
            return reactComponent.type === Chart;
        },
        selfUpdateSeconds: function(reactComponent) {
            return reactComponent.props.pollSeconds;  // Note this is unorthodox.
        }
    });
    if (!defined(terria)) {
        // The chart expand button needs a reference to the Terria instance to add the chart to the catalog.
        throw new DeveloperError('Terria is a required argument of registerCustomComponentTypes.');
    }
    CustomComponents.register(chartComponentType);

    const collapsibleComponentType = new CustomComponentType({
        name: 'collapsible',
        attributes: ['title', 'open'],
        processNode: function(context, node, children) {
            return React.createElement(Collapsible, {
                key: node.attribs.title,
                displayName: node.attribs.title,
                title: node.attribs.title,
                startsOpen: typeof node.attribs.open === 'string' ? JSON.parse(node.attribs.open) : undefined
            }, children);
        }
    });
    CustomComponents.register(collapsibleComponentType);

};

/**
 * @private
 */
function checkAllPropertyKeys(object, allowedKeys) {
    for (const key in object) {
        if (object.hasOwnProperty(key)) {
            if (allowedKeys.indexOf(key) === -1) {
                console.log('Unknown attribute ' + key);
                throw new DeveloperError('Unknown attribute ' + key);
            }
        }
    }
}

module.exports = registerCustomComponentTypes;