/*
Copyright 2019 Matthew Graham <sl@airmail.cc>

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

#include <stdio.h>

void wcout(char *wd, int lines, int words, int bytes, char name[])
{
	while (*wd) switch(*wd++) {
		case 'l':
			printf("%7i", lines);
			break;
		case 'w':
			printf("%7i", words);
			break;
		case 'c':
			printf("%8i", bytes);
			break;
	}
	printf("%12s\n", name);
}

int main(int argc, char *argv[])
{
	int character;
	int linec=0, wordc=0, bytec=0, iter = 1;
	int tlinec=0, twordc=0, tbytec=0;
	unsigned int spaceflag, lineflag, tabflag;
	char *wd = "lwc";
	FILE *filep = stdin;

	if (argc > 1 && argv[1][0]=='-') {
		wd = ++argv[1];
		argc--;
		argv++;
	}

	do {
		lineflag=1;
		if (argc > 1 && (filep = fopen(argv[iter], "r"))==NULL) {
			fprintf(stderr, "wc: unable to open %s\n", argv[iter]);
			continue;
		}
		while ((character = fgetc(filep)) != EOF) {
			bytec++;
			if (character>='A' && character<='z' && (spaceflag==1||lineflag==1||tabflag==1)) {
		        	wordc++;
				spaceflag=0;
				lineflag=0;
				tabflag=0;
			}
			switch(character) {
			case '\n': linec++;
				lineflag=1;
				break;
			case ' ': spaceflag=1;
				break;
			case '\t': tabflag=1;
				break;
			}
		} fclose(filep);

		if (argc > 2) {
			tlinec += linec;
			twordc += wordc;
			tbytec += bytec;
		}
		if (argc > 1)
			wcout(wd, linec, wordc, bytec, argv[iter]);
		else
			wcout(wd, linec, wordc, bytec, "");

		linec = 0;
		wordc = 0;
		bytec = 0;
	} while (++iter<argc);
	if (argc > 2)
		wcout(wd, tlinec, twordc, tbytec, "total");
}