pub(crate) mod overlay;
#[cfg(test)]
#[path = "overlay_tests.rs"]
mod overlay_tests;
pub(crate) mod prompt;
#[cfg(test)]
#[path = "prompt_tests.rs"]
mod prompt_tests;
pub(crate) mod ui_message;
#[cfg(test)]
#[path = "ui_message_tests.rs"]
mod ui_message_tests;
pub(crate) use overlay::{
McpServerTransport, OverlayFocus, OverlayState, UiConfirmOverlay, UiErrorOverlay, UiMcpOverlay,
UiMcpServerInfo, UiMemoryEntry, UiMemoryOverlay, UiOverlay, UiOverlayKind, UiQuestionOverlay,
UiQuestionSurfaceKind, UiSearchMatch, UiSearchOverlay, UiTaskOverlay, UiTaskStepItem,
UiTodoOverlay,
};
#[cfg(test)]
pub(crate) use overlay::{UiQuestionOption, UiQuestionPrompt, UiQuestionToolContext, UiTodoItem};
pub(crate) use prompt::{
PromptMode, PromptMotion, PromptState, PromptSubmission, PromptSubmissionStatus,
QueuedPromptCommand, QueuedPromptCommandKind,
};
#[cfg(test)]
pub(crate) use ui_message::UiMessageKind;
pub(crate) use ui_message::{
UiAssistantMessage, UiErrorMessage, UiMessage, UiMessageBase, UiMessageId, UiStep, UiStepState,
UiSystemMessage, UiSystemMessageLevel, UiThinkingBlock, UiThinkingTiming, UiTokenUsage,
UiToolCall, UiToolCallState, UiToolResult, UiTurnTerminal, UiUserMessage,
};
#[cfg(test)]
mod tests;