1 files changed, 134 insertions, 33 deletions

diff --git a/js/mediators/io-mediator.js b/js/mediators/io-mediator.js
index 65218526..f44c182c 100644
--- a/js/mediators/io-mediator.js
+++ b/js/mediators/io-mediator.js
@@ -41,9 +41,12 @@ exports.IoMediator = Montage.create(Component, {
     },
     ////////////////////////////////////////////////////////////////////
     //
-    appTemplatesUrl: {
+    getAppTemplatesUrlRegEx: {
         enumerable: false,
-        value: new RegExp(chrome.extension.getURL('js/document/templates/montage-html/'), 'gi')
+        value: function () {
+                var regex = new RegExp(chrome.extension.getURL('js/document/templates/montage-html').replace(/\//gi, '\\\/'), 'gi');
+                return regex;
+        }
     },
         ////////////////////////////////////////////////////////////////////
     //
@@ -210,35 +213,51 @@ exports.IoMediator = Montage.create(Component, {
     parseNinjaTemplateToHtml: {
         enumerable: false,
         value: function (template) {
+                var regexRootUrl, rootUrl = this.application.ninja.coreIoApi.rootUrl+escape((this.application.ninja.documentController.documentHackReference.root.split(this.application.ninja.coreIoApi.cloudData.root)[1]));
+                regexRootUrl = new RegExp(rootUrl.replace(/\//gi, '\\\/'), 'gi');
                 //Injecting head and body into old document
-                template.document.content.document.body.innerHTML = template.body;
-                template.document.content.document.head.innerHTML = template.head;
+                template.document.content.document.head.innerHTML = template.head.replace(regexRootUrl, '');
+                template.document.content.document.body.innerHTML = template.body.replace(regexRootUrl, '');
                 //Getting all CSS (style or link) tags
                 var styletags = template.document.content.document.getElementsByTagName('style'),
-                        linktags = template.document.content.document.getElementsByTagName('link');
-                //Looping through link tags and removing file recreated elements
-                for (var j in styletags) {
-                        if (styletags[j].getAttribute) {
-                                if(styletags[j].getAttribute('data-ninja-uri') !== null && !styletags[j].getAttribute('data-ninja-template')) {//TODO: Use querySelectorAll
-                                        try {
-                                                //Checking head first
-                                                template.document.content.document.head.removeChild(styletags[j]);
-                                        } catch (e) {
-                                                try {
-                                                        //Checking body if not in head
-                                                        template.document.content.document.body.removeChild(styletags[j]);
-                                                } catch (e) {
-                                                        //Error, not found!
-                                                }
+                        linktags = template.document.content.document.getElementsByTagName('link'),
+                        toremovetags = [];
+                //Getting styles tags to be removed from document
+                if (styletags.length) {
+                        for (var j=0; j<styletags.length; j++) {
+                                if (styletags[j].getAttribute) {
+                                        if(styletags[j].getAttribute('data-ninja-uri') !== null && !styletags[j].getAttribute('data-ninja-template')) {
+                                                toremovetags.push(styletags[j]);
+                                        } else if (styletags[j].getAttribute('data-ninja-external-url')) {
+                                                toremovetags.push(styletags[j]);
                                         }
-                                        
                                 }
                         }
                 }
-                //TODO: Add logic to only enble tags we disabled
+                //Removing styles tags from document
+                for (var h=0; toremovetags[h]; h++) {
+                        try {
+                                //Checking head first
+                                template.document.content.document.head.removeChild(toremovetags[h]);
+                        } catch (e) {
+                                try {
+                                        //Checking body if not in head
+                                        template.document.content.document.body.removeChild(toremovetags[h]);
+                                } catch (e) {
+                                        //Error, not found!
+                                }
+                        }
+                }
+                //Removing disabled tags from tags that were not originally disabled by user (Ninja enables all)
                 for (var l in linktags) {
                         if (linktags[l].getAttribute && linktags[l].getAttribute('disabled')) {//TODO: Use querySelectorAll
-                                linktags[l].removeAttribute('disabled');
+                                for (var p=0; toremovetags[p]; p++) {
+                                        if (toremovetags[p].getAttribute('href') === linktags[l].getAttribute('href')) {
+                                                if (!toremovetags[p].getAttribute('data-ninja-disabled')) {
+                                                        linktags[l].removeAttribute('disabled');
+                                                }
+                                        }
+                                }
                         }
                 }
                 //Checking for type of save: styles = <style> only | css = <style> and <link> (all CSS)
@@ -252,10 +271,12 @@ exports.IoMediator = Montage.create(Component, {
                                         if (template.styles[i].ownerNode.getAttribute) {
                                                 //Checking for node not to be loaded from file
                                                 if (template.styles[i].ownerNode.getAttribute('data-ninja-uri') === null && !template.styles[i].ownerNode.getAttribute('data-ninja-template')) {
-                                                        //Inseting data from rules array into tag as string
-                                                        docStyles[styleCounter].innerHTML = this.getCssFromRules(template.styles[i].cssRules);
-                                                        //Syncing <style> tags count since it might be mixed with <link>
-                                                        styleCounter++;
+                                                        if(docStyles[styleCounter]) {
+                                                                //Inseting data from rules array into tag as string
+                                                                docStyles[styleCounter].innerHTML = this.getCssFromRules(template.styles[i].cssRules);
+                                                                //Syncing <style> tags count since it might be mixed with <link>
+                                                                styleCounter++;
+                                                        }
                                                 }
                                         }
                                 }
@@ -281,9 +302,11 @@ exports.IoMediator = Montage.create(Component, {
                                         if (template.css[i].ownerNode.getAttribute) {
                                                 if (template.css[i].ownerNode.getAttribute('data-ninja-uri') === null && !template.css[i].ownerNode.getAttribute('data-ninja-template')) {//TODO: Use querySelectorAll
                                                         //Inseting data from rules array into <style> as string
-                                                        docStyles[styleCounter].innerHTML = this.getCssFromRules(template.css[i].cssRules);
-                                                        styleCounter++;
-                                                } else {
+                                                        if (docStyles[styleCounter] && !template.css[i].ownerNode.getAttribute('data-ninja-external-url')) {
+                                                                docStyles[styleCounter].innerHTML = this.getCssFromRules(template.css[i].cssRules);
+                                                                styleCounter++;
+                                                        }
+                                                } else if (!template.css[i].ownerNode.getAttribute('data-ninja-template')){
                                                         //Checking for attributes to be added to tag upon saving
                                                         for (var k in docLinks) {
                                                                 if (docLinks[k].getAttribute) {
@@ -300,8 +323,34 @@ exports.IoMediator = Montage.create(Component, {
                                                                         }
                                                                 }
                                                         }
+                                                                                                                
+                                                        ///////////////////////////////////////////////////////////////////////////////////////////
+                                                        ///////////////////////////////////////////////////////////////////////////////////////////
+                                                        
+                                                        
+                                                        var cleanedCss,
+                                                                dirtyCss = this.getCssFromRules(template.css[i].cssRules),
+                                                                fileUrl = template.css[i].ownerNode.getAttribute('data-ninja-file-url'),
+                                                                fileRootUrl = this.application.ninja.coreIoApi.rootUrl+fileUrl.split(fileUrl.split('/')[fileUrl.split('/').length-1])[0],
+                                                                cleanedCss = dirtyCss.replace(/(\b(?:(?:https?|ftp|file|[A-Za-z]+):\/\/|www\.|ftp\.)(?:\([-A-Z0-9+&@#\/%=~_|$?!:,.]*\)|[-A-Z0-9+&@#\/%=~_|$?!:,.])*(?:\([-A-Z0-9+&@#\/%=~_|$?!:,.]*\)|[A-Z0-9+&@#\/%=~_|$]))/gi, parseNinjaUrl.bind(this));
+                                                                
+                                                        function parseNinjaUrl (url) {
+                                                                if (url.indexOf(this.application.ninja.coreIoApi.rootUrl) !== -1) {
+                                                                        return this.getUrlfromNinjaUrl(url, fileRootUrl, fileUrl);
+                                                                } else {
+                                                                        return url;
+                                                                }
+                                                        }
+                                                        
+                                                        ///////////////////////////////////////////////////////////////////////////////////////////
+                                                        ///////////////////////////////////////////////////////////////////////////////////////////
+                                                        
+                                                        
+                                                        
+                                                        
                                                         //Saving data from rules array converted to string into <link> file
-                                                        var save = this.fio.saveFile({uri: template.css[i].ownerNode.getAttribute('data-ninja-uri'), contents: this.getCssFromRules(template.css[i].cssRules)});
+                                                        var save = this.fio.saveFile({uri: template.css[i].ownerNode.getAttribute('data-ninja-uri'), contents: cleanedCss});
+                                                        //TODO: Add error handling for saving files
                                                 }
                                         }
                                 }
@@ -349,8 +398,60 @@ exports.IoMediator = Montage.create(Component, {
                         //Setting string in tag
                         webgltag.innerHTML = json;
                 }
+                var cleanHTML = template.document.content.document.documentElement.outerHTML.replace(/(\b(?:(?:https?|ftp|file|[A-Za-z]+):\/\/|www\.|ftp\.)(?:\([-A-Z0-9+&@#\/%=~_|$?!:,.]*\)|[-A-Z0-9+&@#\/%=~_|$?!:,.])*(?:\([-A-Z0-9+&@#\/%=~_|$?!:,.]*\)|[A-Z0-9+&@#\/%=~_|$]))/gi, parseNinjaRootUrl.bind(this));
+                //console.log(this.getPrettyHtml(cleanHTML.replace(this.getAppTemplatesUrlRegEx(), '')));
+                function parseNinjaRootUrl (url) {
+                        if (url.indexOf(this.application.ninja.coreIoApi.rootUrl) !== -1) {
+                                return this.getUrlfromNinjaUrl(url, rootUrl, rootUrl.replace(new RegExp((this.application.ninja.coreIoApi.rootUrl).replace(/\//gi, '\\\/'), 'gi'), '')+'file.ext');//Wrong parameters
+                        } else {
+                                return url;
+                        }
+                }                       
+                        //console.log(rootUrl, this.application.ninja.coreIoApi.rootUrl, this.application.ninja.documentController.documentHackReference.root, this.application.ninja.coreIoApi.cloudData.root);
+                //console.log(this.getPrettyHtml(template.document.content.document.documentElement.outerHTML));
+                //return;
+                //
+                return this.getPrettyHtml(cleanHTML.replace(this.getAppTemplatesUrlRegEx(), ''));
+        }
+    },
+    ////////////////////////////////////////////////////////////////////
+    //
+    getUrlfromNinjaUrl: {
+        enumerable: false,
+        value: function (url, fileRootUrl, fileUrl) {
+                //console.log("Params: ", url, fileRootUrl, fileUrl);
+                //console.log("Getting: " + url);
+                //
+                if (url.indexOf(fileRootUrl) !== -1) {
+                        url = url.replace(new RegExp(fileRootUrl.replace(/\//gi, '\\\/'), 'gi'), '');
+                } else {
+                        //TODO: Clean up vars
+                        var assetsDirs = (url.replace(new RegExp((this.application.ninja.coreIoApi.rootUrl).replace(/\//gi, '\\\/'), 'gi'), '')).split('/');
+                        var fileD