Add initial structure

Signed-off-by: trivernis <trivernis@protonmail.com>

.gitignore

@@ -0,0 +1,4 @@
+/target
+bot.db
+.env
+*.env

.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml
+# Editor-based HTTP Client requests
+/httpRequests/

.idea/csv-plugin.xml

@@ -0,0 +1,16 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="CsvFileAttributes">
+    <option name="attributeMap">
+      <map>
+        <entry key="/src/database/scripts/update_tables.sql">
+          <value>
+            <Attribute>
+              <option name="separator" value="," />
+            </Attribute>
+          </value>
+        </entry>
+      </map>
+    </option>
+  </component>
+</project>

.idea/dataSources.xml

@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="DataSourceManagerImpl" format="xml" multifile-model="true">
+    <data-source source="LOCAL" name="bot.db" uuid="4d1d7a8d-098d-4916-b51c-e7bde2613eda">
+      <driver-ref>sqlite.xerial</driver-ref>
+      <synchronize>true</synchronize>
+      <jdbc-driver>org.sqlite.JDBC</jdbc-driver>
+      <jdbc-url>jdbc:sqlite:bot.db</jdbc-url>
+      <working-dir>$ProjectFileDir$</working-dir>
+    </data-source>
+  </component>
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/tobi-rs.iml" filepath="$PROJECT_DIR$/.idea/tobi-rs.iml" />
+    </modules>
+  </component>
+</project>

.idea/sqldialects.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="SqlDialectMappings">
+    <file url="file://$PROJECT_DIR$/src/database/scripts/create_tables.sql" dialect="GenericSQL" />
+    <file url="file://$PROJECT_DIR$/src/database/scripts/update_tables.sql" dialect="SQLite" />
+    <file url="PROJECT" dialect="SQLite" />
+  </component>
+</project>

.idea/tobi-rs.iml

@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="CPP_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
+      <excludeFolder url="file://$MODULE_DIR$/target" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>

Cargo.toml

@@ -0,0 +1,18 @@
+[package]
+name = "tobi-rs"
+version = "0.1.0"
+authors = ["trivernis <trivernis@protonmail.com>"]
+edition = "2018"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+serenity = "0.10.5"
+dotenv = "0.15.0"
+tokio = { version = "1.4.0", features = ["macros", "rt-multi-thread"] }
+serde_rusqlite = "0.26.0"
+rusqlite = "0.24"
+serde_derive = "1.0.125"
+serde = "1.0.125"
+thiserror = "1.0.24"
+parking_lot = "0.11.1"

src/client.rs

@@ -0,0 +1,37 @@
+use crate::database::{get_database, Database};
+use crate::utils::error::{BotError, BotResult};
+use crate::utils::store::{Store, StoreData};
+use serenity::async_trait;
+use serenity::client::EventHandler;
+use serenity::framework::StandardFramework;
+use serenity::Client;
+
+struct Handler;
+
+#[async_trait]
+impl EventHandler for Handler {}
+
+pub async fn get_client() -> BotResult<Client> {
+    let token = dotenv::var("BOT_TOKEN").map_err(|_| BotError::MissingToken)?;
+    let database = get_database()?;
+
+    let client = Client::builder(token).framework(get_framework()).await?;
+    {
+        let mut data = client.data.write().await;
+        data.insert::<Store>(StoreData::new(database))
+    }
+
+    Ok(client)
+}
+
+pub fn get_framework() -> StandardFramework {
+    StandardFramework::default().configure(|c| {
+        c.prefix(
+            dotenv::var("BOT_PREFIX")
+                .unwrap_or("~!".to_string())
+                .as_str(),
+        )
+        .allow_dm(true)
+        .ignore_bots(true)
+    })
+}

src/database/guild.rs

@@ -0,0 +1,13 @@
+use serde_derive::{Deserialize, Serialize};
+
+#[derive(Serialize, Deserialize, Debug, PartialEq)]
+pub struct Guild {
+    guild_id: i32,
+}
+
+#[derive(Serialize, Deserialize, Debug, PartialEq)]
+pub struct GuildSettings {
+    guild_id: i32,
+    key: String,
+    value: String,
+}

src/database/mod.rs

@@ -0,0 +1,34 @@
+use crate::database::scripts::{CREATE_SCRIPT, UPDATE_SCRIPT};
+use crate::utils::error::BotResult;
+use rusqlite::{Connection, NO_PARAMS};
+
+pub mod guild;
+pub mod scripts;
+
+#[derive(Debug)]
+pub struct Database {
+    connection: Connection,
+}
+
+impl Database {
+    pub fn new(connection: Connection) -> Self {
+        Self { connection }
+    }
+
+    /// Initializes the database
+    pub fn init(&self) -> BotResult<()> {
+        self.connection.execute(CREATE_SCRIPT, NO_PARAMS)?;
+        self.connection.execute(UPDATE_SCRIPT, NO_PARAMS)?;
+
+        Ok(())
+    }
+}
+
+pub fn get_database() -> BotResult<Database> {
+    let filename = dotenv::var("DB_NAME").unwrap_or("bot.db".to_string());
+    let connection = rusqlite::Connection::open(filename)?;
+    let database = Database::new(connection);
+    database.init()?;
+
+    Ok(database)
+}

src/database/scripts/create_tables.sql

@@ -0,0 +1,10 @@
+CREATE TABLE IF NOT EXISTS guilds (
+    guild_id INTEGER PRIMARY KEY
+);
+
+CREATE TABLE IF NOT EXISTS guild_settings (
+     guild_id INTEGER NOT NULL,
+     setting_key TEXT NOT NULL,
+     setting_value TEXT,
+     FOREIGN KEY (guild_id) REFERENCES guilds (guild_id)
+);

src/database/scripts/mod.rs

@@ -0,0 +1,2 @@
+pub static CREATE_SCRIPT: &str = include_str!("create_tables.sql");
+pub static UPDATE_SCRIPT: &str = include_str!("update_tables.sql");

src/database/scripts/update_tables.sql

@@ -0,0 +1 @@
+SELECT NULL FROM guilds;

src/main.rs

@@ -0,0 +1,34 @@
+use crate::client::get_client;
+use serenity::client::{Context, EventHandler};
+use serenity::framework::standard::{
+    macros::{command, group},
+    CommandResult,
+};
+use serenity::model::channel::Message;
+
+pub(crate) mod client;
+pub(crate) mod database;
+pub(crate) mod utils;
+
+#[group]
+#[commands(ping)]
+struct General;
+
+struct Handler;
+
+#[tokio::main]
+async fn main() {
+    let mut client = get_client().await.unwrap();
+
+    // start listening for events by starting a single shard
+    if let Err(why) = client.start().await {
+        println!("An error occurred while running the client: {:?}", why);
+    }
+}
+
+#[command]
+async fn ping(ctx: &Context, msg: &Message) -> CommandResult {
+    msg.reply(ctx, "Pong!").await?;
+
+    Ok(())
+}

src/utils/error.rs

@@ -0,0 +1,15 @@
+use thiserror::Error;
+
+pub type BotResult<T> = Result<T, BotError>;
+
+#[derive(Error, Debug)]
+pub enum BotError {
+    #[error("Serenity Error: {0}")]
+    SerenityError(#[from] serenity::Error),
+
+    #[error("Sqlite Error: {0}")]
+    Sqlite(#[from] rusqlite::Error),
+
+    #[error("Missing Bot Token")]
+    MissingToken,
+}

src/utils/mod.rs

@@ -0,0 +1,2 @@
+pub mod error;
+pub mod store;

src/utils/store.rs

@@ -0,0 +1,22 @@
+use crate::database::Database;
+use parking_lot::Mutex;
+use serenity::prelude::TypeMapKey;
+use std::sync::Arc;
+
+pub struct Store;
+
+pub struct StoreData {
+    pub database: Arc<Mutex<Database>>,
+}
+
+impl StoreData {
+    pub fn new(database: Database) -> StoreData {
+        Self {
+            database: Arc::new(Mutex::new(database)),
+        }
+    }
+}
+
+impl TypeMapKey for Store {
+    type Value = StoreData;
+}