1 files changed, 8 insertions, 5 deletions
diff --git a/client/socket.js b/client/socket.js
index d94a968..9c35af5 100644
--- a/client/socket.js
+++ b/client/socket.js
@@ -1,4 +1,4 @@
-import { updateFile } from './actions'
+import { updateFile, updateTask, addFile } from './actions'
 import store from './store'
 
 const socket = io(window.location.origin)
@@ -9,13 +9,16 @@ socket.on('connect', (data) => {
 socket.on('worker', (data) => {
   console.log('worker connected', data)
 })
-socket.on('processed', (data) => {
-  console.log('processed', data)
+socket.on('updateFile', (data) => {
+  console.log('updateFile', data)
   store.dispatch(updateFile(data.file))
 })
-socket.on('completed', (data) => {
-  console.log('completed', data)
+socket.on('updateTask', (data) => {
+  console.log('updateTask', data)
   store.dispatch(updateTask(data.task))
+  if (data.task.output_file) {
+    store.dispatch(addFile(data.task.output_file))
+  }
 })
 
 export default socket