Firmware upgrade now functional

tasks.coffee

@@ -1,4 +1,7 @@
+config = require './config'
 common = require './common'
+db = require './db'
+mongodb = require 'mongodb'
 BATCH_SIZE = 16
 
 exports.STATUS_QUEUED = STATUS_QUEUED = 0
@@ -8,60 +11,64 @@ exports.STATUS_FAULT = STATUS_FAULT = 3
 exports.STATUS_FINISHED = STATUS_FINISHED = 4
 
 
-this.init = (task, methodResponse, cwmpResponse, deviceUpdates) ->
+this.init = (task, methodResponse, callback) ->
   if not task.subtask?
-    task.subtask = {'name' : 'getParameterNames', 'parameterPath' : '', 'nextLevel' : false}
+    task.subtask = {name : 'getParameterNames', parameterPath : '', nextLevel : false}
 
   if task.subtask.name == 'getParameterNames'
     if methodResponse.faultcode?
       task.fault = methodResponse
-      return STATUS_FAULT
+      callback(null, STATUS_FAULT)
+      return
 
-    status = this.getParameterNames(task.subtask, methodResponse, cwmpResponse, deviceUpdates)
-    if status is STATUS_FINISHED
-      parameterNames = []
-      for p in deviceUpdates.parameterNames
-        if not common.endsWith(p[0], '.')
-          parameterNames.push(p[0])
-      task.subtask = {'name' : 'getParameterValues', 'parameterNames' : parameterNames}
-    else if status != STATUS_STARTED
-      throw Error('Unexpected subtask status')
-
-  if task.subtask.name == 'getParameterValues'
+    this.getParameterNames(task.subtask, methodResponse, (err, status, cwmpResponse, deviceUpdates) =>
+      if status is STATUS_FINISHED
+        parameterNames = []
+        for p in deviceUpdates.parameterNames
+          if not common.endsWith(p[0], '.')
+            parameterNames.push(p[0])
+        task.subtask = {name : 'getParameterValues', parameterNames : parameterNames}
+        this.getParameterValues(task.subtask, {}, (err, status, cwmpResponse, deviceUpdates) ->
+          callback(err, status, cwmpResponse, deviceUpdates)
+        )
+      else if status is STATUS_STARTED
+        callback(err, STATUS_STARTED, cwmpResponse, deviceUpdates)
+      else
+        throw Error('Unexpected subtask status')
+    )
+  else if task.subtask.name == 'getParameterValues'
     if methodResponse.faultcode?
       # Ignore GetParameterValues errors. A workaround for the crappy Seewon devices.
-      status = this.getParameterValues(task.subtask, {parameterList : {}}, cwmpResponse, deviceUpdates)
-    else
-      status = this.getParameterValues(task.subtask, methodResponse, cwmpResponse, deviceUpdates)
-    if status is STATUS_FINISHED
-      return STATUS_FINISHED
-    else if status isnt STATUS_STARTED
-      throw Error('Unexpected subtask status')
-
-  return STATUS_STARTED
+      methodResponse = {parameterList : {}}
+    this.getParameterValues(task.subtask, methodResponse, (err, status, cwmpResponse, deviceUpdates) ->
+      callback(err, status, cwmpResponse, deviceUpdates)
+    )
+  else
+    throw Error('Unexpected subtask name')
 
 
-this.getParameterNames = (task, methodResponse, cwmpResponse, deviceUpdates) ->
+this.getParameterNames = (task, methodResponse, callback) ->
   if methodResponse.faultcode?
     task.fault = methodResponse
-    return STATUS_FAULT
+    callback(null, STATUS_FAULT)
+    return
 
   if methodResponse.type is 'GetParameterNamesResponse'
-    deviceUpdates.parameterNames = methodResponse.parameterList
-    return STATUS_FINISHED
-
-  cwmpResponse.methodRequest = {
-    type : 'GetParameterNames',
-    parameterPath : task.parameterPath,
-    nextLevel : if task.nextLevel? then task.nextLevel else false
-  }
-  return STATUS_STARTED
+    callback(null, STATUS_FINISHED, null, {parameterNames : methodResponse.parameterList})
+  else
+    methodRequest = {
+      type : 'GetParameterNames',
+      parameterPath : task.parameterPath,
+      nextLevel : if task.nextLevel? then task.nextLevel else false
+    }
+    callback(null, STATUS_STARTED, {methodRequest : methodRequest})
 
 
-this.getParameterValues = (task, methodResponse, cwmpResponse, deviceUpdates) ->
+this.getParameterValues = (task, methodResponse, callback) ->
   if methodResponse.faultcode?
     task.fault = methodResponse
-    return STATUS_FAULT
+    callback(null, STATUS_FAULT)
+    return
 
   if not task.currentIndex?
     task.currentIndex = 0
@@ -72,23 +79,23 @@ this.getParameterValues = (task, methodResponse, cwmpResponse, deviceUpdates) ->
   names = task.parameterNames.slice(task.currentIndex, task.nextIndex)
 
   if methodResponse.type is 'GetParameterValuesResponse'
-    deviceUpdates.parameterValues = methodResponse.parameterList
+    deviceUpdates = {parameterValues : methodResponse.parameterList}
 
   if names.length == 0
-    return STATUS_FINISHED
-
-  cwmpResponse.methodRequest = {
-    type : 'GetParameterValues',
-    parameterNames : names
-  }
-
-  return STATUS_STARTED
+    callback(null, STATUS_FINISHED, null, deviceUpdates)
+  else
+    methodRequest = {
+      type : 'GetParameterValues',
+      parameterNames : names
+    }
+    callback(null, STATUS_STARTED, {methodRequest : methodRequest}, deviceUpdates)
 
 
-this.setParameterValues = (task, methodResponse, cwmpResponse, deviceUpdates) ->
+this.setParameterValues = (task, methodResponse, callback) ->
   if methodResponse.faultcode?
     task.fault = methodResponse
-    return STATUS_FAULT
+    callback(null, STATUS_FAULT)
+    return
 
   if not task.currentIndex?
     task.currentIndex = 0
@@ -100,47 +107,52 @@ this.setParameterValues = (task, methodResponse, cwmpResponse, deviceUpdates) ->
   values = task.parameterValues.slice(task.currentIndex, task.nextIndex)
 
   if prevValues?
-    deviceUpdates.parameterValues = prevValues
+    deviceUpdates = {parameterValues : prevValues}
 
   if values.length == 0
-    return STATUS_FINISHED
-
-  cwmpResponse.methodRequest = {
-    type : 'SetParameterValues',
-    parameterList : values
-  }
-  return STATUS_STARTED
+    callback(null, STATUS_FINISHED, null, deviceUpdates)
+  else
+    callback(null, STATUS_STARTED, {methodRequest : {type : 'SetParameterValues', parameterList : values}}, deviceUpdates)
 
 
-this.reboot = (task, methodResponse, cwmpResponse, deviceUpdates) ->
+this.reboot = (task, methodResponse, callback) ->
   if methodResponse.faultcode?
     task.fault = methodResponse
-    return STATUS_FAULT
-
+    callback(null, STATUS_FAULT)
+    return
+  
   if methodResponse.type isnt 'RebootResponse'
-    cwmpResponse.methodRequest = {
-      type : 'Reboot'
-    }
-    return STATUS_STARTED
-
-  return STATUS_FINISHED
+    callback(null, STATUS_STARTED, {methodRequest : {type : 'Reboot'}})
+  else
+    callback(null, STATUS_FINISHED)
 
 
-this.download = (task, methodResponse, cwmpResponse, deviceUpdates) ->
+this.download = (task, methodResponse, callback) ->
   if methodResponse.faultcode?
     task.fault = methodResponse
-    return STATUS_FAULT
+    callback(null, STATUS_FAULT)
+    return
 
   if methodResponse.type isnt 'DownloadResponse'
-    cwmpResponse.methodRequest = {
-      type : 'Download',
-      fileType : task.fileType,
-      url : task.url
-    }
-    return STATUS_STARTED
+    db.filesCollection.findOne({_id : mongodb.ObjectID(task.file)}, (err, file) ->
+      if not file?
+        callback('File not found')
+        return
+      else if err?
+        callback(err)
+        return
 
-  return STATUS_FINISHED
+      methodRequest = {
+        type : 'Download',
+        fileType : '1 Firmware Upgrade Image',
+        fileSize : file.length,
+        url : "http://#{config.FILES_IP}:#{config.FILES_PORT}/#{file.filename}"
+      }
+      callback(null, STATUS_STARTED, {methodRequest : methodRequest})
+    )
+  else
+    callback(null, STATUS_FINISHED)
 
 
-exports.task = (task, methodResponse, cwmpResponse, deviceUpdates) ->
-  return this[task.name](task, methodResponse, cwmpResponse, deviceUpdates)
+exports.task = (task, methodResponse, callback) ->
+  this[task.name](task, methodResponse, callback)