Implement TabletBot discord bot

Author: InfinityGhost

Cargo.lock

@@ -7,3 +7,15 @@ authors = [ "InfinityGhost" ]
 edition = "2021"
 
 [dependencies]
+# Serialization
+serde = { version = "1.0", features = ["derive"] }
+serde_json = { version = "1.0" }
+
+# Discord API
+serenity = { version = "0.11", default-features = false, features = ["cache", "client", "gateway", "rustls_backend", "model", "framework", "standard_framework"] }
+tokio = { version = "1.21.2", features = ["macros", "signal", "rt-multi-thread"] }
+
+# Misc
+regex = "1.6.0"
+octocrab = "0.17.0"
+reqwest = "0.11.12"

Cargo.toml

@@ -1,14 +1,24 @@
 { lib
 , rustPlatform
+, pkg-config
+, openssl
 }:
 
 rustPlatform.buildRustPackage rec {
   pname = "tabletbot";
   name = pname;
 
-  src = ./src;
+  nativeBuildInputs = [
+    pkg-config
+  ];
+
+  buildInputs = [
+    openssl
+  ];
+
+  src = ./.;
 
   cargoLock = {
-    lockFile = ./src/Cargo.lock;
+    lockFile = ./Cargo.lock;
   };
 }

default.nix

@@ -3,6 +3,8 @@
 let
   readFileIfExists = path: with pkgs.lib; if pathExists path then readFile path else null;
 
+  default = pkgs.callPackage ./. {};
+
   rustDeps = with pkgs; [
     llvmPackages_latest.llvm
     llvmPackages_latest.bintools
@@ -15,7 +17,7 @@ let
     python3
   ];
 
-  deps = with pkgs; [];
+  deps = default.nativeBuildInputs;
 
   buildPath = toString ./.build/out;
 
@@ -38,12 +40,16 @@ in pkgs.mkShell rec {
   RUSTC_VERSION = readFileIfExists ./rust-toolchain;
   CARGO_HOME = toString ./.build/cargo;
   RUSTUP_HOME = toString ./.build/rustup;
+  TABLETBOT_DATA = toString ./.build/data;
 
   buildInputs = with pkgs; rustDeps ++ deps ++ utils;
 
   shellHook = with pkgs.lib; ''
     export LD_LIBRARY_PATH=${escapeShellArg (makeLibraryPath buildInputs)}
     export PATH=$PATH:''${CARGO_HOME:-~/.cargo}/bin
     export PATH=$PATH:''${RUSTUP_HOME:-~/.rustup}/toolchains/$RUSTC_VERSION-x86_64-unknown-linux-gnu/bin/
+    export PATH=$PATH:${escapeShellArg (makeBinPath buildInputs)}
+
+    [ -r ${TABLETBOT_DATA}/tokens ] && source ${TABLETBOT_DATA}/tokens
   '';
 }

shell.nix

@@ -0,0 +1,52 @@
+use core::panic;
+use serenity::builder::CreateApplicationCommand;
+use serenity::Error;
+use serenity::http::Http;
+use serenity::model::prelude::command::Command;
+use serenity::model::prelude::interaction::application_command::ApplicationCommandInteraction;
+use serenity::prelude::Context;
+use std::sync::Arc;
+use crate::commands::snippets::*;
+use crate::commands::types::*;
+
+pub mod snippets;
+pub mod types;
+
+pub async fn get_commands(ctx: &Context) -> Vec<CreateApplicationCommand> {
+  vec![
+    SnippetCommand::register(ctx).await,
+    SetSnippetCommand::register(ctx).await,
+    RemoveSnippetCommand::register(ctx).await,
+    ExportSnippetCommand::register(ctx).await
+  ]
+}
+
+pub async fn register(ctx: &Context) -> Result<Vec<Command>, Error> {
+  let commands = get_commands(ctx).await;
+
+  Command::set_global_application_commands(&ctx.http, |c|
+    c.set_application_commands(commands)
+  ).await
+}
+
+pub async fn unregister(http: &Arc<Http>) -> Result<(), Error> {
+  if let Ok(commands) = Command::get_global_application_commands(http).await {
+    for command in commands {
+      if let Err(e) = Command::delete_global_application_command(http, command.id).await {
+        return Err(e);
+      }
+    }
+  }
+
+  Ok(())
+}
+
+pub async fn interact(ctx: &Context, command: &ApplicationCommandInteraction) {
+  match command.data.name.as_str() {
+    SNIPPET_NAME => SnippetCommand::invoke(ctx, command),
+    SET_SNIPPET_NAME => SetSnippetCommand::invoke(ctx, command),
+    REMOVE_SNIPPET_NAME => RemoveSnippetCommand::invoke(ctx, command),
+    EXPORT_SNIPPET_NAME => ExportSnippetCommand::invoke(ctx, command),
+    _ => panic!("Invalid interaction command: {}", command.data.name)
+  }.await
+}