removed unnessesary env var (untested)
This commit is contained in:
@@ -32,9 +32,6 @@ use tokio::time;
|
||||
use websocket::send_cmd_json;
|
||||
|
||||
|
||||
// TODO: bot command permissions
|
||||
|
||||
|
||||
#[derive(Parser, Serialize, Clone)]
|
||||
struct Args {
|
||||
#[arg(short = 'p', long, default_value = "2920", help = "Sets the port number, e.g 2200.")]
|
||||
|
||||
+14
-12
@@ -179,24 +179,26 @@ pub async fn http_edit_msg(
|
||||
|
||||
|
||||
pub async fn send_dm(msg: String, args: Args) {
|
||||
let uid = env::var("ASSISTANT_DM_USER");
|
||||
if uid.is_err() { return; }
|
||||
let uids = env::var("ASSISTANT_OWNERS");
|
||||
if uids.is_err() { return; }
|
||||
|
||||
let uids_split: Vec<String> = uids.unwrap()
|
||||
.split(',')
|
||||
.map(|s| s.to_string())
|
||||
.collect();
|
||||
|
||||
let user = UserId::new(uid.unwrap().parse::<u64>().unwrap());
|
||||
|
||||
let token: String;
|
||||
if !args.test {
|
||||
token = std::env::var("ASSISTANT_TOKEN").expect("Missing ASSISTANT_TOKEN env var!");
|
||||
}
|
||||
else {
|
||||
token = std::env::var("ASSISTANT_TOKEN_TEST").expect("Missing ASSISTANT_TOKEN_TEST env var!");
|
||||
};
|
||||
let token: String =
|
||||
if !args.test { std::env::var("ASSISTANT_TOKEN") .expect("Missing ASSISTANT_TOKEN env var!") }
|
||||
else { std::env::var("ASSISTANT_TOKEN_TEST").expect("Missing ASSISTANT_TOKEN_TEST env var!") };
|
||||
|
||||
let http = Http::new(&token);
|
||||
|
||||
let c_msg = CreateMessage::new().content(msg);
|
||||
|
||||
let _ = user.dm(http, c_msg).await;
|
||||
for uid in uids_split {
|
||||
let user = UserId::new(uid.parse::<u64>().unwrap());
|
||||
let _ = user.dm(http.as_ref(), c_msg.clone()).await;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
+1
-1
@@ -134,7 +134,7 @@ async fn handle_message(msg: tungstenite::protocol::Message, args: Args) {
|
||||
if text.starts_with("json:") {
|
||||
let t_json: Value = serde_json::from_str(&text[5..]).unwrap();
|
||||
if t_json.get("error").is_some() {
|
||||
send_dm("Internal Python error!".to_string(), args).await;
|
||||
send_dm("Unknown internal Python error occurred!".to_string(), args).await;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user