Adding (buggy) DelayedTask

1 files changed, 44 insertions, 3 deletions

diff --git a/lib/LuaTask.cc b/lib/LuaTask.cc
index 754b136..bc6266c 100644
--- a/lib/LuaTask.cc
+++ b/lib/LuaTask.cc
@@ -17,7 +17,7 @@
  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */
 
-/* $Id: LuaTask.cc,v 1.19 2007-12-25 15:06:12 pixel Exp $ */
+/* $Id: LuaTask.cc,v 1.20 2008-01-21 04:04:03 pixel Exp $ */
 
 #include <LuaTask.h>
 #include <LuaHandle.h>
@@ -34,7 +34,7 @@
 
 htab * LuaTask::h = hcreate(1);
 
-LuaTask::LuaTask(Lua * _L, const String & _cmd) : L(_L), cmd(_cmd), nargs(0), c(0), b(0) {
+LuaTask::LuaTask(Lua * __L, const String & _cmd) : L(__L), cmd(_cmd), nargs(0), c(0), b(0) {
     LuaTask * top = gettop();
     
     if (top) {
@@ -46,7 +46,7 @@ LuaTask::LuaTask(Lua * _L, const String & _cmd) : L(_L), cmd(_cmd), nargs(0), c(
     stacktop = L->gettop() + 1;
 }
 
-LuaTask::LuaTask(Lua * _L, int _nargs) throw (GeneralException) : L(_L), cmd(""), nargs(_nargs), c(0), b(0) {
+LuaTask::LuaTask(Lua * __L, int _nargs) throw (GeneralException) : L(__L), cmd(""), nargs(_nargs), c(0), b(0) {
     if (gettop())
 	throw GeneralException("Can't run a stack-based LuaTask when there are other tasks waiting.");
 
@@ -282,11 +282,13 @@ int LuaTask::Do() throw (GeneralException) {
 enum TaskMan_functions_t {
     TASKMAN_NB_TASKS = 0,
     TASKMAN_NB_ZOMBIES,
+    TASKMAN_DELAYED_TASK,
 };
 
 struct lua_functypes_t TaskMan_functions[] = {
     { TASKMAN_NB_TASKS,     "NB_Tasks",      0, 0, { } },
     { TASKMAN_NB_ZOMBIES,   "NB_Zombies",    0, 0, { } },
+    { TASKMAN_DELAYED_TASK, "DelayedTask",   1, 2, { BLUA_STRING, BLUA_NUMBER } },
     { -1, 0, 0, 0, 0 }
 };
 
@@ -294,6 +296,7 @@ class sLua_TaskMan : public Base {
   public:
     DECLARE_FUNCTION(TaskMan, TASKMAN_NB_TASKS);
     DECLARE_FUNCTION(TaskMan, TASKMAN_NB_ZOMBIES);
+    DECLARE_FUNCTION(TaskMan, TASKMAN_DELAYED_TASK);
   private:
     static int TaskMan_proceed_statics(Lua * L, int n, int caller);
 };
@@ -303,9 +306,40 @@ void LuaTaskMan::pushstatics(Lua * L) throw (GeneralException) {
 
     PUSH_FUNCTION(TaskMan, TASKMAN_NB_TASKS);
     PUSH_FUNCTION(TaskMan, TASKMAN_NB_ZOMBIES);
+    PUSH_FUNCTION(TaskMan, TASKMAN_DELAYED_TASK);
 }
 
+class DelayedTask : public Task {
+  public:
+      DelayedTask(Lua * __L, const String & _cmd, int _delay) : L(__L), cmd(_cmd), delay(_delay) {
+          printm(M_INFO, "Creating object delayedtask, delay = %i seconds.\n", _delay);
+	  SetBurst();
+      }
+    virtual int Do() throw (GeneralException) {
+        timeval delay_tv = { delay, 0 };
+	switch(current) {
+	case 0:
+	    printm(M_INFO, "DelayTask: initializing.\n");
+            WaitFor(delay);
+	    current = 1;
+	    return TASK_ON_HOLD;
+	case 1:
+    	    printm(M_INFO, "DelayTask: expired, lauching LuaTask with cmd = " + cmd + "\n");
+    	    RemoveTimeout();
+	    new LuaTask(L, cmd);
+	    return TASK_DONE;
+	}
+    }
+  private:
+    Lua * L;
+    String cmd;
+    int delay;
+};
+
 int sLua_TaskMan::TaskMan_proceed_statics(Lua * L, int n, int caller) {
+    String cmd;
+    int delay = 0;
+    
     switch (caller) {
     case TASKMAN_NB_TASKS:
 	return TaskMan::nb_tasks();
@@ -313,6 +347,13 @@ int sLua_TaskMan::TaskMan_proceed_statics(Lua * L, int n, int caller) {
     case TASKMAN_NB_ZOMBIES:
 	return TaskMan::nb_zombies();
         break;
+    case TASKMAN_DELAYED_TASK:
+	cmd = L->tostring(1);
+	if (n == 2)
+	    delay = L->tonumber(2);
+	printm(M_INFO, "Creating delayed task of %i seconds.\n", delay);
+	new DelayedTask(L, cmd, delay);
+	break;
     }
 
     return 1;