1 files changed, 113 insertions, 8 deletions

diff --git a/js/mediators/io-mediator.js b/js/mediators/io-mediator.js
index 56869839..e5eb1fc6 100644
--- a/js/mediators/io-mediator.js
+++ b/js/mediators/io-mediator.js
@@ -195,18 +195,121 @@ exports.IoMediator = Montage.create(Component, {
         }
     },
     ////////////////////////////////////////////////////////////////////
-    //TODO: Expand to allow more templates
+    //TODO: Expand to allow more templates, clean up variables
     parseNinjaTemplateToHtml: {
         enumerable: false,
         value: function (template) {
-                //
+                //Injecting head and body into old document
                 template.document.content.document.body.innerHTML = template.body;
                 template.document.content.document.head.innerHTML = template.head;
-                //TODO: Remove temp fix for styles
-                if (template.style) {
-                        template.document.content.document.head.getElementsByTagName('style')[0].innerHTML = this.getCssFromRules(template.style.cssRules);
+                //Getting all CSS (style or link) tags
+                var styletags = template.document.content.document.getElementsByTagName('style'),
+                        linktags = template.document.content.document.getElementsByTagName('link'),
+                        url = new RegExp(window.location.protocol+'//'+window.location.host+'/js/document/templates/montage-html/', 'gi');
+                //Looping through link tags and removing file recreated elements
+                for (var j in styletags) {
+                        if (styletags[j].getAttribute) {
+                                if(styletags[j].getAttribute('ninjauri') !== null) {//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!
+                                                }
+                                        }
+                                        
+                                }
+                        }
+                }
+                //TODO: Add logic to only enble tags we disabled
+                for (var l in linktags) {
+                        if (linktags[l].getAttribute && linktags[l].getAttribute('disabled')) {//TODO: Use querySelectorAll
+                                linktags[l].removeAttribute('disabled');
+                        }
+                }
+                //Checking for type of save: styles = <style> only | css = <style> and <link> (all CSS)
+                if (template.styles) {
+                        //Getting all style tags
+                        var styleCounter = 0,
+                                docStyles = template.document.content.document.getElementsByTagName('style');
+                        //Looping through all style tags
+                        for(var i in template.styles) {
+                                if (template.styles[i].ownerNode) {
+                                        if (template.styles[i].ownerNode.getAttribute) {
+                                                //Checking for node not to be loaded from file
+                                                if (template.styles[i].ownerNode.getAttribute('ninjauri') === null) {
+                                                        //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++;
+                                                }
+                                        }
+                                }
+                        }
+                } else if (template.css) {
+                        //Getting all style and link tags
+                        var styleCounter = 0,
+                                docStyles = template.document.content.document.getElementsByTagName('style'),
+                                docLinks = template.document.content.document.getElementsByTagName('link');
+                        for(var i in template.css) {
+                                if (template.css[i].ownerNode) {
+                                        if (template.css[i].ownerNode.getAttribute) {
+                                                if (template.css[i].ownerNode.getAttribute('ninjauri') === null) {//TODO: Use querySelectorAll
+                                                        //Inseting data from rules array into <style> as string
+                                                        docStyles[styleCounter].innerHTML = this.getCssFromRules(template.css[i].cssRules);
+                                                        styleCounter++;
+                                                } else {
+                                                        //Saving data from rules array converted to string into <link> file
+                                                        var save = this.fio.saveFile({uri: template.css[i].ownerNode.getAttribute('ninjauri'), contents: this.getCssFromRules(template.css[i].cssRules)});
+                                                }
+                                        }
+                                }
+                        }
                 }
-                return template.document.content.document.documentElement.outerHTML;
+                //Checking for webGL elements in document
+                if (template.webgl.length) {
+                        //
+                        var json, matchingtags = [], webgltag, scripts = template.document.content.document.getElementsByTagName('script');
+                        //
+                        for (var i in scripts) {
+                                if (scripts[i].getAttribute) {
+                                        if (scripts[i].getAttribute('data-ninja-webgl') !== null) {//TODO: Use querySelectorAll
+                                                matchingtags.push(scripts[i]);
+                                        }
+                                }
+                        }
+                        //
+                        if (matchingtags.length) {
+                                if (matchingtags.length === 1) {
+                                        webgltag = matchingtags[0];
+                                } else {
+                                        //TODO: Add logic to handle multiple tags, perhaps combine to one
+                                        webgltag = matchingtags[matchingtags.length-1]; //Saving all data to last one...
+                                }
+                        }
+                        //
+                        if (!webgltag) {
+                                webgltag = template.document.content.document.createElement('script');
+                                webgltag.setAttribute('data-ninja-webgl', 'true');
+                                template.document.content.document.head.appendChild(webgltag);
+                        }
+                        //
+                        json = '\n\t\t({\n\t\t\t"version": "X.X.X.X",\n\t\t\t"data": [';
+                        //
+                        for (var j=0; template.webgl[j]; j++) {
+                                json += '\n\t\t\t\t"'+escape(template.webgl[j])+'"';
+                        }
+                        //
+                        json += ']\n\t\t})\n\t';
+                        //
+                        webgltag.innerHTML = json;
+                }
+                //
+                return template.document.content.document.documentElement.outerHTML.replace(url, '');
         }
     },
     ////////////////////////////////////////////////////////////////////
@@ -215,7 +318,7 @@ exports.IoMediator = Montage.create(Component, {
         enumerable: false,
         value: function (list) {
                 //Variable to store CSS definitions
-                var i, str, css = '';
+                var i, str, url, css = '';
                 //Looping through list
                 if (list && list.length > 0) {
                         //Adding each list item to string and also adding breaks
@@ -227,8 +330,10 @@ exports.IoMediator = Montage.create(Component, {
                                 css += '\n'+str;
                         }
                 }
+                //TODO: Add better logic for creating this string
+                url = new RegExp(window.location.protocol+'//'+window.location.host+'/js/document/templates/montage-html/', 'gi');
                 //Returning the CSS string
-                return css;
+                return css.replace(url, '');
         }
     }
     ////////////////////////////////////////////////////////////////////