Fix leading dots being removed (#59)

* fix leading dots being removed

* only modify auto-generated filenames + added test

src/index.js

@@ -649,12 +649,12 @@ export class DownloaderHelper extends EventEmitter {
                 fileName = `${URL.parse(this.requestURL).hostname}.html`;
             }
         }
-
+        
         return (
             (this.__opts.fileName)
                 ? this.__getFileNameFromOpts(fileName, response)
-                : fileName
-        ).split('.').filter(Boolean).join('.'); // remove any potential trailing '.' (just to be sure)
+                : fileName.replace(/\.*$/, '') // remove any potential trailing '.' (just to be sure)
+        )
     }
 
     /**

test/index.spec.js

@@ -235,5 +235,29 @@ describe('DownloaderHelper', function () {
             });
             expect(result).to.be.equal(newFileName);
         });
+
+        it("should keep leading dots but remove trailing dots for auto-generated file names", function () {
+            const newFileName = '.gitignore.';
+            const expectedFileName = '.gitignore';
+            const dl = new DownloaderHelper('https://google.com/', __dirname, {
+                // fileName: { name: newFileName, ext: true }
+            });
+            const result = dl.__getFileNameFromHeaders({
+                'content-disposition': 'Content-Disposition: attachment; filename="' + newFileName + '"',
+            });
+            expect(result).to.be.equal(expectedFileName);
+        });
+
+        it("should not modify the filename when providing a callback", function () {
+            const newFileName = '.gitignore.';
+            const expectedFileName = newFileName
+            const dl = new DownloaderHelper('https://google.com/', __dirname, {
+                fileName: () => '.gitignore.'
+            });
+            const result = dl.__getFileNameFromHeaders({
+                'content-disposition': 'Content-Disposition: attachment; filename="' + newFileName + '"',
+            });
+            expect(result).to.be.equal(expectedFileName);
+        });
     });
 });