import { h, Component } from 'preact'
import { bindActionCreators } from 'redux'
import { connect } from 'react-redux'
import * as util from '../../util'

import * as samplernnActions from './samplernn.actions'

import DatasetForm from '../../dataset/dataset.form'
import NewDatasetForm from '../../dataset/dataset.new'
import { FileList, FileRow } from '../../common/fileList.component'
import Loading from '../../common/loading.component'

const samplernnModule = {
  name: 'samplernn',
  datatype: 'audio',
}

class SampleRNNDatasets extends Component {
  constructor(props){
    super(props)
    this.fileOptions = this.fileOptions.bind(this)
    this.pickFile = this.pickFile.bind(this)
  }
  componentWillMount(){
    const id = this.props.id
    console.log('load dataset:', id, this.props.id)
    const { match, samplernn, actions } = this.props
    if (id === 'new') return
    if (id) {
      if (parseInt(id)) localStorage.setItem('samplernn.last_id', id)
      if (! samplernn.folder || samplernn.folder.id !== id) {
        actions.load_directories(id)
      }
    }
  }
  pickFile(file){
    console.log('pick', file)
  }
  fileOptions(file){
    if (file.activity === 'url' && !file.dataset) {
      if (this.props.runner.cpu.status !== 'IDLE') {
        return (
          <div className='gray'>
            fetching...
          </div>
        )
      } else {
        return (
          <div className='link' onClick={() => this.fetchURL(file.url)}>
            fetch
          </div>
        )
      }
    }
    return (
      <div>
        <div className='link' onClick={() => this.train(file)}>train</div>
        {file.epoch == 0 && <div className='epochs'>{file.epochs} ep.</div>}
      </div>
    )
  }
  render(){
    const { samplernn, match, history } = this.props
    if (samplernn.loading) {
      return <Loading progress={samplernn.progress} />
    }
    if (!samplernn.data.folders.length) {
      console.log('no folders, redirect to /new')
      return history.push('/samplernn/new/')
    }
    const folder = samplernn.folder
    if (!folder || !folder.name) return
    return (
      <div class='rows params datasets'>
        <div class='row row-heading dataset'>
          <div class='col'>input</div>
          <div class='col'>status</div>
          <div class='col'>checkpoint</div>
          <div class='col'>output</div>
        </div>
        {this.renderGroups()}
      </div>
    )
  }
  renderGroups(){
    const { samplernn, runner, onPickDataset, onPickFile, datasetActions } = this.props
    const folder = samplernn.folder
    const { mapFn, sortFn } = util.sort.orderByFn('date desc')
    const moduleOnGPU = runner.gpu.task.module === samplernnModule.name
    const datasets = folder.datasets.map(mapFn).sort(sortFn).map(pair => {
      const dataset = pair[1]
      const isProcessing = moduleOnGPU && runner.gpu.task.dataset === dataset
      const status = isProcessing ? util.gerund(runner.gpu.task.activity) : ''
      return (
        <div className='row dataset' onClick={() => onPickDataset && onPickDataset(dataset)}>
          {this.props.beforeRow && this.props.beforeRow(dataset)}
          <div className='col'>
            {!!dataset.input.length &&
              <FileList
                files={dataset.input}
                className='input_files'
                fileListClassName=''
                rowClassName='input_file'
                options={this.fileOptions}
                onClick={onPickFile}
              />
            }
          </div>
          <div className={'col quiet ' + (dataset.isBuilt ? 'built' : 'not_built')}>
            {status}
            {this.props.datasetActions && this.props.datasetActions(dataset)}
          </div>
          <div className='col checkpoint'>
            {!!dataset.checkpoints.length &&
              <FileRow
                file={dataset.checkpoints[0]}
                fields={'name date epoch'}
                className='row checkpoint'
              />
            }
          </div>
          <div className='col'>
            {!!dataset.output.length &&
              <FileList
                files={dataset.output}
                orderBy='epoch desc'
                fields={'name date epoch size'}
                onPickFile={onPickFile}
              />
            }
          </div>
          {this.props.afterRow && this.props.afterRow(dataset)}
        </div>
      )
    })
    return datasets
  }
}

const mapStateToProps = state => ({
  samplernn: state.module.samplernn,
  runner: state.system.runner,
  task: state.task,
})

const mapDispatchToProps = (dispatch, ownProps) => ({
  actions: bindActionCreators(samplernnActions, dispatch),
})

export default connect(mapStateToProps, mapDispatchToProps)(SampleRNNDatasets)