Merge pull request #6 from EvanSimpson/allow-any-root

Allow any node type to be root node

Author: chibicode

src/grab-node.js

@@ -8,14 +8,14 @@ import JSONNumberNode from './JSONNumberNode';
 import JSONBooleanNode from './JSONBooleanNode';
 import JSONNullNode from './JSONNullNode';
 
-export default function(key, value, prevValue, theme) {
+export default function(key, value, prevValue, theme, initialExpanded = false) {
   const nodeType = objType(value);
   if (nodeType === 'Object') {
-    return <JSONObjectNode data={value} previousData={prevValue} theme={theme} keyName={key} key={key} />;
+    return <JSONObjectNode data={value} previousData={prevValue} theme={theme} initialExpanded={initialExpanded} keyName={key} key={key} />;
   } else if (nodeType === 'Array') {
-    return <JSONArrayNode data={value} previousData={prevValue} theme={theme} keyName={key} key={key} />;
+    return <JSONArrayNode data={value} previousData={prevValue} theme={theme} initialExpanded={initialExpanded} keyName={key} key={key} />;
   } else if (nodeType === 'Iterable') {
-    return <JSONIterableNode data={value} previousData={prevValue} theme={theme} keyName={key} key={key} />;
+    return <JSONIterableNode data={value} previousData={prevValue} theme={theme} initialExpanded={initialExpanded} keyName={key} key={key} />;
   } else if (nodeType === 'String') {
     return <JSONStringNode keyName={key} previousValue={prevValue} theme={theme} value={value} key={key} />;
   } else if (nodeType === 'Number') {

src/index.js

@@ -5,6 +5,7 @@
 
 import React from 'react';
 import objectType from './obj-type';
+import grabNode from './grab-node';
 import JSONObjectNode from './JSONObjectNode';
 import JSONArrayNode from './JSONArrayNode';
 import solarized from './themes/solarized';
@@ -42,13 +43,8 @@ export default class JSONTree extends React.Component {
 
   render() {
     const nodeType = objectType(this.props.data);
-    let rootNode = false;
     const keyName = this.props.keyName || 'root';
-    if (nodeType === 'Object') {
-      rootNode = <JSONObjectNode theme={this.props.theme} data={this.props.data} previousData={this.props.previousData} keyName={keyName} initialExpanded={true} />;
-    } else if (nodeType === 'Array') {
-      rootNode = <JSONArrayNode theme={this.props.theme} data={this.props.data} previousData={this.props.previousData} initialExpanded={true} keyName={keyName} />;
-    }
+    const rootNode = grabNode(keyName, this.props.data, this.props.previousData, this.props.theme, true);
     return (
       <ul style={{
         ...styles.tree,