class Lookbook::PreviewEntity

def after_render_method
  fetch_config(:after_render)
end

def default_scenario
  visible_scenarios.first
end

def display_options
  global_options = Lookbook.config.preview_display_options
  global_options.deep_merge(fetch_config(:display_options, {}))
end

def file_name_base
  @_file_name_slug ||= file_name(true).gsub(/(_component_preview|component_preview|_preview|preview|component)$/, "")
end

def grouped_scenario_entities
  scenarios = []
  scenario_entities.each.with_index(1) do |entity, i|
    if entity.group.nil?
      entity.default_priority = i
      scenarios << entity
    else
      group_name = entity.group.presence || entity.parent.name.pluralize
      group = scenarios.find do |s|
        s.is_a?(ScenarioGroupEntity) && s.name == Utils.name(group_name)
      end
      if group
        group.add_scenario(entity)
      else
        group = ScenarioGroupEntity.new(group_name, [entity], self)
        group.default_priority = i
        scenarios << group
      end
    end
  end
  scenarios
end

def guess_render_targets
  [preview_class.name.chomp("Preview").constantize&.name]
rescue
  []
end

def hidden_scenarios
  @_hidden_scenarios ||= ScenarioCollection.new(scenarios.select(&:hidden?))
end

def initialize(code_object)
  @code_object = code_object
  @preview_class = code_object.path.constantize
  @file_path = Pathname(code_object.file)
  @base_directories = Engine.preview_paths
  cleaned_path = preview_class.name.underscore.strip
    .gsub(/(_component_preview|_preview)(\..+)?$/, "")
    .gsub(/\/(component_preview|preview|component)(\..+)?$/, "")
  @lookup_path = PathUtils.to_lookup_path(cleaned_path)
end

def inspect_path
  lookbook_inspect_path(lookup_path)
end

def layout
  preview_class.instance_variable_get(:@layout)
end

def load_scenarios
  code_object.groups.any? ? grouped_scenario_entities : scenario_entities
end

def preview_class_name
  preview_class.name
end

def preview_path
  lookbook_preview_path(lookup_path)
end

def render_target
  render_targets.first
end

def render_targets
  @_render_targets ||= RenderTargetCollection.new(scenarios.flat_map(&:render_targets).uniq(&:lookup_path))
end

def scenario(scenario_name)
  scenarios.find { |m| m.name == scenario_name.to_s }
end

def scenario_entities
  public_methods = preview_class.public_instance_methods(false)
  method_objects = code_object.meths.select { |m| public_methods.include?(m.name) }
  method_objects.map.with_index { |code_object, i| ScenarioEntity.new(code_object, self, priority: i) }
end

def scenarios
  @_scenarios ||= ScenarioCollection.new(load_scenarios)
end

def visible_scenarios
  @_visible_scenarios ||= ScenarioCollection.new(scenarios.select(&:visible?))
end