From 098a30bbb02bce2d23982c1c94a2454dd530c42d Mon Sep 17 00:00:00 2001
From: calzoneman <calzoneman@gmail.com>
Date: Sat, 25 May 2013 17:12:08 -0400
Subject: [PATCH] more database work

---
 database-new.js       | 97 +++++++++++++++++++++++++++++++++++--------
 tests/database-new.js | 31 ++++++++++++++
 2 files changed, 111 insertions(+), 17 deletions(-)
 create mode 100644 tests/database-new.js

diff --git a/database-new.js b/database-new.js
index 0b10b0e9..153e9cc2 100644
--- a/database-new.js
+++ b/database-new.js
@@ -1,24 +1,35 @@
 var mysql = require("mysql-libmysqlclient");
-var Config = require("./config");
 var Logger = require("./logger");
 var db = false;
+var SERVER = "";
+var USER = "";
+var DATABASE = "";
+var PASSWORD = "";
+var CONFIG = {};
+
+function setup(cfg) {
+    SERVER = cfg.MYSQL_SERVER;
+    USER = cfg.MYSQL_USER;
+    DATABASE = cfg.MYSQL_DB;
+    PASSWORD = cfg.MYSQL_PASSWORD;
+    CONFIG = cfg;
+}
 
 function getConnection() {
     if(db && db.connectedSync()) {
         return db;
     }
     db = mysql.createConnectionSync();
-    db.connectSync(Config.MYSQL_SERVER, Config.MYSQL_USER,
-                   Config.MYSQL_PASSWORD, Config.MYSQL_DB);
+    db.connectSync(SERVER, USER, PASSWORD, DATABASE);
     if(!db.connectedSync()) {
         //Logger.errlog.log("DB connection failed");
         return false;
     }
-    if(Config.DEBUG) {
+    if(CONFIG.DEBUG) {
         db._querySync = db.querySync;
         db.querySync = function(q) {
             Logger.syslog.log("DEBUG: " + q);
-            this._querySync(q);
+            return this._querySync(q);
         }
     }
     return db;
@@ -114,11 +125,12 @@ function refreshGlobalBans() {
     if(!results) {
         Logger.errlog.log("! Failed to load global bans");
     }
-
-    var rows = results.fetchAllSync();
-    global_bans = {};
-    for(var i = 0; i < rows.length; i++) {
-        global_bans[rows[i].ip] = rows[i].note;
+    else {
+        var rows = results.fetchAllSync();
+        global_bans = {};
+        for(var i = 0; i < rows.length; i++) {
+            global_bans[rows[i].ip] = rows[i].note;
+        }
     }
 }
 
@@ -153,8 +165,7 @@ function globalUnbanIP(ip) {
 
 function registerChannel(name) {
     var db = getConnection();
-    if(!db) {
-        return false;
+    if(!db) { return false;
     }
 
     // Library table
@@ -367,11 +378,63 @@ function getChannelRank(chan, name) {
     return rows[0].rank;
 }
 
-function test() {
+function setChannelRank(chan, name, rank) {
     var db = getConnection();
-    var q = createQuery("INSERT INTO `?` VALUES (?, ?, ?, ?, ?, ?, ?, ?)", ["registrations", null, "bob2", "asdf", 1, "",  0, "", ""]);
-    console.log(q);
-    //console.log(db.querySync(q));
+    if(!db) {
+        return false;
+    }
+
+    var query = createQuery(
+        "INSERT INTO `?` VALUES (?, ?) ON DUPLICATE KEY UPDATE",
+        ["chan_"+chan+"_ranks", rank, name]
+    );
+
+    return db.querySync(query);
 }
 
-test();
+function addToLibrary(chan, media) {
+    var db = getConnection();
+    if(!db) {
+        return false;
+    }
+
+    var query = createQuery(
+        ["INSERT INTO `?` ",
+            "(`id`, `title`, `seconds`, `type`) ",
+         "VALUES ",
+            "(?, ?, ?, ?)"].join(""),
+        ["chan_"+chan+"_library", media.id, media.title, media.seconds, media.type]
+    );
+
+    return db.querySync(query);
+}
+
+function removeFromLibrary(chan, id) {
+    var db = getConnection();
+    if(!db) {
+        return false;
+    }
+
+    var query = createQuery(
+        "DELETE FROM `?` WHERE id=?",
+        ["chan_"+chan+"_library", id]
+    );
+
+    return db.querySync(query);
+}
+
+exports.setup = setup;
+exports.getConnection = getConnection;
+exports.createQuery = createQuery;
+exports.init = init;
+exports.checkGlobalBan = checkGlobalBan;
+exports.refreshGlobalBans = refreshGlobalBans;
+exports.globalBanIP = globalBanIP;
+exports.globalUnbanIP = globalUnbanIP;
+exports.registerChannel = registerChannel;
+exports.loadChannel = loadChannel;
+exports.deleteChannel = deleteChannel;
+exports.getChannelRank = getChannelRank;
+exports.setChannelRank = setChannelRank;
+exports.addToLibrary = addToLibrary;
+exports.removeFromLibrary = removeFromLibrary;
diff --git a/tests/database-new.js b/tests/database-new.js
new file mode 100644
index 00000000..631ca296
--- /dev/null
+++ b/tests/database-new.js
@@ -0,0 +1,31 @@
+var Database = require("../database-new");
+Database.setup(require("../config-testing"));
+var assert = require("assert");
+var db = Database.getConnection();
+// Empty database
+db.realQuerySync("SHOW TABLES;");
+result = db.storeResultSync();
+var tables = [];
+var tmp = result.fetchAllSync({"asArray": true});
+tmp.forEach(function(t) { tables.push(t[0]); });
+db.querySync("DROP TABLE " + tables.join(", "));
+
+Database.init();
+
+// Test global bans
+assert(!Database.checkGlobalBan("127.0.0.1"));
+Database.globalBanIP("192.168.1.12", "test");
+Database.globalBanIP("192.168.2", "test");
+Database.globalBanIP("192.167", "test");
+Database.refreshGlobalBans();
+assert(Database.checkGlobalBan("192.168.1.12"));
+assert(Database.checkGlobalBan("192.168.2.24"));
+assert(Database.checkGlobalBan("192.167.1.15"));
+Database.globalUnbanIP("192.168.1.12");
+Database.globalUnbanIP("192.167");
+Database.refreshGlobalBans();
+assert(!Database.checkGlobalBan("192.168.1.12"));
+assert(!Database.checkGlobalBan("192.167.5.54"));
+
+console.log("[PASS] Global Bans");
+process.exit(0);