iced-yoda/devtools/src/lib.rs

#![allow(missing_docs)]
use iced_debug as debug;
use iced_program as program;
use iced_test as test;
use iced_widget::core;
use iced_widget::runtime;
use iced_widget::runtime::futures;

mod comet;
mod executor;
mod icon;
mod time_machine;
mod widget;

use crate::core::alignment::Horizontal::Right;
use crate::core::border;
use crate::core::keyboard;
use crate::core::theme::{self, Base, Theme};
use crate::core::time::seconds;
use crate::core::window;
use crate::core::{
    Alignment::Center, Color, Element, Font, Length::Fill, Size,
};
use crate::futures::Subscription;
use crate::program::Program;
use crate::runtime::Task;
use crate::runtime::font;
use crate::test::instruction;
use crate::time_machine::TimeMachine;
use crate::widget::{
    Text, bottom_right, button, center, column, container, horizontal_space,
    opaque, row, scrollable, stack, text, text_input, themer,
};

use std::fmt;
use std::thread;

pub fn attach<P: Program + 'static>(program: P) -> Attach<P> {
    Attach { program }
}

/// A [`Program`] with some devtools attached to it.
#[derive(Debug)]
pub struct Attach<P> {
    /// The original [`Program`] managed by these devtools.
    pub program: P,
}

impl<P> Program for Attach<P>
where
    P: Program + 'static,
{
    type State = DevTools<P>;
    type Message = Event<P>;
    type Theme = P::Theme;
    type Renderer = P::Renderer;
    type Executor = P::Executor;

    fn name() -> &'static str {
        P::name()
    }

    fn settings(&self) -> core::Settings {
        self.program.settings()
    }

    fn boot(&self) -> (Self::State, Task<Self::Message>) {
        let (state, boot) = self.program.boot();
        let (devtools, task) = DevTools::new(state);

        (
            devtools,
            Task::batch([
                boot.map(Event::Program),
                task.map(Event::Message),
                font::load(icon::FONT).discard(),
            ]),
        )
    }

    fn update(
        &self,
        state: &mut Self::State,
        message: Self::Message,
    ) -> Task<Self::Message> {
        state.update(&self.program, message)
    }

    fn view<'a>(
        &self,
        state: &'a Self::State,
        window: window::Id,
    ) -> Element<'a, Self::Message, Self::Theme, Self::Renderer> {
        state.view(&self.program, window)
    }

    fn title(&self, state: &Self::State, window: window::Id) -> String {
        state.title(&self.program, window)
    }

    fn subscription(
        &self,
        state: &Self::State,
    ) -> runtime::futures::Subscription<Self::Message> {
        state.subscription(&self.program)
    }

    fn theme(&self, state: &Self::State, window: window::Id) -> Self::Theme {
        state.theme(&self.program, window)
    }

    fn style(&self, state: &Self::State, theme: &Self::Theme) -> theme::Style {
        state.style(&self.program, theme)
    }

    fn scale_factor(&self, state: &Self::State, window: window::Id) -> f64 {
        state.scale_factor(&self.program, window)
    }
}

/// The state of the devtools.
#[allow(missing_debug_implementations)]
pub struct DevTools<P>
where
    P: Program,
{
    state: P::State,
    size: Size,
    mode: Mode,
    show_notification: bool,
    time_machine: TimeMachine<P>,
}

#[derive(Debug, Clone)]
pub enum Message {
    HideNotification,
    Toggle,
    ToggleComet,
    CometLaunched(comet::launch::Result),
    InstallComet,
    Installing(comet::install::Result),
    CancelSetup,
    ChangeWidth(String),
    ChangeHeight(String),
    Record,
    Stop,
    Recorded(core::Event),
}

enum Mode {
    Hidden,
    Open { recorder: Recorder },
    Setup(Setup),
}

struct Recorder {
    instructions: Vec<test::Instruction>,
    is_recording: bool,
}

enum Setup {
    Idle { goal: Goal },
    Running { logs: Vec<String> },
}

enum Goal {
    Installation,
    Update { revision: Option<String> },
}

impl<P> DevTools<P>
where
    P: Program + 'static,
{
    fn new(state: P::State) -> (Self, Task<Message>) {
        (
            Self {
                state,
                size: Size::new(512.0, 512.0),
                mode: Mode::Hidden,
                show_notification: true,
                time_machine: TimeMachine::new(),
            },
            executor::spawn_blocking(|mut sender| {
                thread::sleep(seconds(2));
                let _ = sender.try_send(());
            })
            .map(|_| Message::HideNotification),
        )
    }

    fn title(&self, program: &P, window: window::Id) -> String {
        program.title(&self.state, window)
    }

    fn update(&mut self, program: &P, event: Event<P>) -> Task<Event<P>> {
        match event {
            Event::Message(message) => match message {
                Message::HideNotification => {
                    self.show_notification = false;

                    Task::none()
                }
                Message::Toggle => {
                    match &self.mode {
                        Mode::Hidden => {
                            self.mode = Mode::Open {
                                recorder: Recorder {
                                    instructions: Vec::new(),
                                    is_recording: false,
                                },
                            };
                        }
                        Mode::Open { recorder } if !recorder.is_recording => {
                            self.mode = Mode::Hidden;
                        }
                        Mode::Setup(_) | Mode::Open { .. } => {}
                    }

                    Task::none()
                }
                Message::ToggleComet => {
                    if let Mode::Setup(setup) = &self.mode {
                        if matches!(setup, Setup::Idle { .. }) {
                            self.mode = Mode::Hidden;
                        }

                        Task::none()
                    } else if debug::quit() {
                        Task::none()
                    } else {
                        comet::launch()
                            .map(Message::CometLaunched)
                            .map(Event::Message)
                    }
                }
                Message::CometLaunched(Ok(())) => Task::none(),
                Message::CometLaunched(Err(error)) => {
                    match error {
                        comet::launch::Error::NotFound => {
                            self.mode = Mode::Setup(Setup::Idle {
                                goal: Goal::Installation,
                            });
                        }
                        comet::launch::Error::Outdated { revision } => {
                            self.mode = Mode::Setup(Setup::Idle {
                                goal: Goal::Update { revision },
                            });
                        }
                        comet::launch::Error::IoFailed(error) => {
                            log::error!("comet failed to run: {error}");
                        }
                    }

                    Task::none()
                }
                Message::InstallComet => {
                    self.mode =
                        Mode::Setup(Setup::Running { logs: Vec::new() });

                    comet::install()
                        .map(Message::Installing)
                        .map(Event::Message)
                }
                Message::Installing(Ok(installation)) => {
                    let Mode::Setup(Setup::Running { logs }) = &mut self.mode
                    else {
                        return Task::none();
                    };

                    match installation {
                        comet::install::Event::Logged(log) => {
                            logs.push(log);
                            Task::none()
                        }
                        comet::install::Event::Finished => {
                            self.mode = Mode::Hidden;
                            comet::launch().discard()
                        }
                    }
                }
                Message::Installing(Err(error)) => {
                    let Mode::Setup(Setup::Running { logs }) = &mut self.mode
                    else {
                        return Task::none();
                    };

                    match error {
                        comet::install::Error::ProcessFailed(status) => {
                            logs.push(format!("process failed with {status}"));
                        }
                        comet::install::Error::IoFailed(error) => {
                            logs.push(error.to_string());
                        }
                    }

                    Task::none()
                }
                Message::CancelSetup => {
                    self.mode = Mode::Hidden;

                    Task::none()
                }
                Message::ChangeWidth(width) => {
                    if let Ok(width) = width.parse() {
                        self.size.width = width;
                    }

                    Task::none()
                }
                Message::ChangeHeight(height) => {
                    if let Ok(height) = height.parse() {
                        self.size.height = height;
                    }

                    Task::none()
                }
                Message::Record => {
                    let Mode::Open { recorder } = &mut self.mode else {
                        return Task::none();
                    };

                    recorder.instructions.clear();
                    recorder.is_recording = true;

                    let (state, task) = program.boot();
                    self.state = state;

                    task.map(Event::Program)
                }
                Message::Recorded(event) => {
                    let Mode::Open { recorder } = &mut self.mode else {
                        return Task::none();
                    };

                    let Some(interaction) =
                        instruction::Interaction::from_event(event)
                    else {
                        return Task::none();
                    };

                    if let Some(test::Instruction::Interact(last_interaction)) =
                        recorder.instructions.pop()
                    {
                        let (last_interaction, new_interaction) =
                            last_interaction.merge(interaction);

                        recorder.instructions.push(
                            test::Instruction::Interact(last_interaction),
                        );

                        if let Some(new_interaction) = new_interaction {
                            recorder.instructions.push(
                                test::Instruction::Interact(new_interaction),
                            );
                        }
                    } else {
                        recorder
                            .instructions
                            .push(test::Instruction::Interact(interaction));
                    }

                    Task::none()
                }
                Message::Stop => {
                    let Mode::Open { recorder } = &mut self.mode else {
                        return Task::none();
                    };

                    recorder.is_recording = false;

                    Task::none()
                }
            },
            Event::Program(message) => {
                self.time_machine.push(&message);

                if self.time_machine.is_rewinding() {
                    debug::enable();
                }

                let span = debug::update(&message);
                let task = program.update(&mut self.state, message);
                debug::tasks_spawned(task.units());
                span.finish();

                if self.time_machine.is_rewinding() {
                    debug::disable();
                }

                task.map(Event::Program)
            }
            Event::Command(command) => {
                match command {
                    debug::Command::RewindTo { message } => {
                        self.time_machine.rewind(program, message);
                    }
                    debug::Command::GoLive => {
                        self.time_machine.go_to_present();
                    }
                }

                Task::none()
            }
            Event::Discard => Task::none(),
        }
    }

    fn view(
        &self,
        program: &P,
        window: window::Id,
    ) -> Element<'_, Event<P>, P::Theme, P::Renderer> {
        let state = self.state();

        let view = {
            let view = program.view(state, window);
            let theme = program.theme(state, window);

            let view: Element<'_, _, Theme, _> = themer(theme, view).into();

            if self.time_machine.is_rewinding() {
                view.map(|_| Event::Discard)
            } else {
                view.map(Event::Program)
            }
        };

        let theme = program
            .theme(state, window)
            .palette()
            .map(|palette| Theme::custom("iced devtools", palette))
            .unwrap_or_default();

        let setup = if let Mode::Setup(setup) = &self.mode {
            let stage: Element<'_, _, Theme, P::Renderer> = match setup {
                Setup::Idle { goal } => self::setup(goal),
                Setup::Running { logs } => installation(logs),
            };

            let setup = center(
                container(stage)
                    .padding(20)
                    .max_width(500)
                    .style(container::bordered_box),
            )
            .padding(10)
            .style(|_theme| {
                container::Style::default()
                    .background(Color::BLACK.scale_alpha(0.8))
            });

            Some(setup)
        } else {
            None
        }
        .map(|mode| Element::from(mode).map(Event::Message));

        let notification = self.show_notification.then(|| {
            bottom_right(opaque(
                container(text("Press F12 to open developer tools"))
                    .padding(10)
                    .style(container::dark),
            ))
        });

        let sidebar = if let Mode::Open { recorder } = &self.mode {
            let title = monospace("Developer Tools");

            let recorder = {
                let events = container(if recorder.instructions.is_empty() {
                    Element::from(center(
                        monospace("No instructions recorded yet!")
                            .size(14)
                            .width(Fill)
                            .center(),
                    ))
                } else {
                    scrollable(
                        column(recorder.instructions.iter().map(
                            |instruction| {
                                monospace(instruction.to_string())
                                    .size(10)
                                    .into()
                            },
                        ))
                        .spacing(5),
                    )
                    .spacing(5)
                    .into()
                })
                .width(Fill)
                .height(Fill)
                .style(container::rounded_box)
                .padding(5);

                let controls = {
                    row![
                        button(icon::play().size(14).width(Fill).center()),
                        if recorder.is_recording {
                            button(icon::stop().size(14).width(Fill).center())
                                .on_press(Message::Stop)
                                .style(button::success)
                        } else {
                            button(icon::record().size(14).width(Fill).center())
                                .on_press(Message::Record)
                                .style(button::danger)
                        }
                    ]
                    .spacing(10)
                };

                column![events, controls].spacing(10).align_x(Center)
            };

            let viewport = row![
                text_input("Width", &self.size.width.to_string())
                    .size(14)
                    .on_input(Message::ChangeWidth),
                monospace("x"),
                text_input("Height", &self.size.height.to_string())
                    .size(14)
                    .on_input(Message::ChangeHeight),
            ]
            .spacing(10)
            .align_y(Center);

            let tools = column![
                title,
                labeled("Viewport", viewport),
                labeled("Tester", recorder)
            ]
            .spacing(10);

            let sidebar = container(tools)
                .padding(10)
                .width(250)
                .height(Fill)
                .style(container::dark);

            Some(Element::from(sidebar).map(Event::Message))
        } else {
            None
        };

        let content = row![if let Mode::Open { recorder } = &self.mode {
            let is_recording = recorder.is_recording;

            let status = if is_recording {
                monospace("Recording").style(|theme| text::Style {
                    color: Some(theme.palette().danger),
                })
            } else {
                monospace("Idle").style(|theme| text::Style {
                    color: Some(
                        theme.extended_palette().background.strongest.color,
                    ),
                })
            };

            let viewport = container(
                scrollable(
                    container(if recorder.is_recording {
                        widget::recorder(view)
                            .on_event(|event| {
                                Event::Message(Message::Recorded(event))
                            })
                            .into()
                    } else {
                        view
                    })
                    .width(self.size.width)
                    .height(self.size.height),
                )
                .direction(scrollable::Direction::Both {
                    vertical: scrollable::Scrollbar::default(),
                    horizontal: scrollable::Scrollbar::default(),
                }),
            )
            .style(move |theme| {
                let palette = theme.extended_palette();

                container::Style {
                    border: border::width(2.0).color(if is_recording {
                        palette.danger.base.color
                    } else {
                        palette.background.strongest.color
                    }),
                    ..container::Style::default()
                }
            })
            .padding(10);

            center(column![status, viewport].spacing(10).align_x(Right))
                .padding(10)
                .into()
        } else {
            view
        }]
        .push_maybe(sidebar);

        themer(
            theme,
            stack![content]
                .height(Fill)
                .push_maybe(setup.map(opaque))
                .push_maybe(notification),
        )
        .into()
    }

    fn subscription(&self, program: &P) -> Subscription<Event<P>> {
        let subscription =
            program.subscription(&self.state).map(Event::Program);
        debug::subscriptions_tracked(subscription.units());

        let hotkeys =
            futures::keyboard::on_key_press(|key, _modifiers| match key {
                keyboard::Key::Named(keyboard::key::Named::F12) => {
                    Some(Message::Toggle)
                }
                keyboard::Key::Named(keyboard::key::Named::F11) => {
                    Some(Message::ToggleComet)
                }
                _ => None,
            })
            .map(Event::Message);

        let commands = debug::commands().map(Event::Command);

        Subscription::batch([subscription, hotkeys, commands])
    }

    fn theme(&self, program: &P, window: window::Id) -> P::Theme {
        program.theme(self.state(), window)
    }

    fn style(&self, program: &P, theme: &P::Theme) -> theme::Style {
        program.style(self.state(), theme)
    }

    fn scale_factor(&self, program: &P, window: window::Id) -> f64 {
        program.scale_factor(self.state(), window)
    }

    fn state(&self) -> &P::State {
        self.time_machine.state().unwrap_or(&self.state)
    }
}

pub enum Event<P>
where
    P: Program,
{
    Message(Message),
    Program(P::Message),
    Command(debug::Command),
    Discard,
}

impl<P> fmt::Debug for Event<P>
where
    P: Program,
{
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::Message(message) => message.fmt(f),
            Self::Program(message) => message.fmt(f),
            Self::Command(command) => command.fmt(f),
            Self::Discard => f.write_str("Discard"),
        }
    }
}

#[cfg(feature = "time-travel")]
impl<P> Clone for Event<P>
where
    P: Program,
{
    fn clone(&self) -> Self {
        match self {
            Self::Message(message) => Self::Message(message.clone()),
            Self::Program(message) => Self::Program(message.clone()),
            Self::Command(command) => Self::Command(*command),
            Self::Discard => Self::Discard,
        }
    }
}

fn setup<Renderer>(goal: &Goal) -> Element<'_, Message, Theme, Renderer>
where
    Renderer: program::Renderer + 'static,
{
    let controls = row![
        button(text("Cancel").center().width(Fill))
            .width(100)
            .on_press(Message::CancelSetup)
            .style(button::danger),
        horizontal_space(),
        button(
            text(match goal {
                Goal::Installation => "Install",
                Goal::Update { .. } => "Update",
            })
            .center()
            .width(Fill)
        )
        .width(100)
        .on_press(Message::InstallComet)
        .style(button::success),
    ];

    let command = container(
        monospace(format!(
            "cargo install --locked \\
    --git https://github.com/iced-rs/comet.git \\
    --rev {}",
            comet::COMPATIBLE_REVISION
        ))
        .size(14),
    )
    .width(Fill)
    .padding(5)
    .style(container::dark);

    Element::from(match goal {
        Goal::Installation => column![
            text("comet is not installed!").size(20),
            "In order to display performance \
                metrics, the  comet debugger must \
                be installed in your system.",
            "The comet debugger is an official \
                companion tool that helps you debug \
                your iced applications.",
            column![
                "Do you wish to install it with the \
                    following command?",
                command
            ]
            .spacing(10),
            controls,
        ]
        .spacing(20),
        Goal::Update { revision } => {
            let comparison = column![
                row![
                    "Installed revision:",
                    horizontal_space(),
                    inline_code(revision.as_deref().unwrap_or("Unknown"))
                ]
                .align_y(Center),
                row![
                    "Compatible revision:",
                    horizontal_space(),
                    inline_code(comet::COMPATIBLE_REVISION),
                ]
                .align_y(Center)
            ]
            .spacing(5);

            column![
                text("comet is out of date!").size(20),
                comparison,
                column![
                    "Do you wish to update it with the following \
                        command?",
                    command
                ]
                .spacing(10),
                controls,
            ]
            .spacing(20)
        }
    })
}

fn installation<'a, Renderer>(
    logs: &'a [String],
) -> Element<'a, Message, Theme, Renderer>
where
    Renderer: program::Renderer + 'a,
{
    column![
        text("Installing comet...").size(20),
        container(
            scrollable(
                column(
                    logs.iter().map(|log| { monospace(log).size(12).into() }),
                )
                .spacing(3),
            )
            .spacing(10)
            .width(Fill)
            .height(300)
            .anchor_bottom(),
        )
        .padding(10)
        .style(container::dark)
    ]
    .spacing(20)
    .into()
}

fn inline_code<'a, Renderer>(
    code: impl text::IntoFragment<'a>,
) -> Element<'a, Message, Theme, Renderer>
where
    Renderer: program::Renderer + 'a,
{
    container(monospace(code).size(12))
        .style(|_theme| {
            container::Style::default()
                .background(Color::BLACK)
                .border(border::rounded(2))
        })
        .padding([2, 4])
        .into()
}

fn monospace<'a, Renderer>(
    fragment: impl text::IntoFragment<'a>,
) -> Text<'a, Theme, Renderer>
where
    Renderer: program::Renderer + 'a,
{
    text(fragment).font(Font::MONOSPACE)
}

fn labeled<'a, Message, Renderer>(
    fragment: impl text::IntoFragment<'a>,
    content: impl Into<Element<'a, Message, Theme, Renderer>>,
) -> Element<'a, Message, Theme, Renderer>
where
    Message: 'a,
    Renderer: program::Renderer + 'a,
{
    column![monospace(fragment).size(14), content.into()]
        .spacing(5)
        .into()
}