Added event listeners(not implemented)

# TODO
- fix broken shuffle logic
- implement event listeners
- Many buggy features to fix

Signed-off-by: zhaoxinn <[email protected]>

Author: zxzinn

src/main/java/com/zxzinn/novelai/event/FileOperationEvent.java

@@ -0,0 +1,25 @@
+package com.zxzinn.novelai.event;
+
+import lombok.Getter;
+
+import java.io.File;
+import java.util.EventListener;
+import java.util.EventObject;
+
+@Getter
+public class FileOperationEvent extends EventObject {
+    public enum OperationType {
+        ADDED, DELETED, MODIFIED
+    }
+
+    private final File file;
+    private final OperationType operationType;
+
+    public FileOperationEvent(Object source, File file, OperationType operationType) {
+        super(source);
+        this.file = file;
+        this.operationType = operationType;
+    }
+
+}
+

src/main/java/com/zxzinn/novelai/event/FileOperationListener.java

@@ -0,0 +1,7 @@
+package com.zxzinn.novelai.event;
+
+import java.util.EventListener;
+
+public interface FileOperationListener extends EventListener {
+    void onFileOperation(FileOperationEvent event);
+}

src/main/java/com/zxzinn/novelai/event/GenerationStateChangeEvent.java

@@ -0,0 +1,19 @@
+package com.zxzinn.novelai.event;
+
+import com.zxzinn.novelai.GenerationState;
+import lombok.Getter;
+
+import java.util.EventListener;
+import java.util.EventObject;
+
+@Getter
+public class GenerationStateChangeEvent extends EventObject {
+    private final GenerationState newState;
+
+    public GenerationStateChangeEvent(Object source, GenerationState newState) {
+        super(source);
+        this.newState = newState;
+    }
+
+}
+

src/main/java/com/zxzinn/novelai/event/GenerationStateChangeListener.java

@@ -0,0 +1,7 @@
+package com.zxzinn.novelai.event;
+
+import java.util.EventListener;
+
+public interface GenerationStateChangeListener extends EventListener {
+    void onGenerationStateChange(GenerationStateChangeEvent event);
+}

src/main/java/com/zxzinn/novelai/event/ImageReceivedEvent.java

@@ -0,0 +1,19 @@
+package com.zxzinn.novelai.event;
+
+import lombok.Getter;
+
+import java.awt.image.BufferedImage;
+import java.util.EventListener;
+import java.util.EventObject;
+
+@Getter
+public class ImageReceivedEvent extends EventObject {
+    private final BufferedImage image;
+
+    public ImageReceivedEvent(Object source, BufferedImage image) {
+        super(source);
+        this.image = image;
+    }
+
+}
+

src/main/java/com/zxzinn/novelai/event/ImageReceivedListener.java

@@ -0,0 +1,7 @@
+package com.zxzinn.novelai.event;
+
+import java.util.EventListener;
+
+public interface ImageReceivedListener extends EventListener {
+    void onImageReceived(ImageReceivedEvent event);
+}

src/main/java/com/zxzinn/novelai/event/ParameterChangeEvent.java

@@ -0,0 +1,20 @@
+package com.zxzinn.novelai.event;
+
+import lombok.Getter;
+
+import java.util.EventListener;
+import java.util.EventObject;
+
+@Getter
+public class ParameterChangeEvent extends EventObject {
+    private final String parameterName;
+    private final Object newValue;
+
+    public ParameterChangeEvent(Object source, String parameterName, Object newValue) {
+        super(source);
+        this.parameterName = parameterName;
+        this.newValue = newValue;
+    }
+
+}
+

src/main/java/com/zxzinn/novelai/event/ParameterChangeListener.java

@@ -0,0 +1,7 @@
+package com.zxzinn.novelai.event;
+
+import java.util.EventListener;
+
+public interface ParameterChangeListener extends EventListener {
+    void onParameterChange(ParameterChangeEvent event);
+}

src/main/java/com/zxzinn/novelai/event/PromptUpdateEvent.java

@@ -0,0 +1,20 @@
+package com.zxzinn.novelai.event;
+
+import lombok.Getter;
+
+import java.util.EventListener;
+import java.util.EventObject;
+
+@Getter
+public class PromptUpdateEvent extends EventObject {
+    private final String positivePrompt;
+    private final String negativePrompt;
+
+    public PromptUpdateEvent(Object source, String positivePrompt, String negativePrompt) {
+        super(source);
+        this.positivePrompt = positivePrompt;
+        this.negativePrompt = negativePrompt;
+    }
+
+}
+

src/main/java/com/zxzinn/novelai/event/PromptUpdateListener.java

@@ -0,0 +1,7 @@
+package com.zxzinn.novelai.event;
+
+import java.util.EventListener;
+
+public interface PromptUpdateListener extends EventListener {
+    void onPromptUpdate(PromptUpdateEvent event);
+}

src/main/java/com/zxzinn/novelai/gui/MainGUI.java

@@ -5,6 +5,8 @@
 import com.zxzinn.novelai.api.GenerationRequestBuilder;
 import com.zxzinn.novelai.api.NAIConstants;
 import com.zxzinn.novelai.config.ConfigManager;
+import com.zxzinn.novelai.event.ImageReceivedEvent;
+import com.zxzinn.novelai.event.ImageReceivedListener;
 import com.zxzinn.novelai.gui.common.ImagePreviewPanel;
 import com.zxzinn.novelai.gui.filewindow.FileManagerTab;
 import com.zxzinn.novelai.gui.generation.*;
@@ -27,7 +29,7 @@
 @Log4j2
 @Getter
 @Setter
-public class MainGUI extends JFrame implements UIComponent {
+public class MainGUI extends JFrame implements UIComponent , ImageReceivedListener {
     private static final ConfigManager config = ConfigManager.getInstance();
     public static final int WINDOW_WIDTH = config.getInteger("ui.window.width");
     public static final int WINDOW_HEIGHT = config.getInteger("ui.window.height");
@@ -134,7 +136,7 @@ public void layoutComponents() {
     public void bindEvents() {
         actionComboBox.addActionListener(e -> updateParametersPanel());
         generationControlPanel.setOnGenerateRequested(this::startImageGeneration);
-        generationControlPanel.setOnImageReceived(this::onReceivedImage);
+        generationControlPanel.addImageReceivedListener(this);
     }
 
     private void startImageGeneration() {
@@ -205,7 +207,9 @@ private void refreshPromptPreview() {
         });
     }
 
-    private void onReceivedImage(BufferedImage image) {
+    @Override
+    public void onImageReceived(ImageReceivedEvent event) {
+        BufferedImage image = event.getImage();
         SwingUtilities.invokeLater(() -> {
             try {
                 imagePreviewPanel.setImage(image);

src/main/java/com/zxzinn/novelai/gui/generation/GenerationControlPanel.java

@@ -1,6 +1,8 @@
 package com.zxzinn.novelai.gui.generation;
 
 import com.zxzinn.novelai.GenerationState;
+import com.zxzinn.novelai.event.ImageReceivedEvent;
+import com.zxzinn.novelai.event.ImageReceivedListener;
 import com.zxzinn.novelai.utils.I18nManager;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
@@ -9,6 +11,8 @@
 import javax.swing.*;
 import java.awt.*;
 import java.awt.image.BufferedImage;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.function.Consumer;
 
@@ -18,6 +22,7 @@
 public class GenerationControlPanel extends JPanel {
     private final JButton controlButton;
     private final JComboBox<String> generationCountComboBox;
+    private final List<ImageReceivedListener> imageReceivedListeners = new ArrayList<>();
 
     private GenerationState currentState;
     private final AtomicInteger lastingCount = new AtomicInteger();
@@ -102,9 +107,18 @@ public void updateState(GenerationState newState) {
         });
     }
 
+    public void addImageReceivedListener(ImageReceivedListener listener) {
+        imageReceivedListeners.add(listener);
+    }
+
+    public void removeImageReceivedListener(ImageReceivedListener listener) {
+        imageReceivedListeners.remove(listener);
+    }
+
     public void onImageReceived(BufferedImage image) {
-        if (onImageReceived != null) {
-            onImageReceived.accept(image);
+        ImageReceivedEvent event = new ImageReceivedEvent(this, image);
+        for (ImageReceivedListener listener : imageReceivedListeners) {
+            listener.onImageReceived(event);
         }
     }
 }