Add initial toml support
This commit is contained in:
parent
b74e46ec97
commit
f2f3d2aae5
|
@ -1 +1,2 @@
|
|||
/target
|
||||
config.toml
|
|
@ -175,6 +175,7 @@ dependencies = [
|
|||
"irc",
|
||||
"serenity",
|
||||
"tokio",
|
||||
"toml",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
|
|
|
@ -8,6 +8,7 @@ edition = "2021"
|
|||
[dependencies]
|
||||
anyhow = "1.0.52"
|
||||
irc = "0.15"
|
||||
toml = "0.5"
|
||||
|
||||
[dependencies.tokio]
|
||||
version = "1.15.0"
|
||||
|
|
12
src/main.rs
12
src/main.rs
|
@ -1,4 +1,4 @@
|
|||
use std::{env, sync::Arc};
|
||||
use std::{env, sync::Arc, fs::File, io::Read};
|
||||
|
||||
use serenity::{
|
||||
async_trait,
|
||||
|
@ -18,6 +18,8 @@ use irc::{
|
|||
proto::Command,
|
||||
};
|
||||
|
||||
use toml::Value;
|
||||
|
||||
struct Handler;
|
||||
|
||||
#[async_trait]
|
||||
|
@ -81,7 +83,13 @@ impl TypeMapKey for SenderKey {
|
|||
|
||||
#[tokio::main]
|
||||
async fn main() -> anyhow::Result<()> {
|
||||
let token = env::var("DISCORD_TOKEN").expect("DISCORD_TOKEN not set");
|
||||
let filename = env::args().nth(1).expect("No filename was provided!");
|
||||
let mut data = String::new();
|
||||
File::open(filename)?.read_to_string(&mut data)?;
|
||||
|
||||
let value = data.parse::<Value>()?;
|
||||
|
||||
let token = value["token"].as_str().expect("No token provided!").to_string();
|
||||
|
||||
let mut discord_client = DiscordClient::builder(&token)
|
||||
.event_handler(Handler)
|
||||
|
|
Loading…
Reference in New Issue