/*
 * $Id: setup.c 8230 2010-01-14 20:37:44Z dothebart $
 *
 * WebCit setup utility
 * 
 * (This is basically just an install wizard.  It's not required.)
 *
 */

#include "sysdep.h"
#include "webcit.h"
#include "webserver.h"


#define UI_TEXT		0	/* Default setup type -- text only */
#define UI_DIALOG	2	/* Use the 'dialog' program */
#define UI_SILENT	3	/* Silent running, for use in scripts */

int setup_type;
char setup_directory[SIZ];
int using_web_installer = 0;
char suggested_url[SIZ];

/* some copies... */
int lprintf(int loglevel, const char *format, ...){return 0;}
void wc_printf(const char *format,...){}

void RegisterNS(const char *NSName, long len, 
		int nMinArgs, 
		int nMaxArgs, 
		WCHandlerFunc HandlerFunc,
		WCPreevalFunc PreEvalFunc,
		int ContextRequired){}
void RegisterHeaderHandler(const char *Name, long Len, Header_Evaluator F){}
pthread_key_t MyConKey;

#ifdef ENABLE_NLS

#ifdef HAVE_USELOCALE 
int localeoffset = 1;
#else
int localeoffset = 0;
#endif

#endif
/*
 * Delete an entry from /etc/inittab
 */
void delete_init_entry(char *which_entry)
{
	char *inittab = NULL;
	FILE *fp;
	char buf[SIZ];
	char entry[SIZ];
	char levels[SIZ];
	char state[SIZ];
	char prog[SIZ];
	int rv;

	inittab = strdup("");
	if (inittab == NULL) return;

	fp = fopen("/etc/inittab", "r");
	if (fp == NULL) return;

	while(fgets(buf, sizeof buf, fp) != NULL) {

		if (num_tokens(buf, ':') == 4) {
			extract_token(entry, buf, 0, ':', sizeof entry);
			extract_token(levels, buf, 1, ':', sizeof levels);
			extract_token(state, buf, 2, ':', sizeof state);
			extract_token(prog, buf, 3, ':', sizeof prog); /* includes 0x0a LF */

			if (!strcmp(entry, which_entry)) {
				strcpy(state, "off");	/* disable it */
			}
		}

		inittab = realloc(inittab, strlen(inittab) + strlen(buf) + 2);
		if (inittab == NULL) {
			fclose(fp);
			return;
		}
		
		strcat(inittab, buf);
	}
	fclose(fp);
	fp = fopen("/etc/inittab", "w");
	if (fp != NULL) {
		rv = fwrite(inittab, strlen(inittab), 1, fp);
		fclose(fp);
		kill(1, SIGHUP);	/* Tell init to re-read /etc/inittab */
	}
	free(inittab);
}




/* 
 * Remove any /etc/inittab entries for webcit, because we don't
 * start it that way anymore.
 */
void delete_the_old_way(void) {
	FILE *infp;
	char buf[1024];
	char looking_for[1024];
	int have_entry = 0;
	char entry[1024];
	char prog[1024];
	char init_entry[1024];


	strcpy(init_entry, "");

	/* Determine the fully qualified path name of webcit */
	snprintf(looking_for, sizeof looking_for, "%s/webcit ", setup_directory);

	/* Pound through /etc/inittab line by line.  Set have_entry to 1 if
	 * an entry is found which we believe starts webcit.
	 */
	infp = fopen("/etc/inittab", "r");
	if (infp == NULL) {
		return;
	} else {
		while (fgets(buf, sizeof buf, infp) != NULL) {
			buf[strlen(buf) - 1] = 0;
			extract_token(entry, buf, 0, ':', sizeof entry);
			extract_token(prog, buf, 3, ':', sizeof prog);
			if (!strncasecmp(prog, looking_for,
			   strlen(looking_for))) {
				++have_entry;
				strcpy(init_entry, entry);
			}
		}
		fclose(infp);
	}

	/* Bail out if there's nothing to do. */
	if (!have_entry) return;

	delete_init_entry(init_entry);
}



void cleanup(int exitcode)
{
	exit(exitcode);
}



void title(char *text)
{
	if (setup_type == UI_TEXT) {
		printf("\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n<%s>\n", text);
	}
}




int yesno(char *question, int default_value)
{
	int i = 0;
	int answer = 0;
	char buf[SIZ] = "";

	switch (setup_type) {

	case UI_TEXT:
		do {
			printf("%s\nYes/No [%s] --> ",
				question,
				( default_value ? "Yes" : "No" )
			);
			if (fgets(buf, sizeof buf, stdin))
			{
				answer = tolower(buf[0]);
				if ((buf[0]==0) || (buf[0]==13) || (buf[0]==10))
					answer = default_value;
				else if (answer == 'y')
					answer = 1;
				else if (answer == 'n')
					answer = 0;
			}

		} while ((answer < 0) || (answer > 1));
		break;

	case UI_DIALOG:
		sprintf(buf, "exec %s %s --yesno '%s' 15 75",
			getenv("CTDL_DIALOG"),
			( default_value ? "" : "--defaultno" ),
			question);
		i = system(buf);
		if (i == 0) {
			answer = 1;
		}
		else {
			answer = 0;
		}
		break;

	}
	return (answer);
}




void set_value(char *prompt, char str[])
{
	char buf[SIZ] = "";
	char dialog_result[PATH_MAX];
	char setupmsg[SIZ];
	FILE *fp;
	int rv;

	strcpy(setupmsg, "");

	switch (setup_type) {
	case UI_TEXT:
		title("WebCit setup");
		printf("\n%s\n", prompt);
		printf("This is currently set to:\n%s\n", str);
		printf("Enter new value or press return to leave unchanged:\n");
		if (fgets(buf, sizeof buf, stdin)) {
			buf[strlen(buf) - 1] = 0;
		}
		if (strlen(buf) != 0)
			strcpy(str, buf);
		break;
	case UI_DIALOG:
		CtdlMakeTempFileName(dialog_result, sizeof dialog_result);
		sprintf(buf, "exec %s --inputbox '%s' 19 72 '%s' 2>%s",
			getenv("CTDL_DIALOG"),
			prompt,
			str,
			dialog_result);
		rv = system(buf);
		fp = fopen(dialog_result, "r");
		if (fp != NULL) {
			if (fgets(str, sizeof buf, fp)){
				if (str[strlen(str)-1] == 10) {
					str[strlen(str)-1] = 0;
				}
			}
			fclose(fp);
			unlink(dialog_result);
		}
		break;

	}
}


extern const char *AvailLang[];
int GetLocalePrefs(void)
{
	int nLocales;
	StrBuf *Buf;
	char buf[SIZ];
	char dialog_result[PATH_MAX];
	FILE *fp;
	int i = 0;
	int offs = 0;
	int rv;


	nLocales = 0; 
	while (!IsEmptyStr(AvailLang[nLocales]))
		nLocales++;

	Buf = NewStrBuf();

	StrBufAppendBufPlain(Buf, HKEY("Select the locale webcit should use : \n"), 0);
#ifdef HAVE_USELOCALE 
	StrBufAppendBufPlain(Buf, HKEY(" 0 Let the user select it at the login prompt (default)\n"), 0);
	offs ++;
#endif
	for (i = 0; i < nLocales; i++) {
		StrBufAppendPrintf(Buf, " %ld: %s\n", i + offs, AvailLang[i]);

	}

	switch (setup_type) {
	case UI_TEXT:
		title("WebCit setup");
		printf("\n%s\n", ChrPtr(Buf));
		printf("This is currently set to:\n%ld\n", 0L);
		printf("Enter new value or press return to leave unchanged:\n");
		if (fgets(buf, sizeof buf, stdin))
			return atoi(buf);
		break;

	case UI_DIALOG:
		CtdlMakeTempFileName(dialog_result, sizeof dialog_result);
		sprintf(buf, "exec %s --inputbox '%s' 19 72 '%ld' 2>%s",
			getenv("CTDL_DIALOG"),
			ChrPtr(Buf),
			0L,
			dialog_result);
		rv = system(buf);
		fp = fopen(dialog_result, "r");
		if (fp != NULL) {
			char *str = &buf[0];
			if (fgets(str, sizeof buf, fp)){
				if (str[strlen(str)-1] == 10) {
					str[strlen(str)-1] = 0;
				}
			}
			fclose(fp);
			unlink(dialog_result);
			return atoi(buf);
		}
		break;

	}
	return 0;
}

void important_message(char *title, char *msgtext)
{
	char buf[SIZ];
	int rv;

	switch (setup_type) {

	case UI_TEXT:
		printf("\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n\n");
		printf("       %s \n\n%s\n\n", title, msgtext);
		printf("Press return to continue...");
		if (fgets(buf, sizeof buf, stdin));
		break;

	case UI_DIALOG:
		sprintf(buf, "exec %s --msgbox '%s' 19 72",
			getenv("CTDL_DIALOG"),
			msgtext);
		rv = system(buf);
		break;
	}
}


void display_error(char *error_message)
{
	important_message("Error", error_message);
}

void progress(char *text, long int curr, long int cmax)
{
	static long dots_printed = 0L;
	long a = 0;
	char buf[SIZ];
	static FILE *fp = NULL;

	switch (setup_type) {

	case UI_TEXT:
		if (curr == 0) {
			printf("%s\n", text);
			printf("..........................");
			printf("..........................");
			printf("..........................\r");
			fflush(stdout);
			dots_printed = 0;
		} else if (curr == cmax) {
			printf("\r%79s\n", "");
		} else {
			a = (curr * 100) / cmax;
			a = a * 78;
			a = a / 100;
			while (dots_printed < a) {
				printf("*");
				++dots_printed;
				fflush(stdout);
			}
		}
		break;

	case UI_DIALOG:
		if (curr == 0) {
			sprintf(buf, "exec %s --gauge '%s' 7 72 0",
				getenv("CTDL_DIALOG"),
				text);
			fp = popen(buf, "w");
			if (fp != NULL) {
				fprintf(fp, "0\n");
				fflush(fp);
			}
		} 
		else if (curr == cmax) {
			if (fp != NULL) {
				fprintf(fp, "100\n");
				pclose(fp);
				fp = NULL;
			}
		}
		else {
			a = (curr * 100) / cmax;
			if (fp != NULL) {
				fprintf(fp, "%ld\n", a);
				fflush(fp);
			}
		}
		break;
	}
}




/*
 * install_init_scripts()  -- Create and deploy SysV init scripts.
 *
 */
void install_init_scripts(void)
{
#ifdef ENABLE_NLS
	int localechoice;
#endif
	char question[1024];
	char buf[256];
	char http_port[128];
#ifdef HAVE_OPENSSL
	char https_port[128];
#endif
	char hostname[128];
	char portname[128];
	char command[SIZ];
	struct utsname my_utsname;
	struct stat etcinitd;
	FILE *fp;
	char *initfile = "/etc/init.d/webcit";
	int rv;

	fp = fopen(initfile, "r");
	if (fp != NULL) {
		if (yesno("WebCit already appears to be configured to start at boot.\n"
			  "Would you like to keep your boot configuration as is?\n", 1) == 1) {
			return;
		}
		fclose(fp);
		
	}

	/* Otherwise, prompt the user to create an entry. */
	snprintf(question, sizeof question,
		 "Would you like to automatically start WebCit at boot?"
		);
	if (yesno(question, 1) == 0)
		return;


#ifdef ENABLE_NLS

	localechoice = GetLocalePrefs();

#endif
	/* Defaults */
	sprintf(http_port, "2000");
#ifdef HAVE_OPENSSL
	sprintf(https_port, "443");
#endif
	sprintf(hostname, "uds");
	sprintf(portname, "/usr/local/citadel");

	/* This is a very hackish way of learning the port numbers used
	 * in a previous install, if we are upgrading: read them out of
	 * the existing init script.
	 */
	if ((stat("/etc/init.d/", &etcinitd) == -1) && 
	    (errno == ENOENT))
	{
		if ((stat("/etc/rc.d/init.d/", &etcinitd) == -1) &&
		    (errno == ENOENT))
			initfile = WEBCITDIR"/webcit.init";
		else
			initfile = "/etc/rc.d/init.d/webcit";
	}

	fp = fopen(initfile, "r");
	if (fp != NULL) {
		while (fgets(buf, sizeof buf, fp) != NULL) {
			if (strlen(buf) > 0) {
				buf[strlen(buf)-1] = 0;	/* strip trailing cr */
			}
			if (!strncasecmp(buf, "HTTP_PORT=", 10)) {
				safestrncpy(http_port, &buf[10], sizeof http_port);
			}
#ifdef HAVE_OPENSSL
			if (!strncasecmp(buf, "HTTPS_PORT=", 11)) {
				safestrncpy(https_port, &buf[11], sizeof https_port);
			}
#endif
			if (!strncasecmp(buf, "CTDL_HOSTNAME=", 14)) {
				safestrncpy(hostname, &buf[14], sizeof hostname);
			}
			if (!strncasecmp(buf, "CTDL_PORTNAME=", 14)) {
				safestrncpy(portname, &buf[14], sizeof portname);
			}
		}
		fclose(fp);
	}

	/* Now ask for the port numbers */
	snprintf(question, sizeof question,
		 "On which port do you want WebCit to listen for HTTP "
		 "requests?\n\nYou can use the standard port (80) if you are "
		 "not running another\nweb server (such as Apache), otherwise "
		 "select another port.");
	set_value(question, http_port);
	uname(&my_utsname);
	sprintf(suggested_url, "http://%s:%s/", my_utsname.nodename, http_port);

#ifdef HAVE_OPENSSL
	snprintf(question, sizeof question,
		 "On which port do you want WebCit to listen for HTTPS "
		 "requests?\n\nYou can use the standard port (443) if you are "
		 "not running another\nweb server (such as Apache), otherwise "
		 "select another port.");
	set_value(question, https_port);
#endif

	/* Find out where Citadel is. */
	if ( (using_web_installer) && (getenv("CITADEL") != NULL) ) {
		strcpy(hostname, "uds");
		strcpy(portname, getenv("CITADEL"));
	}
	else {
		snprintf(question, sizeof question,
			 "Is the Citadel service running on the same host as WebCit?");
		if (yesno(question, ((!strcasecmp(hostname, "uds")) ? 1 : 0))) {
			strcpy(hostname, "uds");
			if (atoi(portname) != 0) strcpy(portname, "/usr/local/citadel");
			set_value("In what directory is Citadel installed?", portname);
		}
		else {
			if (!strcasecmp(hostname, "uds")) strcpy(hostname, "127.0.0.1");
			if (atoi(portname) == 0) strcpy(portname, "504");
			set_value("Enter the host name or IP address of your "
				  "Citadel server.", hostname);
			set_value("Enter the port number on which Citadel is "
				  "running (usually 504)", portname);
		}
	}


	fp = fopen(initfile, "w");

	fprintf(fp,	"#!/bin/sh\n"
		"\n"
		"# uncomment this to create coredumps as described in\n"
		"# http://www.citadel.org/doku.php/faq:mastering_your_os:gdb#how.do.i.make.my.system.produce.core-files\n"
		"# ulimit -c unlimited\n"
			"WEBCIT_DIR=%s\n", setup_directory);
	fprintf(fp,	"HTTP_PORT=%s\n", http_port);
#ifdef HAVE_OPENSSL
	fprintf(fp,	"HTTPS_PORT=%s\n", https_port);
#endif
	fprintf(fp,	"CTDL_HOSTNAME=%s\n", hostname);
	fprintf(fp,	"CTDL_PORTNAME=%s\n", portname);

#ifdef ENABLE_NLS
	
	if (localechoice == 0) {
#ifdef HAVE_USELOCALE 
		fprintf(fp, "unset LANG\n");
#else
		fprintf(fp, "export WEBCIT_LANG=c\n");
#endif
	}
	else {
		fprintf(fp, "export WEBCIT_LANG=%s\n", AvailLang[localechoice - localeoffset]);

	}
#else
	fprintf(fp,     "# your system doesn't support locales\n");
#endif
	fprintf(fp,	"\n"
			"\n"
			"case \"$1\" in\n"
			"\n"
			"start)		echo -n \"Starting WebCit... \"\n"
			"		if   $WEBCIT_DIR/webcit "
		                                        "-D/var/run/webcit.pid "
							"-p$HTTP_PORT $CTDL_HOSTNAME $CTDL_PORTNAME\n"
			"		then\n"
			"			echo \"ok\"\n"
			"		else\n"
			"			echo \"failed\"\n"
			"		fi\n");
#ifdef HAVE_OPENSSL
	fprintf(fp,	"		echo -n \"Starting WebCit SSL... \"\n"
			"		if  $WEBCIT_DIR/webcit "
		                                        "-D/var/run/webcit-ssl.pid "
							"-s -p$HTTPS_PORT $CTDL_HOSTNAME $CTDL_PORTNAME\n"
			"		then\n"
			"			echo \"ok\"\n"
			"		else\n"
			"			echo \"failed\"\n"
			"		fi\n");
#endif
	fprintf(fp,	"		;;\n"
			"stop)		echo -n \"Stopping WebCit... \"\n"
			"		if kill `cat /var/run/webcit.pid 2>/dev/null` 2>/dev/null\n"
			"		then\n"
			"			echo \"ok\"\n"
			"		else\n"
			"			echo \"failed\"\n"
			"		fi\n"
			"		rm -f /var/run/webcit.pid 2>/dev/null\n");
#ifdef HAVE_OPENSSL
	fprintf(fp,	"		echo -n \"Stopping WebCit SSL... \"\n"
			"		if kill `cat /var/run/webcit-ssl.pid 2>/dev/null` 2>/dev/null\n"
			"		then\n"
			"			echo \"ok\"\n"
			"		else\n"
			"			echo \"failed\"\n"
			"		fi\n"
			"		rm -f /var/run/webcit-ssl.pid 2>/dev/null\n");
#endif
	fprintf(fp,	"		;;\n"
			"restart)	$0 stop\n"
			"		$0 start\n"
			"		;;\n"
			"*)		echo \"Usage: $0 {start|stop|restart}\"\n"
			"		exit 1\n"
			"		;;\n"
			"esac\n"
	);

	fclose(fp);
	chmod(initfile, 0755);

	/* Set up the run levels. */
	rv = system("/bin/rm -f /etc/rc?.d/[SK]??webcit 2>/dev/null");
	snprintf(command, sizeof(command), "for x in 2 3 4 5 ; do [ -d /etc/rc$x.d ] && ln -s %s /etc/rc$x.d/S84webcit ; done 2>/dev/null", initfile);
	rv = system(command);
	snprintf(command, sizeof(command), "for x in 0 6 S; do [ -d /etc/rc$x.d ] && ln -s %s /etc/rc$x.d/K15webcit ; done 2>/dev/null", initfile);
	rv = system(command);

}




/*
 * Figure out what type of user interface we're going to use
 */
int discover_ui(void)
{

	/* Use "dialog" if we have it */
	if (getenv("CTDL_DIALOG") != NULL) {
		return UI_DIALOG;
	}
		
	return UI_TEXT;
}





int main(int argc, char *argv[])
{
	int a;
	char aaa[256];
	int info_only = 0;
	int rv;

	strcpy(suggested_url, "http://<your_host_name>:<port>/");

	/* set an invalid setup type */
	setup_type = (-1);

	/* Check to see if we're running the web installer */
	if (getenv("CITADEL_INSTALLER") != NULL) {
		using_web_installer = 1;
	}

	/* parse command line args */
	for (a = 0; a < argc; ++a) {
		if (!strncmp(argv[a], "-u", 2)) {
			strcpy(aaa, argv[a]);
			strcpy(aaa, &aaa[2]);
			setup_type = atoi(aaa);
		}
		if (!strcmp(argv[a], "-i")) {
			info_only = 1;
		}
		if (!strcmp(argv[a], "-q")) {
			setup_type = UI_SILENT;
		}
	}


	/* If a setup type was not specified, try to determine automatically
	 * the best one to use out of all available types.
	 */
	if (setup_type < 0) {
		setup_type = discover_ui();
	}
	if (info_only == 1) {
		important_message("WebCit Setup", "Welcome to WebCit setup");
		cleanup(0);
	}

	/* Get started in a valid setup directory. */
	strcpy(setup_directory, WEBCITDIR);
	if ( (using_web_installer) && (getenv("WEBCIT") != NULL) ) {
		strcpy(setup_directory, getenv("WEBCIT"));
	}
	else {
		set_value("In what directory is WebCit installed?",
			setup_directory);
	}
	if (chdir(setup_directory) != 0) {
		important_message("WebCit Setup",
			  "The directory you specified does not exist.");
		cleanup(errno);
	}

	/*
	 * We used to start WebCit by putting it directly into /etc/inittab.
	 * Since some systems are moving away from init, we can't do this anymore.
	 */
	progress("Removing obsolete /etc/inittab entries...", 0, 1);
	delete_the_old_way();
	progress("Removing obsolete /etc/inittab entries...", 1, 1);

	/* Now begin. */
	switch (setup_type) {

	case UI_TEXT:
		printf("\n\n\n"
			"               *** WebCit setup program ***\n\n");
		break;

	}

	/* 
	 * If we're running on SysV, install init scripts.
	 */
	if (!access("/var/run", W_OK)) {
		install_init_scripts();

		if (!access("/etc/init.d/webcit", X_OK)) {
			rv = system("/etc/init.d/webcit stop");
			rv = system("/etc/init.d/webcit start");
		}

		sprintf(aaa,
			"Setup is finished.  You may now log in.\n"
			"Point your web browser at %s\n", suggested_url
		);
		important_message("Setup finished", aaa);
	}

	else {
		important_message("Setup finished",
			"Setup is finished.  You may now start the server.");
	}

	cleanup(0);
	return 0;
}