mirror of https://github.com/calzoneman/sync.git
Continue work on db
This commit is contained in:
calzoneman
parent
44d5f42a36
commit
8822fc5206
182
database.js
182
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;
|
||||
|
|
|
|||
Loading…
Reference in New Issue