Add time range filter buttons to historical data graph

Implemented time range selection buttons (5D, 1M, 3M, 6M, 1Y, 5Y, All) for the stock historical data view. The buttons are positioned on the same line as the Back button, aligned to the right side.

Changes:
- Add TimeRange enum with conversion methods to days and display labels
- Add selected_time_range field to track current filter (defaults to 3M)
- Update load_plot_data to respect selected time range
- Add change_time_range method to switch filters and reload data
- Implement plot_needs_reset flag to fully reset graph view on mode change
- Add UI buttons with visual highlight for selected range
- Use Plot.reset() to clear zoom/pan state when switching ranges

The graph now completely resets (axes, data, zoom, pan) when:
- Clicking any time range button
- Switching between stocks
- Returning to a previously selected time range

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: brainless

indistocks-gui/src/app.rs

@@ -13,6 +13,43 @@ pub enum View {
     Logs,
 }
 
+#[derive(Debug, Clone, Copy, PartialEq)]
+pub enum TimeRange {
+    FiveDays,
+    OneMonth,
+    ThreeMonths,
+    SixMonths,
+    OneYear,
+    FiveYears,
+    All,
+}
+
+impl TimeRange {
+    pub fn to_days(&self) -> Option<i64> {
+        match self {
+            TimeRange::FiveDays => Some(5),
+            TimeRange::OneMonth => Some(30),
+            TimeRange::ThreeMonths => Some(90),
+            TimeRange::SixMonths => Some(180),
+            TimeRange::OneYear => Some(365),
+            TimeRange::FiveYears => Some(365 * 5),
+            TimeRange::All => None, // None means load all data
+        }
+    }
+
+    pub fn label(&self) -> &str {
+        match self {
+            TimeRange::FiveDays => "5D",
+            TimeRange::OneMonth => "1M",
+            TimeRange::ThreeMonths => "3M",
+            TimeRange::SixMonths => "6M",
+            TimeRange::OneYear => "1Y",
+            TimeRange::FiveYears => "5Y",
+            TimeRange::All => "All",
+        }
+    }
+}
+
 pub struct IndistocksApp {
     pub current_view: View,
     pub db_conn: Arc<Mutex<Connection>>,
@@ -35,6 +72,8 @@ pub struct IndistocksApp {
     pub plot_loaded_range: Option<(NaiveDate, NaiveDate)>, // Track what data is currently loaded
     pub plot_earliest_available: Option<NaiveDate>, // Earliest date available in DB for current symbol
     pub plot_loading_in_progress: bool, // Prevent concurrent loads
+    pub selected_time_range: TimeRange, // Current time range filter for the plot
+    pub plot_needs_reset: bool, // Flag to reset plot view on next render
     // Search caching
     pub last_search_query: String,
     pub search_results: Vec<String>,
@@ -86,6 +125,8 @@ impl IndistocksApp {
             plot_loaded_range: None,
             plot_earliest_available: None,
             plot_loading_in_progress: false,
+            selected_time_range: TimeRange::ThreeMonths, // Default to 3 months
+            plot_needs_reset: false,
             last_search_query: String::new(),
             search_results: Vec::new(),
             stocks_price_from: String::new(),
@@ -127,6 +168,7 @@ impl IndistocksApp {
         self.plot_loaded_range = None;
         self.plot_earliest_available = None;
         self.plot_loading_in_progress = false;
+        self.plot_needs_reset = true; // Reset plot view when loading new stock
 
         let conn = self.db_conn.lock().unwrap();
 
@@ -165,9 +207,14 @@ impl IndistocksApp {
             println!("Data available from {} to {} ({} days span, {} data points in DB)",
                 earliest, latest, (latest - earliest).num_days(), total_count);
 
-            // Load last 3 months of data initially
-            let start = latest - chrono::Duration::days(90);
-            let load_from = if start < earliest { earliest } else { start };
+            // Load data based on selected time range
+            let load_from = match self.selected_time_range.to_days() {
+                Some(days) => {
+                    let start = latest - chrono::Duration::days(days);
+                    if start < earliest { earliest } else { start }
+                },
+                None => earliest, // Load all data
+            };
 
             match get_stock_data_in_range(&conn, symbol, load_from, latest) {
                 Ok(data) => {
@@ -242,6 +289,22 @@ impl IndistocksApp {
             self.plot_loading_in_progress = false;
         }
     }
+
+    /// Change the time range and reload data for the current symbol
+    pub fn change_time_range(&mut self, time_range: TimeRange) {
+        self.selected_time_range = time_range;
+
+        // Reset plot state to ensure graph resets
+        self.plot_data.clear();
+        self.plot_loaded_range = None;
+        self.plot_loading_in_progress = false;
+        self.plot_needs_reset = true; // Reset plot view when changing time range
+
+        // Reload the data if a symbol is selected
+        if let Some(symbol) = &self.selected_symbol.clone() {
+            self.load_plot_data(symbol);
+        }
+    }
 }
 
 impl eframe::App for IndistocksApp {

indistocks-gui/src/ui/main_content.rs

@@ -1,4 +1,4 @@
-use crate::app::IndistocksApp;
+use crate::app::{IndistocksApp, TimeRange};
 use chrono::{Datelike, Duration, NaiveDate};
 
 
@@ -23,8 +23,8 @@ pub fn render(ui: &mut egui::Ui, app: &mut IndistocksApp) {
             let (x_fmt, should_filter_ticks) = get_date_format_and_filter(days_diff);
             let x_fmt_clone = x_fmt.clone();
 
-            // Plot the data - use symbol in ID to reset view when switching stocks
-            let plot = egui_plot::Plot::new(format!("price_plot_{}", symbol))
+            // Plot the data - use symbol and time range in ID to reset view when switching stocks or time ranges
+            let mut plot = egui_plot::Plot::new(format!("price_plot_{}_{}", symbol, app.selected_time_range.label()))
                 .height(600.0)
                 .legend(egui_plot::Legend::default())
                 .allow_zoom([true, false])  // Allow horizontal zoom only
@@ -39,6 +39,12 @@ pub fn render(ui: &mut egui::Ui, app: &mut IndistocksApp) {
                         value.y)
                 });
 
+            // Reset plot view if needed (when changing time range or loading new stock)
+            if app.plot_needs_reset {
+                plot = plot.reset();
+                app.plot_needs_reset = false;
+            }
+
             let response = plot.show(ui, |plot_ui| {
                 let points: egui_plot::PlotPoints = app.plot_data
                     .iter()
@@ -98,10 +104,40 @@ pub fn render(ui: &mut egui::Ui, app: &mut IndistocksApp) {
             }
         }
 
-        if ui.button("Back").clicked() {
-            app.selected_symbol = None;
-            app.plot_data.clear();
-        }
+        // Horizontal layout for Back button and time range buttons
+        ui.horizontal(|ui| {
+            if ui.button("Back").clicked() {
+                app.selected_symbol = None;
+                app.plot_data.clear();
+            }
+
+            // Add spacing to push time range buttons to the right
+            ui.with_layout(egui::Layout::right_to_left(egui::Align::Center), |ui| {
+                // Time range buttons (in reverse order because of right_to_left layout)
+                let time_ranges = [
+                    TimeRange::All,
+                    TimeRange::FiveYears,
+                    TimeRange::OneYear,
+                    TimeRange::SixMonths,
+                    TimeRange::ThreeMonths,
+                    TimeRange::OneMonth,
+                    TimeRange::FiveDays,
+                ];
+
+                for time_range in time_ranges.iter().rev() {
+                    let is_selected = app.selected_time_range == *time_range;
+                    let button = if is_selected {
+                        egui::Button::new(time_range.label()).fill(ui.style().visuals.selection.bg_fill)
+                    } else {
+                        egui::Button::new(time_range.label())
+                    };
+
+                    if ui.add(button).clicked() {
+                        app.change_time_range(*time_range);
+                    }
+                }
+            });
+        });
     } else if !app.search_query.is_empty() {
         // Show search results
         ui.heading("Search Results");