Merge pull request #8 from devchat-ai/feat-diff

Feat: Implement Diff View

Author: pplam

src/main/java/ai/devchat/devchat/handler/ViewDiffRequestHandler.java

@@ -1,10 +1,20 @@
 package ai.devchat.devchat.handler;
 
-import ai.devchat.common.Log;
 import ai.devchat.devchat.ActionHandler;
 import ai.devchat.devchat.DevChatActionHandler;
 import com.alibaba.fastjson.JSONObject;
-import com.intellij.psi.PsiFileFactory;
+import com.intellij.diff.DiffContentFactory;
+import com.intellij.diff.DiffManager;
+import com.intellij.diff.requests.SimpleDiffRequest;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.editor.Document;
+import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.editor.SelectionModel;
+import com.intellij.openapi.fileEditor.FileDocumentManager;
+import com.intellij.openapi.fileEditor.FileEditorManager;
+import com.intellij.openapi.fileTypes.FileType;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.vfs.VirtualFile;
 
 public class ViewDiffRequestHandler implements ActionHandler {
     private JSONObject metadata;
@@ -15,14 +25,42 @@ public ViewDiffRequestHandler(DevChatActionHandler devChatActionHandler) {
         this.devChatActionHandler = devChatActionHandler;
     }
 
+
     @Override
     public void executeAction() {
-        Log.info("Handling view diff request");
-        String callbackFunc = metadata.getString("callback");
         String diffContent = payload.getString("content");
+        Project project = devChatActionHandler.getProject();
+
+        ApplicationManager.getApplication().invokeLater(() -> {
+            Editor editor = FileEditorManager.getInstance(project).getSelectedTextEditor();
+            if (editor == null) {
+                // Handle the case when no editor is opened
+                return;
+            }
+            Document document = editor.getDocument();
+            VirtualFile file = FileDocumentManager.getInstance().getFile(document);
+            if (file == null) {
+                // Handle the case when no file corresponds to the document
+                return;
+            }
+
+            FileType fileType = file.getFileType();
+            SelectionModel selectionModel = editor.getSelectionModel();
+            String localContent = selectionModel.hasSelection()
+                    ? selectionModel.getSelectedText()
+                    : document.getText();
+
+            DiffContentFactory contentFactory = DiffContentFactory.getInstance();
+            SimpleDiffRequest diffRequest = new SimpleDiffRequest(
+                    "Code Diff",
+                    contentFactory.create(localContent, fileType),
+                    contentFactory.create(diffContent, fileType),
+                    "Current Code",
+                    "New Code"
+            );
 
-        PsiFileFactory psiFileFactory = PsiFileFactory.getInstance(devChatActionHandler.getProject());
-//        PsiFile psiFile = psiFileFactory.createFileFromText("yourFileName.java", , diffContent);
+            DiffManager.getInstance().showDiff(project, diffRequest);
+        });
     }
 
     @Override