insomnia/packages/insomnia-app/app/ui/components/export-requests/tree.js

// @flow
import * as React from 'react';
import RequestRow from './request-row';
import RequestGroupRow from './request-group-row';
import * as models from '../../../models/index';
import type { Node } from '../modals/export-requests-modal';
import type { Request } from '../../../models/request';
import type { RequestGroup } from '../../../models/request-group';

type Props = {
  root: ?Node,
  handleSetRequestGroupCollapsed: Function,
  handleSetItemSelected: Function,
};

class Tree extends React.PureComponent<Props> {
  renderChildren(node: ?Node) {
    if (node == null) {
      return null;
    }

    if (node.doc.type === models.request.type) {
      // Directly cast to Request will result in error, so cast it to any first.
      const request: Request = ((node.doc: any): Request);
      return (
        <RequestRow
          key={node.doc._id}
          handleSetItemSelected={this.props.handleSetItemSelected}
          isSelected={node.selectedRequests === node.totalRequests}
          request={request}
        />
      );
    }

    if (node.totalRequests === 0) {
      // Don't show empty folders.
      return null;
    }

    const children = node.children.map(child => this.renderChildren(child));

    // Directly cast to RequestGroup will result in error, so cast it to any first.
    const requestGroup: RequestGroup = ((node.doc: any): RequestGroup);
    return (
      <RequestGroupRow
        key={node.doc._id}
        handleSetRequestGroupCollapsed={this.props.handleSetRequestGroupCollapsed}
        handleSetItemSelected={this.props.handleSetItemSelected}
        isCollapsed={node.collapsed}
        totalRequests={node.totalRequests}
        selectedRequests={node.selectedRequests}
        requestGroup={requestGroup}
        children={children}
      />
    );
  }

  render() {
    const { root } = this.props;
    return (
      <ul className="tree__list tree__list-root theme--tree__list">{this.renderChildren(root)}</ul>
    );
  }
}

export default Tree;