#include <sys/types.h>
#include <unistd.h>
#include <signal.h>
#include <getopt.h>
#include <pwd.h>
#include <grp.h>
#include <errno.h>
#include <syslog.h>
#include <iostream>
#include <memory>
#include <typeinfo>
using namespace std;
#include "kingate/fastcgi.h"
#include "kingate/cgi_gateway.h"
using namespace kingate;
#include "sitecing/sitecing_util.h"
#include "sitecing/util.h"
#include "sitecing/sitespace.h"
#include "sitecing/sitecing_interface_cgi.h"
#include "sitecing/cgi_component.h"
#include "sitecing/configuration.h"
#include "sitecing/magic.h"
#include "sitecing/sitecing_exception.h"
#include "sitecing/exception.h"
#include "sitecing/process_manager.h"
using namespace sitecing;

#include "config.h"
#define PHEADER PACKAGE " Version " VERSION
#define PCOPY   "Copyright (c) 2004 Klever Group"

class cdummyClass : public acomponent {
    public:
	void main(int _magic,va_list _args) {}
	void *__the_most_derived_this() { return NULL; }
} cdummyInstance;
class adummyClass : public cgi_component {
    public:
	void main(int _magic,va_list _args) {}
	void *__the_most_derived_this() { return NULL; }
} adummyInstance;

class sitecing_fastcgi_pm : public process_manager {
    public:
	configuration config;
	fcgi_socket *fss;
	semaphore sem;
	bool multi;
	uid_t uid;
	gid_t gid;
	pid_file pidfile;

	sitecing_fastcgi_pm(const string& config_file);
	~sitecing_fastcgi_pm();

	void process(int slot);

	void run();
	void give_up_privs();
};

sitecing_fastcgi_pm::sitecing_fastcgi_pm(const string& config_file)
    : config(config_file), multi(false) {
	if(( (config.flags&configuration::flag_user) || (config.flags&configuration::flag_group) || (config.flags&configuration::flag_chroot) ) && geteuid() )
	    throw konforka::exception(CODEPOINT,"can't use User, Group or Chroot when started without root privileges");
	if(config.flags&configuration::flag_user) {
	    struct passwd *ptmp = getpwnam(config.user.c_str());
	    if(ptmp) {
		uid = ptmp->pw_uid;
	    }else{
		errno=0;
		uid = strtol(config.user.c_str(),NULL,0);
		if(errno)
		    throw konforka::exception(CODEPOINT,"failed to resolve User value to uid");
	    }
	}
	if(config.flags&configuration::flag_group) {
	    struct group *gtmp = getgrnam(config.group.c_str());
	    if(gtmp) {
		gid = gtmp->gr_gid;
	    }else{
		errno=0;
		gid = strtol(config.group.c_str(),NULL,0);
		if(errno)
		    throw konforka::exception(CODEPOINT,"failed to resolve Group value to gid");
	    }
	}
	if(!(config.flags&configuration::flag_root_source))
	    throw konforka::exception(CODEPOINT,"unspecified root for sources");
	if(!(config.flags&configuration::flag_root_intermediate))
	    throw konforka::exception(CODEPOINT,"unspecified root for intermediate files");
	if(!(config.flags&configuration::flag_root_so))
	    throw konforka::exception(CODEPOINT,"unspecified root for shared objects");
	if(config.flags&configuration::flag_min_children)
	    min_children = config.min_children;
	if(config.flags&configuration::flag_max_children)
	    max_children = config.max_children;
	if(config.flags&configuration::flag_min_spare_children)
	    min_spare_children = config.min_spare_children;
	if(config.flags&configuration::flag_max_spare_children)
	    max_spare_children = config.max_spare_children;
	if(max_children<min_spare_children)
	    throw konforka::exception(CODEPOINT,"inconsistent numbers of MaxChildren and MinSpareChildren");
	if(min_children>max_spare_children && max_spare_children>=0)
	    throw konforka::exception(CODEPOINT,"inconsistent numbers of MinChildren and MaxSpareChildren");
	if(config.flags&configuration::flag_multi_process) {
	    multi = config.multi_process;
	}else{
	    if(config.flags&configuration::flag_listen_socket)
		multi = true;
	    else
		multi = false;
	}
	fss = (config.flags&configuration::flag_listen_socket)?new fcgi_socket(config.listen_socket.c_str(),5):new fcgi_socket(0);
	if(!fss)
	    throw konforka::exception(CODEPOINT,"failed to establish listening socket");
	if(config.flags&configuration::flag_daemonize && config.daemonize) {
	    pid_t pf = fork();
	    if(pf<0)
		throw konforka::exception(CODEPOINT,"failed to fork()");
	    if(pf) {
		die_humbly=true;
		_exit(0);
	    }
	}
	if(config.flags&configuration::flag_pid_file) {
	    pidfile.set(config.pid_file);
	}
	if(multi)
	    sem.init();
    }
sitecing_fastcgi_pm::~sitecing_fastcgi_pm() {
    if(fss)
	delete fss;
}

void sitecing_fastcgi_pm::process(int slot) {
    signal(SIGINT,SIG_DFL);
    signal(SIGABRT,SIG_DFL);
    signal(SIGTERM,SIG_DFL);
    give_up_privs();
    scoreboard_slot *sslot = sboard.get_slot(slot);
    try {
	sitespace ss(config);
	fcgi_socket& fs = *fss;
	sitecing_interface_cgi scif(&ss);
	string component_path;
	string action;
	config_options::action_handler_t *action_handler;
	int rpc = 0;
	if(config.flags&configuration::flag_requests_per_child)
	    rpc = config.requests_per_child;
	for(int req=0;(rpc<=0) || (req<rpc);++req) {
	    semaphore_lock sl;
	    if(multi) {
		sslot->state = scoreboard_slot::state_idle;
		sl.sem = &sem;
		sl.lock();
	    }
	    sslot->state = scoreboard_slot::state_accept;
	    fcgi_interface fi(fs);
	    sslot->state = scoreboard_slot::state_processing;
	    if(multi)
		sl.unlock();
	    cgi_gateway gw(fi);
	    ++scif.serial;
	    scif.prepare(&gw);
	    try {
		component_path = konforka::normalize_path(gw.path_info(),konforka::strip_leading_slash|konforka::strip_trailing_slash);
		string full_component_path;
		string sitecing_path_info;
		while(true) {
		    full_component_path = config.root_source+'/'+component_path;
		    if(!access(full_component_path.c_str(),F_OK))
			break;
		    string::size_type sl = component_path.rfind('/');
		    if(sl==string::npos)
			throw konforka::exception(CODEPOINT,"can't find the target component");
		    sitecing_path_info.insert(0,component_path,sl,string::npos);
		    component_path.erase(sl);
		}
		fi.metavars["SITECING_PATH_INFO"]=sitecing_path_info;
		fi.metavars["SITECING_COMPONENT_PATH"]=component_path;
		action = component_path;
		action_handler = config.lookup_action_handler(component_path);
		if(action_handler) {
		    action = action_handler->action;
		}
		string pwd = konforka::dir_name(full_component_path);
		if(chdir(pwd.c_str()))
		    throw konforka::exception(CODEPOINT,"failed to chdir() into document's directory");
		so_component soc = ss.fetch(action,&scif);
		if(action_handler) {
		    soc.ac->run(__magic_action,
			    config.root_source.c_str(), config.root_intermediate.c_str(), config.root_so.c_str(),
			    &(action_handler->args)
			    );
		}else{
		    soc.ac->main(0,NULL);
		}
	    }catch(http_status& hs) {
		scif.headers["Status"] = hs.status+" "+hs.message;
		string hshp = config.lookup_http_status_handler(component_path,hs.status);
		if(!hshp.empty()) {
		    so_component hsh = ss.fetch(hshp,&scif); // TODO: handle error trying to handle status
		    hsh.ac->run(__magic_http_status,config.root_source.c_str(),config.root_intermediate.c_str(),
			    config.root_so.c_str(),action.c_str(),&hs);
		}
	    }catch(compile_error& ce) {
		config_options *co_exception_handler = config.lookup_config(component_path,config_options::flag_exception_handler);
		if(co_exception_handler) {
		    so_component eh = ss.fetch(co_exception_handler->exception_handler,&scif); // TODO: handle error trying to handle error.
		    eh.ac->run(__magic_compile_error,ce.what(),config.root_source.c_str(),config.root_intermediate.c_str(),config.root_so.c_str(),ce.component_path.c_str());
		}else{
		    ce.see(CODEPOINT);
		    throw;
		}
	    }catch(preprocessor_error& pe) {
		config_options *co_exception_handler = config.lookup_config(component_path,config_options::flag_exception_handler);
		if(co_exception_handler) {
		    so_component eh = ss.fetch(co_exception_handler->exception_handler,&scif); // TODO: handle error trying to handle error.
		    eh.ac->run(__magic_preprocess_error,pe.what(),config.root_source.c_str(),config.root_intermediate.c_str(),config.root_so.c_str(),pe.component_name.c_str(),pe.line_number);
		}else{
		    pe.see(CODEPOINT);
		    throw;
		}
	    }catch(exception& e) {
		config_options *co_exception_handler = config.lookup_config(component_path,config_options::flag_exception_handler);
		if(co_exception_handler) {
		    so_component eh = ss.fetch(co_exception_handler->exception_handler,&scif); // TODO: handle error trying to handle error.
		    eh.ac->run(__magic_generic_exception,e.what(),config.root_source.c_str(),config.root_intermediate.c_str(),config.root_so.c_str(),component_path.c_str(),&e);
		}
	    }
	    scif.flush();
	}
    }catch(exception& e) {
	cerr << "->Oops: " << e.what() << endl;
    }
}

void sitecing_fastcgi_pm::run() {
    if(multi)
	manage();
    else
	process(0);
}

void sitecing_fastcgi_pm::give_up_privs() {
    if(config.flags&configuration::flag_chroot) {
	if(chroot(config.chroot.c_str()))
	    throw konforka::exception(CODEPOINT,"failed to chroot()");
    }
    if(config.flags&configuration::flag_group) {
	if((getgid()!=gid) && setgid(gid))
	    throw konforka::exception(CODEPOINT,"failed to setgid()");
    }
    if(config.flags&configuration::flag_user) {
	if((getuid()!=uid) && setuid(uid))
	    throw konforka::exception(CODEPOINT,"failed to setuid()");
    }
}

static sitecing_fastcgi_pm* _process_manager = NULL;

static void lethal_signal_handler(int signum) {
    _process_manager->finishing=true;
}

int main(int argc,char **argv) {
    const char* id = *argv;
    const char* t;
    while(t = index(id,'/')) {
	id=t; id++;
    }
    openlog(id,LOG_PERROR|LOG_PID,LOG_USER);
    try {
	string config_file = "sitecing.conf";
	while(true) {
	    static struct option opts[] = {
		{ "help", no_argument, 0, 'h' },
		{ "usage", no_argument, 0, 'h' },
		{ "version", no_argument, 0, 'V' },
		{ "license", no_argument, 0, 'L' },
		{ "config", required_argument, 0, 'f' },
		{ NULL, 0, 0, 0 }
	    };
	    int c = getopt_long(argc,argv,"f:hVL",opts,NULL);
	    if(c==-1)
		break;
	    switch(c) {
		case 'h':
		    cerr << PHEADER << endl
			<< PCOPY << endl << endl
			<< " -h, --help" << endl
			<< " --usage          display this text" << endl
			<< " -V, --version    display version number" << endl
			<< " -L, --license    show license" << endl
			<< " -f filename, --config=filename" << endl
			<< "                  specify configuration file to use" << endl;
		    exit(0);
		    break;
		case 'V':
		    cerr << VERSION << endl;
		    exit(0);
		    break;
		case 'L':
		    extern const char *COPYING;
		    cerr << COPYING << endl;
		    exit(0);
		    break;
		case 'f':
		    config_file = optarg;
		    break;
		default:
		    cerr << "Huh??" << endl;
		    break;
	    }
	}
	sitecing_fastcgi_pm sfpm(config_file);
	_process_manager = &sfpm;
	signal(SIGINT,lethal_signal_handler);
	signal(SIGABRT,lethal_signal_handler);
	signal(SIGTERM,lethal_signal_handler);
	sfpm.run();
    }catch(exception& e) {
	/* cerr << "Oops: " << e.what() << endl; */
	syslog(LOG_ERR,"uncaught exception: %s",e.what());
    }
    closelog();
}