#include <common.h>

/**
** User function main #5:  exit, fork, exec, write
**
** Iterates spawning copies of userW (and possibly userZ), reporting
** their PIDs as it goes.
**
** Invoked as:  main5  x  n  b
**	 where x is the ID character
**		   n is the iteration count
**		   b is the w&z boolean
*/

USERMAIN(main)
{
	int count = 5; // default iteration count
	char ch = '5'; // default character to print
	int alsoZ = 0; // also do userZ?
	char msgw[] = "*5w*";
	char msgz[] = "*5z*";
	char buf[128];

	// process the command-line arguments
	switch (argc) {
	case 4:
		alsoZ = argv[3][0] == 't';
		// FALL THROUGH
	case 3:
		count = str2int(argv[2], 10);
		// FALL THROUGH
	case 2:
		ch = argv[1][0];
		break;
	default:
		sprint(buf, "main5: argc %d, args: ", argc);
		cwrites(buf);
		for (int i = 0; i <= argc; ++i) {
			sprint(buf, " %s", argv[argc] ? argv[argc] : "(null)");
			cwrites(buf);
		}
		cwrites("\n");
	}

	// update the extra message strings
	msgw[1] = msgz[1] = ch;

	// announce our presence
	write(CHAN_SIO, &ch, 1);

	// set up the argument vector(s)

	// W:  15 iterations, 5-second sleep
	char *argsw[] = { "userW", "W", "15", "5", NULL };

	// Z:  15 iterations
	char *argsz[] = { "userZ", "Z", "15", NULL };

	for (int i = 0; i < count; ++i) {
		write(CHAN_SIO, &ch, 1);
		int whom = spawn(ProgW, argsw);
		if (whom < 1) {
			swrites(msgw);
		}
		if (alsoZ) {
			whom = spawn(ProgZ, argsz);
			if (whom < 1) {
				swrites(msgz);
			}
		}
	}

	exit(0);

	return (42); // shut the compiler up!
}