Add flag support to wc (-l, -w, -c)

Author: zohrehKazemianpour

implement-shell-tools/wc/wc.js

@@ -7,6 +7,9 @@ program
   .description(
     "The wc utility displays the number of lines, words, and bytes contained in each input file, or standard input"
   )
+  .option("-l, --lines", "count lines only")
+  .option("-w, --words", "count words only")
+  .option("-c, --bytes", "count bytes only")
   .argument("<path...>", "The file paths to process");
 
 //interpret the program
@@ -15,7 +18,25 @@ program.parse();
 //initialise totals
 let totalLines = 0;
 let totalWords = 0;
-let totalCharacters = 0;
+let totalBytes = 0;
+
+//check for flags
+const hasLineFlag = program.opts().lines;
+const hasWordFlag = program.opts().words;
+const hasBytesFlag = program.opts().bytes;
+
+// create output format function to avoid repetition
+function formatOutput(lines, words, bytes, path) {
+  if (hasLineFlag) {
+    console.log(`${lines} ${path}`);
+  } else if (hasWordFlag) {
+    console.log(`${words} ${path}`);
+  } else if (hasBytesFlag) {
+    console.log(`${bytes} ${path}`);
+  } else {
+    console.log(`${lines} ${words} ${bytes} ${path}`);
+  }
+}
 
 //process each file
 
@@ -29,17 +50,17 @@ for (const path of program.args) {
   //count words (split by any whitespace)
   const words = content.split(/\s+/).filter((word) => word.length > 0).length;
 
-  //count character
-  const characters = content.length;
+  //count bytes correctly especially important for non-ASCII characters
+  const bytes = Buffer.byteLength(content, "utf-8");
 
   //Add to totals
   totalLines += lines;
   totalWords += words;
-  totalCharacters += characters;
+  totalBytes += bytes;
 
-  console.log(`${lines} ${words} ${characters} ${path}`);
+  formatOutput(lines, words, bytes, path);
 }
 
 if (program.args.length > 1) {
-  console.log(`${totalLines} ${totalWords} ${totalCharacters} total`);
+  formatOutput(totalLines, totalWords, totalBytes, "total");
 }