# frozen_string_literal: true

require "action_view"
require "action_view/base"
require "view_component/collection"
require "view_component/compile_cache"
require "view_component/compiler"
require "view_component/config"
require "view_component/errors"
require "view_component/inline_template"
require "view_component/preview"
require "view_component/request_details"
require "view_component/slotable"
require "view_component/template"
require "view_component/translatable"
require "view_component/with_content_helper"

module ActionView
  class OutputBuffer
    def with_buffer(buf = nil)
      new_buffer = buf || +""
      old_buffer, @raw_buffer = @raw_buffer, new_buffer
      yield
      new_buffer
    ensure
      @raw_buffer = old_buffer
    end
  end
end

module ViewComponent
  class Base
    class << self
      delegate(*ViewComponent::Config.defaults.keys, to: :config)

      # Returns the current config.
      #
      # @return [ActiveSupport::OrderedOptions]
      def config
        module_parents.each do |module_parent|
          next unless module_parent.respond_to?(:config)
          module_parent_config = module_parent.config.try(:view_component)
          return module_parent_config if module_parent_config
        end
        ViewComponent::Config.current
      end
    end

    include ActionView::Helpers
    include Rails.application.routes.url_helpers if defined?(Rails.application.routes)
    include ERB::Escape
    include ActiveSupport::CoreExt::ERBUtil

    include ViewComponent::InlineTemplate
    include ViewComponent::Slotable
    include ViewComponent::Translatable
    include ViewComponent::WithContentHelper

    # For CSRF authenticity tokens in forms
    delegate :form_authenticity_token, :protect_against_forgery?, :config, to: :helpers

    # HTML construction methods
    delegate :output_buffer, :lookup_context, :view_renderer, :view_flow, to: :helpers

    # For Turbo::StreamsHelper
    delegate :formats, :formats=, to: :helpers

    # For Content Security Policy nonces
    delegate :content_security_policy_nonce, to: :helpers

    # Config option that strips trailing whitespace in templates before compiling them.
    class_attribute :__vc_strip_trailing_whitespace, instance_accessor: false, instance_predicate: false, default: false

    attr_accessor :__vc_original_view_context
    attr_reader :current_template

    # Components render in their own view context. Helpers and other functionality
    # require a reference to the original Rails view context, an instance of
    # `ActionView::Base`. Use this method to set a reference to the original
    # view context. Objects that implement this method will render in the component's
    # view context, while objects that don't will render in the original view context
    # so helpers, etc work as expected.
    #
    # @param view_context [ActionView::Base] The original view context.
    # @return [void]
    def set_original_view_context(view_context)
      # noop
    end

    using RequestDetails

    # Including `Rails.application.routes.url_helpers` defines an initializer that accepts (...),
    # so we have to define our own empty initializer to overwrite it.
    def initialize
    end

    # Entrypoint for rendering components.
    #
    # - `view_context`: ActionView context from calling view
    # - `block`: optional block to be captured within the view context
    #
    # Returns HTML that has been escaped by the respective template handler.
    #
    # @return [String]
    def render_in(view_context, &block)
      self.class.__vc_compile(raise_errors: true)

      @view_context = view_context
      @old_virtual_path = view_context.instance_variable_get(:@virtual_path)
      self.__vc_original_view_context ||= view_context

      @output_buffer = view_context.output_buffer

      @lookup_context ||= view_context.lookup_context

      # For content_for
      @view_flow ||= view_context.view_flow

      # For i18n
      @virtual_path ||= virtual_path

      # Describes the inferred request constraints (locales, formats, variants)
      @__vc_requested_details ||= @lookup_context.vc_requested_details

      # For caching, such as #cache_if
      @current_template = nil unless defined?(@current_template)
      old_current_template = @current_template

      if block && defined?(@__vc_content_set_by_with_content)
        raise DuplicateContentError.new(self.class.name)
      end

      @__vc_content_evaluated = false
      @__vc_render_in_block = block
      @view_context.instance_variable_set(:@virtual_path, virtual_path)

      before_render

      if render?
        value = nil

        @output_buffer.with_buffer do
          rendered_template =
            around_render do
              render_template_for(@__vc_requested_details).to_s
            end

          # Avoid allocating new string when output_preamble and output_postamble are blank
          value = if output_preamble.blank? && output_postamble.blank?
            rendered_template
          else
            __vc_safe_output_preamble + rendered_template + __vc_safe_output_postamble
          end
        end

        if ActionView::Base.annotate_rendered_view_with_filenames && current_template.inline_call? && request&.format == :html
          identifier = defined?(Rails.root) ? self.class.identifier.sub("#{Rails.root}/", "") : self.class.identifier
          value = "<!-- BEGIN #{identifier} -->".html_safe + value + "<!-- END #{identifier} -->".html_safe
        end

        value
      else
        ""
      end
    ensure
      view_context.instance_variable_set(:@virtual_path, @old_virtual_path)
      @current_template = old_current_template
    end

    # Subclass components that call `super` inside their template code will cause a
    # double render if they emit the result.
    #
    # ```erb
    # <%= super %> # double-renders
    # <% super %> # doesn't double-render
    # ```
    #
    # `super` also doesn't consider the current variant. `render_parent` renders the
    # parent template considering the current variant and emits the result without
    # double-rendering.
    def render_parent
      render_parent_to_string
      nil
    end

    # Renders the parent component to a string and returns it. This method is meant
    # to be used inside custom #call methods when a string result is desired, eg.
    #
    # ```ruby
    # def call
    #   "<div>#{render_parent_to_string}</div>"
    # end
    # ```
    #
    # When rendering the parent inside an .erb template, use `#render_parent` instead.
    def render_parent_to_string
      @__vc_parent_render_level ||= 0 # ensure a good starting value

      begin
        target_render = self.class.instance_variable_get(:@__vc_ancestor_calls)[@__vc_parent_render_level]
        @__vc_parent_render_level += 1

        target_render.bind_call(self, @__vc_requested_details)
      ensure
        @__vc_parent_render_level -= 1
      end
    end

    # Optional content to be returned before the rendered template.
    #
    # @return [String]
    def output_preamble
      @@default_output_preamble ||= "".html_safe
    end

    # Optional content to be returned after the rendered template.
    #
    # @return [String]
    def output_postamble
      @@default_output_postamble ||= "".html_safe
    end

    # Called before rendering the component. Override to perform operations that
    # depend on having access to the view context, such as helpers.
    #
    # @return [void]
    def before_render
      # noop
    end

    # Called around rendering the component. Override to wrap the rendering of a
    # component in custom instrumentation, etc.
    #
    # @return [void]
    def around_render
      yield
    end

    # Override to determine whether the ViewComponent should render.
    #
    # @return [Boolean]
    def render?
      true
    end

    # Re-use original view_context if we're not rendering a component.
    #
    # As of v4, ViewComponent::Base re-uses the existing view context created
    # by ActionView, meaning the current view context and the original view
    # context are the same object. set_original_view_context is still called
    # to maintain backwards compatibility.
    #
    # @private
    def render(options = {}, args = {}, &block)
      if options.respond_to?(:set_original_view_context)
        options.set_original_view_context(self.__vc_original_view_context)

        # We assume options is a component, so there's no need to evaluate the
        # block in the view context as we do below.
        @view_context.render(options, args, &block)
      elsif block
        __vc_original_view_context.render(options, args) do
          # capture the block output in the view context of the component
          output = capture(&block)

          # Partials are rendered to their own buffer and do not append to the
          # original @output_buffer we retain a reference to in #render_in. This
          # is a problem since the block passed to us here in the #render method
          # is evaluated within the context of ViewComponent::Base, and thus
          # appends to the original @output_buffer. To avoid this, we evaluate the
          # block in the view context instead, which will append to the output buffer
          # created for the partial.
          __vc_original_view_context.capture { output }
        end
      else
        __vc_original_view_context.render(options, args)
      end
    end

    # The current controller. Use sparingly as doing so introduces coupling
    # that inhibits encapsulation & reuse, often making testing difficult.
    #
    # @return [ActionController::Base]
    def controller
      raise ControllerCalledBeforeRenderError if view_context.nil?

      @__vc_controller ||= view_context.controller
    end

    # A proxy through which to access helpers. Use sparingly as doing so introduces
    # coupling that inhibits encapsulation & reuse, often making testing difficult.
    #
    # @return [ActionView::Base]
    def helpers
      raise HelpersCalledBeforeRenderError if view_context.nil?

      # Attempt to re-use the original view_context passed to the first
      # component rendered in the rendering pipeline. This prevents the
      # instantiation of a new view_context via `controller.view_context` which
      # always returns a new instance of the view context class.
      #
      # This allows ivars to remain persisted when using the same helper via
      # `helpers` across multiple components and partials.
      @__vc_helpers ||= __vc_original_view_context || controller.view_context
    end

    if defined?(Rails.env) && (::Rails.env.development? || ::Rails.env.test?)
      # @private
      def method_missing(method_name, *args) # rubocop:disable Style/MissingRespondToMissing
        super
      rescue => e # rubocop:disable Style/RescueStandardError
        e.set_backtrace e.backtrace.tap(&:shift)
        raise e, <<~MESSAGE.chomp if view_context && e.is_a?(NameError) && helpers.respond_to?(method_name)
          #{e.message}

          You may be trying to call a method provided as a view helper. Did you mean `helpers.#{method_name}`?
        MESSAGE

        raise
      end
    end

    # Exposes .virtual_path as an instance method
    #
    # @private
    def virtual_path
      self.class.virtual_path
    end

    # For caching, such as #cache_if
    # @private
    def view_cache_dependencies
      []
    end

    if defined?(Rails::VERSION) && Rails::VERSION::MAJOR == 7 && Rails::VERSION::MINOR == 1
      # Rails expects us to define `format` on all renderables,
      # but we do not know the `format` of a ViewComponent until runtime.
      def format
        nil
      end
    end

    # The current request. Use sparingly as doing so introduces coupling that
    # inhibits encapsulation & reuse, often making testing difficult.
    #
    # @return [ActionDispatch::Request]
    def request
      __vc_request
    end

    # @private
    def __vc_request
      # The current request (if present, as mailers/jobs/etc do not have a request)
      @__vc_request ||= controller.request if controller.respond_to?(:request)
    end

    # The content passed to the component instance as a block.
    #
    # @return [String]
    def content
      @__vc_content_evaluated = true
      return @__vc_content if defined?(@__vc_content)

      @__vc_content =
        if __vc_render_in_block_provided?
          with_captured_virtual_path(@old_virtual_path) do
            view_context.capture(self, &@__vc_render_in_block)
          end
        elsif __vc_content_set_by_with_content_defined?
          @__vc_content_set_by_with_content
        end
    end

    # Whether `content` has been passed to the component.
    #
    # @return [Boolean]
    def content?
      __vc_render_in_block_provided? || __vc_content_set_by_with_content_defined?
    end

    # @private
    # Temporarily sets the virtual path to the captured value, then restores it.
    # This ensures translations and other path-dependent code execute with the correct scope.
    def with_captured_virtual_path(captured_path)
      old_virtual_path = @view_context.instance_variable_get(:@virtual_path)
      @view_context.instance_variable_set(:@virtual_path, captured_path)
      yield
    ensure
      @view_context.instance_variable_set(:@virtual_path, old_virtual_path)
    end

    private

    attr_reader :view_context

    def __vc_render_in_block_provided?
      defined?(@view_context) && @view_context && @__vc_render_in_block
    end

    def __vc_content_set_by_with_content_defined?
      defined?(@__vc_content_set_by_with_content)
    end

    def __vc_maybe_escape_html(text)
      return text if @current_template && !@current_template.html?
      return text if text.blank?

      if text.html_safe?
        text
      else
        yield
        html_escape(text)
      end
    end

    def __vc_safe_output_preamble
      __vc_maybe_escape_html(output_preamble) do
        Kernel.warn("WARNING: The #{self.class} component was provided an HTML-unsafe preamble. The preamble will be automatically escaped, but you may want to investigate.")
      end
    end

    def __vc_safe_output_postamble
      __vc_maybe_escape_html(output_postamble) do
        Kernel.warn("WARNING: The #{self.class} component was provided an HTML-unsafe postamble. The postamble will be automatically escaped, but you may want to investigate.")
      end
    end

    # Configuration for generators.
    #
    # All options under this namespace default to `false` unless otherwise
    # stated.
    #
    # #### #sidecar
    #
    # Always generate a component with a sidecar directory:
    #
    # ```ruby
    # config.view_component.generate.sidecar = true
    # ```
    #
    # #### #stimulus_controller
    #
    # Always generate a Stimulus controller alongside the component:
    #
    # ```ruby
    # config.view_component.generate.stimulus_controller = true
    # ```
    #
    # #### `#typescript`
    #
    # Generate TypeScript files instead of JavaScript files:
    #
    # ```ruby
    # config.view_component.generate.typescript = true
    # ```
    #
    # #### #locale
    #
    # Always generate translations file alongside the component:
    #
    # ```ruby
    # config.view_component.generate.locale = true
    # ```
    #
    # #### #distinct_locale_files
    #
    # Always generate as many translations files as available locales:
    #
    # ```ruby
    # config.view_component.generate.distinct_locale_files = true
    # ```
    #
    # One file will be generated for each configured `I18n.available_locales`,
    # falling back to `[:en]` when no `available_locales` is defined.
    #
    # #### #preview
    #
    # Always generate preview alongside the component:
    #
    # ```ruby
    # config.view_component.generate.preview = true
    # ```
    #
    #  Defaults to `false`.
    #
    # #### ßparent_class
    #
    # Parent class for generated components
    #
    # ```ruby
    # config.view_component.generate.parent_class = "MyBaseComponent"
    # ```
    #
    # Defaults to nil. If this is falsy, generators will use
    # "ApplicationComponent" if defined, "ViewComponent::Base" otherwise.
    #

    class << self
      # The file path of the component Ruby file.
      #
      # @return [String]
      attr_reader :identifier

      # @private
      attr_writer :identifier

      # @private
      attr_accessor :virtual_path

      # Find sidecar files for the given extensions.
      #
      # The provided array of extensions is expected to contain
      # strings starting without the dot, example: `["erb", "haml"]`.
      #
      # For example, one might collect sidecar CSS files that need to be compiled.
      # @param extensions [Array<String>] Extensions of which to return matching sidecar files.
      def sidecar_files(extensions)
        return [] unless identifier

        extensions = extensions.join(",")

        # view files in a directory named like the component
        directory = File.dirname(identifier)
        filename = File.basename(identifier, ".rb")
        component_name = name.demodulize.underscore

        # Add support for nested components defined in the same file.
        #
        # for example
        #
        # class MyComponent < ViewComponent::Base
        #   class MyOtherComponent < ViewComponent::Base
        #   end
        # end
        #
        # Without this, `MyOtherComponent` will not look for `my_component/my_other_component.html.erb`
        nested_component_files =
          if name.include?("::") && component_name != filename
            Dir["#{directory}/#{filename}/#{component_name}.*{#{extensions}}"]
          else
            []
          end

        # view files in the same directory as the component
        sidecar_files = Dir["#{directory}/#{component_name}.*{#{extensions}}"]

        sidecar_directory_files = Dir["#{directory}/#{component_name}/#{filename}.*{#{extensions}}"]

        (sidecar_files - [identifier] + sidecar_directory_files + nested_component_files).uniq
      end

      # Render a component for each element in a collection ([documentation](/guide/collections)):
      #
      # ```ruby
      # render(ProductsComponent.with_collection(@products, foo: :bar))
      # ```
      #
      # @param collection [Enumerable] A list of items to pass the ViewComponent one at a time.
      # @param spacer_component [ViewComponent::Base] Component instance to be rendered between items.
      # @param args [Arguments] Arguments to pass to the ViewComponent every time.
      def with_collection(collection, spacer_component: nil, **args)
        Collection.new(self, collection, spacer_component, **args)
      end

      # @private
      def __vc_compile(raise_errors: false, force: false)
        __vc_compiler.compile(raise_errors: raise_errors, force: force)
      end

      # @private
      def inherited(child)
        # Compile so child will inherit compiled `call_*` template methods that
        # `compile` defines
        __vc_compile

        # Give the child its own personal #render_template_for to protect against the case when
        # eager loading is disabled and the parent component is rendered before the child. In
        # such a scenario, the parent will override ViewComponent::Base#render_template_for,
        # meaning it will not be called for any children and thus not compile their templates.
        if !child.method_defined?(:render_template_for, false) && !child.__vc_compiled?
          child.class_eval <<~RUBY, __FILE__, __LINE__ + 1
            def render_template_for(requested_details)
              # Force compilation here so the compiler always redefines render_template_for.
              # This is mostly a safeguard to prevent infinite recursion.
              self.class.__vc_compile(raise_errors: true, force: true)
              # .__vc_compile replaces this method; call the new one
              render_template_for(requested_details)
            end
          RUBY
        end

        # Derive the source location of the component Ruby file from the call stack.
        # We need to ignore `inherited` frames here as they indicate that `inherited`
        # has been re-defined by the consuming application, likely in ApplicationComponent.
        # We use `base_label` method here instead of `label` to avoid cases where the method
        # owner is included in a prefix like `ApplicationComponent.inherited`.
        child.identifier = caller_locations(1, 10).reject { |l| l.base_label == "inherited" }[0].path
        child.virtual_path = child.name&.underscore

        # Set collection parameter to the extended component
        child.with_collection_parameter(__vc_provided_collection_parameter)

        if method_defined?(:render_template_for, false)
          vc_ancestor_calls = defined?(@__vc_ancestor_calls) ? @__vc_ancestor_calls.dup : []

          vc_ancestor_calls.unshift(instance_method(:render_template_for))
          child.instance_variable_set(:@__vc_ancestor_calls, vc_ancestor_calls)
        end

        super
      end

      # @private
      def __vc_compiled?
        __vc_compiler.compiled?
      end

      # Hook called by the compiler after a component is compiled.
      #
      # Extensions can override this class method to run logic after
      # compilation (e.g., generate helpers, register metadata, etc.).
      #
      # By default, this is a no-op.
      def after_compile
        # no-op by default
      end

      # @private
      def __vc_ensure_compiled
        __vc_compile unless __vc_compiled?
      end

      # @private
      def __vc_compiler
        @__vc_compiler ||= Compiler.new(self)
      end

      # Set the parameter name used when rendering elements of a collection ([documentation](/guide/collections)):
      #
      # ```ruby
      # with_collection_parameter :item
      # ```
      #
      # @param parameter [Symbol] The parameter name used when rendering elements of a collection.
      def with_collection_parameter(parameter)
        @__vc_provided_collection_parameter = parameter
        @__vc_initialize_parameters = nil
      end

      # Strips trailing whitespace from templates before compiling them.
      #
      # ```ruby
      # class MyComponent < ViewComponent::Base
      #   strip_trailing_whitespace
      # end
      # ```
      #
      # @param value [Boolean] Whether to strip newlines.
      def strip_trailing_whitespace(value = true)
        self.__vc_strip_trailing_whitespace = value
      end

      # Whether trailing whitespace will be stripped before compilation.
      #
      # @return [Boolean]
      def strip_trailing_whitespace?
        __vc_strip_trailing_whitespace
      end

      # Ensure the component initializer accepts the
      # collection parameter. By default, we don't
      # validate that the default parameter name
      # is accepted, as support for collection
      # rendering is optional.
      # @private
      def __vc_validate_collection_parameter!(validate_default: false)
        parameter = validate_default ? __vc_collection_parameter : __vc_provided_collection_parameter

        return unless parameter
        return if __vc_initialize_parameter_names.include?(parameter) || __vc_splatted_keyword_argument_present?

        raise MissingCollectionArgumentError.new(name, parameter)
      end

      # Ensure the component initializer doesn't define
      # invalid parameters that could override the framework's
      # methods.
      # @private
      def __vc_validate_initialization_parameters!
        return unless __vc_initialize_parameter_names.include?(:content)

        raise ReservedParameterError.new(name, :content)
      end

      # @private
      def __vc_collection_parameter
        @__vc_provided_collection_parameter ||= name && name.demodulize.underscore.chomp("_component").to_sym
      end

      # @private
      def __vc_collection_counter_parameter
        @__vc_collection_counter_parameter ||= :"#{__vc_collection_parameter}_counter"
      end

      # @private
      def __vc_counter_argument_present?
        __vc_initialize_parameter_names.include?(__vc_collection_counter_parameter)
      end

      # @private
      def __vc_collection_iteration_parameter
        @__vc_collection_iteration_parameter ||= :"#{__vc_collection_parameter}_iteration"
      end

      # @private
      def __vc_iteration_argument_present?
        __vc_initialize_parameter_names.include?(__vc_collection_iteration_parameter)
      end

      private

      def __vc_splatted_keyword_argument_present?
        __vc_initialize_parameters.flatten.include?(:keyrest)
      end

      def __vc_initialize_parameter_names
        @__vc_initialize_parameter_names ||=
          if respond_to?(:attribute_names)
            attribute_names.map(&:to_sym)
          else
            __vc_initialize_parameters.map(&:last)
          end
      end

      def __vc_initialize_parameters
        @__vc_initialize_parameters ||= instance_method(:initialize).parameters
      end

      def __vc_provided_collection_parameter
        @__vc_provided_collection_parameter ||= nil
      end
    end

    ActiveSupport.run_load_hooks(:view_component, self)
  end
end