BetterDiscordApp-v2/lib/Utils.js

281 lines
9.2 KiB
JavaScript

/* 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;