From d7655ebb3c65a8a46f3f304ccc02d47134112bd5 Mon Sep 17 00:00:00 2001
From: Shrikant Kandula <shrikant@appsmith.com>
Date: Wed, 24 Jun 2020 08:28:54 +0000
Subject: [PATCH] Add image location and documentation link fields to plugins

---
 .../com/appsmith/server/domains/Plugin.java   | 11 ++++++++
 .../server/migrations/DatabaseChangelog.java  | 25 +++++++++++++++++++
 2 files changed, 36 insertions(+)

diff --git a/app/server/appsmith-server/src/main/java/com/appsmith/server/domains/Plugin.java b/app/server/appsmith-server/src/main/java/com/appsmith/server/domains/Plugin.java
index 8c6a7d36e0..6a68e39102 100644
--- a/app/server/appsmith-server/src/main/java/com/appsmith/server/domains/Plugin.java
+++ b/app/server/appsmith-server/src/main/java/com/appsmith/server/domains/Plugin.java
@@ -17,6 +17,11 @@ import java.util.List;
 @Document
 public class Plugin extends BaseDomain {
 
+    public enum ResponseType {
+        TABLE,
+        JSON,
+    }
+
     String name;
 
     PluginType type;
@@ -25,6 +30,12 @@ public class Plugin extends BaseDomain {
 
     String jarLocation;
 
+    String iconLocation;
+
+    String documentationLink;
+
+    ResponseType responseType;
+
     List<PluginParameterType> datasourceParams;
 
     List<PluginParameterType> actionParams;
diff --git a/app/server/appsmith-server/src/main/java/com/appsmith/server/migrations/DatabaseChangelog.java b/app/server/appsmith-server/src/main/java/com/appsmith/server/migrations/DatabaseChangelog.java
index 1d4d54b050..93ee95cfd8 100644
--- a/app/server/appsmith-server/src/main/java/com/appsmith/server/migrations/DatabaseChangelog.java
+++ b/app/server/appsmith-server/src/main/java/com/appsmith/server/migrations/DatabaseChangelog.java
@@ -418,4 +418,29 @@ public class DatabaseChangelog {
         );
     }
 
+    @ChangeSet(order = "015", id = "set-plugin-image-and-docs-link", author = "")
+    public void setPluginImageAndDocsLink(MongoTemplate mongoTemplate) {
+        for (Plugin plugin : mongoTemplate.findAll(Plugin.class)) {
+            if ("postgres-plugin".equals(plugin.getPackageName())) {
+                plugin.setIconLocation("https://s3.us-east-2.amazonaws.com/assets.appsmith.com/Postgress.png");
+                plugin.setDocumentationLink("https://www.postgresql.org/docs/12/index.html");
+                plugin.setResponseType(Plugin.ResponseType.TABLE);
+
+            } else if ("restapi-plugin".equals(plugin.getPackageName())) {
+                plugin.setIconLocation("https://s3.us-east-2.amazonaws.com/assets.appsmith.com/RestAPI.png");
+
+            } else if ("mongo-plugin".equals(plugin.getPackageName())) {
+                plugin.setIconLocation("https://s3.us-east-2.amazonaws.com/assets.appsmith.com/MongoDB.png");
+                plugin.setDocumentationLink("https://docs.mongodb.com/manual/reference/command/nav-crud/");
+                plugin.setResponseType(Plugin.ResponseType.JSON);
+
+            } else {
+                continue;
+
+            }
+
+            mongoTemplate.save(plugin);
+        }
+    }
+
 }