From 214a4f8617d3c2fc763d7b0805c461a229b64da0 Mon Sep 17 00:00:00 2001
From: Lorenzo Coppi <lorenzo.coppi@cys4.com>
Date: Wed, 17 Jan 2024 16:45:43 +0100
Subject: [PATCH] refactor: rename MainUI initialization

---
 src/main/java/burp/BurpExtender.java               |  2 +-
 .../java/com/cys4/sensitivediscoverer/MainUI.java  | 14 +++++++++++---
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/main/java/burp/BurpExtender.java b/src/main/java/burp/BurpExtender.java
index 6ed6b5f..ce073b2 100644
--- a/src/main/java/burp/BurpExtender.java
+++ b/src/main/java/burp/BurpExtender.java
@@ -13,7 +13,7 @@ public class BurpExtender implements IBurpExtender {
     public void registerExtenderCallbacks(final IBurpExtenderCallbacks callbacks) {
         try {
             MainUI mainUI = new MainUI(callbacks);
-            mainUI.initialize();
+            mainUI.initializeUI();
 
             callbacks.setExtensionName(mainUI.getNameExtension());
 
diff --git a/src/main/java/com/cys4/sensitivediscoverer/MainUI.java b/src/main/java/com/cys4/sensitivediscoverer/MainUI.java
index 3f35606..fef5021 100644
--- a/src/main/java/com/cys4/sensitivediscoverer/MainUI.java
+++ b/src/main/java/com/cys4/sensitivediscoverer/MainUI.java
@@ -29,8 +29,10 @@ public class MainUI implements ITab {
     private final Properties configProperties;
     private final ScannerOptions scannerOptions;
     private JTabbedPane mainPanel;
+    private boolean interfaceInitialized;
 
     public MainUI(IBurpExtenderCallbacks callbacks) throws Exception {
+        this.interfaceInitialized = false;
         this.callbacks = callbacks;
 
         // setup stdout/stderr
@@ -50,6 +52,10 @@ public MainUI(IBurpExtenderCallbacks callbacks) throws Exception {
         this.extensionsRegexList = RegexSeeder.getExtensionRegexes();
     }
 
+    public boolean isInterfaceInitialized() {
+        return interfaceInitialized;
+    }
+
     public ScannerOptions getScannerOptions() {
         return scannerOptions;
     }
@@ -57,11 +63,11 @@ public ScannerOptions getScannerOptions() {
     /**
      * Main function that initializes the extension and creates the UI, asynchronously
      */
-    public void initialize() {
-        SwingUtilities.invokeLater(this::_initialize);
+    public void initializeUI() {
+        SwingUtilities.invokeLater(this::_initializeUI);
     }
 
-    private void _initialize() {
+    private void _initializeUI() {
         mainPanel = new JTabbedPane();
         LoggerTab loggerTab = new LoggerTab(this);
         mainPanel.addTab(loggerTab.getTabName(), loggerTab.getPanel());
@@ -72,6 +78,8 @@ private void _initialize() {
 
         callbacks.customizeUiComponent(mainPanel);
         callbacks.addSuiteTab(MainUI.this);
+
+        this.interfaceInitialized = true;
     }
 
     /**