From 77b6e2cd2e796d063dbc6d42934143bf45bba8b0 Mon Sep 17 00:00:00 2001
From: vera-liu <vera.liu@airbnb.com>
Date: Thu, 3 Nov 2016 13:53:45 -0700
Subject: [PATCH] Get pivot table working in explore v2 (#1432)

* Use jquery calls in find() and html() of slice mock to
Get pivot_table viz working

* Borrowed code from caravel.js into error callback
---
 .../explorev2/components/ChartContainer.jsx   | 47 +++++++++++++++++--
 1 file changed, 42 insertions(+), 5 deletions(-)

diff --git a/caravel/assets/javascripts/explorev2/components/ChartContainer.jsx b/caravel/assets/javascripts/explorev2/components/ChartContainer.jsx
index 8c8cf38f2d21a..a70b0e1928a51 100644
--- a/caravel/assets/javascripts/explorev2/components/ChartContainer.jsx
+++ b/caravel/assets/javascripts/explorev2/components/ChartContainer.jsx
@@ -1,3 +1,4 @@
+import $ from 'jquery';
 import React, { PropTypes } from 'react';
 import { connect } from 'react-redux';
 import { Panel } from 'react-bootstrap';
@@ -8,12 +9,19 @@ const propTypes = {
   sliceName: PropTypes.string.isRequired,
   vizType: PropTypes.string.isRequired,
   height: PropTypes.string.isRequired,
-  sliceContainerId: PropTypes.string.isRequired,
+  containerId: PropTypes.string.isRequired,
   jsonEndpoint: PropTypes.string.isRequired,
   columnFormats: PropTypes.object,
 };
 
 class ChartContainer extends React.Component {
+  constructor(props) {
+    super(props);
+    this.state = {
+      selector: `#${this.props.containerId}`,
+    };
+  }
+
   componentDidMount() {
     this.renderVis();
   }
@@ -27,8 +35,9 @@ class ChartContainer extends React.Component {
       jsonEndpoint: () => this.props.jsonEndpoint,
 
       container: {
-        html: () => {
+        html: (data) => {
           // this should be a callback to clear the contents of the slice container
+          $(this.state.selector).html(data);
         },
 
         css: () => {
@@ -37,17 +46,45 @@ class ChartContainer extends React.Component {
           // should call callback to adjust height of chart
         },
         height: () => parseInt(this.props.height, 10) - 100,
+
+        find: (classname) => ($(this.state.selector).find(classname)),
+
       },
 
       width: () => this.chartContainerRef.getBoundingClientRect().width,
 
       height: () => parseInt(this.props.height, 10) - 100,
 
-      selector: `#${this.props.sliceContainerId}`,
+      selector: this.state.selector,
 
       done: () => {
         // finished rendering callback
       },
+
+      error(msg, xhr) {
+        let errorMsg = msg;
+        let errHtml = '';
+        try {
+          const o = JSON.parse(msg);
+          if (o.error) {
+            errorMsg = o.error;
+          }
+        } catch (e) {
+          // pass
+        }
+        errHtml = `<div class="alert alert-danger">${errorMsg}</div>`;
+        if (xhr) {
+          const extendedMsg = this.getErrorMsg(xhr);
+          if (extendedMsg) {
+            errHtml += `<div class="alert alert-danger">${extendedMsg}</div>`;
+          }
+        }
+        $(this.state.selector).html(errHtml);
+        this.render();
+        $('span.query').removeClass('disabled');
+        $('.query-and-save button').removeAttr('disabled');
+      },
+
       d3format: (col, number) => {
         // mock d3format function in Slice object in caravel.js
         const format = this.props.columnFormats[col];
@@ -76,7 +113,7 @@ class ChartContainer extends React.Component {
           }
         >
           <div
-            id={this.props.sliceContainerId}
+            id={this.props.containerId}
             ref={(ref) => { this.chartContainerRef = ref; }}
             className={this.props.vizType}
           />
@@ -92,7 +129,7 @@ function mapStateToProps(state) {
   return {
     sliceName: state.sliceName,
     vizType: state.viz.formData.vizType,
-    sliceContainerId: `slice-container-${state.viz.formData.sliceId}`,
+    containerId: `slice-container-${state.viz.formData.sliceId}`,
     jsonEndpoint: state.viz.jsonEndPoint,
     columnFormats: state.viz.columnFormats,
   };