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 { FileList, FileRow } from '../../common/fileList.component'

class SampleRNNDatasets extends Component {
  constructor(props){
    super()
    this.fileOptions = this.fileOptions.bind(this)
    this.pickFile = this.pickFile.bind(this)
    let id = props.match.params.id || localStorage.getItem('samplernn.last_id')
    if (! id && props.location.pathname.match(/\/new\//)) {
      id = 'new'
    } else if (id) {
      localStorage.setItem('samplernn.last_id', id)
    }
    if (id && (! props.samplernn.folder || props.samplernn.folder.id !== id)) {
      props.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 } = this.props
    const folder = samplernn.folder
    if (!folder.name) return
    const { mapFn, sortFn } = util.sort.orderByFn('date desc')
    const datasets = folder.datasets.map(mapFn).sort(sortFn).map(pair => {
      const dataset = pair[1]
      return (
        <div className='row dataset'>
          <div className='col'>
            {!!dataset.input.length &&
              <FileList
                files={dataset.input}
                className='input_files'
                fileListClassName=''
                rowClassName='input_file'
              />
            }
          </div>
          <div className='col quiet'>
            <div>{dataset.isBuilt ? 'cached' : ''}</div>
          </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'}
              />
            }
          </div>
        </div>
      )
    })
    return (
      <div className='app'>
        <DatasetForm folder={folder} canRename canUpload canAddURL />
        <div class='rows params datasets'>
          <div class='row dataset'>
            <div class='col'>input</div>
            <div class='col'></div>
            <div class='col'>checkpoint</div>
            <div class='col'>output</div>
          </div>
          {datasets}
        </div>
      </div>
    )
  }
}

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)