Fixing issue while reading multiple directories (#180)

* Fixing issue while reading multiple directories

* Fixing issue while reading multiple directories

* releasing version

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cparra/apexdocs",
-  "version": "3.1.0",
+  "version": "3.1.1",
   "description": "Library with CLI capabilities to generate documentation for Salesforce Apex classes.",
   "keywords": [
     "apex",

src/application/__tests__/apex-file-reader.spec.ts

@@ -125,4 +125,56 @@ describe('File Reader', () => {
     expect(result[0].content).toBe('public class MyClass{}');
     expect(result[1].content).toBe('public class AnotherClass{}');
   });
+
+  it('returns the file contents for all Apex  when there are multiple directories', async () => {
+    const fileSystem = new TestFileSystem([
+      {
+        type: 'directory',
+        path: '',
+        files: [
+          {
+            type: 'file',
+            path: 'SomeFile.cls',
+            content: 'public class MyClass{}',
+          },
+          {
+            type: 'directory',
+            path: 'subdir',
+            files: [
+              {
+                type: 'file',
+                path: 'AnotherFile.cls',
+                content: 'public class AnotherClass{}',
+              },
+            ],
+          },
+          {
+            type: 'directory',
+            path: 'subdir2',
+            files: [
+              {
+                type: 'file',
+                path: 'SomeFile2.cls',
+                content: 'public class MyClass{}',
+              },
+              {
+                type: 'directory',
+                path: 'subdir',
+                files: [
+                  {
+                    type: 'file',
+                    path: 'AnotherFile2.cls',
+                    content: 'public class AnotherClass{}',
+                  },
+                ],
+              },
+            ],
+          },
+        ],
+      },
+    ]);
+
+    const result = await ApexFileReader.processFiles(fileSystem, '', false);
+    expect(result.length).toBe(4);
+  });
 });

src/application/apex-file-reader.ts

@@ -28,9 +28,9 @@ export class ApexFileReader {
       const currentPath = fileSystem.joinPath(rootPath, filePath);
       if (await fileSystem.isDirectory(currentPath)) {
         paths.push(...(await this.getFilePaths(fileSystem, currentPath)));
-        return paths;
+      } else {
+        paths.push(currentPath);
       }
-      paths.push(currentPath);
     }
     return paths;
   }