wc looping version

Author: Faithy4444

implement-shell-tools/wc/wc.mjs

@@ -7,7 +7,7 @@ program
 .option("-l, --line", "counts number of lines in the file")
 .option("-w, --words", "counts number of words in the file")
 .option("-c, --bytes", "counts number  of bytes in the file")
-.argument("<filepath>")
+.argument("<filepaths...>")
 
 program.parse();
 
@@ -19,37 +19,54 @@ if(args.length === 0){
   program.help();
 }
 
-const path = args[0];
-const content = await fs.readFile(path, "utf-8");
+const countLines = content => content.split(/\r?\n/).length;
+const countWords = content => content.trim().split(/\s+/).filter(Boolean).length;
+const countBytes = content => Buffer.byteLength(content, "utf-8");
 
+// store totals for multiple files
+let totalLines = 0;
+let totalWords = 0;
+let totalBytes = 0;
 
-// function to count the number of lines in a files
-function countLines(content){
-const lines = content.split(/\r?\n/);
-return lines.length;
-}
+for (const path of args) {
+  try {
+    const content = await fs.readFile(path, "utf-8");
 
-//function to count the words
-function countWords(content){
-  const words = content.trim().split(/\s+/);
-  return words.length;
-}
+    const lines = countLines(content);
+    const words = countWords(content);
+    const bytes = countBytes(content);
+
+    totalLines += lines;
+    totalWords += words;
+    totalBytes += bytes;
+
+    // collect counts based on options
+    const output = [];
+    if (opts.line) output.push(lines);
+    if (opts.words) output.push(words);
+    if (opts.bytes) output.push(bytes);
 
-//function to count the bytes
-function countBytes(content){
-  return Buffer.byteLength(content, "utf-8");
+    // if no options are passed, print all
+    if (!opts.line && !opts.words && !opts.bytes) {
+      output.push(lines, words, bytes);
+    }
+
+    console.log(...output, path);
+  } catch (err) {
+    console.error(`Error reading file "${path}": ${err.message}`);
+  }
 }
 
-const lines = countLines(content);
-const words = countWords(content);
-const bytes = countBytes(content);
-
-if (opts.line) {
-  console.log(lines);
-} else if (opts.words) {
-  console.log(words, path);
-} else if (opts.bytes) {
-  console.log(bytes, path);
-} else {
-  console.log(lines, words, bytes, path);
+// print totals if more than one file
+if (args.length > 1) {
+  const totalOutput = [];
+  if (opts.line) totalOutput.push(totalLines);
+  if (opts.words) totalOutput.push(totalWords);
+  if (opts.bytes) totalOutput.push(totalBytes);
+
+  if (!opts.line && !opts.words && !opts.bytes) {
+    totalOutput.push(totalLines, totalWords, totalBytes);
+  }
+
+  console.log(...totalOutput, "total");
 }