#include <fcntl.h>
#include <stdio.h>
#include <errno.h>
#include <unistd.h>
#include <string.h>

/* cmd arguments l - lines c - bytes w - words */
unsigned int l_flag;
unsigned int c_flag;
unsigned int w_flag;

unsigned int words;
unsigned int bytes;
unsigned int lines;

/* Total */
unsigned int twords;
unsigned int tbytes;
unsigned int tlines;

void count(const int fd) {
	char buf[4096];
	ssize_t n = 0;

	int in_word = 1;
	while ((n = read(fd, buf, sizeof(buf))) > 0) {
		bytes += n;

		for (ssize_t i = 0; i < n; i++) {
			if (buf[i] == '\n')
				lines++;

			if (buf[i] == ' ' || buf[i] - '\t' < 5) {
				if (!in_word) {
					in_word = 1;
					words++;
				}
			}

			else
				in_word = 0;
		}
	}

	tbytes += bytes;
	twords += words;
	tlines += lines;
}

void print_count(const char *path, unsigned int plines, unsigned int pwords, unsigned int pbytes) {
	if (l_flag)
		printf("%u ", plines);

	if (w_flag)
		printf("%u ", pwords);

	if (c_flag)
		printf("%u", pbytes);

	printf(" %s\n", path);
}

int main(const int argc, const char **argv) {
	int i;
	for (i = 1; i < argc; i++) {
		if (argv[i][0] != '-')
			break;

		else if (!strcmp(argv[i], "-l"))
			l_flag = 1;

		else if (!strcmp(argv[i], "-c"))
			c_flag = 1;

		else if (!strcmp(argv[i], "-w"))
			w_flag = 1;

		else if (!strcmp(argv[i], "-h")) {
			printf("wc [-l (lines)] [-c (bytes)] [-w (words)] [file1 file2...]\n");
			return 0;
		}
	}


	if (!w_flag && !l_flag && !c_flag) {
		w_flag = 1;
		l_flag = 1;
		c_flag = 1;
	}

	if (i == argc) {
		count(STDIN_FILENO);
		print_count("", lines, words, bytes);
	}

	for (; i < argc; i++) {
		int fd = open(argv[i], O_RDONLY);
		if (fd < 0) {
			fprintf(stderr, "wc: %s\n", strerror(errno));
			return 1;
		}

		count(fd);
		if (i == argc - 1)
			print_count("total", tlines, twords, tbytes);

		else
			print_count(argv[i], lines, words, bytes);

		close(fd);
		words = bytes = lines = 0;
	}

	return 0;
}