feat(rpc agent): add mark collection as rpc option on add datasource

.rubocop.yml

@@ -353,6 +353,7 @@ Metrics/ClassLength:
     - 'packages/forest_admin_datasource_rpc/lib/forest_admin_datasource_rpc/Utils/schema_polling_pool.rb'
     - 'packages/forest_admin_datasource_rpc/lib/forest_admin_datasource_rpc/Utils/sse_client.rb'
     - 'packages/forest_admin_rpc_agent/lib/forest_admin_rpc_agent/routes/sse.rb'
+    - 'packages/forest_admin_rpc_agent/lib/forest_admin_rpc_agent/agent.rb'
 
 Style/OpenStructUse:
   Exclude:

packages/forest_admin_rpc_agent/lib/forest_admin_rpc_agent/agent.rb

@@ -12,6 +12,15 @@ def setup(options)
       @rpc_collections = []
       @cached_schema = nil
       @cached_schema_hash = nil
+      @customizer = ForestAdminRpcAgent::DatasourceCustomizer.new
+    end
+
+    def add_datasource(datasource, options = {})
+      if options[:mark_collections_as_rpc]
+        options[:mark_collections_callback] = ->(ds) { mark_collections_as_rpc(*ds.collections.keys) }
+      end
+
+      super
     end
 
     def send_schema(force: false)

packages/forest_admin_rpc_agent/lib/forest_admin_rpc_agent/datasource_customizer.rb

@@ -0,0 +1,27 @@
+module ForestAdminRpcAgent
+  class DatasourceCustomizer < ForestAdminDatasourceCustomizer::DatasourceCustomizer
+    def add_datasource(datasource, options)
+      @stack.queue_customization(lambda {
+        if options[:include] || options[:exclude]
+          publication_decorator = Decorators::Publication::PublicationDatasourceDecorator.new(datasource)
+          publication_decorator.keep_collections_matching(options[:include], options[:exclude])
+          datasource = publication_decorator
+        end
+
+        if options[:rename]
+          rename_collection_decorator = Decorators::RenameCollection::RenameCollectionDatasourceDecorator.new(
+            datasource
+          )
+          rename_collection_decorator.rename_collections(options[:rename])
+          datasource = rename_collection_decorator
+        end
+
+        options[:mark_collections_callback]&.call(datasource)
+
+        @composite_datasource.add_data_source(datasource)
+      })
+
+      self
+    end
+  end
+end