train link....!

1 files changed, 29 insertions, 5 deletions

diff --git a/app/client/modules/samplernn/datasets.component.js b/app/client/modules/samplernn/datasets.component.js
index 86ecc21..efbf992 100644
--- a/app/client/modules/samplernn/datasets.component.js
+++ b/app/client/modules/samplernn/datasets.component.js
@@ -3,6 +3,8 @@ import { bindActionCreators } from 'redux'
 import { connect } from 'react-redux'
 
 import { actions } from '../../api'
+import * as taskActions from '../../task/task.actions'
+import * as systemActions from '../../system/system.actions'
   // folderActions.index({ module: 'samplernn' })
   // folderActions.show(12)
   // folderActions.create({ module: 'samplernn', name: 'foo' })
@@ -25,6 +27,7 @@ class SampleRNNDatasets extends Component {
     this.handleName = this.handleName.bind(this)
     this.handleUpload = this.handleUpload.bind(this)
     this.handleURL = this.handleURL.bind(this)
+    this.fetchURL = this.fetchURL.bind(this)
     this.fileOptions = this.fileOptions.bind(this)
     props.actions.folder.index({ module: 'samplernn' })
     props.actions.file.index({ module: 'samplernn' })
@@ -78,12 +81,30 @@ class SampleRNNDatasets extends Component {
       url
     })
   }
+  fetchURL(url) {
+    actions.task.start_task({
+      activity: 'cpu',
+      module: 'samplernn',
+      epochs: 1,
+      opt: { url }
+    }, { preempt: true, watch: true })
+  }
   fileOptions(file){
     console.log(file)
     if (file.activity === 'url' && !file.dataset) {
-      return (
-        <div className='gray'>fetching...</div>
-      )
+      if (this.props.runner.cpu.status !== 'IDLE') {
+        return (
+          <div className='gray'>
+            fetching...
+          </div>
+        )
+      } else {
+        return (
+          <div className='link' onClick={this.fetchURL}>
+            fetch
+          </div>
+        )
+      }
     }
     return (
       <div>
@@ -138,14 +159,17 @@ class SampleRNNDatasets extends Component {
 }
 
 const mapStateToProps = state => ({
-  samplernn: state.module.samplernn
+  samplernn: state.module.samplernn,
+  runner: state.system.runner,
+  task: state.task,
 })
 
 const mapDispatchToProps = (dispatch, ownProps) => ({
   actions: {
     folder: bindActionCreators(actions.folder, dispatch),
     file: bindActionCreators(actions.file, dispatch),
-    task: bindActionCreators(actions.task, dispatch),
+    task: bindActionCreators(taskActions, dispatch),
+    system: bindActionCreators(systemActions, dispatch),
   }
 })