use super::ReduceMiddlewareResult;
use crate::{
middleware::{Middleware, ReduceFn},
Store,
};
use std::{fmt::Debug, hash::Hash};
pub enum LogLevel {
Trace,
Debug,
Warn,
Info,
}
impl LogLevel {
pub fn log<S: AsRef<str>>(&self, message: S) {
match self {
LogLevel::Trace => log::trace!("{}", message.as_ref()),
LogLevel::Debug => log::debug!("{}", message.as_ref()),
LogLevel::Warn => log::warn!("{}", message.as_ref()),
LogLevel::Info => log::info!("{}", message.as_ref()),
}
}
}
impl Default for LogLevel {
fn default() -> Self {
LogLevel::Debug
}
}
pub struct SimpleLoggerMiddleware {
log_level: LogLevel,
}
impl SimpleLoggerMiddleware {
pub fn new() -> Self {
SimpleLoggerMiddleware {
log_level: LogLevel::default(),
}
}
pub fn log_level(mut self, log_level: LogLevel) -> Self {
self.log_level = log_level;
self
}
}
impl Default for SimpleLoggerMiddleware {
fn default() -> Self {
SimpleLoggerMiddleware::new()
}
}
impl<State, Action, Event, Effect> Middleware<State, Action, Event, Effect>
for SimpleLoggerMiddleware
where
Event: Clone + Hash + Eq + Debug,
State: Debug,
Action: Debug,
Effect: Debug,
{
fn on_reduce(
&self,
store: &Store<State, Action, Event, Effect>,
action: Option<&Action>,
reduce: ReduceFn<State, Action, Event, Effect>,
) -> ReduceMiddlewareResult<Event, Effect> {
let was_action = match &action {
Some(action) => {
self.log_level
.log(format!("prev state: {:?}", store.state()));
self.log_level.log(format!("action: {:?}", action));
true
}
None => {
self.log_level.log("action: None");
false
}
};
let events = reduce(store, action);
if was_action {
self.log_level
.log(format!("next state: {:?}", store.state()));
}
events
}
fn process_effect(
&self,
_store: &Store<State, Action, Event, Effect>,
effect: Effect,
) -> Option<Effect> {
self.log_level.log(format!("effect: {:?}", effect));
Some(effect)
}
fn on_notify(
&self,
store: &Store<State, Action, Event, Effect>,
events: Vec<Event>,
notify: super::NotifyFn<State, Action, Event, Effect>,
) -> Vec<Event> {
for event in &events {
self.log_level.log(format!("event: {:?}", event));
}
notify(store, events)
}
}