From 44a394a068b23d3eae93dcc480fe51d2097b4a5f Mon Sep 17 00:00:00 2001
From: Andrew Mao <mizzao@gmail.com>
Date: Thu, 14 Aug 2014 00:09:16 -0400
Subject: [PATCH] better fixing of idle time

---
 server/scripts.coffee | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/server/scripts.coffee b/server/scripts.coffee
index 0864621..f2ede53 100644
--- a/server/scripts.coffee
+++ b/server/scripts.coffee
@@ -284,17 +284,20 @@ Meteor.methods
     console.log "#{paidWorkers} workers compensated"
     return
 
-  "cm-fix-asst-idle": (asstId, instanceId) ->
+  "cm-fix-asst-idle": (asstId, instanceId, idleMillis) ->
     TurkServer.checkAdmin()
     check(asstId, String)
     check(instanceId, String)
 
+    modifier = if idleMillis?
+      $set: { "instances.$.idleTime": idleMillis }
+    else
+      $unset: { "instances.$.idleTime": null }
+
     Assignments.update {
       _id: asstId,
       "instances.id": instanceId
-    }, {
-      $unset: { "instances.$.idleTime": null }
-    }
+    }, modifier
 
     return