BetterDiscordApp-v2/lib/Utils.js

/* BetterDiscordApp Utils and Helper functions
 * Version: 1.5
 * Author: Jiiks | http://jiiks.net
 * Date: 25/08/2015 - 09:19
 * Last Updated: 06/05/2016
 * https://github.com/Jiiks/BetterDiscordApp
 */

var https = require('https');
var http = require('http');
var _fs = require('fs');
var _mainWindow;
var eol = require('os').EOL;
var logs = "";

function Utils(mainWindow) {
    _mainWindow = mainWindow;
}

//Get browser mainwindow
function getMainWindow() {
    return _mainWindow;
}

//Download using https
Utils.prototype.download = function(host, path, callback) {
    this.log("Downloading Resource: " + host + path);
    var options = {
        host: host,
        path: path,
        headers: {'user-agent': 'Mozilla/5.0'},
    }

    https.get(options, function(res) {
        var data = "";
        res.on('data', function(chunk) {
            data += chunk;
        });
        res.on("end", function() {
            callback(data);
        });
    }).on("error", function() {
        callback(null);
    });
}
//Download using http
Utils.prototype.downloadHttp = function(url, callback) {
    this.log("Downloading Resource: " + url);
    http.get(url, function(result) {
        var data = '';
        result.on('data', function(chunk) {
            data += chunk;
        });

        result.on('end', function() {
            callback(data);
        });
    }).on('error', function() {
        callback(null);
    });
}

Utils.prototype.getHash = function(beta, callback) {
    var branch = beta ? "beta" : "master";
    this.download("api.github.com", "/repos/Jiiks/BetterDiscordApp/commits/"+branch+"", function(data) {
        callback(JSON.parse(data).sha);
    });
}

Utils.prototype.sendIcpAsync = function(message) {
    this.execJs('betterDiscordIPC.send("asynchronous-message", "'+message+'");');
}

//Get Webcontents
Utils.prototype.getWebContents = function() {
    return getMainWindow().webContents;
}

//Js logger
Utils.prototype.jsLog = function(message, type) {

    switch(type) {
        case "log":
            this.execJs('console.log("BetterDiscord: ' + message + '");');
            break;
        case "warn":
            this.execJs('console.warn("BetterDiscord: ' + message + '");');
            break;
        case "error":
            this.execJs('console.error("BetterDiscord: ' + message + '");');
            break;
    }
}

Utils.prototype.updateLoading = function(message, cur, max) {
    this.log(message);
    this.execJs('document.getElementById("bd-status").innerHTML = "BetterDiscord - '+message+' : ";');
    this.execJs('document.getElementById("bd-pbar").value = '+cur+';');
    this.execJs('document.getElementById("bd-pbar").max = '+max+';');
}

//Logger
Utils.prototype.log = function(message) {
    console.log("[BetterDiscord INF] " + message);
    var d = new Date();
    var ds = ("00" + (d.getDate() + 1)).slice(-2) + "/" + 
    ("00" + d.getMonth()).slice(-2) + "/" + 
    d.getFullYear() + " " + 
    ("00" + d.getHours()).slice(-2) + ":" + 
    ("00" + d.getMinutes()).slice(-2) + ":" + 
    ("00" + d.getSeconds()).slice(-2);
    logs += "[INF]["+ds+"] " + message + eol;
}
Utils.prototype.err = function(err) {
    console.log("[BetterDiscord ERR] " + err.message);
    var d = new Date();
    var ds = ("00" + (d.getDate() + 1)).slice(-2) + "/" + 
    ("00" + d.getMonth()).slice(-2) + "/" + 
    d.getFullYear() + " " + 
    ("00" + d.getHours()).slice(-2) + ":" + 
    ("00" + d.getMinutes()).slice(-2) + ":" + 
    ("00" + d.getSeconds()).slice(-2);
    logs += "[ERR]["+ds+"] " + err.message + eol;
}
Utils.prototype.warn = function(message) {
    console.log("[BetterDiscord WRN] " + message);
    var d = new Date();
    var ds = ("00" + (d.getDate() + 1)).slice(-2) + "/" + 
    ("00" + d.getMonth()).slice(-2) + "/" + 
    d.getFullYear() + " " + 
    ("00" + d.getHours()).slice(-2) + ":" + 
    ("00" + d.getMinutes()).slice(-2) + ":" + 
    ("00" + d.getSeconds()).slice(-2);
    logs += "[WRN]["+ds+"] " + message + eol;
}
Utils.prototype.saveLogs = function(path) {
    try {
        _fs.writeFileSync(path + "/logs.log", logs);
    }catch(err) {}
}

//Execute javascript
Utils.prototype.execJs = function(js) {
    this.getWebContents().executeJavaScript(js);
}

//Parse and execute javascript
Utils.prototype.execJsParse = function(js) {
    this.execJs(js); //TODO
}

//Inject variable
Utils.prototype.injectVar = function(variable, data) {
    this.execJs('var ' + variable + ' = "' + data + '";');
}
Utils.prototype.injectVarRaw = function(variable, data) {
    this.execJs('var ' + variable + ' = ' + data + ';');
}

//Alert
Utils.prototype.alert = function(title, message) {
    var id = 'bdalert-';
    for( var i=0; i < 5; i++ )
        id += "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789".charAt(Math.floor(Math.random() * "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789".length)); 
    var bdAlert = '\
    <div id=\''+id+'\' class=\'modal\' style=\'opacity:1\'>\
        <div class=\'modal-inner\'>\
            <div class=\'markdown-modal\'>\
                <div class=\'markdown-modal-header\'>\
                    <strong style=\'float:left\'><span>BetterDiscord - </span><span>'+title+'</span></strong>\
                    <span></span>\
                    <button class=\'markdown-modal-close\' onclick=document.getElementById(\''+id+'\').remove();></button>\
                </div>\
                <div class=\'scroller-wrap fade\'>\
                    <div style=\'font-weight:700\' class=\'scroller\'>'+message+'</div>\
                </div>\
                <div class=\'markdown-modal-footer\'>\
                    <span style=\'float:right\'> for support.</span>\
                    <a style=\'float:right\' href=\'https://discord.gg/0Tmfo5ZbOR9NxvDd\' target=\'_blank\'>#support</a>\
                    <span style=\'float:right\'>Join </span>\
                </div>\
            </div>\
        </div>\
    </div>\
    ';

    this.execJs("document.body.insertAdjacentHTML('afterbegin', \""+bdAlert+"\");");
}

//Css internal style injector
Utils.prototype.injectStylesheet = function(url) {
    var self = this;
    this.download(url, function(data) {
        var js = 'var style = document.createElement("style"); style.type = "text/css"; style.innerHTML = "'+data+'";';
        self.injectToElementByTag("head", js, "style");
    });
}

Utils.prototype.injectStylesheetSync = function(url, callbackMessage) {
    this.execJs('$("head").append(" <link rel=\'stylesheet\' href=\''+url+'\'> ");');
    this.sendIcpAsync(callbackMessage);
};

Utils.prototype.headStyleSheet = function(url) {
    this.execJs('(function() { var stylesheet = document.createElement("link"); stylesheet.type = "text/css"; document.getElementsByTagName("head")[0].appendChild(stylesheet); stylesheet.href = "'+url+'" })();')
}

Utils.prototype.injectJavaScriptSync = function(url, callbackMessage) {
    this.execJs(' (function() { var script = document.createElement("script"); script.type = "text/javascript"; script.onload = function() { betterDiscordIPC.send("asynchronous-message", "'+callbackMessage+'"); }; document.getElementsByTagName("body")[0].appendChild(script); script.src = "'+url+'"; })(); ');
}

Utils.prototype.injectJavaScript = function(url, jquery) {
    if(!jquery) {
        this.execJs('(function() { var script = document.createElement("script"); script.type = "text/javascript"; document.getElementsByTagName("body")[0].appendChild(script); script.src = "' + url + '"; })();');
    }else {
        this.execJs(' (function() { function injectJs() { var script = document.createElement("script"); script.type = "text/javascript"; document.getElementsByTagName("body")[0].appendChild(script); script.src = "' + url + '"; } function jqDefer() { if(window.jQuery) { injectJs(); }else{ setTimeout(function() { jqDefer(); }, 100) } } jqDefer(); })(); ');
    }
}

Utils.prototype.mkdirSync = function(path) {
    if(!_fs.existsSync(path)) {
        this.log("Directory " + path + " does not exist. Creating");
        _fs.mkdirSync(path);
    }
};

Utils.prototype.attemptSync = function(func, attempts, attempt, message, success, err) {
    var self = this;
    attempt = attempt || 0;
    attempt++;

    if(attempt > attempts) {
        err();
        return;
    }

    setTimeout(function() {
        if(!func()) {
            self.warn(message + ", retrying #" + attempt);
            self.try(func, attempts, attempt, message, success, err);
            return;
        }
    
        success();
    }, 1000);
};

Utils.prototype.attempt = function(func, attempts, attempt, message, success, err) {
    var self = this;
    attempt = attempt || 0;
    attempt++;

    if(attempt > attempts) {
        err();
        return;
    }

    setTimeout(function() {
        func(function(ok) {
            if(!ok) {
                self.warn(message + ", retrying #" + attempt);
                self.try(func, attempts, attempt, message, success, err);
                return;
            }
            success();
        });
    }, 1000);
};

Utils.prototype.openDir = function(path) {
    switch(process.platform) {
        case "win32":
            require("child_process").exec('start "" "' + path + '"');
        break;
        case "darwin":
            require("child_process").exec('open ' + path);
        break;
    }
};

exports.Utils = Utils;