fix: Associate LSP bundle download with app preload instead of project (#36)

* refactor: Associate LSP bundle download with app preload instead of project

Signed-off-by: Arunprasad Rajkumar <[email protected]>

Signed-off-by: Jeff MAURY <[email protected]>

Co-authored-by: Jeff MAURY <[email protected]>

Author: arajkumar

src/main/java/org/jboss/tools/intellij/analytics/AnalyticsPersistentSettings.java

@@ -0,0 +1,31 @@
+package org.jboss.tools.intellij.analytics;
+
+import org.kohsuke.github.GitHub;
+import org.kohsuke.github.GHAsset;
+import org.kohsuke.github.GHRepository;
+import org.kohsuke.github.GHRelease;
+import java.io.IOException;
+
+public class GitHubRelease {
+  private final GHRepository repo;
+
+  public GitHubRelease(final String repository) throws IOException {
+    final GitHub github = GitHub.connectAnonymously();
+    this.repo = github.getRepository(repository);
+  }
+
+  public String getLatestRelease() throws IOException {
+    return this.repo.getLatestRelease().getTagName();
+  }
+
+  public String getDownloadUri(final String releaseLabel, final String fileLabel) throws IOException {
+    final GHRelease release = this.repo.getReleaseByTagName(releaseLabel);
+    final GHAsset asset = release.listAssets()
+                                 .toList()
+                                 .stream()
+                                 .filter(a -> a.getLabel().equals(fileLabel))
+                                 .findFirst()
+                                 .orElseThrow(() -> new IOException(fileLabel + ": unable to download"));
+    return asset.getBrowserDownloadUrl();
+  }
+}

src/main/java/org/jboss/tools/intellij/analytics/GitHubRelease.java

@@ -0,0 +1,50 @@
+package org.jboss.tools.intellij.analytics;
+
+import java.io.File;
+import java.io.IOException;
+
+import com.intellij.ide.plugins.IdeaPluginDescriptor;
+import com.intellij.ide.plugins.PluginManager;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.components.Service;
+import com.intellij.openapi.components.ServiceManager;
+import com.intellij.openapi.diagnostic.Logger;
+import com.intellij.openapi.extensions.PluginId;
+import com.intellij.openapi.progress.ProgressIndicator;
+import com.intellij.openapi.project.Project;
+import com.intellij.util.io.HttpRequests;
+
+public final class GitHubReleaseDownloader {
+  private final IdeaPluginDescriptor descriptor = PluginManager.getPlugin(PluginId.getId("org.jboss.tools.intellij.analytics"));
+  private final String fileName;
+  private final ICookie cookies;
+  private final GitHubRelease release;
+
+  public GitHubReleaseDownloader(final String fileName, final ICookie cookies) throws IOException {
+    this.fileName = fileName;
+    this.cookies = cookies;
+    this.release = new GitHubRelease("fabric8-analytics/fabric8-analytics-lsp-server");
+  }
+
+  private boolean isNewRelease(final String releaseLabel) {
+    final String currentVersion = cookies.getValue(ICookie.Name.LSPVersion);
+    return !releaseLabel.equals(currentVersion);
+  }
+
+  public File download(final ProgressIndicator indicator) throws IOException {
+    final File dest = new File(descriptor.getPath(), fileName);
+    final String latestReleaseTag = this.release.getLatestRelease();
+    if (!isNewRelease(latestReleaseTag) && dest.exists()) {
+      return dest;
+    }
+    final String url = this.release.getDownloadUri(latestReleaseTag, this.fileName);
+    HttpRequests
+      .request(url)
+      .productNameAsUserAgent()
+      .saveToFile(dest, indicator);
+
+    dest.setExecutable(true);
+    cookies.setValue(ICookie.Name.LSPVersion, latestReleaseTag);
+    return dest;
+  }
+}

src/main/java/org/jboss/tools/intellij/analytics/GitHubReleaseDownloader.java

@@ -0,0 +1,10 @@
+package org.jboss.tools.intellij.analytics;
+
+public interface ICookie {
+  enum Name {
+    LSPVersion,
+  }
+
+  void setValue(Name name, String value);
+  String getValue(Name name);
+}

src/main/java/org/jboss/tools/intellij/analytics/ICookie.java

@@ -0,0 +1,49 @@
+package org.jboss.tools.intellij.analytics;
+
+import java.io.File;
+import java.io.IOException;
+import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.application.PreloadingActivity;
+import com.intellij.openapi.progress.ProgressIndicator;
+import com.intellij.openapi.diagnostic.Logger;
+import com.intellij.openapi.components.ServiceManager;
+import org.wso2.lsp4intellij.IntellijLanguageClient;
+
+public final class PreloadLanguageServer extends PreloadingActivity {
+  private static final Logger log = Logger.getInstance(PreloadLanguageServer.class);
+  private final ICookie cookies = ServiceManager.getService(Settings.class);
+
+  private void attachLanguageClient(final File cliFile) {
+    final String[] EXTENSIONS = {"xml", "json", "txt"};
+    final String[] cmds = {cliFile.toString(), "--stdio"};
+    ApplicationManager.getApplication().invokeAndWait(() -> {
+      for (String ext : EXTENSIONS) {
+        AnalyticsLanguageServerDefinition serverDefinition = new AnalyticsLanguageServerDefinition(ext, cmds);
+        IntellijLanguageClient.addServerDefinition(serverDefinition);
+        IntellijLanguageClient.addExtensionManager(ext, serverDefinition);
+      }
+    });
+    log.warn(String.format("lsp registration done %s", cliFile));
+  }
+
+  @Override
+  public void preload(ProgressIndicator indicator) {
+    if (ApplicationManager.getApplication().isUnitTestMode()) {
+      return;
+    }
+    log.debug("lsp preload called");
+    try {
+      final String devUrl = System.getenv("ANALYTICS_LSP_FILE_PATH");
+      File lspBundle;
+      if (devUrl != null) {
+        lspBundle = new File(devUrl);
+      } else {
+        final GitHubReleaseDownloader bundle = new GitHubReleaseDownloader(Platform.current.lspBundleName, cookies);
+        lspBundle = bundle.download(indicator);
+      }
+      attachLanguageClient(lspBundle);
+    } catch(IOException ex) {
+      log.warn("lsp download fail", ex);
+    }
+  }
+}

src/main/java/org/jboss/tools/intellij/analytics/LSPBundle.java

@@ -0,0 +1,53 @@
+package org.jboss.tools.intellij.analytics;
+
+import com.intellij.openapi.components.PersistentStateComponent;
+import com.intellij.openapi.components.RoamingType;
+import com.intellij.openapi.components.State;
+import com.intellij.openapi.components.Service;
+import com.intellij.openapi.components.Storage;
+import com.intellij.util.xmlb.XmlSerializerUtil;
+import com.intellij.util.xmlb.annotations.MapAnnotation;
+
+import java.util.Map;
+import java.util.HashMap;
+
+@Service
+@State(
+    name = "Settings",
+    storages = {
+      @Storage(file = "analytics.settings.xml", roamingType = RoamingType.DISABLED)
+})
+public final class Settings implements ICookie, PersistentStateComponent<Settings> {
+
+    // str representation of ICookie.Name values are key.
+    private Map<String, String> settings = new HashMap();
+
+    @Override
+    public Settings getState() {
+      return this;
+    }
+
+    @Override
+    public void loadState(Settings state) {
+        XmlSerializerUtil.copyBean(state, this);
+    }
+
+    @Override
+    public void setValue(ICookie.Name name, String value) {
+        this.settings.put(name.name(), value);
+    }
+
+    @Override
+    public String getValue(ICookie.Name name) {
+      return this.settings.getOrDefault(name.name(), "");
+    }
+
+    @MapAnnotation
+    public Map<String, String> getSettings() {
+        return settings;
+    }
+
+    public void setSettings(Map<String, String> settings) {
+        this.settings = settings;
+    }
+}

src/main/java/org/jboss/tools/intellij/analytics/LSPBundleException.java

@@ -93,7 +93,8 @@
     <externalAnnotator id="LSPAnnotator-xml" language="XML" implementationClass="org.wso2.lsp4intellij.contributors.annotator.LSPAnnotator"/>
     <externalAnnotator id="LSPAnnotator-json" language="JSON" implementationClass="org.wso2.lsp4intellij.contributors.annotator.LSPAnnotator"/>
     <externalAnnotator id="LSPAnnotator-txt" language="TEXT" implementationClass="org.wso2.lsp4intellij.contributors.annotator.LSPAnnotator"/>
-    <postStartupActivity implementation="org.jboss.tools.intellij.analytics.PostStartupActivity"/>
+    <preloadingActivity implementation="org.jboss.tools.intellij.analytics.PreloadLanguageServer"
+                id="org.jboss.tools.intellij.analytics.PostStartupActivity"/>
   </extensions>
 
   <application-components>