/*
 *
 * init.c - main loop and initialization routines
 *
 * This file is part of zsh, the Z shell.
 *
 * This software is Copyright 1992 by Paul Falstad
 *
 * Permission is hereby granted to copy, reproduce, redistribute or otherwise
 * use this software as long as: there is no monetary profit gained
 * specifically from the use or reproduction of this software, it is not
 * sold, rented, traded or otherwise marketed, and this copyright notice is
 * included prominently in any copy made. 
 *
 * The author make no claims as to the fitness or correctness of this software
 * for any use whatsoever, and it is provided as is. Any use of this software
 * is at the user's own risk. 
 *
 */

#define GLOBALS
#include "zsh.h"
#include <pwd.h>

static int noexitct = 0;

void main(argc,argv,envp) /**/
int argc; char **argv; char **envp;
{
char *zshname;

#ifdef LC_ALL
	setlocale(LC_ALL, "");
#endif
	environ = envp;
	meminit();
	if (!(zshname = strrchr(argv[0], '/')))
		zshname = argv[0];
	else
		zshname++;
	setflags(zshname);
	parseargs(argv);
	setmoreflags();
	setupvals();
	initialize();
	heapalloc();
	runscripts(zshname);
	for(;;)
		{
		do
			loop(1);
		while (tok != ENDINPUT);
		if (!(isset(IGNOREEOF) && interact))
			{
#if 0
			if (interact)
				fputs(islogin ? "logout\n" : "exit\n",stderr);
#endif
			zexit(lastval);
			continue;
			}
		noexitct++;
		if (noexitct >= 10)
			{
			stopmsg = 1;
			zexit(lastval);
			}
		zerrnam("zsh",(!islogin) ? "use 'exit' to exit."
			: "use 'logout' to logout.",NULL,0);
		}
}

/* keep executing lists until EOF found */

void loop(toplevel) /**/
int toplevel;
{
List list;
Heap h = (Heap) peekfirst(heaplist);

	pushheap();
	for(;;)
		{
		freeheap();
		if (interact && isset(SHINSTDIN))
			preprompt();
		hbegin();		/* init history mech */
		intr();			/* interrupts on */
		ainit();			/* init alias mech */
		lexinit();
		errflag = 0;
		if (!(list = parse_event()))
			{				/* if we couldn't parse a list */
			hend();
			if (tok == ENDINPUT && !errflag)
				break;
			continue;
			}
		if (hend())
			{
			if (stopmsg)		/* unset 'you have stopped jobs' flag */
				stopmsg--;
			execlist(list);
			if (toplevel)
				noexitct = 0;
			}
		if (ferror(stderr))
			{
			zerr("write error",NULL,0);
			clearerr(stderr);
			}
		if (subsh)				/* how'd we get this far in a subshell? */
			exit(lastval);
		if ((!interact && errflag) || retflag)
			break;
		if (isset('t') || (lastval && isset(ERREXIT)))
			{
			if (sigtrapped[SIGEXIT])
				dotrap(SIGEXIT);
			exit(lastval);
			}
		}
	while ((Heap) peekfirst(heaplist) != h)
		popheap();
}

void setflags(zshname) /**/
const char *zshname;
{
int c;

	for (c = 0; c != 32; c++)		opts[c] = OPT_UNSET;
	for (c = 32; c != 128; c++)	opts[c] = OPT_INVALID;
	for (c = 'a'; c <= 'z'; c++)	opts[c] = opts[c-'a'+'A'] = OPT_UNSET;
	for (c = '0'; c <= '9'; c++)	opts[c] = OPT_UNSET;
	opts['A'] = OPT_INVALID;
	opts['i'] = (isatty(0)) ? OPT_SET : OPT_UNSET;
	opts[BGNICE] = opts[NOTIFY] = OPT_SET;
	opts[USEZLE] = (interact && SHTTY != -1) ? OPT_SET : OPT_UNSET;
	opts[HASHCMDS] = opts[HASHLISTALL] = opts[HASHDIRS] = OPT_SET;

	/* KSH Mode:
		The following seven options cause zsh to behave more like KSH
		when invoked as "ksh".
		K - don't recognize csh-style history subst
		k - allow interactive comments
		I - don't perform brace expansion
		3 - don't print error for unmatched wildcards
		H - don't query 'rm *'
		y - split parameters using IFS
		KSHOPTIONPRINT - print options ksh-like
		wnp@rcvie.co.at, 1992-05-14
	*/

	if ( strcmp(zshname, "ksh") == 0 )
	{
		opts['K'] = opts['k'] = opts['I'] = opts['3'] = OPT_SET ;
		opts['H'] = opts['y'] = opts[KSHOPTIONPRINT] = OPT_SET ;
	}
}

static char *cmd;

void parseargs(argv) /**/
char **argv;
{
char **x;
int bk = 0,action;
Lklist paramlist;

	hackzero = argzero = *argv;
	opts[LOGINSHELL] = (**(argv++) == '-') ? OPT_SET : OPT_UNSET;
	SHIN = 0;
	while (!bk && *argv && (**argv == '-' || **argv == '+'))
		{
		action = (**argv == '-') ? OPT_SET : OPT_UNSET;
		while (*++*argv) {
			if (opts[(int)**argv] == OPT_INVALID) {
				zerr("bad option: -%c",NULL,**argv);
				exit(1);
			}
			if (bk = **argv == 'b') break;
			if (**argv == 'c') { /* -c command */
				argv++;
				if (!*argv) {
					zerr("string expected after -c",NULL,0);
					exit(1);
				}
				cmd = *argv;
				opts[INTERACTIVE] = OPT_UNSET;
				opts['c'] = OPT_SET;
				break;
			} else if (**argv == 'o') {
				int c;

				if (!*++*argv)
					argv++;
				if (!*argv) {
					zerr("string expected after -o",NULL,0);
					exit(1);
				}
				c = optlookup(*argv);
				if (c == -1)
					zerr("no such option: %s",*argv,0);
				else
					opts[c] = action;
				break;
			} else opts[(int)**argv] = action;
		}
		argv++;
	}
	paramlist = newlist();
	if (*argv)
		{
		if (opts[SHINSTDIN] == OPT_UNSET)
			{
			SHIN = movefd(open(argzero = *argv,O_RDONLY));
			if (SHIN == -1)
				{
				zerr("can't open input file: %s",*argv,0);
				exit(1);
				}
			opts[INTERACTIVE] = OPT_UNSET;
			argv++;
			}
		while (*argv)
			addnode(paramlist,ztrdup(*argv++));
		}
	else
		opts[SHINSTDIN] = OPT_SET;
	pparams = x = zcalloc((countnodes(paramlist)+1)*sizeof(char *));
	while (*x++ = getnode(paramlist));
	free(paramlist);
	argzero = ztrdup(argzero);
}

void setmoreflags() /**/
{
#ifndef NOCLOSEFUNNYFDS
int t0;
#endif
long ttpgrp;

	/* stdout,stderr fully buffered */
#ifdef _IOFBF
	setvbuf(stdout,malloc(BUFSIZ),_IOFBF,BUFSIZ);
	setvbuf(stderr,malloc(BUFSIZ),_IOFBF,BUFSIZ);
#else
	setbuffer(stdout,malloc(BUFSIZ),BUFSIZ);
	setbuffer(stderr,malloc(BUFSIZ),BUFSIZ);
#endif
	subsh = 0;
#ifndef NOCLOSEFUNNYFDS
	/* this works around a bug in some versions of in.rshd */
	for (t0 = 3; t0 != 10; t0++)
		close(t0);
#endif
#ifdef JOB_CONTROL
	opts[MONITOR] = (interact) ? OPT_SET : OPT_UNSET;
	if (jobbing) {
		SHTTY = movefd((isatty(0)) ? dup(0) : open("/dev/tty",O_RDWR));
		if (SHTTY == -1)
			opts[MONITOR] = OPT_UNSET;
		else {
#ifdef TIOCSETD
#ifdef NTTYDISC
			int ldisc = NTTYDISC;
			ioctl(SHTTY, TIOCSETD, &ldisc);
#endif
#endif
			gettyinfo(&shttyinfo);	/* get tty state */
#ifdef sgi
			if (shttyinfo.tio.c_cc[VSWTCH] <= 0) /* hack for irises */
				shttyinfo.tio.c_cc[VSWTCH] = CSWTCH;
#endif
			savedttyinfo = shttyinfo;
		}
#ifdef sgi
		setpgrp(0,getpgrp(0));
#endif
		if ((mypgrp = getpgrp(0)) <= 0)
			opts[MONITOR] = OPT_UNSET;
		else while ((ttpgrp = gettygrp()) != -1 && ttpgrp != mypgrp) {
			sleep(1);
			mypgrp = getpgrp(0);
			if (mypgrp == gettygrp()) break;
			killpg(mypgrp,SIGTTIN);
			mypgrp = getpgrp(0);
		}
	} else
		SHTTY = -1;
#else
	opts[MONITOR] = OPT_UNSET;
	SHTTY = movefd((isatty(0)) ? dup(0) : open("/dev/tty",O_RDWR));
	if (SHTTY != -1) {
		gettyinfo(&shttyinfo);
		savedttyinfo = shttyinfo;
	}
#endif
}

static long
get_baudrate(speedcode)
int speedcode;
{
	switch(speedcode) {
		case B0:	return(0L);
		case B50:	return(50L);
		case B75:	return(75L);
		case B110:	return(110L);
		case B134:	return(134L);
		case B150:	return(150L);
		case B200:	return(200L);
		case B300:	return(300L);
		case B600:	return(600L);
#ifdef _B900
		case _B900:	return(900L);
#endif
		case B1200:	return(1200L);
		case B1800:	return(1800L);
		case B2400:	return(2400L);
#ifdef _B3600
		case _B3600:	return(3600L);
#endif
		case B4800:	return(4800L);
#ifdef _B7200
		case _B7200:	return(7200L);
#endif
		case B9600:	return(9600L);
#ifdef B19200
		case B19200:	return(19200L);
#else
#ifdef EXTA
              case EXTA:      return(19200L);
#endif
#endif
#ifdef B38400
		case B38400:	return(38400L);
#else
#ifdef EXTB
              case EXTB:      return(38400L);
#endif
#endif
		default:	break;
	}
	return(0L);
}

void setupvals() /**/
{
struct passwd *pswd;
char *ptr,*s;

	curhist = 0;
	histsiz = DEFAULT_HISTSIZE;
	lithistsiz = 5;
	inithist();
	mailcheck = logcheck = 60;
	dirstacksize = -1;
	listmax = 100;
	reporttime = -1;
	bangchar = '!';
	hashchar = '#';
	hatchar = '^';
	termok = 0;
	curjob = prevjob = coprocin = coprocout = -1;
	shtimer = time(NULL);	/* init $SECONDS */
	srand((unsigned int) shtimer);
	/* build various hash tables; argument to newhtable is table size */
	aliastab = newhtable(37);
	addreswords();
	paramtab = newhtable(151);
	cmdnamtab = newhtable(37);
	compctltab = newhtable(13);
	initxbindtab();
	nullcmd = ztrdup("cat");
	readnullcmd = ztrdup("more");
	prompt = ztrdup("%m%# ");
	prompt2 = ztrdup("> ");
	prompt3 = ztrdup("?# ");
	prompt4 = ztrdup("+ ");
	sprompt = ztrdup("zsh: correct `%R' to `%r' [nyae]? ");
	term = ztrdup("");
	ppid = getppid();
#ifdef TIO
#if defined(HAS_TCCRAP) && defined(TERMIOS)
	baud = cfgetospeed(&shttyinfo.tio);
	if (baud < 100) baud = get_baudrate(baud); /* aren't "standards" great?? */
#else
	baud = get_baudrate(shttyinfo.tio.c_cflag & CBAUD);
#endif
#else
	baud = get_baudrate(shttyinfo.sgttyb.sg_ospeed);
#endif
#ifdef TIOCGWINSZ
	if (!(columns = shttyinfo.winsize.ws_col))
		columns = 80;
	if (!(lines = shttyinfo.winsize.ws_row))
		lines = 24;
#else
	columns = 80;
	lines = 24;
#endif
	ifs = ztrdup(" \t\n");
	if (pswd = getpwuid(getuid())) {
		username = ztrdup(pswd->pw_name);
		home = ztrdup(pswd->pw_dir);
	} else {
		username = ztrdup("");
		home = ztrdup("/");
	}
	if (ptr = zgetenv("LOGNAME"))
		logname = ztrdup(ptr);
	else
		logname = ztrdup(username);
	timefmt = ztrdup(DEFTIMEFMT);
	watchfmt = ztrdup(DEFWATCHFMT);
	if (!(ttystrname = ztrdup(ttyname(SHTTY))))
		ttystrname = ztrdup("");
	wordchars = ztrdup(DEFWORDCHARS);
	fceditparam = ztrdup(DEFFCEDIT);
	tmpprefix = ztrdup(DEFTMPPREFIX);
	postedit = ztrdup("");
	if (ispwd(home)) pwd = ztrdup(home);
	else if ((ptr = zgetenv("PWD")) && ispwd(ptr)) pwd = ztrdup(ptr);
	else pwd = zgetwd();
	oldpwd = ztrdup(pwd);
	hostnam = zalloc(256);
	underscore = ztrdup("");
	gethostname(hostnam,256);
	mypid = getpid();
	cdpath = mkarray(NULL);
	manpath = mkarray(NULL);
	fignore = mkarray(NULL);
	fpath = mkarray(NULL);
	mailpath = mkarray(NULL);
	watch = mkarray(NULL);
	hosts = mkarray(NULL);
	psvar = mkarray(NULL);
	compctlsetup();
	userdirs = (char **) zcalloc(sizeof(char *)*2);
	usernames = (char **) zcalloc(sizeof(char *)*2);
	userdirsz = 2;
	userdirct = 0;
	adduserdir("",home);
	optarg = ztrdup("");
	zoptind = 1;
	schedcmds = NULL;
	path = (char **) zalloc(4*sizeof *path);
	path[0] = ztrdup("/bin"); path[1] = ztrdup("/usr/bin");
	path[2] = ztrdup("/usr/ucb"); path[3] = NULL;
	inittyptab();
	initlextabs();
	setupparams();
	setparams();
	inittyptab();
	if ((s = zgetenv("EMACS")) && !strcmp(s,"t") && !strcmp(term,"emacs"))
		opts[USEZLE] = OPT_UNSET;
#ifndef HAS_RUSAGE
	times(&shtms);
#endif
}

void compctlsetup() /**/
{
static char
	*hs[] = {"telnet","rlogin","ftp","rup","rusers","rsh",NULL},
	*os[] = {"setopt","unsetopt",NULL},
	*vs[] = {"export","typeset","vared","unset",NULL},
	*cs[] = {"which","builtin",NULL},
	*bs[] = {"bindkey",NULL};

	compctl_process(hs,CC_HOSTS,NULL);
	compctl_process(os,CC_OPTIONS,NULL);
	compctl_process(vs,CC_VARS,NULL);
	compctl_process(bs,CC_BINDINGS,NULL);
	compctl_process(cs,CC_COMMPATH,NULL);
	cc_compos.mask   = CC_COMMPATH;
	cc_default.mask  = CC_FILES;
}

void initialize() /**/
{
int t0;
#ifdef SYSVR4
	static struct sigaction chldaction;
#endif

	breaks = loops = 0;
	lastmailcheck = time(NULL);
	locallist = NULL;
	dirstack = newlist();
	bufstack = newlist();
	newcmdnamtab();
	inbuf = zalloc(inbufsz = 256);
	inbufptr = inbuf+inbufsz;
	inbufct = 0;
#ifndef QDEBUG
	signal(SIGQUIT,SIG_IGN);
#endif
#ifdef RLIM_INFINITY
	for (t0 = 0; t0 != RLIM_NLIMITS; t0++)
		getrlimit(t0,limits+t0);
#endif
	hsubl = hsubr = NULL;
	lastpid = 0;
	bshin = fdopen(SHIN,"r");
#ifdef SYSVR4
	memset(&chldaction, 0, sizeof(chldaction));
	chldaction.sa_handler = handler;
	sigaction(SIGCHLD, &chldaction, NULL);
#else
	signal(SIGHUP,handler);
	signal(SIGCHLD,handler);
#endif
	if (jobbing)
		{
		long ttypgrp;
		while ((ttypgrp = gettygrp()) != -1 && ttypgrp != mypgrp)
			kill(0,SIGTTIN);
		if (ttypgrp == -1)
			{
			opts[MONITOR] = OPT_UNSET;
			}
		else
			{
			signal(SIGTTOU,SIG_IGN);
			signal(SIGTSTP,SIG_IGN);
			signal(SIGTTIN,SIG_IGN);
			signal(SIGPIPE,SIG_IGN);
			attachtty(mypgrp);
			}
		}
	if (interact)
		{
		signal(SIGTERM,SIG_IGN);
#ifdef SIGWINCH
		signal(SIGWINCH,handler);
#endif
		signal(SIGALRM,handler);
		intr();
		}
}

void addreswords() /**/
{
static char *reswds[] = {
	"do", "done", "esac", "then", "elif", "else", "fi", "for", "case",
	"if", "while", "function", "repeat", "time", "until", "exec", "command",
	"select", "coproc", "noglob", "-", "nocorrect", "foreach", "end", NULL
	};
int t0;

	for (t0 = 0; reswds[t0]; t0++)
		addhperm(reswds[t0],mkanode(NULL,-1-t0),aliastab,(FFunc) 0);
}

void runscripts(zshname) /**/
char *zshname;
{
	/*
	   KSH Mode:
	   if called as "ksh", we source the standard
	   sh/ksh scripts:
	   wnp@rcvie.co.at 1992/05/14
	 */

	if ( strcmp(zshname, "ksh") == 0 )
	{
		if (islogin) source("/etc/profile") ;
		if (islogin) sourcehome(".profile");
		source(getsparam("ENV"));
	}
	else if (isset(NORCS))
	{
#ifdef GLOBALZSHENV
		source(GLOBALZSHENV);
#endif
	}
	else
	{
#ifdef GLOBALZSHENV
		source(GLOBALZSHENV);
#endif
		if (isset(NORCS)) {
#ifdef GLOBALZPROFILE
			if (islogin) source(GLOBALZPROFILE);
#endif
#ifdef GLOBALZSHRC
			if (! isset(NORCS))
				source(GLOBALZSHRC);
#endif
#ifdef GLOBALZLOGIN
			if (islogin && ! isset(NORCS))
				source(GLOBALZLOGIN);
#endif
		} else {
			sourcehome(".zshenv");
			if (interact && ! isset(NORCS)) {
				if (islogin) {
#ifdef GLOBALZPROFILE
					source(GLOBALZPROFILE);
					if (! isset(NORCS))
#endif
						sourcehome(".zprofile");
				}
#ifdef GLOBALZSHRC
				if (! isset(NORCS))
					source(GLOBALZSHRC);
				if (! isset(NORCS))
#endif
					sourcehome(".zshrc");
				if (islogin && ! isset(NORCS)) {
#ifdef GLOBALZLOGIN
					source(GLOBALZLOGIN);
					if (! isset(NORCS))
#endif
						sourcehome(".zlogin");
				}
			}
		}
	}
	if (isset('c'))
		{
		if (SHIN >= 10)
			close(SHIN);
		SHIN = movefd(open("/dev/null",O_RDONLY));
		execstring(cmd);
		stopmsg = 1;
		zexit(lastval);
		}
	if (interact && ! isset(NORCS))
		readhistfile(getsparam("HISTFILE"),0);
#ifdef TIOCSWINSZ
	if (!(columns = shttyinfo.winsize.ws_col))
		columns = 80;
	if (!(lines = shttyinfo.winsize.ws_row))
		lines = 24;
#endif
}

void ainit() /**/
{
	alstackind = 0;		/* reset alias stack */
	alstat = 0;
	isfirstln = 1;
}