1 files changed, 24 insertions, 12 deletions
diff --git a/app/client/modules/samplernn/index.js b/app/client/modules/samplernn/index.js
index 2a9d64d..85c99e4 100644
--- a/app/client/modules/samplernn/index.js
+++ b/app/client/modules/samplernn/index.js
@@ -1,5 +1,8 @@
 import { h, Component } from 'preact'
 import { Route, Link } from 'react-router-dom'
+
+import actions from '../../actions'
+
 import util from '../../util'
 
 import SampleRNNNew from './views/samplernn.new'
@@ -8,18 +11,27 @@ import SampleRNNImport from './views/samplernn.import'
 import SampleRNNResults from './views/samplernn.results'
 import SampleRNNGraph from './views/samplernn.graph'
 
-function router () {
-  document.body.style.backgroundImage = 'linear-gradient(' + (util.randint(40)+40) + 'deg, #eef, #fef)'
-  return (
-    <section>
-      <Route exact path='/samplernn/new' component={SampleRNNNew} />
-      <Route exact path='/samplernn/datasets' component={SampleRNNShow} />
-      <Route exact path='/samplernn/datasets/:id' component={SampleRNNShow} />
-      <Route exact path='/samplernn/import' component={SampleRNNImport} />
-      <Route exact path='/samplernn/results' component={SampleRNNResults} />
-      <Route exact path='/samplernn/graph' component={SampleRNNGraph} />
-    </section>
-  )
+class router {
+  componentWillMount(){
+    actions.system.changeTool('samplernn')
+    document.body.style.backgroundImage = 'linear-gradient(' + (util.randint(40)+40) + 'deg, #eef, #fef)'
+  }
+  componentWillReceiveProps(){
+    actions.system.changeTool('samplernn')
+    document.body.style.backgroundImage = 'linear-gradient(' + (util.randint(40)+40) + 'deg, #eef, #fef)'
+  }
+  render(){
+    return (
+      <section>
+        <Route exact path='/samplernn/new' component={SampleRNNNew} />
+        <Route exact path='/samplernn/datasets' component={SampleRNNShow} />
+        <Route exact path='/samplernn/datasets/:id' component={SampleRNNShow} />
+        <Route exact path='/samplernn/import' component={SampleRNNImport} />
+        <Route exact path='/samplernn/results' component={SampleRNNResults} />
+        <Route exact path='/samplernn/graph' component={SampleRNNGraph} />
+      </section>
+    )
+  }
 }
 
 function links(){