use anyhow::Result; use hyperlog_core::log::GraphItem; use ratatui::{prelude::*, widgets::*}; use crate::{ command_parser::Commands, commander, components::movement_graph::GraphItemType, models::Msg, state::SharedState, }; use super::{ movement_graph::{MovementGraph, MovementGraphItem}, render_graph::summarize::SummarizeRenderGraph, }; #[derive(Clone, Debug)] pub enum FilterBy { NotDone, None, } impl Default for FilterBy { fn default() -> Self { Self::NotDone } } #[derive(Default, Clone, Debug)] pub struct DisplayOptions { pub filter_by: FilterBy, } pub struct GraphExplorer<'a> { state: SharedState, pub inner: GraphExplorerState<'a>, } pub struct GraphExplorerState<'a> { root: String, current_path: Option<&'a str>, current_position: Vec, display_options: DisplayOptions, graph: Option, } impl<'a> GraphExplorer<'a> { pub fn new(root: String, state: SharedState) -> Self { Self { state, inner: GraphExplorerState::<'a> { root, current_path: None, current_position: Vec::new(), graph: None, display_options: DisplayOptions::default(), }, } } pub fn update_graph(&mut self) -> Result<&mut Self> { let now = std::time::SystemTime::now(); let graph = self .state .querier .get( &self.inner.root, self.inner .current_path .map(|p| p.split('.').collect::>()) .unwrap_or_default(), ) .ok_or(anyhow::anyhow!("graph should've had an item"))?; self.inner.graph = Some(graph); let elapsed = now.elapsed()?; tracing::trace!("Graph.update_graph took: {}nanos", elapsed.as_nanos()); Ok(self) } fn linearize_graph(&self) -> Option { tracing::trace!("current display options: {:?}", self.inner.display_options); self.inner .graph .clone() .map(|g| MovementGraph::new(g, &self.inner.display_options)) } /// Will only incrmeent to the next level /// /// Current: 0.1.0 /// Available: 0.1.0.[0,1,2] /// Choses: 0.1.0.0 else nothing pub(crate) fn move_right(&mut self) -> Result<()> { if let Some(graph) = self.linearize_graph() { tracing::debug!("graph: {:?}", graph); let position_items = &self.inner.current_position; if let Some(next_item) = graph.next_right(position_items) { self.inner.current_position.push(next_item.index); tracing::trace!("found next item: {:?}", self.inner.current_position); } } Ok(()) } /// Will only incrmeent to the next level /// /// Current: 0.1.0 /// Available: 0.[0,1,2].0 /// Choses: 0.1 else nothing pub(crate) fn move_left(&mut self) -> Result<()> { if let Some(last) = self.inner.current_position.pop() { tracing::trace!( "found last item: {:?}, popped: {}", self.inner.current_position, last ); } Ok(()) } /// Will move up if a sibling exists, or up to the most common sibling between sections /// /// Current: 0.1.1 /// Available: 0.[0.[0,1],1.[0,1]] /// Chose: 0.1.0 again 0.0 We don't choose a subitem in the next three instead we just find the most common sibling pub(crate) fn move_up(&mut self) -> Result<()> { if let Some(graph) = self.linearize_graph() { let position_items = &self.inner.current_position; if let Some(next_item) = graph.next_up(position_items) { self.inner.current_position = next_item; tracing::trace!("found next up: {:?}", self.inner.current_position) } } Ok(()) } /// Will move down if a sibling exists, or down to the most common sibling between sections /// /// Current: 0.0.0 /// Available: 0.[0.[0,1],1.[0,1]] /// Chose: 0.0.1 again 0.1 pub(crate) fn move_down(&mut self) -> Result<()> { if let Some(graph) = self.linearize_graph() { let position_items = &self.inner.current_position; if let Some(next_item) = graph.next_down(position_items) { self.inner.current_position = next_item; tracing::trace!("found next down: {:?}", self.inner.current_position) } } Ok(()) } pub(crate) fn get_current_path(&self) -> Vec { let graph = self.linearize_graph(); let position_items = &self.inner.current_position; if let Some(graph) = graph { graph.to_current_path(position_items) } else { Vec::new() } } fn get_current_item(&self) -> Option { let graph = self.linearize_graph(); if let Some(graph) = graph { graph.get_graph_item(&self.inner.current_position).cloned() } else { None } } pub fn execute_command(&mut self, command: &Commands) -> anyhow::Result> { match command { Commands::Archive => { if !self.get_current_path().is_empty() { tracing::debug!("archiving path: {:?}", self.get_current_path()) } } Commands::CreateSection { name } => { if !name.is_empty() { let mut path = self.get_current_path(); path.push(name.replace(" ", "-").replace(".", "-")); self.state .commander .execute(commander::Command::CreateSection { root: self.inner.root.clone(), path, })?; } } Commands::Edit => { if let Some(item) = self.get_current_item() { let path = self.get_current_path(); tracing::debug!( "found item to edit: path: {}, item: {}", path.join("."), item.name ); match item.item_type { GraphItemType::Section => { todo!("cannot edit section at the moment") } GraphItemType::Item { .. } => { if let Some(item) = self.state.querier.get(&self.inner.root, path) { if let GraphItem::Item { .. } = item { return Ok(Some(Msg::OpenEditItemDialog { item })); } } } } } } Commands::ShowAll => { self.inner.display_options.filter_by = FilterBy::None; } Commands::HideDone => { self.inner.display_options.filter_by = FilterBy::NotDone; } _ => (), } self.update_graph()?; Ok(None) } pub(crate) fn interact(&mut self) -> anyhow::Result<()> { if !self.get_current_path().is_empty() { tracing::info!("toggling state of items"); self.state .commander .execute(commander::Command::ToggleItem { root: self.inner.root.to_string(), path: self.get_current_path(), })?; } self.update_graph()?; Ok(()) } } impl<'a> StatefulWidget for GraphExplorer<'a> { type State = GraphExplorerState<'a>; fn render(self, area: Rect, buf: &mut Buffer, state: &mut Self::State) { let Rect { height, .. } = area; let _height = height as usize; if let Some(graph) = &state.graph { let movement_graph: MovementGraph = MovementGraph::new(graph.clone(), &state.display_options); let lines = movement_graph.render_graph(&state.current_position); let para = Paragraph::new(lines); para.render(area, buf); } } }