#include	"mk.h"

#define		MKFILE		"mkfile"

static char *version = "@(#)mk general release 4 (plan 9)";
int debug;
Rule *rules, *metarules;
int nproclimit;
int nflag = 0;
int tflag = 0;
int iflag = 0;
int kflag = 0;
int aflag = 0;
char *explain = 0;
Word *target1;
int nreps = 1;
Job *jobs;
char *whatif = 0;
char shell[] = SHELL;
char shellname[] = SHELL;
Biobuf stdout;
Rule *patrule;
#ifdef	PROF
short buf[10000];
#endif	PROF

void
main(int argc, char **argv)
{
	Word *w;
	char *s;
	char *files[256], **f = files, **ff;
	int sflag = 0;
	int i;
	int tfd = -1;
	Biobuf tb;
	Bufblock *buf;
	static char temp[] = "/tmp/mkargXXXXXX";

	/*
	 *  start with a copy of the current environment variables
	 *  instead of sharing them
	 */
	rfork(RFENVG);

	Binit(&stdout, 1, OWRITE);
	buf = newbuf();
	USED(argc);
	for(argv++; *argv && (**argv == '-'); argv++)
	{
		bufcpy(buf, argv[0], strlen(argv[0]));
		insert(buf, ' ');
		switch(argv[0][1])
		{
		case 'a':
			aflag = 1;
			break;
		case 'd':
			if(*(s = &argv[0][2]))
				while(*s) switch(*s++)
				{
				case 'p':	debug |= D_PARSE; break;
				case 'g':	debug |= D_GRAPH; break;
				case 'e':	debug |= D_EXEC; break;
				}
			else
				debug = 0xFFFF;
			break;
		case 'e':
			explain = &argv[0][2];
			break;
		case 'f':
			if(*++argv == 0)
				usage();
			*f++ = *argv;
			bufcpy(buf, argv[0], strlen(argv[0]));
			insert(buf, ' ');
			break;
		case 'i':
			iflag = 1;
			break;
		case 'k':
			kflag = 1;
			break;
		case 'n':
			nflag = 1;
			break;
		case 's':
			sflag = 1;
			break;
		case 't':
			tflag = 1;
			break;
		case 'w':
			if(argv[0][2])
				whatif = &argv[0][2];
			else {
				if(*++argv == 0)
					usage();
				whatif = &argv[0][0];
			}
			break;
		default:
			usage();
		}
	}
#ifdef	PROF
	{
		extern etext();
		monitor(main, etext, buf, sizeof buf, 300);
	}
#endif	PROF

	if(aflag)
		iflag = 1;
	for(i = strlen(shell)-1; i >= 0; i--)
			if(shell[i] == '/')
				break;
	strcpy(shellname, shell+i+1);
	account();
	syminit();
	initenv();

	/*
		assignment args become null strings
	*/
	for(i = 0; argv[i]; i++) if(utfrune(argv[i], '=')){
		bufcpy(buf, argv[i], strlen(argv[i]));
		insert(buf, ' ');
		if(tfd < 0){
			mktemp(temp);
			close(CREAT(temp, 0600));
			if((tfd = open(temp, 2)) < 0){
				perror(temp);
				Exit();
			}
			Binit(&tb, tfd, OWRITE);
		}
		Bprint(&tb, "%s\n", argv[i]);
		*argv[i] = 0;
	}
	if(tfd >= 0){
		Bflush(&tb);
		LSEEK(tfd, 0L, 0);
		parse("command line args", tfd, 1, 1);
		UNLINK(temp);
	}

	if (buf->current != buf->start) {
		buf->current--;
		insert(buf, 0);
	}
	symlook("MKFLAGS", S_VAR, (char *) stow(buf->start));
	buf->current = buf->start;
	for(i = 0; argv[i]; i++){
		if(*argv[i] == 0) continue;
		if(i)
			insert(buf, ' ');
		bufcpy(buf, argv[i], strlen(argv[i]));
	}
	insert(buf, 0);
	symlook("MKARGS", S_VAR, (char *) stow(buf->start));
	freebuf(buf);

	if(f == files){
		if(access(MKFILE, 4) == 0)
			parse(MKFILE, open(MKFILE, 0), 0, 1);
	} else
		for(ff = files; ff < f; ff++)
			parse(*ff, open(*ff, 0), 0, 1);
	if(DEBUG(D_PARSE)){
		dumpw("default targets", target1);
		dumpr("rules", rules);
		dumpr("metarules", metarules);
		dumpv("variables");
	}
	if(whatif)
		timeinit(whatif);
	execinit();
	/* skip assignment args */
	while(*argv && (**argv == 0))
		argv++;
	sigcatch();
	if(*argv == 0){
		if(target1)
			for(w = target1; w; w = w->next)
				mk(w->s);
		else {
			fprint(2, "mk: nothing to mk\n");
			Exit();
		}
	} else {
		if(sflag){
			for(; *argv; argv++)
				if(**argv)
					mk(*argv);
		} else {
			Word *head, *tail, *t;

			/* fake a new rule with all the args as prereqs */
			tail = 0;
			for(; *argv; argv++)
				if(**argv){
					if(tail == 0)
						tail = t = newword(*argv);
					else {
						t->next = newword(*argv);
						t = t->next;
					}
				}
			if(tail->next == 0)
				mk(tail->s);
			else {
				head = newword("command line arguments");
				addrules(head, tail, strdup(""), VIR, inline, 1, (char *)0);
				mk(head->s);
			}
		}
	}
	exits(0);
}

void
usage(void)
{

	fprint(2, "Usage: mk [-f file] [-n] [-a] [-e] [-t] [-k] [-i] [-d[egp]] [targets ...]\n");
	Exit();
}

char *
Malloc(int n)
{
	register char *s;

	if(s = malloc(n))
		return(s);
	fprint(2, "mk: cannot alloc %d bytes\n", n);
	Exit();
	return((char *)0);	/* shut cyntax up */
}

char *
Realloc(char *s, int n)
{
	if(s = realloc(s, n))
		return(s);
	fprint(2, "mk: cannot alloc %d bytes\n", n);
	Exit();
	return((char *)0);	/* shut cyntax up */
}

void
Exit(void)
{
	while(wait(0) >= 0)
		;
	exits("error");
}

/*
char *
strndup(char *s, unsigned n)
{
	register char *goo;

	goo = Malloc(n);
	memmove(goo, s, (COUNT)n);
	return(goo);
}
*/

void
assert(char *s, int n)
{
	if(!n){
		fprint(2, "mk: Assertion ``%s'' failed.\n", s);
		Exit();
	}
}

void
regerror(char *s)
{
	if(patrule)
		fprint(2, "mk: %s:%d: regular expression error; %s\n",
			patrule->file, patrule->line, s);
	else
		fprint(2, "mk: %s:%d: regular expression error; %s\n",
			infile, inline, s);
	Exit();
}