Continue work on db

database.js

@@ -24,6 +24,8 @@ var Database = function (cfg) {
         }
         conn.end();
     });
+
+    this.global_ipbans = {};
 };
 
 Database.prototype.query = function (query, sub, callback) {
@@ -41,6 +43,9 @@ Database.prototype.query = function (query, sub, callback) {
         } else {
             function cback(err, res) {
                 if(err) {
+                    if(self.cfg["debug"]) {
+                        console.log(err);
+                    }
                     callback("Database failure", null);
                 } else {
                     callback(null, res);
@@ -97,6 +102,183 @@ Database.prototype.init = function () {
             console.log("Created registrations table");
         }
     });
+
+    // Create global bans table
+    query = ["CREATE TABLE IF NOT EXISTS `global_bans` (",
+                "`ip` VARCHAR(15) NOT NULL,",
+                "`note` VARCHAR(255) NOT NULL,",
+                "PRIMARY KEY (`ip`))",
+             "ENGINE = MyISAM;"].join("");
+
+    self.query(query, function (err, res) {
+        if(err) {
+            Logger.errlog.log("! Failed to create global ban table");
+        } else if(self.cfg["debug"]) {
+            console.log("Created global ban table");
+        }
+    });
+
+    // Create password reset table
+    query = ["CREATE TABLE IF NOT EXISTS `password_reset` (",
+                "`ip` VARCHAR(15) NOT NULL,",
+                "`name` VARCHAR(20) NOT NULL,",
+                "`hash` VARCHAR(64) NOT NULL,",
+                "`email` VARCHAR(255) NOT NULL,",
+                "`expire` BIGINT NOT NULL,",
+                "PRIMARY KEY (`name`))",
+             "ENGINE = MyISAM;"].join("");
+
+    self.query(query, function (err, res) {
+        if(err) {
+            Logger.errlog.log("! Failed to create password reset table");
+        } else if(self.cfg["debug"]) {
+            console.log("Created password reset table");
+        }
+    });
+
+    // Create user playlist table
+    query = ["CREATE TABLE IF NOT EXISTS `user_playlists` (",
+                "`user` VARCHAR(20) NOT NULL,",
+                "`name` VARCHAR(255) NOT NULL,",
+                "`contents` MEDIUMTEXT NOT NULL,",
+                "`count` INT NOT NULL,",
+                "`time` INT NOT NULL,",
+                "PRIMARY KEY (`user`, `name`))",
+             "ENGINE = MyISAM;"].join("");
+
+    self.query(query, function (err, res) {
+        if(err) {
+            Logger.errlog.log("! Failed to create user playlist table");
+        } else if(self.cfg["debug"]) {
+            console.log("Created user playlist table");
+        }
+    });
+
+    // Create user aliases table
+    query = ["CREATE TABLE IF NOT EXISTS `aliases` (",
+                "`visit_id` INT NOT NULL AUTO_INCREMENT,",
+                "`ip` VARCHAR(15) NOT NULL,",
+                "`name` VARCHAR(20) NOT NULL,",
+                "`time` BIGINT NOT NULL,",
+                "PRIMARY KEY (`visit_id`), INDEX (`ip`))",
+             "ENGINE = MyISAM;"].join("");
+
+    self.query(query, function (err, res) {
+        if(err) {
+            Logger.errlog.log("! Failed to create user aliases table");
+        } else if(self.cfg["debug"]) {
+            console.log("Created user aliases table");
+        }
+    });
+
+    // Create action log table
+    query = ["CREATE TABLE IF NOT EXISTS `actionlog` (",
+                "`ip` VARCHAR(15) NOT NULL,",
+                "`name` VARCHAR(20) NOT NULL,",
+                "`action` VARCHAR(255) NOT NULL,",
+                "`args` TEXT NOT NULL,",
+                "`time` BIGINT NOT NULL,",
+                "PRIMARY KEY (`ip`, `time`), INDEX (`action`))",
+             "ENGINE = MyISAM;"].join("");
+
+    self.query(query, function (err, res) {
+        if(err) {
+            Logger.errlog.log("! Failed to create action log table");
+        } else if(self.cfg["debug"]) {
+            console.log("Created action log table");
+        }
+    });
+
+    // Create stats table
+    query = ["CREATE TABLE IF NOT EXISTS `stats` (",
+                "`time` BIGINT NOT NULL,",
+                "`usercount` INT NOT NULL,",
+                "`chancount` INT NOT NULL,",
+                "`mem` INT NOT NULL,",
+                "PRIMARY KEY (`time`))",
+             "ENGINE = MyISAM;"].join("");
+
+    self.query(query, function (err, res) {
+        if(err) {
+            Logger.errlog.log("! Failed to create stats table");
+        } else if(self.cfg["debug"]) {
+            console.log("Created stats table");
+        }
+    });
+
+    // Refresh global IP bans
+    self.getGlobalIPBans();
 };
 
+/* REGION global bans */
+
+Database.prototype.isGlobalIPBanned = function (ip, callback) {
+    if(typeof callback !== "function")
+        return;
+    const re = /(\d+)\.(\d+)\.(\d+)\.(\d+)/;
+    // Account for range banning
+    var s16 = ip.replace(re, "$1.$2");
+    var s24 = ip.replace(re, "$1.$2.$3");
+
+    var banned = ip in this.global_ipbans ||
+                 s16 in this.global_ipbans ||
+                 s24 in this.global_ipbans;
+    
+    callback(null, banned);
+};
+
+Database.prototype.getGlobalIPBans = function (callback) {
+    if(typeof callback !== "function")
+        callback = function () { }
+    var self = this;
+    self.query("SELECT * FROM global_bans WHERE 1", function (err, res) {
+        if(err) {
+            callback(err, null);
+            return;
+        }
+
+        self.global_ipbans = {};
+        for(var i in res) {
+            self.global_ipbans[res[i].ip] = res[i].note;
+        }
+
+        callback(null, self.global_ipbans);
+    });
+};
+
+Database.prototype.setGlobalIPBan = function (ip, reason, callback) {
+    if(typeof callback !== "function")
+        callback = function () { }
+    var query = "INSERT INTO global_bans VALUES (?, ?)" + 
+                " ON DUPLICATE KEY UPDATE note=?";
+    var self = this;
+    self.query(query, [ip, reason, reason], function (err, res) {
+        if(err) {
+            callback(err, null);
+            return;
+        }
+
+        self.getGlobalIPBans();
+        callback(null, res);
+    });
+};
+
+Database.prototype.clearGlobalIPBan = function (ip, callback) {
+    if(typeof callback !== "function")
+        callback = function () { }
+    var self = this;
+
+    var query = "DELETE FROM global_bans WHERE ip=?";
+    self.query(query, [ip], function (err, res) {
+        if(err) {
+            callback(err, null);
+            return;
+        }
+
+        callback(null, res);
+    });
+};
+
+/* END REGION */
+
 module.exports = Database;