Contents

# frozen_string_literal: true

module HasStates
  class Configuration
    class ModelConfiguration
      attr_reader :model_class, :state_types

      def initialize(model_class)
        @model_class = model_class
        @state_types = {}
      end

      def state_type(name)
        type = StateTypeConfiguration.new(name)

        yield(type) if block_given?

        @state_types[name.to_s] = type

        generate_state_type_scope(name)
        generate_status_predicates(type.statuses)
        generate_state_type_status_predicates(name, type.statuses)
      end

      private

      def generate_state_type_scope(state_type)
        HasStates::State.scope state_type, -> { where(state_type: state_type) }
      end

      def generate_status_predicates(statuses)
        statuses.each do |status_name|
          HasStates::State.define_method(:"#{status_name}?") do
            status == status_name
          end
        end
      end

      def generate_state_type_status_predicates(state_type, statuses)
        statuses.each do |status_name|
          @model_class.define_method(:"#{state_type}_#{status_name}?") do
            states.where(state_type: state_type, status: status_name).exists?
          end
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
stateful_models-0.0.2	lib/has_states/configuration/model_configuration.rb
stateful_models-0.0.1	lib/has_states/configuration/model_configuration.rb