/** * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * KIND, either express or implied. See the License for the * specific language governing permissions and limitations * under the License. */ import React from 'react'; import PropTypes from 'prop-types'; import { OverlayTrigger, Popover, ListGroup, ListGroupItem } from 'react-bootstrap'; import { connect } from 'react-redux'; import { t } from '@superset-ui/translation'; import { getChartKey } from '../../exploreUtils'; import { runAnnotationQuery } from '../../../chart/chartAction'; import InfoTooltipWithTrigger from '../../../components/InfoTooltipWithTrigger'; import AnnotationLayer from './AnnotationLayer'; const propTypes = { colorScheme: PropTypes.string.isRequired, annotationError: PropTypes.object, annotationQuery: PropTypes.object, vizType: PropTypes.string, validationErrors: PropTypes.array, name: PropTypes.string.isRequired, actions: PropTypes.object, value: PropTypes.arrayOf(PropTypes.object), onChange: PropTypes.func, refreshAnnotationData: PropTypes.func, }; const defaultProps = { vizType: '', value: [], annotationError: {}, annotationQuery: {}, onChange: () => {}, }; class AnnotationLayerControl extends React.PureComponent { constructor(props) { super(props); this.addAnnotationLayer = this.addAnnotationLayer.bind(this); this.removeAnnotationLayer = this.removeAnnotationLayer.bind(this); } componentWillReceiveProps(nextProps) { const { name, annotationError, validationErrors, value } = nextProps; if (Object.keys(annotationError).length && !validationErrors.length) { this.props.actions.setControlValue(name, value, Object.keys(annotationError)); } if (!Object.keys(annotationError).length && validationErrors.length) { this.props.actions.setControlValue(name, value, []); } } addAnnotationLayer(annotationLayer) { const annotation = annotationLayer; let annotations = this.props.value.slice(); const i = annotations.findIndex(x => x.name === (annotation.oldName || annotation.name)); delete annotation.oldName; if (i > -1) { annotations[i] = annotation; } else { annotations = annotations.concat(annotation); } this.props.refreshAnnotationData(annotation); this.props.onChange(annotations); } removeAnnotationLayer(annotation) { const annotations = this.props.value.slice() .filter(x => x.name !== annotation.oldName); this.props.onChange(annotations); } renderPopover(parent, annotation, error) { const id = !annotation ? '_new' : annotation.name; return ( this.refs[parent].hide()} /> ); } renderInfo(anno) { const { annotationError, annotationQuery } = this.props; if (annotationQuery[anno.name]) { return ( ); } if (annotationError[anno.name]) { return ( ); } if (!anno.show) { return Hidden ; } return ''; } render() { const annotations = this.props.value.map((anno, i) => ( {anno.name} {this.renderInfo(anno)} )); return (
{annotations}   {t('Add Annotation Layer')}
); } } AnnotationLayerControl.propTypes = propTypes; AnnotationLayerControl.defaultProps = defaultProps; // Tried to hook this up through stores/control.jsx instead of using redux // directly, could not figure out how to get access to the color_scheme function mapStateToProps({ charts, explore }) { const chartKey = getChartKey(explore); const chart = charts[chartKey] || charts[0] || {}; return { colorScheme: (explore.controls || {}).color_scheme.value, annotationError: chart.annotationError, annotationQuery: chart.annotationQuery, vizType: explore.controls.viz_type.value, }; } function mapDispatchToProps(dispatch) { return { refreshAnnotationData: annotationLayer => dispatch(runAnnotationQuery(annotationLayer)), }; } export default connect(mapStateToProps, mapDispatchToProps)(AnnotationLayerControl);