Added support for multiple flags with file input

main
oabrivard 2 years ago
parent f747fbad46
commit 259ea60013

@ -160,22 +160,24 @@ func main() {
byteCounter = fileByteCounter byteCounter = fileByteCounter
} }
switch { if *lFlag || flag.NFlag() == 0 {
case *cFlag:
fmt.Print("\t", byteCounter(f))
case *lFlag:
fmt.Print("\t", lineCounter(f)) fmt.Print("\t", lineCounter(f))
case *wFlag: f.Seek(0, 0)
}
if *wFlag || flag.NFlag() == 0 {
fmt.Print("\t", wordCounter(f)) fmt.Print("\t", wordCounter(f))
case *mFlag:
fmt.Print("\t", charCounter(f))
default:
b := byteCounter(f)
f.Seek(0, 0) f.Seek(0, 0)
l := lineCounter(f) }
if *cFlag || flag.NFlag() == 0 {
fmt.Print("\t", byteCounter(f))
f.Seek(0, 0)
}
if *mFlag {
fmt.Print("\t", charCounter(f))
f.Seek(0, 0) f.Seek(0, 0)
w := wordCounter(f)
fmt.Print("\t", l, "\t", w, "\t", b)
} }
if len(fileName) > 0 { if len(fileName) > 0 {

Loading…
Cancel
Save