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'

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')
    console.log('load dataset:', id)
    if (id === 'new') return
    if (id) {
      localStorage.setItem('samplernn.last_id', id)
      if (! props.samplernn.folder || props.samplernn.folder.id !== id) {
        console.log('load directories')
        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, match, history } = this.props
    const folder = samplernn.folder
    if (match.params.id === 'new') {
      return <NewDatasetForm module={{name: 'samplernn', datatype: 'audio'}} history={history} />        
    }
    if (samplernn.loading) {
      return <span>Loading</span>
    }
    if (!folder || !samplernn.data.folders.length) {
      return history.push('/samplernn/datasets/new/')
    }
    return (
      <div className='app'>
        <div class='heading'>
          <h1>{folder.name}</h1>
        </div>
        {folder.name !== 'unsorted' &&
          <DatasetForm
            title='Add Files'
            folder={folder}
            canUpload canAddURL
          />
        }
        {this.renderDataset()}
      </div>
    )
  }
  renderDataset(){
    return (
      <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>
        {this.renderGroups()}
      </div>
    )
  }
  renderGroups(){
    const { samplernn } = this.props
    const folder = samplernn.folder
    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 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)