feat: use attach to existing instead of choose source

Author: robertmarsal

reai_toolkit/features/ai_decompiler/ai_decompiler.py

@@ -87,7 +87,7 @@ def start_ai_decompiler(self, bv: BinaryView, options: Dict) -> None:
             callback = options.get("callback")
             analysis_id = self.config.get_analysis_id(bv)
             if not analysis_id:
-                raise Exception("Analysis not found. Please choose one using 'Choose Source' feature.")
+                raise Exception("Analysis not found. Please choose one using the 'Attach to existing' feature.")
             function_id = get_function_id_by_addr_util(bv, function.start, self.config)
 
             with revengai.ApiClient(self.config.api_config) as api_client:

reai_toolkit/features/choose_source/__init__.py

@@ -7,19 +7,19 @@ class ChooseSourceFeature(BaseAuthFeature):
     def __init__(self, config=None):
         super().__init__(config)
         self.choose_source = ChooseSource(config)
-        log_info("RevEng.AI | Choose Source Feature initialized")
+        log_info("RevEng.AI | Attach to existing Feature initialized")
 
     def register(self):
         PluginCommand.register(
-            "RevEng.AI\\Analysis\​​Choose Source",
+            "RevEng.AI\\Analysis\​​Attach to existing",
             "Choose a source for the binary analysis",
             self.show_choose_source_dialog,
             self.is_valid
         )
-        log_info("RevEng.AI | Choose Source Feature registered")
+        log_info("RevEng.AI | Attach to existing Feature registered")
 
     def show_choose_source_dialog(self, bv: BinaryView):
-        log_info("RevEng.AI | Opening Choose Source dialog")
+        log_info("RevEng.AI | Opening Attach to existing dialog")
         dialog = ChooseSourceDialog(self.config, self.choose_source, bv)
         dialog.exec_()
 

reai_toolkit/features/choose_source/choose_source.py

@@ -31,7 +31,7 @@ def choose_source(self, bv: BinaryView, chose: str):
 
             return True, "Binary ID changed successfully."
         except Exception as e:
-            log_error(f"RevEng.AI | Failed to choose source: {str(e)}")
+            log_error(f"RevEng.AI | Failed to attach to existing analysis: {str(e)}")
             return False, str(e)
 
     def get_analysis(self, bv: BinaryView):

reai_toolkit/features/choose_source/choose_source_dialog.py

@@ -16,7 +16,7 @@ def __init__(self, config, choose_source, bv):
         self.init_ui()
 
     def init_ui(self):
-        self.setWindowTitle("RevEng.AI: Choose Source")
+        self.setWindowTitle("RevEng.AI: Attach to existing")
         self.setMinimumWidth(500)
         layout = QVBoxLayout()
 
@@ -106,10 +106,10 @@ def _on_analysis_loaded(self, success, analysis):
     def _choose_source(self):
         if not self.combo.currentText():
             log_error("RevEng.AI | Source selection is required")
-            QMessageBox.warning(self, "RevEng.AI Choose Source", "Please select a source for analysis.", QMessageBox.Ok)
+            QMessageBox.warning(self, "RevEng.AI: Attach to exiting", "Please select a source for analysis.", QMessageBox.Ok)
             return
 
-        self.progress = create_progress_dialog(self, "RevEng.AI Choose Source", "Choosing source...")
+        self.progress = create_progress_dialog(self, "RevEng.AI: Attach to existing", "Choosing source...")
         self.progress.show()
         QCoreApplication.processEvents()
 
@@ -121,9 +121,9 @@ def _on_choose_source_finished(self, success, message):
         self.progress.close()
 
         if success:
-            QMessageBox.information(self, "RevEng.AI Choose Source", message, QMessageBox.Ok)
+            QMessageBox.information(self, "RevEng.AI: Attach to existing", message, QMessageBox.Ok)
             self.accept()
         else:
-            log_error(f"RevEng.AI | Failed to choose source: {message}")
-            QMessageBox.critical(self, "RevEng.AI Choose Source Error", f"Failed to choose source: {message}", QMessageBox.Ok)
+            log_error(f"RevEng.AI | Failed to attach to existing analysis: {message}")
+            QMessageBox.critical(self, "RevEng.AI: Attach to existing", f"Failed to choose source: {message}", QMessageBox.Ok)
             self.reject()

reai_toolkit/features/match_current_function/match_current_function.py

@@ -54,7 +54,7 @@ def parse_confidence(item):
 
             analysis_id = self.config.get_analysis_id(bv)
             if not analysis_id:
-                raise Exception("Analysis not found. Please choose one using 'Choose Source' feature.")
+                raise Exception("Analysis not found. Please choose one using the 'Attach to existing' feature.")
 
             if self.cancelled.is_set():
                 return False, "Operation cancelled"

reai_toolkit/features/match_functions/match_functions.py

@@ -44,7 +44,7 @@ def parse_confidence(item):
 
             analysis_id = self.config.get_analysis_id(bv)
             if not analysis_id:
-                raise Exception("Analysis not found. Please choose one using 'Choose Source' feature.")
+                raise Exception("Analysis not found. Please choose one using the 'Attach to existing' feature.")
 
             if self.cancelled.is_set():
                 return False, "Operation cancelled"

reai_toolkit/features/view_function_in_portal/view_function_in_portal.py

@@ -24,7 +24,7 @@ def view_function_in_portal(self, bv: BinaryView, options: Dict) -> None:
 
             analysis_id = self.config.get_analysis_id(bv)
             if not analysis_id:
-                raise Exception("Analysis not found. Please choose one using 'Choose Source' feature.")
+                raise Exception("Analysis not found. Please choose one using the 'Attach to existing' feature.")
 
             with revengai.ApiClient(self.config.api_config) as api_client:
                 api_instance = revengai.AnalysesResultsMetadataApi(api_client)