feat: display notification on plugin updates

Author: carlrobertoh

src/main/java/ee/carlrobert/codegpt/CodeGPTPlugin.java

@@ -5,11 +5,15 @@
 import com.intellij.ide.plugins.PluginManagerCore;
 import com.intellij.openapi.application.PathManager;
 import com.intellij.openapi.extensions.PluginId;
+import com.intellij.openapi.progress.ProgressIndicator;
 import com.intellij.openapi.project.Project;
-import ee.carlrobert.codegpt.telemetry.core.util.Directories;
+import com.intellij.openapi.updateSettings.impl.PluginDownloader;
+import com.intellij.openapi.updateSettings.impl.UpdateChecker;
 import java.io.File;
 import java.nio.file.Path;
 import java.nio.file.Paths;
+import java.util.Collection;
+import java.util.Optional;
 import org.jetbrains.annotations.NotNull;
 
 public final class CodeGPTPlugin {
@@ -46,4 +50,18 @@ private CodeGPTPlugin() {
   public static @NotNull String getProjectIndexStorePath(@NotNull Project project) {
     return getIndexStorePath() + File.separator + project.getName();
   }
+
+  public static Optional<PluginDownloader> tryFindAvailableUpdate(
+      @NotNull ProgressIndicator indicator) {
+    return findAvailableUpdates(indicator).stream()
+        .filter((update) -> CODEGPT_ID.equals(update.getId()))
+        .findFirst();
+  }
+
+  private static @NotNull Collection<PluginDownloader> findAvailableUpdates(
+      @NotNull ProgressIndicator indicator) {
+    return UpdateChecker.getInternalPluginUpdates(null, indicator)
+        .getPluginUpdates()
+        .getAllEnabled();
+  }
 }

src/main/java/ee/carlrobert/codegpt/CodeGPTUpdateStartupActivity.java

@@ -0,0 +1,68 @@
+package ee.carlrobert.codegpt;
+
+import com.intellij.notification.NotificationAction;
+import com.intellij.notification.NotificationType;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.progress.ProgressIndicator;
+import com.intellij.openapi.progress.Task;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.startup.StartupActivity;
+import com.intellij.openapi.updateSettings.impl.UpdateChecker;
+import com.intellij.openapi.updateSettings.impl.UpdateSettings;
+import com.intellij.util.concurrency.AppExecutorUtil;
+import ee.carlrobert.codegpt.settings.configuration.ConfigurationState;
+import ee.carlrobert.codegpt.util.OverlayUtil;
+import java.util.concurrent.TimeUnit;
+import org.jetbrains.annotations.NotNull;
+
+public class CodeGPTUpdateStartupActivity implements StartupActivity.Background {
+
+  @Override
+  public void runActivity(@NotNull Project project) {
+    schedulePluginUpdateChecks(project);
+  }
+
+  private void schedulePluginUpdateChecks(Project project) {
+    AppExecutorUtil.getAppScheduledExecutorService().scheduleWithFixedDelay(() -> {
+      if (project != null && ConfigurationState.getInstance().isCheckForPluginUpdates()) {
+        new CheckForUpdatesTask(project).queue();
+      }
+
+    }, 0, 4L, TimeUnit.HOURS);
+  }
+
+  private static class CheckForUpdatesTask extends Task.Backgroundable {
+
+    public CheckForUpdatesTask(@NotNull Project project) {
+      super(project, CodeGPTBundle.get("checkForUpdatesTask.title"), true);
+    }
+
+    private static void installCodeGPTUpdate(Project project) {
+      var settingsCopy = new UpdateSettings();
+      var settingsState = settingsCopy.getState();
+      settingsState.copyFrom(UpdateSettings.getInstance().getState());
+      settingsState.setCheckNeeded(true);
+      settingsState.setPluginsCheckNeeded(true);
+      settingsState.setShowWhatsNewEditor(true);
+      settingsState.setThirdPartyPluginsAllowed(true);
+      UpdateChecker.updateAndShowResult(project, settingsCopy);
+    }
+
+    public void run(@NotNull ProgressIndicator indicator) {
+      if (!myProject.isDisposed()) {
+        CodeGPTPlugin.tryFindAvailableUpdate(indicator)
+            .ifPresent((update) -> OverlayUtil.getDefaultNotification(
+                    CodeGPTBundle.get("checkForUpdatesTask.notification.message"),
+                    NotificationType.IDE_UPDATE)
+                .addAction(NotificationAction.createSimpleExpiring(
+                    CodeGPTBundle.get("checkForUpdatesTask.notification.installButton"),
+                    () -> ApplicationManager.getApplication()
+                        .executeOnPooledThread(() -> installCodeGPTUpdate(myProject))))
+                .addAction(NotificationAction.createSimpleExpiring(
+                    CodeGPTBundle.get("checkForUpdatesTask.notification.hideButton"),
+                    () -> ConfigurationState.getInstance().setCheckForPluginUpdates(false)))
+                .notify(myProject));
+      }
+    }
+  }
+}

src/main/java/ee/carlrobert/codegpt/settings/configuration/ConfigurationComponent.java

@@ -41,6 +41,7 @@ public class ConfigurationComponent {
 
   private final JPanel mainPanel;
   private final JBTable table;
+  private final JBCheckBox checkForPluginUpdatesCheckBox;
   private final JBCheckBox openNewTabCheckBox;
   private final JBCheckBox methodNameGenerationCheckBox;
   private final JBCheckBox autoFormattingCheckBox;
@@ -99,6 +100,9 @@ public void changedUpdate(DocumentEvent e) {
     systemPromptTextArea.setColumns(60);
     systemPromptTextArea.setRows(3);
 
+    checkForPluginUpdatesCheckBox = new JBCheckBox(
+        CodeGPTBundle.get("configurationConfigurable.checkForPluginUpdates.label"),
+        configuration.isCheckForPluginUpdates());
     openNewTabCheckBox = new JBCheckBox(
         CodeGPTBundle.get("configurationConfigurable.openNewTabCheckBox.label"),
         configuration.isCreateNewChatOnEachAction());
@@ -112,6 +116,7 @@ public void changedUpdate(DocumentEvent e) {
     mainPanel = FormBuilder.createFormBuilder()
         .addComponent(tablePanel)
         .addVerticalGap(4)
+        .addComponent(checkForPluginUpdatesCheckBox)
         .addComponent(openNewTabCheckBox)
         .addComponent(methodNameGenerationCheckBox)
         .addComponent(autoFormattingCheckBox)
@@ -256,6 +261,14 @@ public void setMaxTokens(int maxTokens) {
     maxTokensField.setValue(maxTokens);
   }
 
+  public boolean isCheckForPluginUpdates() {
+    return checkForPluginUpdatesCheckBox.isSelected();
+  }
+
+  public void setCheckForPluginUpdates(boolean checkForUpdates) {
+    checkForPluginUpdatesCheckBox.setSelected(checkForUpdates);
+  }
+
   public boolean isCreateNewChatOnEachAction() {
     return openNewTabCheckBox.isSelected();
   }

src/main/java/ee/carlrobert/codegpt/settings/configuration/ConfigurationConfigurable.java

@@ -37,6 +37,8 @@ public boolean isModified() {
         || configurationComponent.getMaxTokens() != configuration.getMaxTokens()
         || configurationComponent.getTemperature() != configuration.getTemperature()
         || !configurationComponent.getSystemPrompt().equals(configuration.getSystemPrompt())
+        || configurationComponent.isCheckForPluginUpdates()
+        != configuration.isCheckForPluginUpdates()
         || configurationComponent.isCreateNewChatOnEachAction()
         != configuration.isCreateNewChatOnEachAction()
         || configurationComponent.isMethodNameGenerationEnabled()
@@ -52,6 +54,7 @@ public void apply() {
     configuration.setMaxTokens(configurationComponent.getMaxTokens());
     configuration.setTemperature(configurationComponent.getTemperature());
     configuration.setSystemPrompt(configurationComponent.getSystemPrompt());
+    configuration.setCheckForPluginUpdates(configurationComponent.isCheckForPluginUpdates());
     configuration.setCreateNewChatOnEachAction(
         configurationComponent.isCreateNewChatOnEachAction());
     configuration.setMethodNameGenerationEnabled(
@@ -67,6 +70,7 @@ public void reset() {
     configurationComponent.setMaxTokens(configuration.getMaxTokens());
     configurationComponent.setTemperature(configuration.getTemperature());
     configurationComponent.setSystemPrompt(configuration.getSystemPrompt());
+    configurationComponent.setCheckForPluginUpdates(configuration.isCheckForPluginUpdates());
     configurationComponent.setCreateNewChatOnEachAction(
         configuration.isCreateNewChatOnEachAction());
     configurationComponent.setDisableMethodNameGeneration(

src/main/java/ee/carlrobert/codegpt/settings/configuration/ConfigurationState.java

@@ -20,6 +20,7 @@ public class ConfigurationState implements PersistentStateComponent<Configuratio
   private String systemPrompt = COMPLETION_SYSTEM_PROMPT;
   private int maxTokens = 1000;
   private double temperature = 0.1;
+  private boolean checkForPluginUpdates = true;
   private boolean createNewChatOnEachAction;
   private boolean ignoreGitCommitTokenLimit;
   private boolean methodNameGenerationEnabled = true;
@@ -81,6 +82,14 @@ public void setTableData(Map<String, String> tableData) {
     this.tableData = tableData;
   }
 
+  public boolean isCheckForPluginUpdates() {
+    return checkForPluginUpdates;
+  }
+
+  public void setCheckForPluginUpdates(boolean checkForPluginUpdates) {
+    this.checkForPluginUpdates = checkForPluginUpdates;
+  }
+
   public boolean isIgnoreGitCommitTokenLimit() {
     return ignoreGitCommitTokenLimit;
   }

src/main/java/ee/carlrobert/codegpt/util/OverlayUtil.java

@@ -35,9 +35,12 @@
 
 public class OverlayUtil {
 
+  public static Notification getDefaultNotification(String content, NotificationType type) {
+    return new Notification("CodeGPT Notification Group", "CodeGPT", content, type);
+  }
+
   public static void showNotification(String content, NotificationType type) {
-    Notifications.Bus.notify(
-        new Notification("CodeGPT Notification Group", "CodeGPT", content, type));
+    Notifications.Bus.notify(getDefaultNotification(content, type));
   }
 
   public static int showFileStructureDialog(

src/main/resources/META-INF/plugin.xml

@@ -10,6 +10,7 @@
     </projectListeners>
 
     <extensions defaultExtensionNs="com.intellij">
+        <postStartupActivity implementation="ee.carlrobert.codegpt.CodeGPTUpdateStartupActivity"/>
         <postStartupActivity implementation="ee.carlrobert.codegpt.PluginStartupActivity"/>
         <applicationConfigurable id="settings.codegpt" parentId="tools" displayName="CodeGPT"
             instance="ee.carlrobert.codegpt.settings.SettingsConfigurable"/>

src/main/resources/messages/codegpt.properties

@@ -66,6 +66,7 @@ configurationConfigurable.table.header.actionColumnLabel=Action
 configurationConfigurable.table.header.promptColumnLabel=Prompt
 configurationConfigurable.table.action.revertToDefaults.text=Revert to Defaults
 configurationConfigurable.table.action.addKeymap.text=Add Shortcut
+configurationConfigurable.checkForPluginUpdates.label=Check for plugin updates automatically
 configurationConfigurable.openNewTabCheckBox.label=Open a new chat on each action
 configurationConfigurable.enableMethodNameGeneration.label=Enable method name lookup suggestions
 configurationConfigurable.autoFormatting.label=Enable automatic code formatting
@@ -128,4 +129,8 @@ service.llama.title=LLaMA C/C++ Port (Free, Local)
 validation.error.fieldRequired=This field is required.
 validation.error.invalidEmail=The email you entered is invalid.
 validation.error.mustBeNumber=Value must be number.
-validation.error.mustBeBetweenZeroAndOne=Value must be between 0 and 1.
+validation.error.mustBeBetweenZeroAndOne=Value must be between 0 and 1.
+checkForUpdatesTask.title=Checking for CodeGPT update...
+checkForUpdatesTask.notification.message=An update for CodeGPT is available.
+checkForUpdatesTask.notification.installButton=Install update
+checkForUpdatesTask.notification.hideButton=Do not show again