Merge pull request #4 from hankei6km/csb-yhio9

Support multiple files

Author: hankei6km

src/cli.test.ts

@@ -11,14 +11,17 @@ describe('cli()', () => {
     stderr.on('data', errData);
     expect(
       await cli({
-        filename: 'test/assets/test.txt',
+        filenames: ['test/assets/test1.txt', 'test/assets/test2.txt'],
         stdout,
         stderr
       })
     ).toEqual(0);
-    expect(outData.mock.calls.length).toEqual(1);
+    expect(outData.mock.calls.length).toEqual(2);
     expect(outData.mock.calls[0][0].toString('utf8')).toEqual(
-      'test/assets/test.txt: 15 chars\n'
+      'test/assets/test1.txt: 15 chars\n'
+    );
+    expect(outData.mock.calls[1][0].toString('utf8')).toEqual(
+      'test/assets/test2.txt: 17 chars\n'
     );
     expect(errData.mock.calls.length).toEqual(0);
   });
@@ -31,12 +34,15 @@ describe('cli()', () => {
     stderr.on('data', errData);
     expect(
       await cli({
-        filename: 'test/assets/fail.txt',
+        filenames: ['test/assets/test1.txt', 'test/assets/fail.txt'],
         stdout,
         stderr
       })
     ).toEqual(1);
-    expect(outData.mock.calls.length).toEqual(0);
+    expect(outData.mock.calls.length).toEqual(1);
+    expect(outData.mock.calls[0][0].toString('utf8')).toEqual(
+      'test/assets/test1.txt: 15 chars\n'
+    );
     expect(errData.mock.calls.length).toEqual(2);
     expect(errData.mock.calls[0][0].toString('utf8')).toEqual(
       "Error: ENOENT: no such file or directory, open 'test/assets/fail.txt'"

src/cli.ts

@@ -2,14 +2,18 @@ import { Writable } from 'stream';
 import countChars from './count';
 
 type Opts = {
-  filename: string;
+  filenames: string[];
   stdout: Writable;
   stderr: Writable;
 };
-const cli = async ({ filename, stdout, stderr }: Opts): Promise<number> => {
+const cli = async ({ filenames, stdout, stderr }: Opts): Promise<number> => {
   try {
-    const count = await countChars(filename);
-    stdout.write(`${filename}: ${count} chars\n`);
+    const len = filenames.length;
+    for (let i = 0; i < len; i++) {
+      const filename = filenames[i];
+      const count = await countChars(filename);
+      stdout.write(`${filename}: ${count} chars\n`);
+    }
   } catch (err) {
     stderr.write(err.toString());
     stderr.write('\n');

src/count.test.ts

@@ -2,7 +2,7 @@ import countChars from './count';
 
 describe('countChars()', () => {
   it('should return count of content', async () => {
-    expect(await countChars('test/assets/test.txt')).toEqual(15);
+    expect(await countChars('test/assets/test1.txt')).toEqual(15);
   });
   it('should reject when file not found', async () => {
     await expect(countChars('test/assets/fail.txt')).rejects.toThrow('ENOENT:');

src/main.ts

@@ -12,7 +12,7 @@ const argv = yargs(hideBin(process.argv))
 (async () => {
   process.exit(
     await cli({
-      filename: argv._[0] as string,
+      filenames: argv._ as string[],
       stdout: process.stdout,
       stderr: process.stderr
     })

test/assets/test.txt test/assets/test1.txttest/assets/test.txt renamed to test/assets/test1.txt

@@ -0,0 +1,2 @@
+12345678
+abcdefg

test/assets/test2.txt

@@ -11,7 +11,8 @@
     "outDir": "dist"
   },
   "exclude": [
-    "node_modules"
+    "node_modules",
+    "src/**/*.test.ts"
   ],
   "include": [
     "src/**/*.ts",