forked from projectmoon/tenebrous-dicebot
Remove actix, move state to RwLock. Update dependencies.
This commit is contained in:
parent
1ef3b50a6e
commit
9268314421
File diff suppressed because it is too large
Load Diff
12
Cargo.toml
12
Cargo.toml
|
@ -22,22 +22,12 @@ async-trait = "0.1"
|
||||||
url = "2.1"
|
url = "2.1"
|
||||||
dirs = "3.0"
|
dirs = "3.0"
|
||||||
indoc = "1.0"
|
indoc = "1.0"
|
||||||
actix = "0.10"
|
|
||||||
actix-rt = "1.1"
|
|
||||||
combine = "4.3"
|
combine = "4.3"
|
||||||
sled = "0.34"
|
sled = "0.34"
|
||||||
zerocopy = "0.3"
|
zerocopy = "0.3"
|
||||||
byteorder = "1.3"
|
byteorder = "1.3"
|
||||||
futures = "0.3"
|
futures = "0.3"
|
||||||
|
|
||||||
|
|
||||||
# The versioning of the matrix SDK follows its Cargo.toml. The SDK and
|
|
||||||
# macros are on master, but it imports the common and base from 0.1.0.
|
|
||||||
# https://github.com/matrix-org/matrix-rust-sdk/blob/master/matrix_sdk/Cargo.toml
|
|
||||||
matrix-sdk = { git = "https://github.com/matrix-org/matrix-rust-sdk", rev = "master" }
|
matrix-sdk = { git = "https://github.com/matrix-org/matrix-rust-sdk", rev = "master" }
|
||||||
matrix-sdk-common-macros = { git = "https://github.com/matrix-org/matrix-rust-sdk", rev = "master" }
|
|
||||||
matrix-sdk-common = { git = "https://github.com/matrix-org/matrix-rust-sdk", rev = "0.1.0" }
|
|
||||||
matrix-sdk-base = { git = "https://github.com/matrix-org/matrix-rust-sdk", rev = "0.1.0", default_features = false }
|
|
||||||
|
|
||||||
[dependencies.serde]
|
[dependencies.serde]
|
||||||
version = "1"
|
version = "1"
|
||||||
|
@ -45,7 +35,7 @@ features = ['derive']
|
||||||
|
|
||||||
[dependencies.tokio]
|
[dependencies.tokio]
|
||||||
version = "0.2"
|
version = "0.2"
|
||||||
features = ["rt-core", "rt-util", "macros", "time", "signal"]
|
features = [ "full" ]
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
tempfile = "3.1"
|
tempfile = "3.1"
|
||||||
|
|
|
@ -1,25 +0,0 @@
|
||||||
use crate::config::Config;
|
|
||||||
use crate::db::Database;
|
|
||||||
use actix::prelude::*;
|
|
||||||
use state::DiceBotState;
|
|
||||||
use std::sync::Arc;
|
|
||||||
|
|
||||||
pub mod state;
|
|
||||||
|
|
||||||
pub struct Actors {
|
|
||||||
global_state: Addr<DiceBotState>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Actors {
|
|
||||||
pub fn new(config: &Arc<Config>, _db: &Database) -> Actors {
|
|
||||||
let global_state = DiceBotState::new(&config);
|
|
||||||
|
|
||||||
Actors {
|
|
||||||
global_state: global_state.start(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn global_state(&self) -> Addr<DiceBotState> {
|
|
||||||
self.global_state.clone()
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,16 +1,16 @@
|
||||||
//Needed for nested Result handling from tokio. Probably can go away after 1.47.0.
|
//Needed for nested Result handling from tokio. Probably can go away after 1.47.0.
|
||||||
#![type_length_limit = "7605144"]
|
#![type_length_limit = "7605144"]
|
||||||
use actix::prelude::*;
|
|
||||||
use chronicle_dicebot::actors::Actors;
|
|
||||||
use chronicle_dicebot::bot::DiceBot;
|
use chronicle_dicebot::bot::DiceBot;
|
||||||
use chronicle_dicebot::config::*;
|
use chronicle_dicebot::config::*;
|
||||||
use chronicle_dicebot::db::Database;
|
use chronicle_dicebot::db::Database;
|
||||||
use chronicle_dicebot::error::BotError;
|
use chronicle_dicebot::error::BotError;
|
||||||
|
use chronicle_dicebot::state::DiceBotState;
|
||||||
use env_logger::Env;
|
use env_logger::Env;
|
||||||
use log::error;
|
use log::error;
|
||||||
use std::sync::Arc;
|
use std::sync::{Arc, RwLock};
|
||||||
|
use tokio::prelude::*;
|
||||||
|
|
||||||
#[actix_rt::main]
|
#[tokio::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
env_logger::from_env(Env::default().default_filter_or("chronicle_dicebot=info,dicebot=info"))
|
env_logger::from_env(Env::default().default_filter_or("chronicle_dicebot=info,dicebot=info"))
|
||||||
.init();
|
.init();
|
||||||
|
@ -28,13 +28,12 @@ async fn run() -> Result<(), BotError> {
|
||||||
|
|
||||||
let cfg = Arc::new(read_config(config_path)?);
|
let cfg = Arc::new(read_config(config_path)?);
|
||||||
let db = Database::new(&sled::open(cfg.database_path())?);
|
let db = Database::new(&sled::open(cfg.database_path())?);
|
||||||
let actors = Actors::new(&cfg, &db);
|
let state = Arc::new(RwLock::new(DiceBotState::new(&cfg)));
|
||||||
|
|
||||||
match DiceBot::new(&cfg, actors, &db) {
|
match DiceBot::new(&cfg, &state, &db) {
|
||||||
Ok(bot) => bot.run().await?,
|
Ok(bot) => bot.run().await?,
|
||||||
Err(e) => println!("Error connecting: {:?}", e),
|
Err(e) => println!("Error connecting: {:?}", e),
|
||||||
};
|
};
|
||||||
|
|
||||||
System::current().stop();
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
31
src/bot.rs
31
src/bot.rs
|
@ -1,10 +1,10 @@
|
||||||
use crate::actors::state::LogSkippedOldMessages;
|
|
||||||
use crate::actors::Actors;
|
|
||||||
use crate::commands::execute_command;
|
use crate::commands::execute_command;
|
||||||
use crate::config::*;
|
use crate::config::*;
|
||||||
use crate::context::Context;
|
use crate::context::Context;
|
||||||
use crate::db::Database;
|
use crate::db::Database;
|
||||||
use crate::error::BotError;
|
use crate::error::BotError;
|
||||||
|
use crate::state::DiceBotState;
|
||||||
|
use async_trait::async_trait;
|
||||||
use dirs;
|
use dirs;
|
||||||
use log::{debug, error, info, trace, warn};
|
use log::{debug, error, info, trace, warn};
|
||||||
use matrix_sdk::{
|
use matrix_sdk::{
|
||||||
|
@ -16,11 +16,11 @@ use matrix_sdk::{
|
||||||
},
|
},
|
||||||
Client, ClientConfig, EventEmitter, JsonStore, SyncRoom, SyncSettings,
|
Client, ClientConfig, EventEmitter, JsonStore, SyncRoom, SyncSettings,
|
||||||
};
|
};
|
||||||
use matrix_sdk_common_macros::async_trait;
|
//use matrix_sdk_common_macros::async_trait;
|
||||||
use std::clone::Clone;
|
use std::clone::Clone;
|
||||||
use std::ops::Sub;
|
use std::ops::Sub;
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
use std::sync::Arc;
|
use std::sync::{Arc, RwLock};
|
||||||
use std::time::{Duration, SystemTime};
|
use std::time::{Duration, SystemTime};
|
||||||
use url::Url;
|
use url::Url;
|
||||||
|
|
||||||
|
@ -33,8 +33,8 @@ pub struct DiceBot {
|
||||||
/// The matrix client.
|
/// The matrix client.
|
||||||
client: Client,
|
client: Client,
|
||||||
|
|
||||||
/// Actors used by the dice bot to manage internal state.
|
/// State of the dicebot
|
||||||
actors: Actors,
|
state: Arc<RwLock<DiceBotState>>,
|
||||||
|
|
||||||
/// Active database layer
|
/// Active database layer
|
||||||
db: Database,
|
db: Database,
|
||||||
|
@ -61,11 +61,15 @@ impl DiceBot {
|
||||||
/// actor. This function returns a Result because it is possible
|
/// actor. This function returns a Result because it is possible
|
||||||
/// for client creation to fail for some reason (e.g. invalid
|
/// for client creation to fail for some reason (e.g. invalid
|
||||||
/// homeserver URL).
|
/// homeserver URL).
|
||||||
pub fn new(config: &Arc<Config>, actors: Actors, db: &Database) -> Result<Self, BotError> {
|
pub fn new(
|
||||||
|
config: &Arc<Config>,
|
||||||
|
state: &Arc<RwLock<DiceBotState>>,
|
||||||
|
db: &Database,
|
||||||
|
) -> Result<Self, BotError> {
|
||||||
Ok(DiceBot {
|
Ok(DiceBot {
|
||||||
client: create_client(&config)?,
|
client: create_client(&config)?,
|
||||||
config: config.clone(),
|
config: config.clone(),
|
||||||
actors: actors,
|
state: state.clone(),
|
||||||
db: db.clone(),
|
db: db.clone(),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
@ -146,11 +150,12 @@ async fn should_process<'a>(
|
||||||
) -> Result<(String, String), BotError> {
|
) -> Result<(String, String), BotError> {
|
||||||
//Ignore messages that are older than configured duration.
|
//Ignore messages that are older than configured duration.
|
||||||
if !check_message_age(event, bot.config.oldest_message_age()) {
|
if !check_message_age(event, bot.config.oldest_message_age()) {
|
||||||
let res = bot.actors.global_state().send(LogSkippedOldMessages).await;
|
let state_check = bot.state.read().unwrap();
|
||||||
|
if !((*state_check).logged_skipped_old_messages()) {
|
||||||
if let Err(e) = res {
|
drop(state_check);
|
||||||
error!("Actix error: {:?}", e);
|
let mut state = bot.state.write().unwrap();
|
||||||
};
|
(*state).skipped_old_messages();
|
||||||
|
}
|
||||||
|
|
||||||
return Err(BotError::ShouldNotProcessError);
|
return Err(BotError::ShouldNotProcessError);
|
||||||
}
|
}
|
||||||
|
|
|
@ -37,9 +37,6 @@ pub enum BotError {
|
||||||
#[error("future canceled")]
|
#[error("future canceled")]
|
||||||
FutureCanceledError,
|
FutureCanceledError,
|
||||||
|
|
||||||
#[error("tokio task join error")]
|
|
||||||
TokioTaskJoinError(#[from] tokio::task::JoinError),
|
|
||||||
|
|
||||||
//de = deserialization
|
//de = deserialization
|
||||||
#[error("toml parsing error")]
|
#[error("toml parsing error")]
|
||||||
TomlParsingError(#[from] toml::de::Error),
|
TomlParsingError(#[from] toml::de::Error),
|
||||||
|
@ -47,9 +44,6 @@ pub enum BotError {
|
||||||
#[error("i/o error: {0}")]
|
#[error("i/o error: {0}")]
|
||||||
IoError(#[from] std::io::Error),
|
IoError(#[from] std::io::Error),
|
||||||
|
|
||||||
#[error("actor mailbox error: {0}")]
|
|
||||||
ActixMailboxError(#[from] actix::MailboxError),
|
|
||||||
|
|
||||||
#[error("parsing error")]
|
#[error("parsing error")]
|
||||||
ParserError(#[from] combine::error::StringStreamError),
|
ParserError(#[from] combine::error::StringStreamError),
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,3 @@
|
||||||
pub mod actors;
|
|
||||||
pub mod bot;
|
pub mod bot;
|
||||||
pub mod cofd;
|
pub mod cofd;
|
||||||
pub mod commands;
|
pub mod commands;
|
||||||
|
@ -10,4 +9,5 @@ pub mod error;
|
||||||
mod help;
|
mod help;
|
||||||
mod parser;
|
mod parser;
|
||||||
pub mod roll;
|
pub mod roll;
|
||||||
|
pub mod state;
|
||||||
pub mod variables;
|
pub mod variables;
|
||||||
|
|
|
@ -1,30 +1,14 @@
|
||||||
use crate::config::*;
|
use crate::config::*;
|
||||||
use actix::prelude::*;
|
|
||||||
use log::info;
|
use log::info;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
#[derive(Message)]
|
|
||||||
#[rtype(result = "bool")]
|
|
||||||
pub struct LogSkippedOldMessages;
|
|
||||||
|
|
||||||
/// Holds state of the dice bot, for anything requiring mutable
|
/// Holds state of the dice bot, for anything requiring mutable
|
||||||
/// transitions. This is a simple mutable trait whose values represent
|
/// transitions. This is a simple mutable trait whose values represent
|
||||||
/// the current state of the dicebot. It provides mutable methods to
|
/// the current state of the dicebot. It provides mutable methods to
|
||||||
/// change state.
|
/// change state.
|
||||||
pub struct DiceBotState {
|
pub struct DiceBotState {
|
||||||
logged_skipped_old_messages: bool,
|
logged_skipped_old_messages: bool,
|
||||||
config: Arc<Config>,
|
_config: Arc<Config>,
|
||||||
}
|
|
||||||
|
|
||||||
impl Actor for DiceBotState {
|
|
||||||
type Context = Context<Self>;
|
|
||||||
|
|
||||||
fn started(&mut self, _ctx: &mut Self::Context) {
|
|
||||||
info!(
|
|
||||||
"Oldest allowable message time is {} seconds ago",
|
|
||||||
&self.config.oldest_message_age()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl DiceBotState {
|
impl DiceBotState {
|
||||||
|
@ -32,10 +16,14 @@ impl DiceBotState {
|
||||||
pub fn new(config: &Arc<Config>) -> DiceBotState {
|
pub fn new(config: &Arc<Config>) -> DiceBotState {
|
||||||
DiceBotState {
|
DiceBotState {
|
||||||
logged_skipped_old_messages: false,
|
logged_skipped_old_messages: false,
|
||||||
config: config.clone(),
|
_config: config.clone(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn logged_skipped_old_messages(&self) -> bool {
|
||||||
|
self.logged_skipped_old_messages
|
||||||
|
}
|
||||||
|
|
||||||
/// Log and record that we have skipped some old messages. This
|
/// Log and record that we have skipped some old messages. This
|
||||||
/// method will log once, and then no-op from that point on.
|
/// method will log once, and then no-op from that point on.
|
||||||
pub fn skipped_old_messages(&mut self) {
|
pub fn skipped_old_messages(&mut self) {
|
||||||
|
@ -46,12 +34,3 @@ impl DiceBotState {
|
||||||
self.logged_skipped_old_messages = true;
|
self.logged_skipped_old_messages = true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Handler<LogSkippedOldMessages> for DiceBotState {
|
|
||||||
type Result = bool;
|
|
||||||
|
|
||||||
fn handle(&mut self, _msg: LogSkippedOldMessages, _ctx: &mut Context<Self>) -> Self::Result {
|
|
||||||
self.skipped_old_messages();
|
|
||||||
self.logged_skipped_old_messages
|
|
||||||
}
|
|
||||||
}
|
|
Loading…
Reference in New Issue