tui_dispatch_core/

store.rs

//! Centralized state store with reducer pattern

use crate::Action;
use std::marker::PhantomData;

/// A reducer function that handles actions and mutates state
///
/// Returns `true` if the state changed and a re-render is needed.
pub type Reducer<S, A> = fn(&mut S, A) -> bool;

/// Default maximum nested middleware dispatch depth.
pub(crate) const DEFAULT_MAX_DISPATCH_DEPTH: usize = 16;
/// Default maximum number of actions processed per top-level dispatch.
pub(crate) const DEFAULT_MAX_DISPATCH_ACTIONS: usize = 10_000;

/// Configurable limits for middleware dispatch.
///
/// Both limits should be at least `1` for dispatch to make progress.
/// If either value is `0`, all dispatch attempts fail with [`DispatchError`].
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct DispatchLimits {
    /// Maximum nested dispatch depth from middleware injection.
    pub max_depth: usize,
    /// Maximum attempted actions during a single top-level dispatch.
    ///
    /// This includes actions cancelled by `Middleware::before`.
    pub max_actions: usize,
}

impl Default for DispatchLimits {
    fn default() -> Self {
        Self {
            max_depth: DEFAULT_MAX_DISPATCH_DEPTH,
            max_actions: DEFAULT_MAX_DISPATCH_ACTIONS,
        }
    }
}

/// Error returned when middleware dispatch exceeds configured limits.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum DispatchError {
    /// Nested middleware injection exceeded `DispatchLimits::max_depth`.
    DepthExceeded {
        max_depth: usize,
        action: &'static str,
    },
    /// Processed actions exceeded `DispatchLimits::max_actions`.
    ActionBudgetExceeded {
        max_actions: usize,
        processed: usize,
        action: &'static str,
    },
}

impl std::fmt::Display for DispatchError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            DispatchError::DepthExceeded { max_depth, action } => write!(
                f,
                "middleware dispatch depth limit exceeded (max_depth={max_depth}, action={action})"
            ),
            DispatchError::ActionBudgetExceeded {
                max_actions,
                processed,
                action,
            } => write!(
                f,
                "middleware dispatch action budget exceeded (max_actions={max_actions}, processed={processed}, action={action})"
            ),
        }
    }
}

impl std::error::Error for DispatchError {}

pub(crate) fn check_dispatch_limits(
    limits: DispatchLimits,
    dispatch_depth: usize,
    processed: usize,
    action: &'static str,
) -> Result<(), DispatchError> {
    if dispatch_depth >= limits.max_depth {
        return Err(DispatchError::DepthExceeded {
            max_depth: limits.max_depth,
            action,
        });
    }

    if processed >= limits.max_actions {
        return Err(DispatchError::ActionBudgetExceeded {
            max_actions: limits.max_actions,
            processed,
            action,
        });
    }

    Ok(())
}

#[inline]
pub(crate) fn debug_assert_valid_dispatch_limits(limits: DispatchLimits) {
    debug_assert!(
        limits.max_depth >= 1 && limits.max_actions >= 1,
        "DispatchLimits requires max_depth >= 1 and max_actions >= 1"
    );
}

pub(crate) trait MiddlewareDispatchDriver<A: Action> {
    type Output;

    fn before(&mut self, action: &A) -> bool;
    fn reduce(&mut self, action: A) -> Self::Output;
    /// Called only when the root action is cancelled in `before()`.
    ///
    /// Cancelled non-root actions are discarded because they merge as identity
    /// values into their parent results.
    fn cancelled_output(&mut self) -> Self::Output;
    fn after(&mut self, action: &A, result: &Self::Output) -> Vec<A>;
    fn merge_child(&mut self, parent: &mut Self::Output, child: Self::Output);
}

enum DispatchFrame<A: Action, O> {
    Pending(A),
    Entered {
        result: O,
        injected: std::vec::IntoIter<A>,
    },
}

pub(crate) fn run_iterative_middleware_dispatch<A, D>(
    limits: DispatchLimits,
    action: A,
    driver: &mut D,
) -> Result<D::Output, DispatchError>
where
    A: Action,
    D: MiddlewareDispatchDriver<A>,
{
    let mut processed = 0usize;
    let mut stack = vec![DispatchFrame::<A, D::Output>::Pending(action)];

    while let Some(frame) = stack.pop() {
        match frame {
            DispatchFrame::Pending(action) => {
                let depth = stack.len();
                check_dispatch_limits(limits, depth, processed, action.name())?;
                processed += 1;

                if !driver.before(&action) {
                    if !stack.is_empty() {
                        continue;
                    }
                    return Ok(driver.cancelled_output());
                }

                let result = driver.reduce(action.clone());
                let injected = driver.after(&action, &result).into_iter();
                stack.push(DispatchFrame::Entered { result, injected });
            }
            DispatchFrame::Entered {
                result,
                mut injected,
            } => {
                if let Some(injected_action) = injected.next() {
                    stack.push(DispatchFrame::Entered { result, injected });
                    stack.push(DispatchFrame::Pending(injected_action));
                    continue;
                }

                if let Some(DispatchFrame::Entered {
                    result: parent_result,
                    ..
                }) = stack.last_mut()
                {
                    driver.merge_child(parent_result, result);
                    continue;
                }

                return Ok(result);
            }
        }
    }

    unreachable!("dispatch stack should not drain before a root result is returned")
}

/// Compose a reducer by routing actions to focused handlers.
///
/// # When to Use
///
/// For most reducers, a flat `match` is simpler and clearer. Use this macro when:
/// - Your reducer exceeds ~500 lines and splitting improves organization
/// - You have **context-aware routing** (e.g., vim normal vs command mode)
/// - Handlers live in **separate modules** and you want clean composition
///
/// # Syntax
///
/// ```text
/// reducer_compose!(state, action, {
///     // Arms are tried in order, first match wins
///     category "name" => handler,      // Route by action category
///     Action::Specific => handler,     // Route by pattern match
///     _ => fallback_handler,           // Catch-all (required last)
/// })
///
/// // With context (e.g., for modal/mode-aware routing):
/// reducer_compose!(state, action, context, {
///     context Mode::Command => handle_command,  // Route by context value
///     category "nav" => handle_nav,
///     _ => handle_default,
/// })
/// ```
///
/// # Arm Types
///
/// **`category "name" => handler`** - Routes actions where
/// `ActionCategory::category(&action) == Some("name")`. Requires
/// `#[action(infer_categories)]` on your action enum.
///
/// **`context Value => handler`** - Routes when the context expression equals
/// `Value`. Only available in the 4-argument form.
///
/// **`Pattern => handler`** - Standard pattern match (e.g., `Action::Quit`,
/// `Action::Input(_)`).
///
/// **`_ => handler`** - Catch-all fallback. Must be last.
///
/// # Handler Signature
///
/// All handlers must have the same signature:
/// ```text
/// fn handler(state: &mut S, action: A) -> R
/// ```
/// Where `R` is typically `bool` or `ReducerResult<E>`.
///
/// # Category Inference
///
/// With `#[action(infer_categories)]`, categories are inferred from action
/// names by taking the prefix before the verb:
///
/// | Action | Verb | Category |
/// |--------|------|----------|
/// | `NavScrollUp` | Scroll | `"nav"` |
/// | `SearchQuerySubmit` | Submit | `"search_query"` |
/// | `WeatherDidLoad` | Did | `"weather"` |
/// | `Quit` | (none) | `None` |
///
/// For predictable categories, use explicit `#[category = "name"]` attributes.
///
/// # Example
///
/// ```ignore
/// fn reducer(state: &mut AppState, action: Action, mode: Mode) -> bool {
///     reducer_compose!(state, action, mode, {
///         // Command mode gets priority
///         context Mode::Command => handle_command,
///         // Then route by category
///         category "nav" => handle_navigation,
///         category "search" => handle_search,
///         // Specific actions
///         Action::Quit => |_, _| false,
///         // Everything else
///         _ => handle_ui,
///     })
/// }
///
/// fn handle_navigation(state: &mut AppState, action: Action) -> bool {
///     match action {
///         Action::NavUp => { state.cursor -= 1; true }
///         Action::NavDown => { state.cursor += 1; true }
///         _ => false,
///     }
/// }
/// ```
#[macro_export]
macro_rules! reducer_compose {
    // 3-argument form must come first to prevent $context:expr from matching the braces
    ($state:expr, $action:expr, { $($arms:tt)+ }) => {{
        let __state = $state;
        let __action_input = $action;
        let __context = ();
        $crate::reducer_compose!(@accum __state, __action_input, __context; () $($arms)+)
    }};
    ($state:expr, $action:expr, $context:expr, { $($arms:tt)+ }) => {{
        let __state = $state;
        let __action_input = $action;
        let __context = $context;
        $crate::reducer_compose!(@accum __state, __action_input, __context; () $($arms)+)
    }};
    (@accum $state:ident, $action:ident, $context:ident; ($($out:tt)*) category $category:expr => $handler:expr, $($rest:tt)+) => {
        $crate::reducer_compose!(
            @accum $state, $action, $context;
            (
                $($out)*
                __action if $crate::ActionCategory::category(&__action) == Some($category) => {
                    ($handler)($state, __action)
                },
            )
            $($rest)+
        )
    };
    (@accum $state:ident, $action:ident, $context:ident; ($($out:tt)*) context $context_value:expr => $handler:expr, $($rest:tt)+) => {
        $crate::reducer_compose!(
            @accum $state, $action, $context;
            (
                $($out)*
                __action if $context == $context_value => {
                    ($handler)($state, __action)
                },
            )
            $($rest)+
        )
    };
    (@accum $state:ident, $action:ident, $context:ident; ($($out:tt)*) _ => $handler:expr, $($rest:tt)+) => {
        $crate::reducer_compose!(
            @accum $state, $action, $context;
            (
                $($out)*
                __action => {
                    ($handler)($state, __action)
                },
            )
            $($rest)+
        )
    };
    (@accum $state:ident, $action:ident, $context:ident; ($($out:tt)*) $pattern:pat $(if $guard:expr)? => $handler:expr, $($rest:tt)+) => {
        $crate::reducer_compose!(
            @accum $state, $action, $context;
            (
                $($out)*
                __action @ $pattern $(if $guard)? => {
                    ($handler)($state, __action)
                },
            )
            $($rest)+
        )
    };
    (@accum $state:ident, $action:ident, $context:ident; ($($out:tt)*) category $category:expr => $handler:expr $(,)?) => {
        match $action {
            $($out)*
            __action if $crate::ActionCategory::category(&__action) == Some($category) => {
                ($handler)($state, __action)
            }
        }
    };
    (@accum $state:ident, $action:ident, $context:ident; ($($out:tt)*) context $context_value:expr => $handler:expr $(,)?) => {
        match $action {
            $($out)*
            __action if $context == $context_value => {
                ($handler)($state, __action)
            }
        }
    };
    (@accum $state:ident, $action:ident, $context:ident; ($($out:tt)*) _ => $handler:expr $(,)?) => {
        match $action {
            $($out)*
            __action => {
                ($handler)($state, __action)
            }
        }
    };
    (@accum $state:ident, $action:ident, $context:ident; ($($out:tt)*) $pattern:pat $(if $guard:expr)? => $handler:expr $(,)?) => {
        match $action {
            $($out)*
            __action @ $pattern $(if $guard)? => {
                ($handler)($state, __action)
            }
        }
    };
}

/// Centralized state store with Redux-like reducer pattern
///
/// The store holds the application state and provides a single point
/// for state mutations through the `dispatch` method.
///
/// # Type Parameters
/// * `S` - The application state type
/// * `A` - The action type (must implement `Action`)
///
/// # Example
/// ```
/// use tui_dispatch_core::{Action, Store};
///
/// #[derive(Clone, Debug)]
/// enum MyAction { Increment, Decrement }
///
/// impl Action for MyAction {
///     fn name(&self) -> &'static str {
///         match self {
///             MyAction::Increment => "Increment",
///             MyAction::Decrement => "Decrement",
///         }
///     }
/// }
///
/// #[derive(Default)]
/// struct AppState { counter: i32 }
///
/// fn reducer(state: &mut AppState, action: MyAction) -> bool {
///     match action {
///         MyAction::Increment => { state.counter += 1; true }
///         MyAction::Decrement => { state.counter -= 1; true }
///     }
/// }
///
/// let mut store = Store::new(AppState::default(), reducer);
/// store.dispatch(MyAction::Increment);
/// assert_eq!(store.state().counter, 1);
/// ```
pub struct Store<S, A: Action> {
    state: S,
    reducer: Reducer<S, A>,
    _marker: PhantomData<A>,
}

impl<S, A: Action> Store<S, A> {
    /// Create a new store with initial state and reducer
    pub fn new(state: S, reducer: Reducer<S, A>) -> Self {
        Self {
            state,
            reducer,
            _marker: PhantomData,
        }
    }

    /// Dispatch an action to the store
    ///
    /// The reducer will be called with the current state and action.
    /// Returns `true` if the state changed and a re-render is needed.
    pub fn dispatch(&mut self, action: A) -> bool {
        (self.reducer)(&mut self.state, action)
    }

    /// Get a reference to the current state
    pub fn state(&self) -> &S {
        &self.state
    }

    /// Get a mutable reference to the state
    ///
    /// Use this sparingly - prefer dispatching actions for state changes.
    /// This is useful for initializing state or for cases where the
    /// action pattern doesn't fit well.
    pub fn state_mut(&mut self) -> &mut S {
        &mut self.state
    }
}

/// Store with middleware support
///
/// Wraps a `Store` and allows middleware to intercept actions
/// before and after they are processed by the reducer.
pub struct StoreWithMiddleware<S, A: Action, M: Middleware<S, A>> {
    store: Store<S, A>,
    middleware: M,
    dispatch_limits: DispatchLimits,
}

impl<S, A: Action, M: Middleware<S, A>> StoreWithMiddleware<S, A, M> {
    /// Create a new store with middleware
    pub fn new(state: S, reducer: Reducer<S, A>, middleware: M) -> Self {
        Self {
            store: Store::new(state, reducer),
            middleware,
            dispatch_limits: DispatchLimits::default(),
        }
    }

    /// Override middleware dispatch limits.
    pub fn with_dispatch_limits(mut self, limits: DispatchLimits) -> Self {
        debug_assert_valid_dispatch_limits(limits);
        self.dispatch_limits = limits;
        self
    }

    /// Current middleware dispatch limits.
    pub fn dispatch_limits(&self) -> DispatchLimits {
        self.dispatch_limits
    }

    /// Dispatch an action through middleware and store
    ///
    /// This wraps [`Self::try_dispatch`] and panics if dispatch limits are exceeded.
    /// Use `try_dispatch` to handle overflow as a typed error.
    pub fn dispatch(&mut self, action: A) -> bool {
        self.try_dispatch(action)
            .unwrap_or_else(|error| panic!("middleware dispatch failed: {error}"))
    }

    /// Dispatch an action through middleware and store.
    ///
    /// The action passes through `middleware.before()` (which can cancel it),
    /// then the reducer, then `middleware.after()` (which can inject follow-up actions).
    /// Injected actions go through the full pipeline in depth-first order.
    ///
    /// Returns [`DispatchError`] if configured depth or action budget limits are exceeded.
    ///
    /// This operation is not transactional: if overflow happens in an injected chain,
    /// earlier actions in the chain may have already mutated state.
    ///
    /// Action budget accounting includes attempted dispatches that are later cancelled by
    /// `Middleware::before`.
    pub fn try_dispatch(&mut self, action: A) -> Result<bool, DispatchError> {
        let mut driver = StoreDispatchDriver {
            store: &mut self.store,
            middleware: &mut self.middleware,
        };
        run_iterative_middleware_dispatch(self.dispatch_limits, action, &mut driver)
    }

    /// Get a reference to the current state
    pub fn state(&self) -> &S {
        self.store.state()
    }

    /// Get a mutable reference to the state
    pub fn state_mut(&mut self) -> &mut S {
        self.store.state_mut()
    }

    /// Get a reference to the middleware
    pub fn middleware(&self) -> &M {
        &self.middleware
    }

    /// Get a mutable reference to the middleware
    pub fn middleware_mut(&mut self) -> &mut M {
        &mut self.middleware
    }
}

struct StoreDispatchDriver<'a, S, A: Action, M: Middleware<S, A>> {
    store: &'a mut Store<S, A>,
    middleware: &'a mut M,
}

impl<S, A: Action, M: Middleware<S, A>> MiddlewareDispatchDriver<A>
    for StoreDispatchDriver<'_, S, A, M>
{
    type Output = bool;

    fn before(&mut self, action: &A) -> bool {
        self.middleware.before(action, &self.store.state)
    }

    fn reduce(&mut self, action: A) -> Self::Output {
        self.store.dispatch(action)
    }

    fn cancelled_output(&mut self) -> Self::Output {
        false
    }

    fn after(&mut self, action: &A, result: &Self::Output) -> Vec<A> {
        self.middleware.after(action, *result, &self.store.state)
    }

    fn merge_child(&mut self, parent: &mut Self::Output, child: Self::Output) {
        *parent |= child;
    }
}

/// Middleware trait for intercepting actions
///
/// Implement this trait to add logging, persistence, throttling, or other
/// cross-cutting concerns to your store. Middleware can:
///
/// - **Observe**: inspect actions and state (logging, analytics, persistence)
/// - **Cancel**: return `false` from `before()` to prevent the action from reaching the reducer
/// - **Inject**: return follow-up actions from `after()` that are dispatched through the full pipeline
///
/// # Cancel
///
/// Return `false` from `before()` to cancel the action — the reducer is never called and
/// `after()` is not invoked. Useful for throttling, validation, and auth guards.
///
/// # Inject
///
/// Return actions from `after()` to trigger follow-up dispatches. Injected actions go through
/// the full middleware + reducer pipeline. Dispatch limits prevent runaway loops.
///
/// Useful for cascading behavior: moving a card to "Done" triggers a notification,
/// without the move reducer knowing about notifications.
pub trait Middleware<S, A: Action> {
    /// Called before the action is dispatched to the reducer.
    ///
    /// Return `true` to proceed with dispatch, `false` to cancel.
    fn before(&mut self, action: &A, state: &S) -> bool;

    /// Called after the action is processed by the reducer.
    ///
    /// Return any follow-up actions to dispatch through the full pipeline.
    fn after(&mut self, action: &A, state_changed: bool, state: &S) -> Vec<A>;
}

/// A no-op middleware that does nothing
#[derive(Debug, Clone, Copy, Default)]
pub struct NoopMiddleware;

impl<S, A: Action> Middleware<S, A> for NoopMiddleware {
    fn before(&mut self, _action: &A, _state: &S) -> bool {
        true
    }
    fn after(&mut self, _action: &A, _state_changed: bool, _state: &S) -> Vec<A> {
        vec![]
    }
}

/// Middleware that logs actions via the `tracing` crate.
///
/// Requires the `tracing` feature.
#[cfg(feature = "tracing")]
#[derive(Debug, Clone, Default)]
pub struct LoggingMiddleware {
    /// Whether to log before dispatch
    pub log_before: bool,
    /// Whether to log after dispatch
    pub log_after: bool,
}

#[cfg(feature = "tracing")]
impl LoggingMiddleware {
    /// Create a new logging middleware with default settings (log after only)
    pub fn new() -> Self {
        Self {
            log_before: false,
            log_after: true,
        }
    }

    /// Create a logging middleware that logs both before and after
    pub fn verbose() -> Self {
        Self {
            log_before: true,
            log_after: true,
        }
    }
}

#[cfg(feature = "tracing")]
impl<S, A: Action> Middleware<S, A> for LoggingMiddleware {
    fn before(&mut self, action: &A, _state: &S) -> bool {
        if self.log_before {
            tracing::debug!(action = %action.name(), "Dispatching action");
        }
        true
    }

    fn after(&mut self, action: &A, state_changed: bool, _state: &S) -> Vec<A> {
        if self.log_after {
            tracing::debug!(
                action = %action.name(),
                state_changed = state_changed,
                "Action processed"
            );
        }
        vec![]
    }
}

/// Compose multiple middleware into a single middleware
pub struct ComposedMiddleware<S, A: Action> {
    middlewares: Vec<Box<dyn Middleware<S, A>>>,
}

impl<S, A: Action> std::fmt::Debug for ComposedMiddleware<S, A> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("ComposedMiddleware")
            .field("middlewares_count", &self.middlewares.len())
            .finish()
    }
}

impl<S, A: Action> Default for ComposedMiddleware<S, A> {
    fn default() -> Self {
        Self::new()
    }
}

impl<S, A: Action> ComposedMiddleware<S, A> {
    /// Create a new composed middleware
    pub fn new() -> Self {
        Self {
            middlewares: Vec::new(),
        }
    }

    /// Add a middleware to the composition
    pub fn add<M: Middleware<S, A> + 'static>(&mut self, middleware: M) {
        self.middlewares.push(Box::new(middleware));
    }
}

impl<S, A: Action> Middleware<S, A> for ComposedMiddleware<S, A> {
    fn before(&mut self, action: &A, state: &S) -> bool {
        for middleware in &mut self.middlewares {
            if !middleware.before(action, state) {
                return false;
            }
        }
        true
    }

    fn after(&mut self, action: &A, state_changed: bool, state: &S) -> Vec<A> {
        let mut injected = Vec::new();
        // Call in reverse order for proper nesting
        for middleware in self.middlewares.iter_mut().rev() {
            injected.extend(middleware.after(action, state_changed, state));
        }
        injected
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::ActionCategory;

    #[derive(Default)]
    struct TestState {
        counter: i32,
    }

    #[derive(Clone, Debug)]
    enum TestAction {
        Increment,
        Decrement,
        NoOp,
    }

    impl Action for TestAction {
        fn name(&self) -> &'static str {
            match self {
                TestAction::Increment => "Increment",
                TestAction::Decrement => "Decrement",
                TestAction::NoOp => "NoOp",
            }
        }
    }

    fn test_reducer(state: &mut TestState, action: TestAction) -> bool {
        match action {
            TestAction::Increment => {
                state.counter += 1;
                true
            }
            TestAction::Decrement => {
                state.counter -= 1;
                true
            }
            TestAction::NoOp => false,
        }
    }

    #[test]
    fn test_store_dispatch() {
        let mut store = Store::new(TestState::default(), test_reducer);

        assert!(store.dispatch(TestAction::Increment));
        assert_eq!(store.state().counter, 1);

        assert!(store.dispatch(TestAction::Increment));
        assert_eq!(store.state().counter, 2);

        assert!(store.dispatch(TestAction::Decrement));
        assert_eq!(store.state().counter, 1);
    }

    #[test]
    fn test_store_noop() {
        let mut store = Store::new(TestState::default(), test_reducer);

        assert!(!store.dispatch(TestAction::NoOp));
        assert_eq!(store.state().counter, 0);
    }

    #[test]
    fn test_store_state_mut() {
        let mut store = Store::new(TestState::default(), test_reducer);

        store.state_mut().counter = 100;
        assert_eq!(store.state().counter, 100);
    }

    #[derive(Default)]
    struct CountingMiddleware {
        before_count: usize,
        after_count: usize,
    }

    impl<S, A: Action> Middleware<S, A> for CountingMiddleware {
        fn before(&mut self, _action: &A, _state: &S) -> bool {
            self.before_count += 1;
            true
        }

        fn after(&mut self, _action: &A, _state_changed: bool, _state: &S) -> Vec<A> {
            self.after_count += 1;
            vec![]
        }
    }

    #[test]
    fn test_store_with_middleware() {
        let mut store = StoreWithMiddleware::new(
            TestState::default(),
            test_reducer,
            CountingMiddleware::default(),
        );

        store.dispatch(TestAction::Increment);
        store.dispatch(TestAction::Increment);

        assert_eq!(store.middleware().before_count, 2);
        assert_eq!(store.middleware().after_count, 2);
        assert_eq!(store.state().counter, 2);
    }

    struct SelfInjectingMiddleware;

    impl Middleware<TestState, TestAction> for SelfInjectingMiddleware {
        fn before(&mut self, _action: &TestAction, _state: &TestState) -> bool {
            true
        }

        fn after(
            &mut self,
            action: &TestAction,
            _state_changed: bool,
            _state: &TestState,
        ) -> Vec<TestAction> {
            vec![action.clone()]
        }
    }

    #[test]
    fn test_try_dispatch_depth_exceeded() {
        let mut store =
            StoreWithMiddleware::new(TestState::default(), test_reducer, SelfInjectingMiddleware)
                .with_dispatch_limits(DispatchLimits {
                    max_depth: 2,
                    max_actions: 100,
                });

        let err = store.try_dispatch(TestAction::Increment).unwrap_err();
        assert_eq!(
            err,
            DispatchError::DepthExceeded {
                max_depth: 2,
                action: "Increment",
            }
        );
        assert_eq!(store.state().counter, 2);
    }

    #[test]
    fn test_try_dispatch_action_budget_exceeded() {
        let mut store =
            StoreWithMiddleware::new(TestState::default(), test_reducer, SelfInjectingMiddleware)
                .with_dispatch_limits(DispatchLimits {
                    max_depth: 32,
                    max_actions: 2,
                });

        let err = store.try_dispatch(TestAction::Increment).unwrap_err();
        assert_eq!(
            err,
            DispatchError::ActionBudgetExceeded {
                max_actions: 2,
                processed: 2,
                action: "Increment",
            }
        );
        assert_eq!(store.state().counter, 2);
    }

    struct FiniteCascadeMiddleware {
        target: i32,
    }

    impl Middleware<TestState, TestAction> for FiniteCascadeMiddleware {
        fn before(&mut self, _action: &TestAction, _state: &TestState) -> bool {
            true
        }

        fn after(
            &mut self,
            action: &TestAction,
            _state_changed: bool,
            state: &TestState,
        ) -> Vec<TestAction> {
            if matches!(action, TestAction::Increment) && state.counter < self.target {
                vec![TestAction::Increment]
            } else {
                vec![]
            }
        }
    }

    #[test]
    fn test_try_dispatch_deep_finite_chain_succeeds() {
        let target = 512usize;
        let mut store = StoreWithMiddleware::new(
            TestState::default(),
            test_reducer,
            FiniteCascadeMiddleware {
                target: target as i32,
            },
        )
        .with_dispatch_limits(DispatchLimits {
            max_depth: target + 1,
            max_actions: target + 1,
        });

        let changed = store.try_dispatch(TestAction::Increment).unwrap();
        assert!(changed);
        assert_eq!(store.state().counter, target as i32);
    }

    #[derive(Default)]
    struct OrderingState {
        order: Vec<&'static str>,
    }

    #[derive(Clone, Debug)]
    enum OrderingAction {
        Root,
        Left,
        Right,
        Leaf,
    }

    impl Action for OrderingAction {
        fn name(&self) -> &'static str {
            match self {
                OrderingAction::Root => "Root",
                OrderingAction::Left => "Left",
                OrderingAction::Right => "Right",
                OrderingAction::Leaf => "Leaf",
            }
        }
    }

    fn ordering_reducer(state: &mut OrderingState, action: OrderingAction) -> bool {
        state.order.push(action.name());
        true
    }

    struct OrderingMiddleware;

    impl Middleware<OrderingState, OrderingAction> for OrderingMiddleware {
        fn before(&mut self, _action: &OrderingAction, _state: &OrderingState) -> bool {
            true
        }

        fn after(
            &mut self,
            action: &OrderingAction,
            _state_changed: bool,
            _state: &OrderingState,
        ) -> Vec<OrderingAction> {
            match action {
                OrderingAction::Root => vec![OrderingAction::Left, OrderingAction::Right],
                OrderingAction::Left => vec![OrderingAction::Leaf],
                OrderingAction::Right | OrderingAction::Leaf => vec![],
            }
        }
    }

    #[test]
    fn test_try_dispatch_injection_order_is_depth_first() {
        let mut store = StoreWithMiddleware::new(
            OrderingState::default(),
            ordering_reducer,
            OrderingMiddleware,
        )
        .with_dispatch_limits(DispatchLimits {
            max_depth: 8,
            max_actions: 8,
        });

        let changed = store.try_dispatch(OrderingAction::Root).unwrap();
        assert!(changed);
        assert_eq!(store.state().order, vec!["Root", "Left", "Leaf", "Right"]);
    }

    #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
    enum ComposeContext {
        Default,
        Command,
    }

    #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
    enum ComposeCategory {
        Nav,
        Search,
        Uncategorized,
    }

    #[derive(Clone, Debug)]
    enum ComposeAction {
        NavUp,
        Search,
        Other,
    }

    impl Action for ComposeAction {
        fn name(&self) -> &'static str {
            match self {
                ComposeAction::NavUp => "NavUp",
                ComposeAction::Search => "Search",
                ComposeAction::Other => "Other",
            }
        }
    }

    impl ActionCategory for ComposeAction {
        type Category = ComposeCategory;

        fn category(&self) -> Option<&'static str> {
            match self {
                ComposeAction::NavUp => Some("nav"),
                ComposeAction::Search => Some("search"),
                ComposeAction::Other => None,
            }
        }

        fn category_enum(&self) -> Self::Category {
            match self {
                ComposeAction::NavUp => ComposeCategory::Nav,
                ComposeAction::Search => ComposeCategory::Search,
                ComposeAction::Other => ComposeCategory::Uncategorized,
            }
        }
    }

    fn handle_nav(state: &mut usize, _action: ComposeAction) -> &'static str {
        *state += 1;
        "nav"
    }

    fn handle_command(state: &mut usize, _action: ComposeAction) -> &'static str {
        *state += 10;
        "command"
    }

    fn handle_search(state: &mut usize, _action: ComposeAction) -> &'static str {
        *state += 100;
        "search"
    }

    fn handle_default(state: &mut usize, _action: ComposeAction) -> &'static str {
        *state += 1000;
        "default"
    }

    fn composed_reducer(
        state: &mut usize,
        action: ComposeAction,
        context: ComposeContext,
    ) -> &'static str {
        crate::reducer_compose!(state, action, context, {
            category "nav" => handle_nav,
            context ComposeContext::Command => handle_command,
            ComposeAction::Search => handle_search,
            _ => handle_default,
        })
    }

    #[test]
    fn test_reducer_compose_routes_category() {
        let mut state = 0;
        let result = composed_reducer(&mut state, ComposeAction::NavUp, ComposeContext::Command);
        assert_eq!(result, "nav");
        assert_eq!(state, 1);
    }

    #[test]
    fn test_reducer_compose_routes_context() {
        let mut state = 0;
        let result = composed_reducer(&mut state, ComposeAction::Other, ComposeContext::Command);
        assert_eq!(result, "command");
        assert_eq!(state, 10);
    }

    #[test]
    fn test_reducer_compose_routes_pattern() {
        let mut state = 0;
        let result = composed_reducer(&mut state, ComposeAction::Search, ComposeContext::Default);
        assert_eq!(result, "search");
        assert_eq!(state, 100);
    }

    #[test]
    fn test_reducer_compose_routes_fallback() {
        let mut state = 0;
        let result = composed_reducer(&mut state, ComposeAction::Other, ComposeContext::Default);
        assert_eq!(result, "default");
        assert_eq!(state, 1000);
    }

    // Test 3-argument form (no context)
    fn composed_reducer_no_context(state: &mut usize, action: ComposeAction) -> &'static str {
        crate::reducer_compose!(state, action, {
            category "nav" => handle_nav,
            ComposeAction::Search => handle_search,
            _ => handle_default,
        })
    }

    #[test]
    fn test_reducer_compose_3arg_category() {
        let mut state = 0;
        let result = composed_reducer_no_context(&mut state, ComposeAction::NavUp);
        assert_eq!(result, "nav");
        assert_eq!(state, 1);
    }

    #[test]
    fn test_reducer_compose_3arg_pattern() {
        let mut state = 0;
        let result = composed_reducer_no_context(&mut state, ComposeAction::Search);
        assert_eq!(result, "search");
        assert_eq!(state, 100);
    }

    #[test]
    fn test_reducer_compose_3arg_fallback() {
        let mut state = 0;
        let result = composed_reducer_no_context(&mut state, ComposeAction::Other);
        assert_eq!(result, "default");
        assert_eq!(state, 1000);
    }
}