Skip to content

Commit

Permalink
refactor: rename MainUI initialization
Browse files Browse the repository at this point in the history
  • Loading branch information
beryxz committed Jan 17, 2024
1 parent 33e9889 commit 214a4f8
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 4 deletions.
2 changes: 1 addition & 1 deletion src/main/java/burp/BurpExtender.java
Original file line number Diff line number Diff line change
Expand Up @@ -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());

Expand Down
14 changes: 11 additions & 3 deletions src/main/java/com/cys4/sensitivediscoverer/MainUI.java
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -50,18 +52,22 @@ public MainUI(IBurpExtenderCallbacks callbacks) throws Exception {
this.extensionsRegexList = RegexSeeder.getExtensionRegexes();
}

public boolean isInterfaceInitialized() {
return interfaceInitialized;
}

public ScannerOptions getScannerOptions() {
return scannerOptions;
}

/**
* 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());
Expand All @@ -72,6 +78,8 @@ private void _initialize() {

callbacks.customizeUiComponent(mainPanel);
callbacks.addSuiteTab(MainUI.this);

this.interfaceInitialized = true;
}

/**
Expand Down

0 comments on commit 214a4f8

Please sign in to comment.