/*
 *  Baltisot
 *  Copyright (C) 1999-2007 Nicolas "Pixel" Noble
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

/* $Id: LuaTask.cc,v 1.17 2007-10-12 15:18:26 pixel Exp $ */

#include <LuaTask.h>
#include <LuaHandle.h>
#include <TaskMan.h>
#ifndef LUATASK_OMIT_HTTPCLIENT
#include <HttpClient.h>
#endif
#ifndef LUATASK_OMIT_MAILCLIENT
#include <MailClient.h>
#endif
#ifndef LUATASK_OMIT_COMMAND
#include <CopyJob.h>
#endif

htab * LuaTask::h = hcreate(1);

LuaTask::LuaTask(Lua * _L, const String & _cmd) : L(_L), cmd(_cmd), nargs(0), c(0), b(0) {
    LuaTask * top = gettop();
    
    if (top) {
        WaitFor(top);
    } else {
        SetBurst();
    }
    settop(this);
    stacktop = L->gettop() + 1;
}

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.");

    SetBurst();
    settop(this);
    stacktop = L->gettop() - nargs;
}

LuaTask::~LuaTask() {
    if (gettop() == this) {
        settop(0);
    }
}

LuaTask * LuaTask::gettop() {
    if (!hfind(h, (Uint8 *) &L, sizeof(L))) {
        hadd(h, (Uint8 *) &L, sizeof(L), 0);
    }
    
    return (LuaTask *) hstuff(h);
}

void LuaTask::settop(LuaTask * v) {
    if (!hfind(h, (Uint8 *) &L, sizeof(L)))
	return;
    if (v) {
	hstuff(h) = v;
    } else {
	hdel(h); 
    }
}

String LuaTask::GetName() {
    return "LuaTask(" + cmd + ")";
}

std::vector<String> LuaTask::array_to_vector(int index) throw (GeneralException) {
    std::vector<String> r;
    
    if (L->isnil(index))
        return r;
    
    if (!L->istable(index)) {
        throw GeneralException("Argument isn't a table.");
    }
    
    L->push();
    while(L->next(index) != 0) {
        r.push_back(L->tostring(-1));
        L->pop();
    }
    
    return r;
}

int LuaTask::Do() throw (GeneralException) {
    bool yielded;
    switch (current) {
    case 0:
        current = 1;
        if (cmd != "") {
            yielded = L->resume(cmd);
        } else {
            yielded = L->resume(nargs);
        }
    case 2:
	nargs = 0;
#ifndef LUATASK_OMIT_HTTPCLIENT
        if (task == "HttpClient") {
            LuaBuffer o(b);
            o.pushdestruct(L);
            nargs = 1;
        }
#endif

#ifndef LUATASK_OMIT_MAILCLIENT
        if (task == "MailClient") {
            L->push(dynamic_cast<MailClient *>(c)->GetStatus());
            nargs = 1;
            delete b;
            b = 0;
        }
#endif

#ifndef LUATASK_OMIT_COMMAND
        if (task == "Command") {
            delete p;
            LuaBuffer o(b);
            o.pushdestruct(L);
            L->push((lua_Number) TaskMan::Estatus());
            nargs = 2;
        }
#endif

        if (!c)
            delete c;
        c = 0;

	if (current != 1)
            yielded = L->resume(nargs);
    case 1:
        if (!yielded)
            return TASK_DONE;
        current = 2;
        if (L->gettop() >= 1) {
            task = L->tostring(1);
            if (task == "") {
                L->error("Must precise a task-type to execute.");
                return TASK_DONE;
#ifndef LUATASK_OMIT_HTTPCLIENT
            } else if (task == "HttpClient") {
                String url = L->tostring(2);
                t_headers headers;
		Variables vars;
		
		if (L->gettop() >= 4) {
		    if (!L->istable(4)) {
			L->error("Incorrect parameters to HttpClient.");
			return TASK_DONE;
		    }
		    // read variables
		    L->push();
		    while (L->next(4) != 0) {
			String v = L->tostring(-2) + "=" + L->tostring(-1);
			vars.Add(v);
			L->pop();
		    }
		}
		if (L->gettop() >= 3) {
		    if (!L->istable(3)) {
			L->error("Incorrect parameters to HttpClient.");
			return TASK_DONE;
		    }
		    // read headers
		    L->push();
		    while (L->next(3) != 0) {
			String h = L->tostring(-2) + ": " + L->tostring(-1);
			headers.push_back(h);
			L->pop();
		    }
		}

                b = new Buffer(true);

                //headers.push_back("User-Agent: Mozilla/5.0 (Windows; U; Windows NT 5.1; fr; rv:1.8.1) Gecko/20061010 Firefox/2.0");
                c = new HttpClient(url, b, "", headers, vars);
                WaitFor(c);
                Suspend(TASK_ON_HOLD);
#endif
#ifndef LUATASK_OMIT_MAILCLIENT
            } else if (task == "MailClient") {
                String smtp, subject, from, body;
                MailClient::strings_t headers, tos, ccs, bccs, fakes;
                
                if (L->gettop() != 9) {
                    L->error("Incorrect parameters to MailClient.");
                    return TASK_DONE;
                }
                smtp = L->tostring(1);
                subject = L->tostring(2);
                from = L->tostring(3);
                try {
                    headers = array_to_vector(4);
                    tos = array_to_vector(5);
                    ccs = array_to_vector(6);
                    bccs = array_to_vector(7);
                    fakes = array_to_vector(8);
                } catch (GeneralException e) {
                    L->error("Incorrect parameters to MailClient.");
                }
                body = L->tostring(9);
                
                b = new Buffer(true);
                (*b) << body;
                c = new MailClient(smtp, subject, from, headers, tos, ccs, bccs, fakes, b);
                WaitFor(c);
                Suspend(TASK_ON_HOLD);
#endif
#ifndef LUATASK_OMIT_COMMAND
            } else if (task == "Command") {
                pid_t pid;
                int i;
		
                char * cmd;
                char ** args, ** ptr;
                
                cmd = L->tostring(2).strdup();

                args = (char **) malloc(sizeof(char *) * (L->gettop()));
                args[0] = cmd;
                for (i = 3; i <= L->gettop(); i++) {
                    args[i - 2] = L->tostring(i).strdup();
                }
                args[L->gettop() - 1] = 0;
                
                b = new Buffer(true);
                p = new InPipe();
                p->SetNonBlock();
                
                if (!(pid = fork())) {
                    p->Hook();
                    execvp(cmd, args);
                }
                c = new CopyJob(p, b);
                for (i = 3; i <= L->gettop(); i++) {
                    free(args[i - 2]);
                }
                free(cmd);
                current = 3;
                WaitFor(pid);
                Suspend(TASK_ON_HOLD);
#endif
            } else {
                L->error("Unknow requested task: " + task);
                return TASK_DONE;
            }
        } else {
            return TASK_DONE;
        }
        break;
#ifndef LUATASK_OMIT_COMMAND
    case 3:
        current = 2;
        WaitFor(c);
        p->HalfClose();
        return TASK_ON_HOLD;
#endif
    }

    return TASK_ON_HOLD;
}