#include "../lua/lua.h"
#define CLEANUP_KEEP 10 /* keep this many topics and kicks per channel around */
-#define CLEANUP_INTERVAL 84600 /* db cleanup interval (in seconds) */
+#define CLEANUP_INTERVAL 86400 /* db cleanup interval (in seconds) */
#define CLEANUP_INACTIVE_DAYS 30 /* disable channels where nothing happened for this many days */
+#define CLEANUP_DELETE_DAYS 5 /* delete data for channels that have been disabled for this many days */
MODULE_VERSION("");
}
a4statsdb->createtable(a4statsdb, NULL, NULL,
- "CREATE TABLE ? (id SERIAL PRIMARY KEY, name VARCHAR(256) UNIQUE, timestamp INT DEFAULT 0, active INT DEFAULT 1, deleted INT DEFAULT 0, privacy INT DEFAULT 2, "
+ "CREATE TABLE ? (id SERIAL PRIMARY KEY, name VARCHAR(256) UNIQUE, timestamp INT DEFAULT 0, active INT DEFAULT 1, deleted INT DEFAULT 0, privacy INT DEFAULT 1, "
"h0 INT DEFAULT 0, h1 INT DEFAULT 0, h2 INT DEFAULT 0, h3 INT DEFAULT 0, h4 INT DEFAULT 0, h5 INT DEFAULT 0, "
"h6 INT DEFAULT 0, h7 INT DEFAULT 0, h8 INT DEFAULT 0, h9 INT DEFAULT 0, h10 INT DEFAULT 0, h11 INT DEFAULT 0, "
"h12 INT DEFAULT 0, h13 INT DEFAULT 0, h14 INT DEFAULT 0, h15 INT DEFAULT 0, h16 INT DEFAULT 0, h17 INT DEFAULT 0, "
"h18 INT DEFAULT 0, h19 INT DEFAULT 0, h20 INT DEFAULT 0, h21 INT DEFAULT 0, h22 INT DEFAULT 0, h23 INT DEFAULT 0)", "T", "channels");
a4statsdb->createtable(a4statsdb, NULL, NULL,
- "CREATE TABLE ? (channelid INT, kicker VARCHAR(128), kickerid INT, victim VARCHAR(128), victimid INT, timestamp INT, reason VARCHAR(256))", "T", "kicks");
+ "CREATE TABLE ? (channelid INT, kicker VARCHAR(128), kickerid INT, victim VARCHAR(128), victimid INT, timestamp INT, reason VARCHAR(256),"
+ "FOREIGN KEY (channelid) REFERENCES ? (id) ON DELETE CASCADE)", "TT", "kicks", "channels");
a4statsdb->squery(a4statsdb, "CREATE INDEX kicks_channelid_index ON ? (channelid)", "T", "kicks");
a4statsdb->squery(a4statsdb, "CREATE INDEX kicks_timestamp_index ON ? (timestamp)", "T", "kicks");
a4statsdb->createtable(a4statsdb, NULL, NULL,
- "CREATE TABLE ? (channelid INT, setby VARCHAR(128), setbyid INT, timestamp INT, topic VARCHAR(512))", "T", "topics");
+ "CREATE TABLE ? (channelid INT, setby VARCHAR(128), setbyid INT, timestamp INT, topic VARCHAR(512),"
+ "FOREIGN KEY (channelid) REFERENCES ? (id) ON DELETE CASCADE)", "TT", "topics", "channels");
a4statsdb->squery(a4statsdb, "CREATE INDEX topics_channelid_index ON ? (channelid)", "T", "topics");
"h18 INT DEFAULT 0, h19 INT DEFAULT 0, h20 INT DEFAULT 0, h21 INT DEFAULT 0, h22 INT DEFAULT 0, h23 INT DEFAULT 0, "
"last VARCHAR(512), quote VARCHAR(512), quotereset INT DEFAULT 0, mood_happy INT DEFAULT 0, mood_sad INT DEFAULT 0, questions INT DEFAULT 0, yelling INT DEFAULT 0, caps INT DEFAULT 0, "
"slaps INT DEFAULT 0, slapped INT DEFAULT 0, highlights INT DEFAULT 0, kicks INT DEFAULT 0, kicked INT DEFAULT 0, ops INT DEFAULT 0, deops INT DEFAULT 0, actions INT DEFAULT 0, skitzo INT DEFAULT 0, foul INT DEFAULT 0, "
- "firstseen INT DEFAULT 0, curnick VARCHAR(16))", "T", "users");
+ "firstseen INT DEFAULT 0, curnick VARCHAR(16), FOREIGN KEY (channelid) REFERENCES ? (id) ON DELETE CASCADE)", "TT", "users", "channels");
a4statsdb->squery(a4statsdb, "CREATE INDEX users_account_index ON ? (account)", "T", "users");
a4statsdb->squery(a4statsdb, "CREATE INDEX users_accountid_index ON ? (accountid)", "T", "users");
a4statsdb->squery(a4statsdb, "CREATE INDEX users_channelid_lines_index ON ? (channelid, lines)", "T", "users");
a4statsdb->createtable(a4statsdb, NULL, NULL,
- "CREATE TABLE ? (channelid INT, first VARCHAR(128), firstid INT, second VARCHAR(128), secondid INT, seen INT, score INT DEFAULT 1)", "T", "relations");
+ "CREATE TABLE ? (channelid INT, first VARCHAR(128), firstid INT, second VARCHAR(128), secondid INT, seen INT, score INT DEFAULT 1,"
+ "FOREIGN KEY (channelid) REFERENCES ? (id) ON DELETE CASCADE)", "TT", "relations", "channels");
a4statsdb->squery(a4statsdb, "CREATE INDEX relations_channelid_index ON ? (channelid)", "T", "relations");
a4statsdb->squery(a4statsdb, "CREATE INDEX relations_score_index ON ? (score)", "T", "relations");
free(dci);
}
-static void a4stats_fetch_user_cb(const struct DBAPIResult *result, void *uarg) {
- db_callback_info *dci = uarg;
- time_t seen = 0, quotereset = 0;
-
- if (result) {
- if (result->success) {
- while (result->next(result)) {
- seen = (result->get(result, 0)) ? (time_t)strtoul(result->get(result, 0), NULL, 10) : 0;
- quotereset = (result->get(result, 1)) ? (time_t)strtoul(result->get(result, 1), NULL, 10) : 0;
- }
- }
-
- result->clear(result);
- }
-
- if (dci->interp) {
- lua_vpcall(dci->interp, dci->callback, "llR", (long)seen, (long)quotereset, dci->uarg_index);
- luaL_unref(dci->interp, LUA_REGISTRYINDEX, dci->uarg_index);
- }
-
- a4stats_delete_dci(dci);
-}
-
-static int a4stats_lua_fetch_user(lua_State *ps) {
- const char *account, *callback;
- unsigned long channelid, accountid;
- db_callback_info *dci;
-
- if (!lua_islong(ps, 1) || !lua_isstring(ps, 2) || !lua_isnumber(ps, 3) || !lua_isstring(ps, 4))
- LUA_RETURN(ps, LUA_FAIL);
-
- channelid = lua_tonumber(ps, 1);
- account = lua_tostring(ps, 2);
- accountid = lua_tonumber(ps, 3);
- callback = lua_tostring(ps, 4);
-
- dci = malloc(sizeof(*dci));
- dci->interp = ps;
-
- strncpy(dci->callback, callback, sizeof(dci->callback));
- dci->callback[sizeof(dci->callback) - 1] = '\0';
-
- lua_pushvalue(ps, 5);
- dci->uarg_index = luaL_ref(ps, LUA_REGISTRYINDEX);
-
- a4statsdb->query(a4statsdb, a4stats_fetch_user_cb, dci, "SELECT seen, quotereset FROM ? WHERE channelid = ? AND (accountid != 0 AND accountid = ? OR accountid = 0 AND account = ?)", "TUUs", "users", channelid, accountid, account);
-
- LUA_RETURN(ps, LUA_OK);
-}
-
typedef struct user_update_info {
int stage;
char *update;
free(uui->update);
free(uui->account);
free(uui);
- return;
+ goto a4_uuc_return;
}
a4statsdb->query(a4statsdb, a4stats_update_user_cb, uui, "INSERT INTO ? (channelid, account, accountid, firstseen) VALUES (?, ?, ?, ?)", "TUsUt", "users", uui->channelid, uui->account, uui->accountid, time(NULL));
}
+
+a4_uuc_return:
+ if (result)
+ result->clear(result);
}
static int a4stats_lua_update_user(lua_State *ps) {
a4statsdb->query(a4statsdb, a4stats_update_relation_cb, rui, "UPDATE ? SET score = score + 1, seen = ? "
"WHERE channelid = ? AND first = ? AND firstid = ? AND second = ? AND secondid = ?",
"TtUsUsU", "relations", time(NULL), rui->channelid, rui->first, rui->firstid, rui->second, rui->secondid);
- return;
+ goto a4_urc_return;
} else if (rui->stage == 2 && result && result->affected == 0) {
a4statsdb->query(a4statsdb, a4stats_update_relation_cb, rui, "INSERT INTO ? (channelid, first, firstid, second, secondid, seen) VALUES (?, ?, ?, ?, ?, ?)",
"TUsUsUt", "relations", rui->channelid, rui->first, rui->firstid, rui->second, rui->secondid, time(NULL));
- return;
+ goto a4_urc_return;
}
if (!result || result->affected == 0)
free(rui->first);
free(rui->second);
free(rui);
+
+a4_urc_return:
+ if (result)
+ result->clear(result);
}
static int a4stats_lua_update_relation(lua_State *ps) {
LUA_RETURN(ps, LUA_OK);
}
-typedef struct {
- time_t now;
+static struct {
+ time_t start;
unsigned long pending;
unsigned long topicskicks;
unsigned long disabled;
+ unsigned long deleted;
} cleanupdata;
-static void a4stats_cleanupdb_check_done(cleanupdata *cud) {
- if (!--cud->pending)
- controlwall(NO_OPER, NL_CLEANUP, "Deleted %lu old topics and kicks. %lu channels disabled.",
- cud->topicskicks, cud->disabled);
-}
-
-static void a4stats_cleanupdb_topicskicks(const struct DBAPIResult *result, void *arg) {
- cleanupdata *cud = arg;
-
- if (result)
- cud->topicskicks += result->affected;
-
- a4stats_cleanupdb_check_done(cud);
+static void a4stats_cleanupdb_got_result(void) {
+ if (!--cleanupdata.pending) {
+ controlwall(NO_OPER, NL_CLEANUP, "CLEANUPA4STATS: Deleted %lu old topics and kicks. Disabled %lu inactive channels. Deleted data for %lu channels.",
+ cleanupdata.topicskicks, cleanupdata.disabled, cleanupdata.deleted);
+ cleanupdata.start = 0;
+ }
}
-static void a4stats_cleanupdb_disabled(const struct DBAPIResult *result, void *arg) {
- cleanupdata *cud = arg;
+static void a4stats_cleanupdb_cb_countrows(const struct DBAPIResult *result, void *arg) {
+ unsigned long *counter = arg;
- if (result)
- cud->disabled += result->affected;
+ if (result) {
+ *counter += result->affected;
+ result->clear(result);
+ }
- a4stats_cleanupdb_check_done(cud);
+ a4stats_cleanupdb_got_result();
}
-static void a4stats_cleanupdb_cb(const struct DBAPIResult *result, void *arg) {
- cleanupdata *cud = arg;
+static void a4stats_cleanupdb_cb_active(const struct DBAPIResult *result, void *null) {
unsigned long channelid;
time_t seen;
-
- if (result && result->success) {
+
+ if (!result)
+ return;
+
+ if (result->success) {
while (result->next(result)) {
channelid = strtoul(result->get(result, 0), NULL, 10);
seen = (time_t)strtoul(result->get(result, 2), NULL, 10);
if (!seen)
seen = (time_t)strtoul(result->get(result, 1), NULL, 10);
- if (seen < cud->now - CLEANUP_INACTIVE_DAYS * 86400) {
- /* delete inactive channels */
- cud->pending++;
- a4statsdb->query(a4statsdb, a4stats_cleanupdb_disabled, cud,
+ if (seen < cleanupdata.start - CLEANUP_INACTIVE_DAYS * 86400) {
+ /* disable inactive channels */
+ cleanupdata.pending++;
+ a4statsdb->query(a4statsdb, a4stats_cleanupdb_cb_countrows, &cleanupdata.disabled,
"UPDATE ? SET active = 0, deleted = ? WHERE id = ? AND active = 1",
- "TtU", "channels", cud->now, channelid);
+ "TtU", "channels", cleanupdata.start, channelid);
} else {
/* cleanup old kicks/topics */
- cud->pending++;
- a4statsdb->query(a4statsdb, a4stats_cleanupdb_topicskicks, cud,
+ cleanupdata.pending++;
+ a4statsdb->query(a4statsdb, a4stats_cleanupdb_cb_countrows, &cleanupdata.topicskicks,
"DELETE FROM ? WHERE channelid = ? AND timestamp <= "
"(SELECT timestamp FROM ? WHERE channelid = ? ORDER BY timestamp DESC OFFSET ? LIMIT 1)",
"TUTUU", "kicks", channelid, "kicks", channelid, (unsigned long)CLEANUP_KEEP);
- cud->pending++;
- a4statsdb->query(a4statsdb, a4stats_cleanupdb_topicskicks, cud,
+ cleanupdata.pending++;
+ a4statsdb->query(a4statsdb, a4stats_cleanupdb_cb_countrows, &cleanupdata.topicskicks,
"DELETE FROM ? WHERE channelid = ? AND timestamp <= "
"(SELECT timestamp FROM ? WHERE channelid = ? ORDER BY timestamp DESC OFFSET ? LIMIT 1)",
"TUTUU", "topics", channelid, "topics", channelid, (unsigned long)CLEANUP_KEEP);
}
}
}
+ result->clear(result);
}
static void a4stats_cleanupdb(void *null) {
- static cleanupdata cud;
+ controlwall(NO_OPER, NL_CLEANUP, "Starting a4stats_db cleanup.");
- cud.now = time(NULL);
- cud.pending = 0;
- cud.topicskicks = 0;
- cud.disabled = 0;
+ if (cleanupdata.start != 0) {
+ controlwall(NO_OPER, NL_CLEANUP, "a4stats cleanup already in progress.");
+ return;
+ }
- controlwall(NO_OPER, NL_CLEANUP, "Starting a4stats_db cleanup.");
- a4statsdb->query(a4statsdb, a4stats_cleanupdb_cb, &cud,
+ cleanupdata.start = time(NULL);
+ cleanupdata.pending = 0;
+ cleanupdata.topicskicks = 0;
+ cleanupdata.disabled = 0;
+ cleanupdata.deleted = 0;
+
+ a4statsdb->query(a4statsdb, a4stats_cleanupdb_cb_active, NULL,
"SELECT id, timestamp, MAX(users.seen) FROM ? LEFT JOIN ? AS users ON id = users.channelid WHERE active = 1 GROUP BY id",
"TT", "channels", "users");
+ cleanupdata.pending++;
+ a4statsdb->query(a4statsdb, a4stats_cleanupdb_cb_countrows, &cleanupdata.deleted,
+ "DELETE FROM ? WHERE active = 0 AND deleted < ?", "Tt", "channels", (time_t)(cleanupdata.start - CLEANUP_DELETE_DAYS * 84600));
}
static void a4stats_fetch_channels_cb(const struct DBAPIResult *result, void *uarg) {
LUA_RETURN(ps, LUA_OK);
}
+static int a4stats_lua_set_privacy(lua_State *ps) {
+ const char *channel;
+ unsigned long privacy;
+
+ if (!lua_isstring(ps, 1) || !lua_isnumber(ps, 2))
+ LUA_RETURN(ps, LUA_FAIL);
+
+ channel = lua_tostring(ps, 1);
+ privacy = lua_tonumber(ps, 2);
+
+ a4statsdb->squery(a4statsdb, "UPDATE ? SET privacy = ? WHERE name = ?", "TUs", "channels", privacy, channel);
+ LUA_RETURN(ps, LUA_OK);
+}
+
static int a4stats_lua_add_kick(lua_State *ps) {
unsigned long channelid, kickerid, victimid;
const char *kicker, *victim, *reason;
LUA_RETURN(ps, LUA_OK);
}
+static int a4stats_lua_db_begin(lua_State *ps) {
+ a4statsdb->query(a4statsdb, NULL, NULL, "BEGIN TRANSACTION", "");
+ LUA_RETURN(ps, LUA_OK);
+}
+
+static int a4stats_lua_db_commit(lua_State *ps) {
+ a4statsdb->query(a4statsdb, NULL, NULL, "COMMIT TRANSACTION", "");
+ LUA_RETURN(ps, LUA_OK);
+}
+
static void a4stats_hook_loadscript(int hooknum, void *arg) {
void **args = arg;
lua_State *l = args[1];
lua_register(l, "a4_enable_channel", a4stats_lua_enable_channel);
lua_register(l, "a4_disable_channel", a4stats_lua_disable_channel);
+ lua_register(l, "a4_set_privacy", a4stats_lua_set_privacy);
lua_register(l, "a4_fetch_channels", a4stats_lua_fetch_channels);
lua_register(l, "a4_add_kick", a4stats_lua_add_kick);
lua_register(l, "a4_add_topic", a4stats_lua_add_topic);
lua_register(l, "a4_add_line", a4stats_lua_add_line);
- lua_register(l, "a4_fetch_user", a4stats_lua_fetch_user);
lua_register(l, "a4_update_user", a4stats_lua_update_user);
lua_register(l, "a4_update_relation", a4stats_lua_update_relation);
lua_register(l, "a4_escape_string", a4stats_lua_escape_string);
+ lua_register(l, "a4_db_begin", a4stats_lua_db_begin);
+ lua_register(l, "a4_db_commit", a4stats_lua_db_commit);
}
#define lua_unregister(L, n) (lua_pushnil(L), lua_setglobal(L, n))
lua_unregister(l->l, "a4_enable_channel");
lua_unregister(l->l, "a4_disable_channel");
+ lua_unregister(l->l, "a4_set_privacy");
lua_unregister(l->l, "a4_fetch_channels");
lua_unregister(l->l, "a4_add_kick");
lua_unregister(l->l, "a4_add_topic");
lua_unregister(l->l, "a4_update_user");
lua_unregister(l->l, "a4_update_relation");
lua_unregister(l->l, "a4_escape_string");
+ lua_unregister(l->l, "a4_db_begin");
+ lua_unregister(l->l, "a4_db_commit");
}
deregisterhook(HOOK_LUA_LOADSCRIPT, a4stats_hook_loadscript);
projects / irc / quakenet / newserv.git / blobdiff