add search by text path

Author: zsqw123

Cargo.lock

@@ -24,6 +24,7 @@ java_asm_macro = { path = "asm_macro", version = "0.0.15" } # JAVA_ASM_VERSION
 java_asm = { path = "asm", version = "0.0.15" } # JAVA_ASM_VERSION
 java_asm_server = { path = "asm_server", version = "0.0.15" } # JAVA_ASM_VERSION
 
+trie-rs = "0.4.2"
 enum_dispatch = "0.3.13"
 zip = "2.2.0"
 

Cargo.toml

@@ -1,5 +1,6 @@
 use crate::app::EguiApp;
-use egui::{Context, Ui};
+use egui::containers::PopupCloseBehavior;
+use egui::{popup_below_widget, Context, Ui};
 use java_asm_server::AsmServer;
 use std::ops::Deref;
 
@@ -35,16 +36,40 @@ impl EguiApp {
     }
 
     fn file_path_input(&mut self, ui: &mut Ui) {
-        let server_locked = self.server.lock();
-        let Some(server) = server_locked.deref() else { return; };
-        let accessor = server.accessor.lock();
-        let Some(accessor) = accessor.deref() else { return; };
-        // let all_classes: Vec<String> = accessor.read_classes().iter().map(|s| s.to_string()).collect();
         let mut locked_top = self.server_app.top().lock();
         let Some(file_path) = &mut locked_top.file_path else { return; };
-        ui.label(file_path.to_string());
-        // ui.add(AutoCompleteTextEdit::new(
-        //     file_path, all_classes,
-        // ));
+        let edit_path_ui = ui.text_edit_singleline(file_path);
+
+        let popup_id = ui.make_persistent_id("file_path_popup");
+        if edit_path_ui.gained_focus() {
+            let server_locked = self.server.lock();
+            let Some(server) = server_locked.deref() else { return; };
+            server.search(&mut locked_top);
+            ui.memory_mut(|m| m.open_popup(popup_id));
+        }
+
+        let search_results = locked_top.search_result.clone();
+        drop(locked_top);
+
+        if search_results.is_empty() { return; }
+        popup_below_widget(
+            ui, popup_id, &edit_path_ui,
+            PopupCloseBehavior::CloseOnClickOutside, |ui| {
+                ui.vertical(|ui| {
+                    for result in search_results {
+                        if ui.label(result.to_string()).clicked() {
+                            let server_locked = self.server.lock();
+                            let Some(server) = server_locked.deref() else { return; };
+                            let mut content_locked = self.server_app.content().lock();
+                            let mut locked_top = self.server_app.top().lock();
+                            let accessor = server.accessor.lock();
+                            let Some(accessor) = accessor.deref() else { return; };
+                            server.switch_or_open_lock_free(
+                                &result, accessor, &mut content_locked, &mut locked_top,
+                            );
+                        }
+                    }
+                })
+            });
     }
 }

asm_egui/src/top_bar.rs

@@ -22,3 +22,4 @@ enum_dispatch = { workspace = true }
 
 tokio = { workspace = true }
 parking_lot = { workspace = true }
+trie-rs = { workspace = true }

asm_server/Cargo.toml

@@ -2,6 +2,7 @@ use crate::impls::apk_load::read_apk;
 use crate::server::OpenFileError;
 use crate::ui::{AppContainer, DirInfo, Left};
 use crate::{AccessorEnum, AccessorMut, AsmServer, ServerMut};
+use java_asm::StrRef;
 use log::info;
 use std::io::{Read, Seek};
 use std::ops::DerefMut;
@@ -14,6 +15,7 @@ use zip::ZipArchive;
 
 pub enum ServerMessage {
     Progress(ProgressMessage),
+    SearchResult(SearchResultMessage),
 }
 
 pub struct ProgressMessage {
@@ -22,6 +24,10 @@ pub struct ProgressMessage {
     pub in_loading: bool,
 }
 
+pub struct SearchResultMessage {
+    pub result: Vec<StrRef>,
+}
+
 pub struct FileOpenContext {
     pub path: String,
     pub start_time: Instant,
@@ -47,6 +53,9 @@ impl AsmServer {
                         server_ref.loading_state.in_loading = progress.in_loading;
                         server_ref.on_progress_update(&render_target);
                     }
+                    ServerMessage::SearchResult(result) => {
+                        Self::on_search_result(result, &render_target);
+                    }
                 }
             }
         });
@@ -83,6 +92,13 @@ impl AsmServer {
         (*top_mut).loading_state = current_loading_state.clone();
     }
 
+    fn on_search_result(result: SearchResultMessage, render_target: &AppContainer) {
+        let result = result.result;
+        let mut top = render_target.top().lock();
+        let top_mut = top.deref_mut();
+        (*top_mut).search_result = result;
+    }
+
     fn render_to_app(&self, app: AppContainer) {
         let classes = self.read_classes();
         let start = Instant::now();

asm_server/src/impls/server.rs

@@ -4,6 +4,7 @@ use java_asm::smali::SmaliNode;
 use java_asm::{DescriptorRef, StrRef};
 use parking_lot::Mutex;
 use std::sync::Arc;
+use trie_rs::Trie;
 
 pub mod server;
 
@@ -16,10 +17,14 @@ pub struct AsmServer {
     pub loading_state: LoadingState,
     // when in loading state, the accessor is None.
     pub accessor: AccessorMut,
+    classes: ArcNullable<Vec<StrRef>>,
+    trie: ArcNullable<Trie<u8>>,
 }
 
-pub type ServerMut = Arc<Mutex<Option<AsmServer>>>;
-type AccessorMut = Arc<Mutex<Option<AccessorEnum>>>;
+pub type ArcNullable<T> = Arc<Mutex<Option<T>>>;
+
+pub type ServerMut = ArcNullable<AsmServer>;
+type AccessorMut = ArcNullable<AccessorEnum>;
 
 #[derive(Default, Clone, Debug)]
 pub struct LoadingState {

asm_server/src/lib.rs

@@ -1,14 +1,15 @@
 use crate::impls::server::FileOpenContext;
 use crate::impls::util::new_tokio_thread;
 use crate::ui::{AppContainer, Content, Tab, Top};
-use crate::{Accessor, AccessorEnum, AsmServer, LoadingState, ServerMut};
+use crate::{Accessor, AccessorEnum, ArcNullable, AsmServer, LoadingState, ServerMut};
 use java_asm::smali::SmaliNode;
 use java_asm::{AsmErr, StrRef};
 use log::{error, info};
 use std::fs::File;
-use std::ops::{Deref, DerefMut};
+use std::ops::Deref;
 use std::sync::Arc;
 use std::time::Instant;
+use trie_rs::{Trie, TrieBuilder};
 use zip::result::ZipError;
 
 impl AsmServer {
@@ -20,9 +21,35 @@ impl AsmServer {
                 err: None,
             },
             accessor: Default::default(),
+            classes: Default::default(),
+            trie: Default::default(),
         }
     }
 
+    pub fn get_classes(&self) -> &ArcNullable<Vec<StrRef>> {
+        let mut current = self.classes.lock();
+        if current.is_some() { return &self.classes; }
+        let accessor_locked = self.accessor.lock();
+        let Some(accessor) = accessor_locked.deref() else { return &self.classes; };
+        let classes = accessor.read_classes();
+        current.replace(classes);
+        &self.classes
+    }
+
+    pub fn get_trie(&self) -> &ArcNullable<Trie<u8>> {
+        let mut current = self.trie.lock();
+        if current.is_some() { return &self.trie; }
+        let classes_locked = self.get_classes().lock();
+        let Some(classes) = classes_locked.deref() else { return &self.trie; };
+        let mut trie_builder = TrieBuilder::new();
+        for class in classes.iter() {
+            trie_builder.push(class.to_string());
+        };
+        let trie = trie_builder.build();
+        current.replace(trie);
+        &self.trie
+    }
+
     pub fn smart_open(server: ServerMut, path: &str, render_target: AppContainer) {
         let context = FileOpenContext { path: path.to_string(), start_time: Instant::now() };
         new_tokio_thread(|runtime| async move {
@@ -74,7 +101,10 @@ impl AsmServer {
         }
 
         let smali = accessor.read_content(file_key);
-        let Some(smali) = smali else { return; };
+        let Some(smali) = smali else {
+            error!("content with key: `{file_key}` not found.");
+            return;
+        };
         let current_tab = Tab {
             selected: false,
             file_key: Arc::from(file_key),
@@ -87,6 +117,14 @@ impl AsmServer {
 
         top.file_path = Some(file_key.to_string());
     }
+
+    pub fn search(&self, top: &mut Top) {
+        let Some(query) = &top.file_path else { return; };
+        let trie_locked = self.get_trie().lock();
+        let Some(trie) = trie_locked.deref() else { return; };
+        let results: Vec<String> = trie.predictive_search(query).collect();
+        top.search_result = results.into_iter().map(StrRef::from).collect();
+    }
 }
 
 

asm_server/src/server.rs

@@ -6,7 +6,7 @@ use crate::ui::log::LogHolder;
 use crate::ui::AbsFile::{Dir, File};
 use crate::LoadingState;
 use java_asm::smali::SmaliNode;
-use java_asm::StrRef;
+use java_asm::{DescriptorRef, StrRef};
 use ::log::Level;
 use parking_lot::Mutex;
 use std::collections::BTreeMap;
@@ -41,6 +41,7 @@ impl AppContainer {
 pub struct Top {
     pub loading_state: LoadingState,
     pub file_path: Option<String>,
+    pub search_result: Vec<StrRef>,
 }
 
 #[derive(Default, Clone, Debug)]