add comment triggers

Author: aminya

src/index.js

@@ -1,7 +1,7 @@
 'use strict';
 
 // remove for file that only have 'not strict'
-function removeSelective(path, t) {
+function removeSelective(path, t, directiveTriggers, commentTriggers) {
     if (path.node.value.value === 'use strict') {
         const siblings = path.container // get all the sibilings (code in the same level) of 'use strict'
 
@@ -12,6 +12,37 @@ function removeSelective(path, t) {
                     break;
                 }
             }
+
+            // remove based on comment triggers
+            if (commentTriggers.length) {
+                if (sibling.leadingComments) {
+                    let foundInLeadingComment = false;
+                    for (const leadingComment of sibling.leadingComments) {
+                        if (commentTriggers.includes(leadingComment.value.trim())) {
+                            path.remove(); // remove 'use strict'
+                            foundInLeadingComment = true;
+                            break;
+                        }
+                    }
+                    if (foundInLeadingComment) {
+                        break;
+                    }
+                }
+                if (sibling.trailingComments) {
+                    let foundInTrailingComment = false;
+                    for (const trailingComment of sibling.trailingComments) {
+                        if (commentTriggers.includes(trailingComment.value.trim())) {
+                            path.remove(); // remove 'use strict'
+                            foundInTrailingComment = true;
+                            break;
+                        }
+                    }
+                    if (foundInTrailingComment) {
+                        break;
+                    }
+                }
+            }
+
         }
     }
 }
@@ -32,7 +63,7 @@ module.exports = function ({ types: t }) {
                 if (state.opts.removeAll) {
                     removeAll(path) // remove for all files
                 } else {
-                    removeSelective(path, t) // remove for file that only have 'not strict'
+                    removeSelective(path, t, state.opts.directiveTriggers, state.opts.commentTriggers) // remove for file that only have 'not strict'
                 }
             },
         },