chore: Google sheet shared drive support added behind a flag

app/server/appsmith-interfaces/src/main/java/com/appsmith/external/enums/FeatureFlagEnum.java

@@ -14,6 +14,7 @@ public enum FeatureFlagEnum {
     APP_NAVIGATION_LOGO_UPLOAD,
     release_embed_hide_share_settings_enabled,
     rollout_datasource_test_rate_limit_enabled,
+    release_google_sheets_shared_drive_support_enabled,
 
     // Deprecated CE flags over here
     release_git_autocommit_feature_enabled,

app/server/appsmith-interfaces/src/main/java/com/appsmith/external/plugins/PluginExecutor.java

@@ -247,6 +247,50 @@ default Mono<ActionExecutionResult> executeParameterizedWithMetrics(
                 .tap(Micrometer.observation(observationRegistry));
     }
 
+    // TODO: Following methods of executeParameterizedWithFlags, executeParameterizedWithMetricsAndFlags,
+    // triggerWithFlags are
+    // added
+    // to support feature flags in the plugin modules. Current implementation of featureFlagService is only available in
+    // server module
+    // and not available in any of the plugin modules due to dependencies on SessionUserService, TenantService etc.
+    // Hence, these methods are added to support feature flags in the plugin modules.
+    // Ideal solution would be to move featureFlagService and its dependencies to the shared interface module
+    // But this is a bigger change and will be done in future. Current change of passing flags was done to resolve
+    // release blocker
+    // https://github.com/appsmithorg/appsmith/issues/37714
+    // Once thorogh testing of shared drive support is done, we can remove this tech debt of passing feature flags like
+    // this.
+    default Mono<ActionExecutionResult> executeParameterizedWithFlags(
+            C connection,
+            ExecuteActionDTO executeActionDTO,
+            DatasourceConfiguration datasourceConfiguration,
+            ActionConfiguration actionConfiguration,
+            Map<String, Boolean> featureFlagMap) {
+        return this.executeParameterized(connection, executeActionDTO, datasourceConfiguration, actionConfiguration);
+    }
+
+    default Mono<ActionExecutionResult> executeParameterizedWithMetricsAndFlags(
+            C connection,
+            ExecuteActionDTO executeActionDTO,
+            DatasourceConfiguration datasourceConfiguration,
+            ActionConfiguration actionConfiguration,
+            ObservationRegistry observationRegistry,
+            Map<String, Boolean> featureFlagMap) {
+        return this.executeParameterizedWithFlags(
+                        connection, executeActionDTO, datasourceConfiguration, actionConfiguration, featureFlagMap)
+                .tag("plugin", this.getClass().getName())
+                .name(ACTION_EXECUTION_PLUGIN_EXECUTION)
+                .tap(Micrometer.observation(observationRegistry));
+    }
+
+    default Mono<TriggerResultDTO> triggerWithFlags(
+            C connection,
+            DatasourceConfiguration datasourceConfiguration,
+            TriggerRequestDTO request,
+            Map<String, Boolean> featureFlagMap) {
+        return this.trigger(connection, datasourceConfiguration, request);
+    }
+
     /**
      * This function is responsible for preparing the action and datasource configurations to be ready for execution.
      *

app/server/appsmith-plugins/googleSheetsPlugin/src/main/java/com/external/config/ExecutionMethod.java

@@ -77,7 +77,9 @@ default Mono<Object> executePrerequisites(MethodConfig methodConfig, OAuth2 oaut
         return Mono.just(true);
     }
 
-    WebClient.RequestHeadersSpec<?> getExecutionClient(WebClient webClient, MethodConfig methodConfig);
+    default WebClient.RequestHeadersSpec<?> getExecutionClient(WebClient webClient, MethodConfig methodConfig) {
+        return null;
+    }
 
     default JsonNode transformExecutionResponse(
             JsonNode response, MethodConfig methodConfig, Set<String> userAuthorizedSheetIds) {
@@ -102,4 +104,9 @@ default Map<DataType, DataType> getDataTypeConversionMap() {
         conversionMap.put(DataType.FLOAT, DataType.DOUBLE);
         return conversionMap;
     }
+
+    default WebClient.RequestHeadersSpec<?> getExecutionClientWithFlags(
+            WebClient webClient, MethodConfig methodConfig, Map<String, Boolean> featureFlagMap) {
+        return getExecutionClient(webClient, methodConfig);
+    }
 }

app/server/appsmith-plugins/googleSheetsPlugin/src/main/java/com/external/config/FileListMethod.java

@@ -1,5 +1,6 @@
 package com.external.config;
 
+import com.appsmith.external.enums.FeatureFlagEnum;
 import com.appsmith.external.exceptions.pluginExceptions.AppsmithPluginException;
 import com.external.constants.ErrorMessages;
 import com.external.enums.GoogleSheetMethodEnum;
@@ -24,8 +25,9 @@
  * API reference: https://developers.google.com/sheets/api/guides/migration#list_spreadsheets_for_the_authenticated_user
  */
 public class FileListMethod implements ExecutionMethod, TriggerMethod {
-
     ObjectMapper objectMapper;
+    private final String SHARED_DRIVE_PARAMS =
+            "&includeItemsFromAllDrives=true&supportsAllDrives=true&corpora=allDrives";
 
     public FileListMethod(ObjectMapper objectMapper) {
         this.objectMapper = objectMapper;
@@ -37,10 +39,16 @@ public boolean validateExecutionMethodRequest(MethodConfig methodConfig) {
     }
 
     @Override
-    public WebClient.RequestHeadersSpec<?> getExecutionClient(WebClient webClient, MethodConfig methodConfig) {
+    public WebClient.RequestHeadersSpec<?> getExecutionClientWithFlags(
+            WebClient webClient, MethodConfig methodConfig, Map<String, Boolean> featureFlagMap) {
+        // TODO: Flags are needed here for google sheets integration to support shared drive behind a flag
+        // Once thoroughly tested, this flag can be removed
+        Boolean isSharedDriveSupportEnabled = featureFlagMap.getOrDefault(
+                FeatureFlagEnum.release_google_sheets_shared_drive_support_enabled.name(), Boolean.FALSE);
         UriComponentsBuilder uriBuilder = getBaseUriBuilder(
                 this.BASE_DRIVE_API_URL,
-                "?includeItemsFromAllDrives=true&supportsAllDrives=true&orderBy=name&q=mimeType%3D'application%2Fvnd.google-apps.spreadsheet'%20and%20trashed%3Dfalse",
+                "?orderBy=name&q=mimeType%3D'application%2Fvnd.google-apps.spreadsheet'%20and%20trashed%3Dfalse"
+                        + (isSharedDriveSupportEnabled.equals(Boolean.TRUE) ? SHARED_DRIVE_PARAMS : ""),
                 true);
 
         return webClient
@@ -74,8 +82,9 @@ public boolean validateTriggerMethodRequest(MethodConfig methodConfig) {
     }
 
     @Override
-    public WebClient.RequestHeadersSpec<?> getTriggerClient(WebClient webClient, MethodConfig methodConfig) {
-        return this.getExecutionClient(webClient, methodConfig);
+    public WebClient.RequestHeadersSpec<?> getTriggerClientWithFlags(
+            WebClient webClient, MethodConfig methodConfig, Map<String, Boolean> featureFlagMap) {
+        return this.getExecutionClientWithFlags(webClient, methodConfig, featureFlagMap);
     }
 
     @Override

app/server/appsmith-plugins/googleSheetsPlugin/src/main/java/com/external/config/TriggerMethod.java

@@ -3,6 +3,7 @@
 import com.fasterxml.jackson.databind.JsonNode;
 import org.springframework.web.reactive.function.client.WebClient;
 
+import java.util.Map;
 import java.util.Set;
 
 /**
@@ -21,7 +22,17 @@ public interface TriggerMethod {
     /**
      * Returns with the specification required to hit that particular trigger request
      */
-    WebClient.RequestHeadersSpec<?> getTriggerClient(WebClient webClient, MethodConfig methodConfig);
+    default WebClient.RequestHeadersSpec<?> getTriggerClient(WebClient webClient, MethodConfig methodConfig) {
+        return null;
+    }
+
+    /**
+     * Returns with the specification required to hit that particular trigger request
+     */
+    default WebClient.RequestHeadersSpec<?> getTriggerClientWithFlags(
+            WebClient webClient, MethodConfig methodConfig, Map<String, Boolean> featureFlagMap) {
+        return getTriggerClient(webClient, methodConfig);
+    }
 
     /**
      * Transforms the response from the end point into an Appsmith friendly structure

app/server/appsmith-plugins/googleSheetsPlugin/src/main/java/com/external/plugins/GoogleSheetsPlugin.java

@@ -81,6 +81,23 @@ public Mono<ActionExecutionResult> executeParameterized(
                 ExecuteActionDTO executeActionDTO,
                 DatasourceConfiguration datasourceConfiguration,
                 ActionConfiguration actionConfiguration) {
+            return executeParameterizedWithFlags(
+                    connection, executeActionDTO, datasourceConfiguration, actionConfiguration, null);
+        }
+
+        @Override
+        public Mono<TriggerResultDTO> trigger(
+                Void connection, DatasourceConfiguration datasourceConfiguration, TriggerRequestDTO request) {
+            return triggerWithFlags(connection, datasourceConfiguration, request, null);
+        }
+
+        @Override
+        public Mono<ActionExecutionResult> executeParameterizedWithFlags(
+                Void connection,
+                ExecuteActionDTO executeActionDTO,
+                DatasourceConfiguration datasourceConfiguration,
+                ActionConfiguration actionConfiguration,
+                Map<String, Boolean> featureFlagMap) {
 
             log.debug(Thread.currentThread().getName() + ": executeParameterized() called for GoogleSheets plugin.");
             boolean smartJsonSubstitution;
@@ -133,13 +150,14 @@ public Mono<ActionExecutionResult> executeParameterized(
 
             prepareConfigurationsForExecution(executeActionDTO, actionConfiguration, datasourceConfiguration);
 
-            return this.executeCommon(connection, datasourceConfiguration, actionConfiguration);
+            return this.executeCommon(connection, datasourceConfiguration, actionConfiguration, featureFlagMap);
         }
 
         public Mono<ActionExecutionResult> executeCommon(
                 Void connection,
                 DatasourceConfiguration datasourceConfiguration,
-                ActionConfiguration actionConfiguration) {
+                ActionConfiguration actionConfiguration,
+                Map<String, Boolean> featureFlagMap) {
 
             log.debug(Thread.currentThread().getName() + ": executeCommon() called for GoogleSheets plugin.");
             // Initializing object for error condition
@@ -185,7 +203,7 @@ public Mono<ActionExecutionResult> executeCommon(
                     // method
                     .flatMap(res -> {
                         return executionMethod
-                                .getExecutionClient(client, methodConfig)
+                                .getExecutionClientWithFlags(client, methodConfig, featureFlagMap)
                                 .headers(headers -> headers.set(
                                         "Authorization",
                                         "Bearer "
@@ -319,8 +337,11 @@ public Object substituteValueInInput(
         }
 
         @Override
-        public Mono<TriggerResultDTO> trigger(
-                Void connection, DatasourceConfiguration datasourceConfiguration, TriggerRequestDTO request) {
+        public Mono<TriggerResultDTO> triggerWithFlags(
+                Void connection,
+                DatasourceConfiguration datasourceConfiguration,
+                TriggerRequestDTO request,
+                Map<String, Boolean> featureFlagMap) {
             log.debug(Thread.currentThread().getName() + ": trigger() called for GoogleSheets plugin.");
             final TriggerMethod triggerMethod = GoogleSheetsMethodStrategy.getTriggerMethod(request, objectMapper);
             MethodConfig methodConfig = new MethodConfig(request);
@@ -343,7 +364,7 @@ public Mono<TriggerResultDTO> trigger(
                     validateAndGetUserAuthorizedSheetIds(datasourceConfiguration, methodConfig);
 
             return triggerMethod
-                    .getTriggerClient(client, methodConfig)
+                    .getTriggerClientWithFlags(client, methodConfig, featureFlagMap)
                     .headers(headers -> headers.set(
                             "Authorization",
                             "Bearer " + oauth2.getAuthenticationResponse().getToken()))

app/server/appsmith-server/src/main/java/com/appsmith/server/plugins/solutions/PluginTriggerSolutionCEImpl.java

@@ -9,6 +9,7 @@
 import com.appsmith.server.helpers.PluginExecutorHelper;
 import com.appsmith.server.repositories.PluginRepository;
 import com.appsmith.server.services.ConfigService;
+import com.appsmith.server.services.FeatureFlagService;
 import com.appsmith.server.services.TenantService;
 import com.appsmith.server.solutions.DatasourceTriggerSolution;
 import org.apache.commons.lang3.StringUtils;
@@ -28,18 +29,21 @@ public class PluginTriggerSolutionCEImpl implements PluginTriggerSolutionCE {
     private final PluginRepository pluginRepository;
     private final ConfigService configService;
     private final TenantService tenantService;
+    private final FeatureFlagService featureFlagService;
 
     public PluginTriggerSolutionCEImpl(
             DatasourceTriggerSolution datasourceTriggerSolution,
             PluginExecutorHelper pluginExecutorHelper,
             PluginRepository pluginRepository,
             ConfigService configService,
-            TenantService tenantService) {
+            TenantService tenantService,
+            FeatureFlagService featureFlagService) {
         this.datasourceTriggerSolution = datasourceTriggerSolution;
         this.pluginExecutorHelper = pluginExecutorHelper;
         this.pluginRepository = pluginRepository;
         this.configService = configService;
         this.tenantService = tenantService;
+        this.featureFlagService = featureFlagService;
     }
 
     /**
@@ -74,6 +78,11 @@ public Mono<TriggerResultDTO> trigger(
         Mono<PluginExecutor> pluginExecutorMono =
                 pluginMono.flatMap(plugin -> pluginExecutorHelper.getPluginExecutor(Mono.just(plugin)));
 
+        // TODO: Flags are needed here for google sheets integration to support shared drive behind a flag
+        // Once thoroughly tested, this flag can be removed
+        Map<String, Boolean> featureFlagMap =
+                featureFlagService.getCachedTenantFeatureFlags().getFeatures();
+
         /*
          * Since there is no datasource provided, we are passing the Datasource Context connection and datasourceConfiguration as null.
          * We will leave the execution to respective plugin executor.
@@ -83,8 +92,8 @@ public Mono<TriggerResultDTO> trigger(
             PluginExecutor pluginExecutor = pair.getT2();
             setHeadersToTriggerRequest(plugin, httpHeaders, triggerRequestDTO);
             return setTenantAndInstanceId(triggerRequestDTO)
-                    .flatMap(updatedTriggerRequestDTO ->
-                            ((PluginExecutor<Object>) pluginExecutor).trigger(null, null, updatedTriggerRequestDTO));
+                    .flatMap(updatedTriggerRequestDTO -> ((PluginExecutor<Object>) pluginExecutor)
+                            .triggerWithFlags(null, null, updatedTriggerRequestDTO, featureFlagMap));
         });
     }
 

app/server/appsmith-server/src/main/java/com/appsmith/server/plugins/solutions/PluginTriggerSolutionImpl.java

@@ -3,6 +3,7 @@
 import com.appsmith.server.helpers.PluginExecutorHelper;
 import com.appsmith.server.repositories.PluginRepository;
 import com.appsmith.server.services.ConfigService;
+import com.appsmith.server.services.FeatureFlagService;
 import com.appsmith.server.services.TenantService;
 import com.appsmith.server.solutions.DatasourceTriggerSolution;
 import org.springframework.stereotype.Component;
@@ -14,7 +15,14 @@ public PluginTriggerSolutionImpl(
             PluginExecutorHelper pluginExecutorHelper,
             PluginRepository pluginRepository,
             ConfigService configService,
-            TenantService tenantService) {
-        super(datasourceTriggerSolution, pluginExecutorHelper, pluginRepository, configService, tenantService);
+            TenantService tenantService,
+            FeatureFlagService featureFlagService) {
+        super(
+                datasourceTriggerSolution,
+                pluginExecutorHelper,
+                pluginRepository,
+                configService,
+                tenantService,
+                featureFlagService);
     }
 }

app/server/appsmith-server/src/main/java/com/appsmith/server/solutions/ActionExecutionSolutionImpl.java

@@ -13,6 +13,7 @@
 import com.appsmith.server.services.AuthenticationValidator;
 import com.appsmith.server.services.ConfigService;
 import com.appsmith.server.services.DatasourceContextService;
+import com.appsmith.server.services.FeatureFlagService;
 import com.appsmith.server.services.SessionUserService;
 import com.appsmith.server.services.TenantService;
 import com.appsmith.server.solutions.ce.ActionExecutionSolutionCEImpl;
@@ -42,7 +43,8 @@ public ActionExecutionSolutionImpl(
             ConfigService configService,
             TenantService tenantService,
             CommonConfig commonConfig,
-            ActionExecutionSolutionHelper actionExecutionSolutionHelper) {
+            ActionExecutionSolutionHelper actionExecutionSolutionHelper,
+            FeatureFlagService featureFlagService) {
         super(
                 newActionService,
                 actionPermission,
@@ -63,6 +65,7 @@ public ActionExecutionSolutionImpl(
                 configService,
                 tenantService,
                 commonConfig,
-                actionExecutionSolutionHelper);
+                actionExecutionSolutionHelper,
+                featureFlagService);
     }
 }

app/server/appsmith-server/src/main/java/com/appsmith/server/solutions/DatasourceTriggerSolutionImpl.java

@@ -7,6 +7,7 @@
 import com.appsmith.server.services.AuthenticationValidator;
 import com.appsmith.server.services.ConfigService;
 import com.appsmith.server.services.DatasourceContextService;
+import com.appsmith.server.services.FeatureFlagService;
 import com.appsmith.server.services.TenantService;
 import com.appsmith.server.solutions.ce.DatasourceTriggerSolutionCEImpl;
 import lombok.extern.slf4j.Slf4j;
@@ -28,7 +29,8 @@ public DatasourceTriggerSolutionImpl(
             DatasourcePermission datasourcePermission,
             EnvironmentPermission environmentPermission,
             ConfigService configService,
-            TenantService tenantService) {
+            TenantService tenantService,
+            FeatureFlagService featureFlagService) {
         super(
                 datasourceService,
                 datasourceStorageService,
@@ -40,6 +42,7 @@ public DatasourceTriggerSolutionImpl(
                 datasourcePermission,
                 environmentPermission,
                 configService,
-                tenantService);
+                tenantService,
+                featureFlagService);
     }
 }