#include "include.h"
#include <ctype.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

int cmode, wmode, lmode;

void output(const char *str, size_t lines, size_t words, size_t chars) {
  if (lmode) {
    printf("%d", lines);
  }

  if (wmode) {
    if (lmode) {
      putchar(' ');
    }

    printf("%d", words);
  }

  if (cmode) {
    if (lmode) {
      putchar(' ');
    }

    printf("%d", chars);
  }

  if (str) {
    printf(" %s", str);
  }

  putchar('\n');
}

void wc(int argc, char **argv) {
  int fd = fd_stdin;
  size_t total_chars, total_words, total_lines;
  total_chars = total_words = total_lines = 0;

  if (NULL == *argv) {
    goto read_stdin;
  }

  for (; *argv; argv++) {
    int new_word = 1;
    size_t chars = 0, words = chars, lines = chars;

    fd = open(*argv, O_RDONLY, 0);
  read_stdin:
    for (char c; read(fd, &c, 1);) {
      if (cmode) {
        chars++;
      }

      if (wmode) {
        if (isspace(c)) {
          new_word = 1;
        } else if (new_word) {
          words++;
          new_word = 0;
        }
      }

      if (lmode) {
        if ('\n' == c) {
          lines++;
        }
      }
    }
    close(fd);
    if (lmode) {
      total_lines += lines;
    }
    if (wmode) {
      total_words += words;
    }
    if (cmode) {
      total_chars += chars;
    }

    output(*argv, lines, words, chars);
    if (NULL == *argv) {
      return; // We read from stdin
    }
  }

  if (argc > 1) {
    output("total", total_lines, total_words, total_chars);
  }
}

int wc_main(int argc, char **argv) {
  cmode = wmode = lmode = 0;
  for (argc--; *++argv && '-' == **argv; argc--) {
    switch (*(*argv + 1)) {
    case 'l':
      lmode = 1;
      break;
    case 'w':
      wmode = 1;
      break;
    case 'c':
      cmode = 1;
      break;
    }
  }

  if (!(lmode || cmode || wmode)) {
    cmode = wmode = lmode = 1;
  }

  wc(argc, argv);
  return 0;
}