Added string replacement to image and link selector

Author: jwillmer

extension/devtools/views/SelectorEdit.html

@@ -192,13 +192,24 @@
         </div>
     </div>    
 
-        <div class="form-group feature feature-delay">
-            <label for="delay" class="col-sm-2 control-label">Delay (ms)</label>
+    <div class="form-group feature feature-delay">
+        <label for="delay" class="col-sm-2 control-label">Delay (ms)</label>
 
-            <div class="col-sm-8">
-                <input type="text" class="form-control" name="delay" id="delay" placeholder="delay" value="{{selector.delay}}">
+        <div class="col-sm-8">
+            <input type="text" class="form-control" name="delay" id="delay" placeholder="delay" value="{{selector.delay}}">
+        </div>
+    </div>
+
+    <div class="form-group feature feature-stringReplacement">
+        <label for="replace" class="col-sm-2 control-label">Replace</label>
+
+        <div class="col-sm-8">
+            <div class="input-group">
+                <label>Replace text:<input type="text" class="form-control col-sm-6" name="replaceString" id="replaceString" placeholder="Replace text" value="{{selector.replaceString}}"></label>
+                <label>Replacement text:<input type="text" class="form-control col-sm-6" name="replacementString" id="replacementString" placeholder="Replacement text" value="{{selector.replacementString}}"></label>
             </div>
         </div>
+    </div>
 
         <div class="form-group">
             <label for="parentSelectors" class="col-sm-2 control-label">Parent Selectors</label>

extension/scripts/Controller.js

@@ -906,6 +906,10 @@ SitemapController.prototype = {
 		var $columnHeaders = $("#edit-selector .column-header");
 		var $columnNames = $("#edit-selector .column-name");
         var $columnExtracts = $("#edit-selector .column-extract");
+        var stringReplacement = {
+            replaceString: $("#edit-selector [name=replaceString]").val(),
+            replacementString: $("#edit-selector [name=replacementString]").val()
+        };
         var textmanipulation = {
             removeHtml: $("#edit-selector [name=removeHtml]").is(":checked"),
             trimText: $("#edit-selector [name=trimText]").is(":checked"),
@@ -947,7 +951,8 @@ SitemapController.prototype = {
 			parentSelectors: parentSelectors,
 			columns:columns,
 			delay:delay,
-            textmanipulation: textmanipulation
+            textmanipulation: textmanipulation,
+            stringReplacement: stringReplacement
 		});
 		return newSelector;
 	},

extension/scripts/Selector/SelectorImage.js

@@ -38,6 +38,17 @@ var SelectorImage = {
                     src = src ? src[2] : "";
                 }  
 
+                if (this.stringReplacement && this.stringReplacement.replaceString) {
+                    var replace;
+                    var replacement = this.stringReplacement.stringReplacement || "";
+                    try {
+                        var regex = new RegExp(this.stringReplacement.replaceString, 'gm');
+                        replace = regex.test(src) ? regex : this.stringReplacement.replaceString;
+                    } catch (e) { replace = this.stringReplacement.replaceString; }
+
+                    src = src.replace(replace, replacement);
+                }
+
                 data[this.id + '-src'] = src;
 
 				// download image if required
@@ -122,7 +133,7 @@ var SelectorImage = {
 	},
 
 	getFeatures: function () {
-		return ['multiple', 'delay', 'downloadImage']
+        return ['multiple', 'delay', 'downloadImage', 'stringReplacement']
 	},
 
 	getItemCSSSelector: function() {

extension/scripts/Selector/SelectorLink.js

@@ -36,13 +36,25 @@ var SelectorLink = {
 
 			deferredDataExtractionCalls.push(function(element) {
 
+                var href = element.href;
+                if (this.stringReplacement && this.stringReplacement.replaceString) {
+                    var replace;
+                    var replacement = this.stringReplacement.stringReplacement || "";
+                    try {
+                        var regex = new RegExp(this.stringReplacement.replaceString, 'gm');
+                        replace = regex.test(href) ? regex : this.stringReplacement.replaceString;
+                    } catch (e) { replace = this.stringReplacement.replaceString; }
+
+                    href = href.replace(replace, replacement);
+                }
+
 				var deferredData = $.Deferred();
+                var data = {};
 
-				var data = {};
 				data[this.id] = $(element).text();
 				data._followSelectorId = this.id;
-				data[this.id + '-href'] = element.href;
-				data._follow = element.href;
+				data[this.id + '-href'] = href;
+				data._follow = href;
 				deferredData.resolve(data);
 
 				return deferredData;
@@ -65,7 +77,7 @@ var SelectorLink = {
 	},
 
 	getFeatures: function () {
-		return ['multiple', 'delay']
+        return ['multiple', 'delay', 'stringReplacement']
 	},
 
 	getItemCSSSelector: function() {