Add table columns if none specified

Author: dmt0

src/EditorControls.js

@@ -6,6 +6,7 @@ import {
   shamefullyClearAxisTypes,
   shamefullyAdjustAxisRef,
   shamefullyAdjustGeo,
+  shamefullyAddTableColumns,
 } from './shame';
 import {EDITOR_ACTIONS} from './lib/constants';
 import isNumeric from 'fast-isnumeric';
@@ -62,6 +63,7 @@ class EditorControls extends Component {
 
         shamefullyClearAxisTypes(graphDiv, payload);
         shamefullyAdjustAxisRef(graphDiv, payload);
+        shamefullyAddTableColumns(graphDiv, payload);
 
         for (let i = 0; i < payload.traceIndexes.length; i++) {
           for (const attr in payload.update) {

src/shame.js

@@ -82,3 +82,21 @@ export const shamefullyAdjustGeo = ({layout: {geo = {}}}, {update}) => {
     update['geo.center'] = {};
   }
 };
+
+export const shamefullyAddTableColumns = (graphDiv, {traceIndexes, update}) => {
+  if (
+    update['cells.values'] &&
+    (!graphDiv.data[traceIndexes[0]].header ||
+      !graphDiv.data[traceIndexes[0]].header.valuessrc)
+  ) {
+    update['header.values'] = update['cells.valuessrc'];
+  } else if (update['header.values'] === null) {
+    update['header.values'] =
+      graphDiv.data[traceIndexes[0]].cells.valuessrc || null;
+  } else if (
+    update['cells.values'] === null &&
+    !graphDiv.data[traceIndexes[0]].header.valuessrc
+  ) {
+    update['header.values'] = null;
+  }
+};