feat: expose plugin to datasource-customizer

packages/forest_admin_agent/lib/forest_admin_agent/serializer/forest_serializer_override.rb

@@ -60,7 +60,10 @@ def self.find_recursive_relationships(root_object, root_inclusion_tree, results,
  objects.each do |obj|
  relation = ForestAdminAgent::Facades::Container.datasource.get_collection(options[:class_name]).schema[:fields][attribute_name]
  relation_class_name = ForestAdminAgent::Facades::Container.datasource.get_collection(relation.foreign_collection).name
- obj_serializer = JSONAPI::Serializer.find_serializer(obj, options)
+ option_relation = options.clone
+ option_relation[:class_name] = relation_class_name
+ obj_serializer = JSONAPI::Serializer.find_serializer(obj, option_relation)
+
  # Use keys of ['posts', '1'] for the results to enforce uniqueness.
  # Spec: A compound document MUST NOT include more than one resource object for each
  # type and id pair.
@@ -200,7 +203,6 @@ def self.serialize(objects, options = {})
  # of the internal special merging logic.
  find_recursive_relationships(obj, inclusion_tree, relationship_data, passthrough_options)
  end
-
  result['included'] = relationship_data.map do |_, data|
  included_passthrough_options = {}
  included_passthrough_options[:base_url] = passthrough_options[:base_url]

packages/forest_admin_datasource_active_record/lib/forest_admin_datasource_active_record/collection.rb

@@ -47,11 +47,10 @@ def delete(_caller, filter)
 
  def fetch_fields
  @model.columns_hash.each do |column_name, column|
- # TODO: check is not sti column
  field = ForestAdminDatasourceToolkit::Schema::ColumnSchema.new(
  column_type: get_column_type(@model, column),
  filter_operators: operators_for_column_type(get_column_type(@model, column)),
- is_primary_key: column_name == @model.primary_key,
+ is_primary_key: column_name == @model.primary_key || @model.primary_key.include?(column_name),
  is_read_only: false,
  is_sortable: true,
  default_value: column.default,

packages/forest_admin_datasource_active_record/lib/forest_admin_datasource_active_record/datasource.rb

@@ -5,6 +5,7 @@ class Datasource < ForestAdminDatasourceToolkit::Datasource
  def initialize(db_config = {})
  super()
  @models = []
+ @habtm_models = {}
  init_orm(db_config)
  generate
  end
@@ -19,11 +20,52 @@ def generate
  end
 
  def fetch_model(model)
- @models << model unless model.abstract_class? || @models.include?(model) || !model.table_exists?
+ if model.name.start_with?('HABTM_')
+ build_habtm(model)
+ else
+ @models << model unless model.abstract_class? || @models.include?(model) || !model.table_exists?
+ end
  end
 
  def init_orm(db_config)
  ActiveRecord::Base.establish_connection(db_config)
  end
+
+ def build_habtm(model)
+ if @habtm_models.key?(model.table_name)
+ @habtm_models[model.table_name].left_reflection = model.right_reflection
+ # when the second model is added, we can push the HABTM model to the models list
+ @models << make_through_model(
+ model.table_name,
+ [
+ @habtm_models[model.table_name].left_reflection,
+ @habtm_models[model.table_name].right_reflection
+ ]
+ )
+ else
+ @habtm_models[model.table_name] = model
+ end
+ end
+
+ def make_through_model(table_name, associations)
+ through_model = Class.new(ActiveRecord::Base) do
+ class << self
+ attr_accessor :name, :table_name
+ end
+
+ def self.add_association(name, options)
+ belongs_to name, required: false, **options
+ end
+ end
+
+ through_model.name = table_name.singularize
+ through_model.table_name = table_name
+ through_model.primary_key = [associations[0].foreign_key, associations[1].foreign_key]
+ associations.each do |association|
+ through_model.add_association(association.name, association.options)
+ end
+
+ through_model
+ end
  end
 end

packages/forest_admin_datasource_active_record/lib/forest_admin_datasource_active_record/utils/query.rb

@@ -107,9 +107,7 @@ def build_select
  def apply_select
  unless @projection.nil?
  @query = @query.select(@select.join(', '))
- @query = @query.eager_load(@projection.relations.keys.map(&:to_sym))
- # TODO: replace eager_load by joins because eager_load select ALL columns of relation
- # @query = @query.joins(@projection.relations.keys.map(&:to_sym))
+ @query = @query.joins(@projection.relations.keys.map(&:to_sym))
  end
 
  @query

packages/forest_admin_datasource_customizer/lib/forest_admin_datasource_customizer/datasource_customizer.rb

@@ -27,8 +27,6 @@ def datasource(logger)
 
  def add_datasource(datasource, options)
  @stack.queue_customization(lambda {
- # TODO: add call logger
-
  if options[:include] || options[:exclude]
  publication_decorator = Decorators::Publication::PublicationDatasourceDecorator.new(datasource)
  publication_decorator.keep_collections_matching(options[:include], options[:exclude])
@@ -61,7 +59,7 @@ def add_chart(name, &definition)
  end
 
  def use(plugin, options)
-# TODO: to implement
+push_customization { plugin.new.run(self, nil, options) }
  end
 
  def customize_collection(name, handle)

packages/forest_admin_datasource_toolkit/lib/forest_admin_datasource_toolkit/collection.rb

@@ -16,9 +16,9 @@ def initialize(datasource, name, native_driver = nil)
  countable: false,
  searchable: false,
  charts: [],
- segments: {}
+ segments: {},
+ actions: {}
  }
- @actions = {}
  end
 
  def enable_count
@@ -50,9 +50,9 @@ def add_fields(fields)
  end
 
  def add_action(name, action)
- raise Exceptions::ForestException, "Action #{name} already defined in collection" if @actions[key]
+ raise Exceptions::ForestException, "Action #{name} already defined in collection" if @schema[:actions].key?(name)
 
-@actions[name] = action
+schema[:actions][name] = action
  end
 
  def render_chart(_caller, name, _record_id)