main.go

package main

import (
	"flag"
	"fmt"
	"os"
)

var (
	byteCount = flag.Bool("c", false, "print the byte count")
	charCount = flag.Bool("m", false, "print the char count")
	lineCount = flag.Bool("l", false, "print the line count")
	wordCount = flag.Bool("w", false, "print the word count")
	debug     = flag.Bool("d", false, "print debug info")
	verbose   = flag.Bool("v", false, "print verbose output")
)

func main() {
	flag.Parse()
	if flag.NArg() == 0 { // TODO : Iterate through all the files passed instead of just processing the first one, should be easy
		flag.Usage()
		os.Exit(1)
	}
	fname := flag.Arg(0)

	if givenFileIsADir(fname) {
		fmt.Printf("wc: %s: Is a directory\n", fname)
		os.Exit(1)
	}
	toPrint := 0 // how many values to print
	var bc, cc, lc, wc int
	if *byteCount {
		bc = getByteCount(fname)
		toPrint++
	}
	if *charCount {
		cc = getCharCount(fname)
		toPrint++
	}
	if *lineCount {
		lc = getLineCount(fname)
		toPrint++
	}
	if *wordCount {
		wc = getWordCount(fname)
		toPrint++
	}
	if !*byteCount && !*charCount && !*lineCount && !*wordCount {
		toPrint += 4
		bc, cc, lc, wc = getByteCount(fname), getCharCount(fname), getLineCount(fname), getWordCount(fname)
	}

	output(bc, lc, cc, wc, toPrint, fname)

}