Remove Quartz Scheduler, You.com model change topic, theme utils, and include other basic refactoring

2026-05-15 09:42:30 +00:00 · 2023-11-21 22:47:09 +02:00 · 2023-11-21 22:47:09 +02:00 · 53bdbcd4f5
commit 53bdbcd4f5
parent 73870cca40
57 changed files with 385 additions and 507 deletions
--- a/src/main/java/ee/carlrobert/codegpt/util/OverlayUtil.java
+++ b/src/main/java/ee/carlrobert/codegpt/util/OverlayUtil.java
@ -0,0 +1,166 @@
+package ee.carlrobert.codegpt.util;
+
+import static com.intellij.openapi.ui.Messages.CANCEL;
+import static com.intellij.openapi.ui.Messages.OK;
+import static ee.carlrobert.codegpt.Icons.DefaultIcon;
+import static java.lang.String.format;
+import static java.util.Objects.requireNonNull;
+
+import com.intellij.execution.ExecutionBundle;
+import com.intellij.notification.Notification;
+import com.intellij.notification.NotificationType;
+import com.intellij.notification.Notifications;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.ui.DialogBuilder;
+import com.intellij.openapi.ui.DoNotAskOption;
+import com.intellij.openapi.ui.MessageDialogBuilder;
+import com.intellij.openapi.ui.MessageType;
+import com.intellij.openapi.ui.Messages;
+import com.intellij.openapi.ui.popup.Balloon;
+import com.intellij.openapi.ui.popup.Balloon.Position;
+import com.intellij.openapi.ui.popup.JBPopupFactory;
+import com.intellij.ui.awt.RelativePoint;
+import com.intellij.ui.components.JBLabel;
+import com.intellij.util.ui.JBFont;
+import com.intellij.util.ui.JBUI;
+import ee.carlrobert.codegpt.CodeGPTBundle;
+import ee.carlrobert.codegpt.conversations.ConversationsState;
+import ee.carlrobert.codegpt.indexes.FolderStructureTreePanel;
+import ee.carlrobert.codegpt.settings.configuration.ConfigurationState;
+import java.awt.Point;
+import java.awt.event.MouseEvent;
+import javax.swing.JComponent;
+import org.jetbrains.annotations.NotNull;
+
+public class OverlayUtil {
+
+  public static void showNotification(String content, NotificationType type) {
+    Notifications.Bus.notify(
+        new Notification("CodeGPT Notification Group", "CodeGPT", content, type));
+  }
+
+  public static int showFileStructureDialog(
+      Project project,
+      FolderStructureTreePanel folderStructureTreePanel) {
+    var dialogBuilder = new DialogBuilder(project);
+    dialogBuilder.setNorthPanel(JBUI.Panels.simplePanel(new JBLabel(
+            "<html>"
+                + "<p>Indexing files enables direct queries related to your codebase.</p>"
+                + "<br/>"
+                + "<p>File indexing occurs locally on your computer; "
+                + "no files are sent to any 3rd party services.</p>"
+                + "<p>For additional information, refer to the "
+                + "<a href=\"https://google.com\">CodeGPT documentation</a>.</p>"
+                + "</html>")
+            .setCopyable(true)
+            .setAllowAutoWrapping(true)
+            .withFont(JBFont.medium()))
+        .withBorder(JBUI.Borders.emptyBottom(12)));
+    dialogBuilder.setCenterPanel(folderStructureTreePanel.getPanel());
+    dialogBuilder.addOkAction().setText("Start Indexing");
+    dialogBuilder.addCancelAction();
+    dialogBuilder.setTitle("Choose Files for Indexing");
+    return dialogBuilder.show();
+  }
+
+  public static int showDeleteConversationDialog() {
+    return Messages.showYesNoDialog(
+        CodeGPTBundle.get("dialog.deleteConversation.description"),
+        CodeGPTBundle.get("dialog.deleteConversation.title"),
+        DefaultIcon);
+  }
+
+  public static int showTokenLimitExceededDialog() {
+    return MessageDialogBuilder.okCancel(
+            CodeGPTBundle.get("dialog.tokenLimitExceeded.title"),
+            CodeGPTBundle.get("dialog.tokenLimitExceeded.description"))
+        .yesText(CodeGPTBundle.get("dialog.continue"))
+        .noText(CodeGPTBundle.get("dialog.cancel"))
+        .icon(DefaultIcon)
+        .doNotAsk(new DoNotAskOption.Adapter() {
+          @Override
+          public void rememberChoice(boolean isSelected, int exitCode) {
+            if (isSelected) {
+              ConversationsState.getInstance().discardAllTokenLimits();
+            }
+          }
+
+          @NotNull
+          @Override
+          public String getDoNotShowMessage() {
+            return ExecutionBundle.message("don.t.ask.again");
+          }
+
+          @Override
+          public boolean shouldSaveOptionsOnCancel() {
+            return true;
+          }
+        })
+        .guessWindowAndAsk() ? OK : CANCEL;
+  }
+
+  public static int showTokenSoftLimitWarningDialog(int tokenCount) {
+    return MessageDialogBuilder.okCancel(
+            CodeGPTBundle.get("dialog.tokenSoftLimitExceeded.title"),
+            format(CodeGPTBundle.get("dialog.tokenSoftLimitExceeded.description"), tokenCount))
+        .yesText(CodeGPTBundle.get("dialog.continue"))
+        .noText(CodeGPTBundle.get("dialog.cancel"))
+        .icon(DefaultIcon)
+        .doNotAsk(new DoNotAskOption.Adapter() {
+          @Override
+          public void rememberChoice(boolean isSelected, int exitCode) {
+            if (isSelected) {
+              ConfigurationState.getInstance().setIgnoreGitCommitTokenLimit(true);
+            }
+          }
+
+          @NotNull
+          @Override
+          public String getDoNotShowMessage() {
+            return ExecutionBundle.message("don.t.ask.again");
+          }
+
+          @Override
+          public boolean shouldSaveOptionsOnCancel() {
+            return true;
+          }
+        })
+        .guessWindowAndAsk() ? OK : CANCEL;
+  }
+
+  public static void showSelectedEditorSelectionWarning(AnActionEvent event) {
+    var locationOnScreen = ((MouseEvent) event.getInputEvent()).getLocationOnScreen();
+    locationOnScreen.y = locationOnScreen.y - 16;
+
+    showWarningBalloon(
+        EditorUtil.getSelectedEditor(requireNonNull(event.getProject())) == null
+            ? "Unable to locate a selected editor"
+            : "Please select a target code before proceeding",
+        locationOnScreen);
+  }
+
+  public static void showWarningBalloon(String content, Point locationOnScreen) {
+    showBalloon(content, MessageType.WARNING, locationOnScreen);
+  }
+
+  public static void showInfoBalloon(String content, Point locationOnScreen) {
+    showBalloon(content, MessageType.INFO, locationOnScreen);
+  }
+
+  private static void showBalloon(String content, MessageType messageType, Point locationOnScreen) {
+    JBPopupFactory.getInstance()
+        .createHtmlTextBalloonBuilder(content, messageType, null)
+        .setFadeoutTime(2500)
+        .createBalloon()
+        .show(RelativePoint.fromScreen(locationOnScreen), Balloon.Position.above);
+  }
+
+  public static void showBalloon(String content, MessageType messageType, JComponent component) {
+    JBPopupFactory.getInstance()
+        .createHtmlTextBalloonBuilder(content, messageType, null)
+        .setFadeoutTime(2500)
+        .createBalloon()
+        .show(RelativePoint.getSouthOf(component), Position.below);
+  }
+}