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

import * as pix2pixhdActions from '../pix2pixhd.actions'
import * as pix2pixhdTasks from '../pix2pixhd.tasks'

import { Loading, CurrentTask, FileList, FileRow } from '../../../common'
import DatasetForm from '../../../dataset/dataset.form'
import NewDatasetForm from '../../../dataset/dataset.new'
import UploadStatus from '../../../dataset/upload.status'

import DatasetComponent from '../../../dataset/dataset.component'

import pix2pixhdModule from '../pix2pixhd.module'

class Pix2PixHDShow extends Component {
  constructor(props){
    super(props)
    this.datasetActions = this.datasetActions.bind(this)
  }
  componentWillMount(){
    const id = this.props.match.params.id || localStorage.getItem('pix2pixhd.last_id')
    console.log('load dataset:', id)
    const { match, pix2pixhd, actions } = this.props
    if (id === 'new') return
    if (id) {
      if (parseInt(id)) localStorage.setItem('pix2pixhd.last_id', id)
      if (! pix2pixhd.folder || pix2pixhd.folder.id !== id) {
        actions.load_directories(id)
      }
    } else {
      this.props.history.push('/pix2pixhd/new/')
    }
  }
  render(){
    const { pix2pixhd, match, history } = this.props
    const { folderLookup } = (pix2pixhd.data || {})
    const folder = (folderLookup || {})[pix2pixhd.folder_id] || {}

    return (
      <div className='app pix2pixhd'>
        <div class='heading'>
          <div class='spaced'>
            <h1>{folder ? folder.name : <Loading />}</h1>
            <UploadStatus />
          </div>
        </div>
        <div className='row'>
        {folder && folder.name && folder.name !== 'unsorted' &&
          <DatasetForm
            title='Add Files'
            module={pix2pixhdModule}
            folder={folder}
            canUpload canAddURL
          />
        }
          <div>
            <UploadStatus />
            <CurrentTask />
          </div>
        </div>

        <DatasetComponent
          loading={pix2pixhd.loading}
          progress={pix2pixhd.progress}
          id={pix2pixhd.folder_id}
          module={pix2pixhdModule}
          data={pix2pixhd.data}
          folder={folder}
          history={history}
          onPickFile={(file, e) => {
            e.preventDefault()
            e.stopPropagation()
            console.log('picked a file', file)
          }}
          datasetActions={this.datasetActions}
        />
      </div>
    )
  }
  datasetActions(dataset, isFetching=false, isProcessing=false){
    const { pix2pixhd, remote } = this.props
    const input = pix2pixhd.data.fileLookup[dataset.input[0]]
    if (! input) return null
    if (input.name && input.name.match(/(gif|jpe?g|png)$/i)) return null
    return (
      <div>
        <div class={'actions'}>
          <span class='link' onClick={() => remote.train_task(dataset, pix2pixhd.folder_id, 1)}>train</span>
          <span class='link' onClick={() => remote.train_task(dataset, pix2pixhd.folder_id, 5)}>5x</span>
          <span class='link' onClick={() => remote.train_task(dataset, pix2pixhd.folder_id, 10)}>10x</span>
          <span class='link' onClick={() => remote.train_task(dataset, pix2pixhd.folder_id, 20)}>20x</span>
          <span class='link' onClick={() => remote.train_task(dataset, pix2pixhd.folder_id, 50)}>50x</span>
        </div>
        {dataset.isBuilt
        ? <div class='subtext'>
            {'fetched '}
            <span class='link' onClick={() => remote.clear_cache_task(dataset)}>rm</span>
          </div>
        : isFetching
        ? <div class='subtext'>
            {'fetching'}
          </div>
        : <div class='subtext'>
            <span class='link' onClick={() => remote.fetch_task(input.url, input.id, dataset.name)}>fetch</span>
          </div>
        }
      </div>
    )
  }
}

const mapStateToProps = state => ({
  pix2pixhd: state.module.pix2pixhd,
})

const mapDispatchToProps = (dispatch, ownProps) => ({
  actions: bindActionCreators(pix2pixhdActions, dispatch),
  remote: bindActionCreators(pix2pixhdTasks, dispatch),
})

export default connect(mapStateToProps, mapDispatchToProps)(Pix2PixHDShow)