lbry-desktop/ui/component/common/file-exporter.jsx

// @flow
import * as ICONS from 'constants/icons';

import React from 'react';
import Button from 'component/button';
import parseData from 'util/parse-data';
import { remote } from 'electron';
import path from 'path';
// @if TARGET='app'
import fs from 'fs';
// @endif

type Props = {
  data: Array<any>,
  title: string,
  label: string,
  defaultPath?: string,
  filters: Array<string>,
  onFileCreated?: string => void,
  disabled: boolean,
};

class FileExporter extends React.PureComponent<Props> {
  static defaultProps = {
    filters: [],
  };

  constructor() {
    super();
    (this: any).handleButtonClick = this.handleButtonClick.bind(this);
  }

  handleFileCreation(filename: string, data: any) {
    const { onFileCreated } = this.props;
    // @if TARGET='app'
    fs.writeFile(filename, data, err => {
      if (err) throw err;
      // Do something after creation

      if (onFileCreated) {
        onFileCreated(filename);
      }
    });
    // @endif
  }

  handleButtonClick() {
    const { title, data, defaultPath, filters } = this.props;

    const options = {
      title,
      defaultPath,
      filters: [
        {
          name: 'CSV',
          extensions: ['csv'],
        },
        {
          name: 'JSON',
          extensions: ['json'],
        },
      ],
    };

    remote.dialog.showSaveDialog(remote.getCurrentWindow(), options, filename => {
      // User hit cancel so do nothing:
      if (!filename) return;
      // Get extension and remove initial dot
      // @if TARGET='app'
      const format = path.extname(filename).replace(/\./g, '');
      // @endif
      // Parse data to string with the chosen format
      const parsed = parseData(data, format, filters);
      // Write file
      if (parsed) {
        this.handleFileCreation(filename, parsed);
      }
    });
  }

  render() {
    const { label, disabled } = this.props;
    return (
      <Button
        button="primary"
        disabled={disabled}
        icon={ICONS.DOWNLOAD}
        label={label || __('Export')}
        onClick={this.handleButtonClick}
      />
    );
  }
}

export default FileExporter;