Update action displayed in notifications

Author: manodnyab

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/core/notifications/CustomizeNotificationsUi.kt

@@ -5,13 +5,32 @@ package software.aws.toolkits.jetbrains.core.notifications
 
 import com.intellij.icons.AllIcons
 import com.intellij.ide.BrowserUtil
+import com.intellij.ide.plugins.IdeaPluginDescriptor
+import com.intellij.ide.plugins.PluginManagerCore
+import com.intellij.notification.NotificationAction
 import com.intellij.openapi.actionSystem.AnAction
 import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.options.ShowSettingsUtil
+import com.intellij.openapi.progress.ProgressIndicator
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.ui.Messages
 import com.intellij.ui.EditorNotificationPanel
+import com.intellij.util.concurrency.annotations.RequiresBackgroundThread
+import software.aws.toolkits.core.utils.debug
+import software.aws.toolkits.core.utils.error
+import software.aws.toolkits.core.utils.getLogger
+import software.aws.toolkits.jetbrains.AwsPlugin
 import software.aws.toolkits.jetbrains.AwsToolkit
+import software.aws.toolkits.jetbrains.core.plugin.PluginUpdateManager.Companion.ID_ACTION_AUTO_UPDATE_SETTINGS
+import software.aws.toolkits.jetbrains.core.plugin.PluginUpdateManager.Companion.LOG
+import software.aws.toolkits.jetbrains.core.plugin.PluginUpdateManager.Companion.SOURCE_AUTO_UPDATE_FINISH_NOTIFY
+import software.aws.toolkits.jetbrains.core.plugin.PluginUpdateManager.Companion.updatePlugin
+import software.aws.toolkits.jetbrains.settings.AwsSettingsSharedConfigurable
+import software.aws.toolkits.jetbrains.utils.notifyInfo
 import software.aws.toolkits.resources.AwsCoreBundle
+import software.aws.toolkits.telemetry.Component
+import software.aws.toolkits.telemetry.ToolkitTelemetry
 
 fun checkSeverity(notificationSeverity: String): NotificationSeverity = when (notificationSeverity) {
     "Critical" -> NotificationSeverity.CRITICAL
@@ -37,7 +56,7 @@ object NotificationManager {
             if (action.type == "UpdateExtension") {
                 add(
                     NotificationActionList(AwsCoreBundle.message("notification.update")) {
-                        // TODO: Add update logic
+
                     }
                 )
             }
@@ -96,6 +115,42 @@ object NotificationManager {
 
         return panel
     }
+
+    @RequiresBackgroundThread
+    fun updatePlugins(progressIndicator: ProgressIndicator, plugin: AwsPlugin) {
+        val pluginInfo = AwsToolkit.PLUGINS_INFO[plugin] ?: return
+        val pluginDescriptor = pluginInfo.descriptor as? IdeaPluginDescriptor ?: return
+        val pluginName = pluginInfo.name
+        try {
+            if (!PluginManagerCore.isPluginInstalled(pluginDescriptor.pluginId)) {
+               // LOG.debug { "$pluginName is not detected as installed, not performing auto-update" }
+                return
+            }
+
+            if (!updatePlugin(pluginDescriptor, progressIndicator)) return
+
+        } catch (e: Exception) {
+            getLogger<NotificationManager>().debug(e) { "Unable to update $pluginName" }
+
+            return
+        } catch (e: Error) {
+            getLogger<NotificationManager>().error { "Unable to update $pluginName" }
+            return
+        }
+
+        if (plugin == AwsPlugin.CORE) return
+        notifyInfo(
+            title = AwsCoreBundle.message("aws.notification.auto_update.title", pluginName),
+            content = AwsCoreBundle.message("aws.settings.auto_update.notification.message"),
+            project = null,
+            notificationActions = listOf(
+                NotificationAction.createSimpleExpiring(AwsCoreBundle.message("aws.settings.auto_update.notification.yes")) {
+                    ApplicationManager.getApplication().restart()
+                },
+                NotificationAction.createSimpleExpiring(AwsCoreBundle.message("aws.settings.auto_update.notification.no")) {}
+            )
+        )
+    }
 }
 
 data class NotificationActionList(

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/core/plugin/PluginUpdateManager.kt

@@ -200,7 +200,7 @@ class PluginUpdateManager : Disposable {
         // TODO: Optimize this to only search the result for AWS plugins
         fun getUpdateInfo(): Collection<PluginDownloader> = UpdateChecker.getPluginUpdates() ?: emptyList()
 
-        internal fun updatePlugin(pluginDescriptor: IdeaPluginDescriptor, progressIndicator: ProgressIndicator): Boolean {
+        fun updatePlugin(pluginDescriptor: IdeaPluginDescriptor, progressIndicator: ProgressIndicator): Boolean {
             val pluginName = pluginDescriptor.name
 
             // wasUpdatedWithRestart means that, it was an update and it needs to restart to apply