refactor: modularize app source by splitting src/app.rs into separate UI, state, input, and capture modules.

Author: sguzman

scratch/add_state_imports.py

@@ -0,0 +1,14 @@
+with open('src/app/state.rs', 'r') as f:
+    state_rs = f.read()
+
+imports = """
+use crate::engine::{BrowserEngine, EngineEvent, EngineStatus, FocusState, RenderSurfaceMetadata, SecurityWarningKind};
+use crate::extraction::extract_entities;
+use std::time::Duration;
+use std::sync::{Arc, RwLock};
+"""
+
+state_rs = imports + state_rs
+
+with open('src/app/state.rs', 'w') as f:
+    f.write(state_rs)

scratch/extract_capture.py

@@ -0,0 +1,38 @@
+import re
+
+with open('src/app/mod.rs', 'r') as f:
+    mod_rs = f.read()
+
+funcs_to_extract = [
+    "fn update_render_frame",
+    "fn capture_frame_to_disk",
+    "fn save_snapshot_to_disk",
+    "fn resolve_capture_dir",
+    "fn sample_pixel_rgba"
+]
+
+extracted_funcs = []
+
+for func in funcs_to_extract:
+    start_idx = mod_rs.find(f"    {func}")
+    if start_idx != -1:
+        brace_count = 0
+        end_idx = -1
+        for i in range(start_idx, len(mod_rs)):
+            if mod_rs[i] == '{': brace_count += 1
+            elif mod_rs[i] == '}':
+                brace_count -= 1
+                if brace_count == 0:
+                    end_idx = i + 1
+                    break
+        
+        extracted = mod_rs[start_idx:end_idx]
+        extracted = extracted.replace("fn ", "pub(super) fn ")
+        extracted_funcs.append(extracted)
+        mod_rs = mod_rs[:start_idx] + mod_rs[end_idx:]
+
+with open('src/app/capture.rs', 'w') as f:
+    f.write("use super::*;\nuse crate::engine::{RenderSurfaceMetadata};\nuse eframe::egui::ColorImage;\n\nimpl BrazenApp {\n" + "\n\n".join(extracted_funcs) + "\n}\n")
+
+with open('src/app/mod.rs', 'w') as f:
+    f.write(mod_rs)

scratch/extract_input.py

@@ -0,0 +1,31 @@
+import re
+
+with open('src/app/mod.rs', 'r') as f:
+    mod_rs = f.read()
+
+# Find forward_input_events
+start_idx = mod_rs.find("    fn forward_input_events(&mut self, ctx: &eframe::egui::Context) {")
+if start_idx != -1:
+    brace_count = 0
+    end_idx = -1
+    for i in range(start_idx, len(mod_rs)):
+        if mod_rs[i] == '{': brace_count += 1
+        elif mod_rs[i] == '}':
+            brace_count -= 1
+            if brace_count == 0:
+                end_idx = i + 1
+                break
+    
+    input_fn = mod_rs[start_idx:end_idx]
+    # change from `fn forward_input_events` to `pub(super) fn forward_input_events`
+    input_fn = input_fn.replace("fn forward_input_events", "pub(super) fn forward_input_events")
+    
+    # Write to input.rs
+    with open('src/app/input.rs', 'w') as f:
+        f.write("use super::*;\nuse crate::engine::{InputEvent, AlphaMode};\n\nimpl BrazenApp {\n" + input_fn + "\n}\n")
+        
+    # Remove from mod.rs
+    mod_rs = mod_rs[:start_idx] + mod_rs[end_idx:]
+
+with open('src/app/mod.rs', 'w') as f:
+    f.write(mod_rs)

scratch/extract_shell_state.py

@@ -0,0 +1,27 @@
+import re
+
+with open('src/app/mod.rs', 'r') as f:
+    mod_rs = f.read()
+
+# Extract ShellState struct and impl
+struct_match = re.search(r'#\[derive\([^)]*\)\]\s*pub struct ShellState \{.*?\}', mod_rs, flags=re.DOTALL)
+impl_match = re.search(r'impl ShellState \{.*?^\}', mod_rs, flags=re.MULTILINE|re.DOTALL)
+
+struct_str = struct_match.group(0) if struct_match else ""
+impl_str = impl_match.group(0) if impl_match else ""
+
+if struct_str:
+    mod_rs = mod_rs.replace(struct_str, "")
+if impl_str:
+    mod_rs = mod_rs.replace(impl_str, "")
+
+with open('src/app/mod.rs', 'w') as f:
+    f.write(mod_rs)
+
+with open('src/app/state.rs', 'r') as f:
+    state_rs = f.read()
+
+state_rs += "\n\n" + struct_str + "\n\n" + impl_str
+
+with open('src/app/state.rs', 'w') as f:
+    f.write(state_rs)

scratch/extract_tests.py

@@ -0,0 +1,31 @@
+import re
+
+with open('src/app/mod.rs', 'r') as f:
+    mod_rs = f.read()
+
+# Extract tests
+start_idx = mod_rs.find("#[cfg(test)]\nmod tests {")
+if start_idx != -1:
+    brace_count = 0
+    end_idx = -1
+    for i in range(start_idx, len(mod_rs)):
+        if mod_rs[i] == '{': brace_count += 1
+        elif mod_rs[i] == '}':
+            brace_count -= 1
+            if brace_count == 0:
+                end_idx = i + 1
+                break
+    
+    test_block = mod_rs[start_idx:end_idx]
+    
+    # Strip `mod tests { ... }` wrappers to put in tests.rs
+    test_content = re.sub(r'^#\[cfg\(test\)\]\s*mod tests\s*\{', '', test_block)
+    test_content = test_content[:-1] # remove last brace
+    
+    with open('src/app/tests.rs', 'w') as f:
+        f.write("use super::*;\n" + test_content)
+        
+    mod_rs = mod_rs[:start_idx] + "\n#[cfg(test)]\nmod tests;\n" + mod_rs[end_idx:]
+
+with open('src/app/mod.rs', 'w') as f:
+    f.write(mod_rs)

scratch/fix_errors.py

@@ -0,0 +1,67 @@
+import re
+
+# Fix panels.rs
+with open('src/app/panels.rs', 'r') as f:
+    panels_rs = f.read()
+
+panels_rs = panels_rs.replace("crate::app::frame_average_ms", "crate::app::ui_main::frame_average_ms")
+
+with open('src/app/panels.rs', 'w') as f:
+    f.write(panels_rs)
+
+
+# Fix ui_components.rs
+with open('src/app/ui_components.rs', 'r') as f:
+    ui_comp = f.read()
+
+ui_comp = ui_comp.replace("use super::state::*;", "use super::state::*;\nuse crate::app::PaletteCommand;\nuse crate::engine::EngineStatus;")
+
+with open('src/app/ui_components.rs', 'w') as f:
+    f.write(ui_comp)
+
+# Fix monitor_chatgpt_mcp in ui_main.rs
+# I'll just append it to impl BrazenApp in ui_main.rs
+monitor_code = """
+    pub(super) fn monitor_chatgpt_mcp(&mut self) {
+        if !self.shell_state.observe_dom || !self.shell_state.control_terminal {
+            return;
+        }
+        
+        let Some(snapshot) = &self.shell_state.dom_snapshot else { return; };
+        
+        // Use scraper to find <client mcp="terminal">...</client>
+        let fragment = scraper::Html::parse_fragment(snapshot);
+        let selector = scraper::Selector::parse("client[mcp=\\"terminal\\"]").unwrap();
+        
+        for element in fragment.select(&selector) {
+            let command = element.text().collect::<String>();
+            let command = command.trim();
+            if !command.is_empty() && !self.processed_mcp_commands.contains(command) {
+                tracing::info!(target: "brazen::automation", command = %command, "Found new MCP terminal command in DOM");
+                
+                // Run command
+                if let Some(tx) = &self.terminal_tx {
+                    let _ = tx.send(command.to_string());
+                }
+                
+                // Mark as processed
+                self.processed_mcp_commands.insert(command.to_string());
+            }
+        }
+    }
+"""
+
+with open('src/app/ui_main.rs', 'r') as f:
+    ui_main_rs = f.read()
+
+# insert before last } of impl BrazenApp
+last_brace = ui_main_rs.find("impl eframe::App")
+if last_brace != -1:
+    # go back to find the } of impl BrazenApp
+    idx = ui_main_rs.rfind("}", 0, last_brace)
+    ui_main_rs = ui_main_rs[:idx] + monitor_code + ui_main_rs[idx:]
+
+with open('src/app/ui_main.rs', 'w') as f:
+    f.write(ui_main_rs)
+
+print("Errors fixed")

scratch/fix_state_imports.py

@@ -0,0 +1,15 @@
+with open('src/app/state.rs', 'r') as f:
+    state_rs = f.read()
+
+imports = """
+use chrono::Utc;
+use crate::session::{NavigationEntry, SessionSnapshot, load_session};
+use crate::engine::EngineLoadStatus;
+use crate::profile_db::ProfileDb;
+use std::time::Instant;
+"""
+
+state_rs = imports + state_rs
+
+with open('src/app/state.rs', 'w') as f:
+    f.write(state_rs)

scratch/fix_structs.py

@@ -0,0 +1,54 @@
+import re
+
+with open('src/app/mod.rs', 'r') as f:
+    mod_rs = f.read()
+
+# Delete duplicate structs/enums from mod.rs
+structs_to_delete = [
+    "WorkspaceLayout",
+    "UiTheme",
+    "UiDensity",
+    "LayoutPreset",
+    "SettingsTab",
+    "LeftPanelTab",
+    "WorkspacePanels",
+    "ReadingQueueItem",
+    "ExtractedEntity"
+]
+
+for s in structs_to_delete:
+    # regex to delete `enum X { ... }` or `struct X { ... }`
+    mod_rs = re.sub(r'#\[derive\([^)]*\)\]\s*(pub )?(enum|struct)\s+' + s + r'\s*\{[^}]*\}', '', mod_rs, flags=re.MULTILINE)
+    # also remove `impl Default for WorkspacePanels`
+    if s == "WorkspacePanels":
+        mod_rs = re.sub(r'impl Default for WorkspacePanels\s*\{.*?(?=\n\n|\Z)', '', mod_rs, flags=re.MULTILINE|re.DOTALL)
+
+# Find DiagnosticTab and move it
+diag_tab_match = re.search(r'#\[derive\([^)]*\)\]\s*(pub )?enum DiagnosticTab\s*\{[^}]*\}', mod_rs)
+diag_tab_str = ""
+if diag_tab_match:
+    diag_tab_str = diag_tab_match.group(0)
+    # make it pub
+    diag_tab_str = diag_tab_str.replace("enum DiagnosticTab", "pub enum DiagnosticTab")
+    mod_rs = mod_rs[:diag_tab_match.start()] + mod_rs[diag_tab_match.end():]
+
+# Make PaletteCommand pub
+mod_rs = mod_rs.replace("enum PaletteCommand", "pub enum PaletteCommand")
+
+with open('src/app/mod.rs', 'w') as f:
+    f.write(mod_rs)
+
+with open('src/app/state.rs', 'r') as f:
+    state_rs = f.read()
+
+# Remove `use super::DiagnosticTab;` from state.rs
+state_rs = state_rs.replace("use super::DiagnosticTab;", "")
+
+# add DiagnosticTab
+if diag_tab_str:
+    state_rs += "\n" + diag_tab_str + "\n"
+
+with open('src/app/state.rs', 'w') as f:
+    f.write(state_rs)
+
+print("Fixed structs in mod.rs and state.rs")

scratch/fix_ui_main.py

@@ -0,0 +1,25 @@
+with open('src/app/ui_main.rs', 'r') as f:
+    ui_main = f.read()
+
+# Fix frame_average_ms
+ui_main = ui_main.replace("""fn frame_average_ms(times: &VecDeque<f32>) -> Option<f32> {
+    if times.is_empty() {
+        return None;
+    }
+    let sum: f32 = times.iter().copied().sum();
+    Some(sum / times.len() as f32)
+
+    pub(super) fn monitor_chatgpt_mcp(&mut self) {""", """pub(super) fn frame_average_ms(times: &VecDeque<f32>) -> Option<f32> {
+    if times.is_empty() {
+        return None;
+    }
+    let sum: f32 = times.iter().copied().sum();
+    Some(sum / times.len() as f32)
+}
+
+impl super::BrazenApp {
+    pub(super) fn monitor_chatgpt_mcp(&mut self) {""")
+
+with open('src/app/ui_main.rs', 'w') as f:
+    f.write(ui_main)
+