ForestAdmin/forest-rails

View on GitHub
lib/forest_liana/collection.rb

Summary

Maintainability
C
7 hrs
Test Coverage
F
58%
module ForestLiana::Collection
  extend ActiveSupport::Concern

  module ClassMethods
    attr_accessor :active_record_class
    attr_accessor :collection_name
    attr_accessor :is_read_only
    attr_accessor :is_searchable

    def collection(collection_name, opts = {})
      self.collection_name = find_name(collection_name).to_s
      self.is_read_only = opts[:read_only] || false
      self.is_searchable = opts[:is_searchable] || false

      # NOTICE: Creates dynamically the serializer if it's a Smart Collection.
      if smart_collection? &&
          !ForestLiana::UserSpace.const_defined?(serializer_name)

        ForestLiana.names_overriden[self] = self.collection_name.to_s

        ForestLiana::SerializerFactory.new(is_smart_collection: true)
          .serializer_for(self)
      end
    end

    def action(name, opts = {})
      opts[:name] = name
      model.actions << ForestLiana::Model::Action.new(opts)
    end

    def segment(name, opts = {}, &block)
      opts[:name] = name
      model.segments << ForestLiana::Model::Segment.new(opts, &block)
    end

    def search_fields(fields)
      model.search_fields = fields
    end

    def field(name, opts, &block)
      # TODO: Handle empty name

      if opts.key?(:isRequired)
        FOREST_LOGGER.warn "DEPRECATION WARNING: isRequired on field \"#{name}\" is deprecated. Please use is_required."
        opts[:is_required] = !!opts[:isRequired]
        opts.delete(:isRequired)
      end
      if opts.key?(:isReadOnly)
        FOREST_LOGGER.warn "DEPRECATION WARNING: isReadOnly on field \"#{name}\" is deprecated. Please use is_read_only."
        opts[:is_read_only] = !!opts[:isReadOnly]
        opts.delete(:isReadOnly)
      end
      if opts.key?(:isFilterable)
        FOREST_LOGGER.warn "DEPRECATION WARNING: isFilterable on field \"#{name}\" is deprecated. Please use is_filterable."
        opts[:is_filterable] = !!opts[:isFilterable]
        opts.delete(:isFilterable)
      end
      if opts.key?(:isSortable)
        FOREST_LOGGER.warn "DEPRECATION WARNING: isSortable on field \"#{name}\" is deprecated. Please use is_sortable."
        opts[:is_sortable] = !!opts[:isSortable]
        opts.delete(:isSortable)
      end

      opts[:is_read_only] = true unless opts.has_key?(:is_read_only)
      opts[:is_read_only] = false if opts.has_key?(:set)
      opts[:is_required] = false unless opts.has_key?(:is_required)
      opts[:type] = "String" unless opts.has_key?(:type)
      opts[:default_value] = nil unless opts.has_key?(:default_value)
      opts[:integration] = nil unless opts.has_key?(:integration)
      opts[:reference] = nil unless opts.has_key?(:reference)
      opts[:inverse_of] = nil unless opts.has_key?(:inverse_of)
      opts[:relationships] = nil unless opts.has_key?(:relationships)
      opts[:widget] = nil unless opts.has_key?(:widget)
      opts[:validations] = [] unless opts.has_key?(:validations)

      model.fields << opts.merge({
        field: name,
        is_filterable: !!opts[:is_filterable],
        is_sortable: !!opts[:is_sortable],
        is_virtual: true
      })

      define_method(name) { self.data[name] } if smart_collection?

      if serializer_name && ForestLiana::UserSpace.const_defined?(
          serializer_name)
        ForestLiana::UserSpace.const_get(serializer_name).class_eval do
          if block
            # NOTICE: Smart Field case.
            compute_value = lambda do |object|
              begin
                object.instance_eval(&block)
              rescue => exception
                FOREST_REPORTER.report exception
                FOREST_LOGGER.error "Cannot retrieve the " + name.to_s + " value because of an " \
                  "internal error in the getter implementation: " + exception.message
                nil
              end
            end

            attribute(name, &compute_value)
          else
            # NOTICE: Smart Collection field case.
            attribute(name)
          end
        end
      end
    end

    def has_many(name, opts, &block)
      model.fields << opts.merge({
        field: name,
        is_virtual: true,
        is_searchable: false,
        type: ['String']
      })

      define_method(name) { self.data[name] } if smart_collection?

      if serializer_name && ForestLiana::UserSpace.const_defined?(
          serializer_name)
        ForestLiana::UserSpace.const_get(serializer_name).class_eval do
          has_many(name, name: name)
        end
      end
    end

    def belongs_to(name, opts, &block)
      model.fields << opts.merge({
        field: name,
        is_virtual: true,
        is_searchable: false,
        type: 'String'
      })

      define_method(name) { self.data[name] } if smart_collection?

      if serializer_name && ForestLiana::UserSpace.const_defined?(
          serializer_name)
        ForestLiana::UserSpace.const_get(serializer_name).class_eval do
          has_one(name, name: name, include_data: true, &block)
        end
      end
    end

    private

    def find_name(collection_name)
      # TODO: Remove once lianas prior to 2.0.0 are not supported anymore.
      model = ForestLiana.models.find { |collection| collection.try(:table_name) == collection_name.to_s }
      if model
        collection_name_new = ForestLiana.name_for(model)
        FOREST_LOGGER.warn "DEPRECATION WARNING: Collection names are now based on the models " \
          "names. Please rename the collection '#{collection_name.to_s}' of your Forest " \
          "customisation in '#{collection_name_new}'."
        return collection_name_new
      end

      # TODO: Remove once lianas prior to 2.0.0 are not supported anymore.
      model = ForestLiana.names_old_overriden.invert[collection_name.to_s]
      if model
        collection_name_new = ForestLiana.name_for(model)
        FOREST_LOGGER.warn "DEPRECATION WARNING: Collection names are now based on the models " \
          "names. Please rename the collection '#{collection_name.to_s}' of your Forest " \
          "customisation in '#{collection_name_new}'."
        return collection_name_new
      end

      collection_name.to_s
    end

    def model
      collection = ForestLiana.apimap.find do |object|
        object.name.to_s == self.collection_name.try(:to_s)
      end

      if collection.blank?
        collection = ForestLiana::Model::Collection.new({
          name: self.collection_name.to_s,
          is_read_only: self.is_read_only,
          is_searchable: self.is_searchable,
          is_virtual: true,
          fields: []
        })

        ForestLiana.apimap << collection
      end

      collection
    end

    def active_record_class
      ForestLiana::SchemaUtils.find_model_from_collection_name(
        self.collection_name)
    end

    def serializer_name
      if smart_collection?
        "#{collection_name.to_s.classify}Serializer"
      else
        component_prefix = ForestLiana.component_prefix(active_record_class)
        serializer_name = "#{component_prefix}Serializer"

        "ForestLiana::UserSpace::#{serializer_name}"
      end
    end

    def serializer_name_for_reference(reference)
      association = opts[:reference].split('.').first
      component_prefix = association.classify
      serializer_name = "#{component_prefix}Serializer"

      "ForestLiana::UserSpace::#{serializer_name}"
    end

    def smart_collection?
      !active_record_class
    end
  end

  attr_accessor :data

  def initialize(data)
    self.data = data
  end
end