forked from karx/dircord
Add initial toml support
This commit is contained in:
parent
b74e46ec97
commit
f2f3d2aae5
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -1 +1,2 @@
|
||||||
/target
|
/target
|
||||||
|
config.toml
|
1
Cargo.lock
generated
1
Cargo.lock
generated
|
@ -175,6 +175,7 @@ dependencies = [
|
||||||
"irc",
|
"irc",
|
||||||
"serenity",
|
"serenity",
|
||||||
"tokio",
|
"tokio",
|
||||||
|
"toml",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
|
|
|
@ -8,6 +8,7 @@ edition = "2021"
|
||||||
[dependencies]
|
[dependencies]
|
||||||
anyhow = "1.0.52"
|
anyhow = "1.0.52"
|
||||||
irc = "0.15"
|
irc = "0.15"
|
||||||
|
toml = "0.5"
|
||||||
|
|
||||||
[dependencies.tokio]
|
[dependencies.tokio]
|
||||||
version = "1.15.0"
|
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::{
|
use serenity::{
|
||||||
async_trait,
|
async_trait,
|
||||||
|
@ -18,6 +18,8 @@ use irc::{
|
||||||
proto::Command,
|
proto::Command,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use toml::Value;
|
||||||
|
|
||||||
struct Handler;
|
struct Handler;
|
||||||
|
|
||||||
#[async_trait]
|
#[async_trait]
|
||||||
|
@ -81,7 +83,13 @@ impl TypeMapKey for SenderKey {
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() -> anyhow::Result<()> {
|
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)
|
let mut discord_client = DiscordClient::builder(&token)
|
||||||
.event_handler(Handler)
|
.event_handler(Handler)
|
||||||
|
|
Loading…
Reference in a new issue