2021-08-19 14:19:14 +00:00
import collections from "../collections.js" ;
import logger from "../logger.js" ;
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
import sqlite3 from "better-sqlite3" ;
2021-07-02 04:42:12 +00:00
const connection = sqlite3 ( process . env . DB . replace ( "sqlite://" , "" ) ) ;
2021-08-19 14:19:14 +00:00
export async function setup ( ) {
2021-07-02 04:42:12 +00:00
let counts ;
try {
counts = connection . prepare ( "SELECT * FROM counts" ) . all ( ) ;
} catch {
2021-08-11 21:54:43 +00:00
connection . prepare ( "CREATE TABLE counts ( command VARCHAR NOT NULL PRIMARY KEY, count integer NOT NULL )" ) . run ( ) ;
2021-07-02 04:42:12 +00:00
counts = [ ] ;
}
2021-08-11 15:00:35 +00:00
try {
connection . prepare ( "SELECT * FROM tags" ) . all ( ) ;
} catch {
connection . prepare ( "CREATE TABLE tags ( guild_id VARCHAR(30) NOT NULL, name text NOT NULL, content text NOT NULL, author VARCHAR(30) NOT NULL, UNIQUE(guild_id, name) )" ) . run ( ) ;
}
2021-07-02 04:42:12 +00:00
if ( ! counts ) {
for ( const command of collections . commands . keys ( ) ) {
connection . prepare ( "INSERT INTO counts (command, count) VALUES (?, ?)" ) . run ( command , 0 ) ;
}
} else {
const exists = [ ] ;
for ( const command of collections . commands . keys ( ) ) {
const count = connection . prepare ( "SELECT * FROM counts WHERE command = ?" ) . get ( command ) ;
if ( ! count ) {
connection . prepare ( "INSERT INTO counts (command, count) VALUES (?, ?)" ) . run ( command , 0 ) ;
}
exists . push ( command ) ;
}
2021-08-14 21:15:21 +00:00
2021-07-02 04:42:12 +00:00
for ( const { command } of counts ) {
if ( ! exists . includes ( command ) ) {
connection . prepare ( "DELETE FROM counts WHERE command = ?" ) . run ( command ) ;
}
}
}
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function stop ( ) {
2021-07-02 04:42:12 +00:00
connection . close ( ) ;
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function fixGuild ( guild ) {
2021-07-02 04:42:12 +00:00
let guildDB ;
try {
guildDB = connection . prepare ( "SELECT * FROM guilds WHERE guild_id = ?" ) . get ( guild . id ) ;
} catch {
2021-08-13 03:28:09 +00:00
connection . prepare ( "CREATE TABLE guilds ( guild_id VARCHAR(30) NOT NULL PRIMARY KEY, prefix VARCHAR(15) NOT NULL, disabled text NOT NULL, disabled_commands text NOT NULL )" ) . run ( ) ;
2021-07-02 04:42:12 +00:00
}
if ( ! guildDB ) {
logger . log ( ` Registering guild database entry for guild ${ guild . id } ... ` ) ;
return await this . addGuild ( guild ) ;
}
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function addCount ( command ) {
2021-07-02 04:42:12 +00:00
connection . prepare ( "UPDATE counts SET count = count + 1 WHERE command = ?" ) . run ( command ) ;
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function getCounts ( ) {
2021-07-02 04:42:12 +00:00
const counts = connection . prepare ( "SELECT * FROM counts" ) . all ( ) ;
const countObject = { } ;
for ( const { command , count } of counts ) {
countObject [ command ] = count ;
}
return countObject ;
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function disableCommand ( guild , command ) {
2021-08-13 03:28:09 +00:00
const guildDB = await this . getGuild ( guild ) ;
connection . prepare ( "UPDATE guilds SET disabled_commands = ? WHERE guild_id = ?" ) . run ( JSON . stringify ( ( guildDB . disabledCommands ? [ ... JSON . parse ( guildDB . disabledCommands ) , command ] : [ command ] ) . filter ( ( v ) => v !== undefined ) ) , guild ) ;
collections . disabledCmdCache . set ( guild , guildDB . disabled _commands ? [ ... JSON . parse ( guildDB . disabledCommands ) , command ] : [ command ] . filter ( ( v ) => v !== undefined ) ) ;
2021-08-19 14:19:14 +00:00
}
2021-08-13 03:28:09 +00:00
2021-08-19 14:19:14 +00:00
export async function enableCommand ( guild , command ) {
2021-08-13 03:28:09 +00:00
const guildDB = await this . getGuild ( guild ) ;
const newDisabled = guildDB . disabledCommands ? JSON . parse ( guildDB . disabledCommands ) . filter ( item => item !== command ) : [ ] ;
connection . prepare ( "UPDATE guilds SET disabled_commands = ? WHERE guild_id = ?" ) . run ( JSON . stringify ( newDisabled ) , guild ) ;
collections . disabledCmdCache . set ( guild , newDisabled ) ;
2021-08-19 14:19:14 +00:00
}
2021-08-13 03:28:09 +00:00
2021-08-19 14:19:14 +00:00
export async function disableChannel ( channel ) {
2021-07-02 04:42:12 +00:00
const guildDB = await this . getGuild ( channel . guild . id ) ;
connection . prepare ( "UPDATE guilds SET disabled = ? WHERE guild_id = ?" ) . run ( JSON . stringify ( [ ... JSON . parse ( guildDB . disabled ) , channel . id ] ) , channel . guild . id ) ;
collections . disabledCache . set ( channel . guild . id , [ ... JSON . parse ( guildDB . disabled ) , channel . id ] ) ;
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function enableChannel ( channel ) {
2021-07-02 04:42:12 +00:00
const guildDB = await this . getGuild ( channel . guild . id ) ;
const newDisabled = JSON . parse ( guildDB . disabled ) . filter ( item => item !== channel . id ) ;
connection . prepare ( "UPDATE guilds SET disabled = ? WHERE guild_id = ?" ) . run ( JSON . stringify ( newDisabled ) , channel . guild . id ) ;
collections . disabledCache . set ( channel . guild . id , newDisabled ) ;
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function getTag ( guild , tag ) {
2021-08-14 13:00:16 +00:00
const tagResult = connection . prepare ( "SELECT * FROM tags WHERE guild_id = ? AND name = ?" ) . get ( guild , tag ) ;
return tagResult ? { content : tagResult . content , author : tagResult . author } : undefined ;
2021-08-19 14:19:14 +00:00
}
2021-08-14 13:00:16 +00:00
2021-08-19 14:19:14 +00:00
export async function getTags ( guild ) {
2021-08-11 15:00:35 +00:00
const tagArray = connection . prepare ( "SELECT * FROM tags WHERE guild_id = ?" ) . all ( guild ) ;
const tags = { } ;
if ( ! tagArray ) return [ ] ;
for ( const tag of tagArray ) {
tags [ tag . name ] = { content : tag . content , author : tag . author } ;
}
return tags ;
2021-08-19 14:19:14 +00:00
}
2021-08-11 01:25:29 +00:00
2021-08-19 14:19:14 +00:00
export async function setTag ( name , content , guild ) {
2021-08-11 15:00:35 +00:00
const tag = {
id : guild . id ,
name : name ,
content : content . content ,
author : content . author
} ;
connection . prepare ( "INSERT INTO tags (guild_id, name, content, author) VALUES (@id, @name, @content, @author)" ) . run ( tag ) ;
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function removeTag ( name , guild ) {
2021-08-11 15:00:35 +00:00
connection . prepare ( "DELETE FROM tags WHERE guild_id = ? AND name = ?" ) . run ( guild . id , name ) ;
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function editTag ( name , content , guild ) {
2021-08-11 15:00:35 +00:00
connection . prepare ( "UPDATE tags SET content = ?, author = ? WHERE guild_id = ? AND name = ?" ) . run ( content . content , content . author , guild . id , name ) ;
2021-08-19 14:19:14 +00:00
}
2021-08-11 01:25:29 +00:00
2021-08-19 14:19:14 +00:00
export async function setPrefix ( prefix , guild ) {
2021-07-02 04:42:12 +00:00
connection . prepare ( "UPDATE guilds SET prefix = ? WHERE guild_id = ?" ) . run ( prefix , guild . id ) ;
collections . prefixCache . set ( guild . id , prefix ) ;
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function addGuild ( guild ) {
2021-07-02 04:42:12 +00:00
const query = await this . getGuild ( guild ) ;
if ( query ) return query ;
const guildObject = {
id : guild . id ,
prefix : process . env . PREFIX ,
disabled : "[]" ,
2021-08-13 03:28:09 +00:00
disabledCommands : "[]"
2021-07-02 04:42:12 +00:00
} ;
2021-08-18 20:41:35 +00:00
connection . prepare ( "INSERT INTO guilds (guild_id, prefix, disabled, disabled_commands) VALUES (@id, @prefix, @disabled, @disabledCommands)" ) . run ( guildObject ) ;
2021-07-02 04:42:12 +00:00
return guildObject ;
2021-08-19 14:19:14 +00:00
}
2021-07-02 04:42:12 +00:00
2021-08-19 14:19:14 +00:00
export async function getGuild ( query ) {
2021-07-02 04:42:12 +00:00
try {
return connection . prepare ( "SELECT * FROM guilds WHERE guild_id = ?" ) . get ( query ) ;
} catch {
return ;
}
2021-08-19 14:19:14 +00:00
}