feat: add the ability to ignore by extension

ahmadalfy · ahmadalfy · commit a390853a9a9a · 2020-02-25T15:06:47.000+02:00
diff --git a/lib/index.js b/lib/index.js
@@ -9,6 +9,10 @@ module.exports = function (options) {
     options.classIgnore = []
   }
 
+  if (!options.extensionIgnore) {
+    options.extensionIgnore = []
+  }
+
   if (options.replaceExtension === undefined) {
     options.replaceExtension = false
   }
@@ -17,7 +21,10 @@ module.exports = function (options) {
     tree.match([{ tag: 'img' }, { tag: 'amp-img' }], function (imgNode) {
       if (imgNode.skip) return imgNode
       var classes = imgNode.attrs && imgNode.attrs.class && (imgNode.attrs.class.split(' ') || [])
-      var isIgnore = options.classIgnore.filter(className => classes.includes(className)).length > 0
+      var extension = imgNode.attrs.src.split('.').pop()
+      var isIgnore =
+        (options.classIgnore.filter(className => classes.includes(className)).length > 0) ||
+        (options.extensionIgnore.filter(fileExtension => fileExtension === extension).length > 0)
       if (isIgnore) return imgNode
       switch (imgNode.tag) {
         case 'amp-img':
diff --git a/test/fixtures/ignore-extension.expected.html b/test/fixtures/ignore-extension.expected.html
@@ -0,0 +1,9 @@
+<!doctype html>
+<html>
+    <body>
+        <img src="photo.svg">
+        <img src="photo.gif">
+        <picture><source type="image/webp" srcset="photo.jpg.webp"><img src="photo.jpg"></picture>
+        <picture><source type="image/webp" srcset="photo.png.webp"><img src="photo.png"></picture>
+    </body>
+</html>
diff --git a/test/fixtures/ignore-extension.html b/test/fixtures/ignore-extension.html
@@ -0,0 +1,9 @@
+<!doctype html>
+<html>
+    <body>
+        <img src="photo.svg">
+        <img src="photo.gif">
+        <img src="photo.jpg">
+        <img src="photo.png">
+    </body>
+</html>
diff --git a/test/test.js b/test/test.js
@@ -39,6 +39,12 @@ test('Class ignore', (t) => {
   })
 })
 
+test('Extension ignore', (t) => {
+  return compare(t, 'ignore-extension', {
+    extensionIgnore: ['gif', 'svg']
+  })
+})
+
 function compare (t, name, options) {
   const html = readFileSync(path.join(fixtures, `${name}.html`), 'utf8')
   const expected = readFileSync(path.join(fixtures, `${name}.expected.html`), 'utf8')
