module MotionPrime
  module ScreenSectionsMixin
    extend ::MotionSupport::Concern

    include HasClassFactory
    include HasNormalizer

    def self.included(base)
      base.class_attribute :_section_options
    end

    attr_accessor :_action_section_options

    def main_section
      @main_section || all_sections.first
    end

    def main_section=(value)
      @main_section = value
    end

    def all_sections
      Array.wrap(@sections.try(:values))
    end

    def all_sections_with_main
      (all_sections + [main_section]).compact.uniq
    end

    def set_section(name, options = {})
      self._action_section_options ||= {}
      self._action_section_options[name.to_sym] = options
    end
    alias_method :section, :set_section

    def set_sections_wrapper(value)
      self.class.set_sections_wrapper(value)
    end

    def refresh
      all_sections_with_main.each { |s| s.try(:reload) }
    end

    protected
      def add_sections
        @main_section ||= nil
        create_sections
        render_sections
      end

      def create_sections
        section_options = self.class.section_options.merge(action_section_options)
        return unless section_options
        @sections = {}
        section_options.map do |name, options|
          if options[:instance]
            section = options[:instance]
          else
            section = create_section(name, options.clone)
          end
          @sections[name] = section if section
        end
      end

      def create_section(name, options)
        section_class = class_factory("#{name}_section")
        options = normalize_options(options).merge(screen: self)
        !options.has_key?(:if) || options[:if] ? section_class.new(options) : nil
      end

      def action_section_options
        _action_section_options || {}
      end

      def sections_wrapper
        self.class.sections_wrapper
      end

      def render_sections
        return unless @sections.present?
        table_wrap = sections_wrapper.nil? ? all_sections.count > 1 : sections_wrapper
        if table_wrap
          table_class = table_wrap.is_a?(TrueClass) ? MotionPrime::TableSection : table_class
          @main_section = table_class.new(model: all_sections, screen: self)
          @main_section.render
        else
          all_sections.each do |section|
            section.render
          end
        end
      end

    module ClassMethods
      def sections_wrapper
        @sections_wrapper
      end

      def set_sections_wrapper(value)
        @sections_wrapper = value
      end

      def section_options
        _section_options || {}
      end

      def section(name, options = {})
        self._section_options ||= {}
        self._section_options[name.to_sym] = options

        define_method name do
          @sections[name]
        end
      end
    end
  end
end