add copy last code block menu item

Author: pariskol

src/main/java/gr/sqlbrowserfx/nodes/ChatGptWebView.java

@@ -7,6 +7,7 @@
 import gr.sqlbrowserfx.utils.JavaFXUtils;
 import javafx.scene.control.ContextMenu;
 import javafx.scene.control.MenuItem;
+import javafx.scene.control.SeparatorMenuItem;
 import javafx.scene.input.Clipboard;
 import javafx.scene.input.ClipboardContent;
 import javafx.scene.input.KeyCode;
@@ -77,8 +78,16 @@ public class ChatGptWebView extends BorderPane implements ContextMenuOwner, Inpu
 			btn.click();
         })();
 	""";
-
-	private final WebView webView = new WebView();;
+	
+	private final String getLastCodeJs = """
+			(function() {
+				const codeElements = document.querySelectorAll('code');
+				const lastCode = codeElements[codeElements.length - 1];
+				return lastCode.textContent;
+	        })();
+		""";
+	
+	private final WebView webView = new WebView();
 
 	public ChatGptWebView() {	    
 	    var webEngine = webView.getEngine();
@@ -92,6 +101,7 @@ public ChatGptWebView() {
 
 	    this.setContextMenu();
 	    this.setInputMap();
+	    
 	    this.setCenter(webView);
 	}
 
@@ -152,6 +162,10 @@ private void clickAskButton() {
 		this.getEngine().executeScript(this.clickAskButtonJs);
 	}
 
+	public String getAiGeneratedCode() {
+		return (String) this.getEngine().executeScript(this.getLastCodeJs);
+	}
+
 	private void copySelectedTextToClipoboard() {
 		var text = (String) webView.getEngine().executeScript(this.selectedTextJs);
 	    var clipboard = Clipboard.getSystemClipboard();
@@ -160,15 +174,26 @@ private void copySelectedTextToClipoboard() {
 	    clipboard.setContent(content);
 	}
 
+	private void copyLastCodeBlockToClipboard() {
+		var text = (String) webView.getEngine().executeScript(this.getLastCodeJs);
+	    var clipboard = Clipboard.getSystemClipboard();
+	    var content = new ClipboardContent();
+	    content.putString(text);
+	    clipboard.setContent(content);
+	}
+	
 	@Override
 	public ContextMenu createContextMenu() {
 		var copySelectedHtmlText = new MenuItem("Copy Selected Text", JavaFXUtils.createIcon("/icons/copy.png"));
 	    copySelectedHtmlText.setOnAction(copyAction -> this.copySelectedTextToClipoboard());
 
+		var copyLastCodeBlock = new MenuItem("Copy Last Code Block", JavaFXUtils.createIcon("/icons/var.png"));
+		copyLastCodeBlock.setOnAction(copyAction -> this.copyLastCodeBlockToClipboard());
+	    
 	    var refresh = new MenuItem("Refresh", JavaFXUtils.createIcon("/icons/refresh.png"));
 	    refresh.setOnAction(copyAction -> this.getEngine().reload());
 
-	    return new ContextMenu(copySelectedHtmlText, refresh);
+	    return new ContextMenu(copySelectedHtmlText, copyLastCodeBlock, new SeparatorMenuItem(), refresh);
 	}
 
 }

src/test/java/sqlbrowserfx/ChatGptWebViewTestGui.java

@@ -2,8 +2,11 @@
 
 import java.io.IOException;
 
+import org.fxmisc.richtext.CodeArea;
+
 import gr.sqlbrowserfx.nodes.ChatGptWebView;
 import javafx.application.Application;
+import javafx.application.Platform;
 import javafx.scene.Scene;
 import javafx.scene.control.Button;
 import javafx.scene.layout.VBox;
@@ -15,8 +18,13 @@ public class ChatGptWebViewTestGui extends Application {
 	public void start(Stage primaryStage) throws IOException {
 		var chatGptWebView = new ChatGptWebView();
 	    var pasteButton = new Button("test");
+	    var codeButton = new Button("code");
+	    var codeArea = new CodeArea();
+	    codeButton.setOnAction(event -> {
+	    	System.out.println(chatGptWebView.getAiGeneratedCode());
+	    });
 	    pasteButton.setOnAction(event -> {
-	    	String safeText = "This is a test text pasted from java"
+	    	String safeText = "Generate only code and only one code block. Generate an sql query"
 	    	        .replace("\\", "\\\\")
 	    	        .replace("\"", "\\\"")
 	    	        .replace("\n", "\\n")
@@ -35,9 +43,21 @@ public void start(Stage primaryStage) throws IOException {
 	    	    """.formatted(safeText);
 
 	    	    chatGptWebView.getEngine().executeScript(js);
+	        	new Thread(() -> {
+	        		try {
+	        			Thread.sleep(3000);
+	        			System.out.println("Getting code");
+	        			Platform.runLater(() -> {
+	        				codeArea.appendText(chatGptWebView.getAiGeneratedCode());
+	        			});
+	        		} catch(Exception e) {
+	        			// Ignore
+	        		}
+	        	}).start();
+	        	
 	    });
 		Scene scene = new Scene(
-				new VBox(pasteButton, chatGptWebView)
+				new VBox(pasteButton, codeButton, chatGptWebView, codeArea)
 		);
 		scene.getStylesheets().add("/styles/flat-dark.css");
 		primaryStage.setScene(scene);