support customized dependency code snippet in metadata and ui fix.

Author: wangmingliang-ms

PluginsAndFeatures/azure-toolkit-for-intellij/azure-sdk-reference-book/src/main/java/com/microsoft/azure/toolkit/intellij/azuresdk/model/AzureSdkArtifactEntity.java

@@ -11,42 +11,58 @@
 import lombok.Setter;
 
 import java.util.Map;
+import java.util.Objects;
 
 @Getter
 @Setter
 @NoArgsConstructor
 @JsonIgnoreProperties(ignoreUnknown = true)
 public class AzureSdkArtifactEntity {
-    public static final String DEPENDENCY_TYPE_MAVEN = "Maven";
-    public static final String DEPENDENCY_TYPE_GRADLE = "Gradle";
-
     private String artifactId;
     private String groupId;
     private String versionGA;
     private String versionPreview;
     private String type;
     private Map<String, String> links;
+    private Map<String, String> dependencyPattern;
+
+    public String getDependencySnippet(DependencyType type, String version) {
+        final String strType = type.getName().toLowerCase();
+        if (Objects.nonNull(dependencyPattern) && dependencyPattern.containsKey(strType)) {
+            return dependencyPattern.get(strType);
+        }
+        return getDefaultDependencySnippet(type, version);
+    }
 
-    public String generateDependencySnippet(String type, String version) {
-        if (DEPENDENCY_TYPE_GRADLE.equals(type)) {
+    private String getDefaultDependencySnippet(final DependencyType type, final String version) {
+        if (DependencyType.GRADLE == type) {
             return String.format("implementation '%s:%s:%s'", this.groupId, this.artifactId, version);
         }
         return String.join("",
-            "<dependency>\n",
-            "    <groupId>", this.groupId, "</groupId>\n",
-            "    <artifactId>", this.artifactId, "</artifactId>\n",
-            "    <version>", version, "</version>\n",
-            "</dependency>"
+                "<dependency>\n",
+                "    <groupId>", this.groupId, "</groupId>\n",
+                "    <artifactId>", this.artifactId, "</artifactId>\n",
+                "    <version>", version, "</version>\n",
+                "</dependency>"
         );
     }
 
-    public String getLink(String type) {
-        return this.links.get(type);
-    }
-
     public static class Type {
         public static final String SPRING = "spring";
         public static final String CLIENT = "client";
         public static final String MANAGEMENT = "mgmt";
     }
+
+    @Getter
+    public enum DependencyType {
+        MAVEN("Maven", "xml"), GRADLE("Gradle", "gradle");
+
+        private final String name;
+        private final String fileExt;
+
+        DependencyType(final String name, final String fileExt) {
+            this.name = name;
+            this.fileExt = fileExt;
+        }
+    }
 }

PluginsAndFeatures/azure-toolkit-for-intellij/azure-sdk-reference-book/src/main/java/com/microsoft/azure/toolkit/intellij/azuresdk/referencebook/AzureSdkArtifactGroupPanel.java

@@ -14,11 +14,14 @@
 import com.intellij.openapi.actionSystem.DefaultActionGroup;
 import com.intellij.openapi.actionSystem.impl.ActionToolbarImpl;
 import com.intellij.openapi.editor.impl.DocumentImpl;
+import com.intellij.openapi.fileTypes.FileType;
+import com.intellij.openapi.fileTypes.ex.FileTypeManagerEx;
 import com.intellij.openapi.ide.CopyPasteManager;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.project.ProjectManager;
 import com.intellij.ui.EditorTextField;
 import com.microsoft.azure.toolkit.intellij.azuresdk.model.AzureSdkArtifactEntity;
+import com.microsoft.azure.toolkit.intellij.azuresdk.model.AzureSdkArtifactEntity.DependencyType;
 import icons.GradleIcons;
 import icons.OpenapiIcons;
 import lombok.Getter;
@@ -32,6 +35,9 @@
 import java.util.List;
 import java.util.function.Consumer;
 
+import static com.microsoft.azure.toolkit.intellij.azuresdk.model.AzureSdkArtifactEntity.DependencyType.GRADLE;
+import static com.microsoft.azure.toolkit.intellij.azuresdk.model.AzureSdkArtifactEntity.DependencyType.MAVEN;
+
 public class AzureSdkArtifactGroupPanel {
     @Getter
     private JPanel contentPanel;
@@ -42,7 +48,7 @@ public class AzureSdkArtifactGroupPanel {
     private final List<AzureSdkArtifactDetailPanel> artifactPnls = new ArrayList<>();
     private AzureSdkArtifactEntity pkg;
     private String version;
-    private String type;
+    private DependencyType type = MAVEN;
 
     public void setData(@Nonnull final List<? extends AzureSdkArtifactEntity> artifacts) {
         this.clear();
@@ -66,12 +72,13 @@ private void clear() {
     private void onPackageOrVersionSelected(AzureSdkArtifactEntity pkg, String version) {
         this.pkg = pkg;
         this.version = version;
-        this.viewer.setText(pkg.generateDependencySnippet(type, version));
+        this.viewer.setText(pkg.getDependencySnippet(type, version));
     }
 
-    private void onDependencyTypeSelected(String type) {
+    private void onDependencyTypeSelected(DependencyType type) {
         this.type = type;
-        this.viewer.setText(pkg.generateDependencySnippet(type, version));
+        final FileType fileType = FileTypeManagerEx.getInstance().getFileTypeByExtension(type.getFileExt());
+        this.viewer.setNewDocumentAndFileType(fileType, new DocumentImpl(pkg.getDependencySnippet(type, version)));
     }
 
     private EditorTextField buildCodeViewer() {
@@ -130,34 +137,34 @@ private void createUIComponents() {
      * @see com.intellij.application.options.schemes.AbstractSchemesPanel
      */
     private static class DependencyTypeSelector extends DefaultActionGroup {
-        private final Consumer<? super String> onTypeSelected;
-        private String selectedType;
+        private final Consumer<? super DependencyType> onTypeSelected;
+        private DependencyType selectedType;
 
-        private DependencyTypeSelector(Consumer<? super String> onTypeSelected) {
+        private DependencyTypeSelector(Consumer<? super DependencyType> onTypeSelected) {
             super();
             setPopup(true);
             this.onTypeSelected = onTypeSelected;
-            final AnAction maven = createAction(AzureSdkArtifactEntity.DEPENDENCY_TYPE_MAVEN, OpenapiIcons.RepositoryLibraryLogo, this::setSelectedType);
-            final AnAction gradle = createAction(AzureSdkArtifactEntity.DEPENDENCY_TYPE_GRADLE, GradleIcons.Gradle, this::setSelectedType);
+            final AnAction maven = createAction(MAVEN.getName(), OpenapiIcons.RepositoryLibraryLogo, () -> this.setSelectedType(MAVEN));
+            final AnAction gradle = createAction(GRADLE.getName(), GradleIcons.Gradle, () -> this.setSelectedType(GRADLE));
             this.addAll(maven, gradle);
         }
 
-        private void setSelectedType(String type) {
+        private void setSelectedType(DependencyType type) {
             this.selectedType = type;
             this.onTypeSelected.accept(type);
         }
 
         @Override
         public void update(@NotNull final AnActionEvent e) {
-            final Icon icon = AzureSdkArtifactEntity.DEPENDENCY_TYPE_GRADLE.equals(selectedType) ? GradleIcons.Gradle : OpenapiIcons.RepositoryLibraryLogo;
+            final Icon icon = GRADLE == selectedType ? GradleIcons.Gradle : OpenapiIcons.RepositoryLibraryLogo;
             e.getPresentation().setIcon(icon);
         }
 
-        private AnAction createAction(final String type, final Icon icon, final Consumer<? super String> onTypeSelected) {
-            return new AnAction(type, null, icon) {
+        private AnAction createAction(final String name, final Icon icon, final Runnable onSelected) {
+            return new AnAction(name, null, icon) {
                 @Override
                 public void actionPerformed(@NotNull final AnActionEvent e) {
-                    onTypeSelected.accept(type);
+                    onSelected.run();
                 }
             };
         }