turboladen/tailor

require_relative '../ruler'

class Tailor
  module Rulers
    class SpacesAfterLparenRuler < Tailor::Ruler
      def initialize(config, options)
        super(config, options)
        add_lexer_observers :comment, :ignored_nl, :lparen, :nl
        @lparen_columns = []
      end

      def comment_update(token, lexed_line, _, lineno, column)
        if token =~ /\n$/
          log 'Found comment with trailing newline.'
          ignored_nl_update(lexed_line, lineno, column)
        end
      end

      def ignored_nl_update(lexed_line, lineno, _)
        check_spaces_after_lparen(lexed_line, lineno)
      end

      def lparen_update(_, column)
        @lparen_columns << column
      end

      def nl_update(lexed_line, lineno, column)
        ignored_nl_update(lexed_line, lineno, column)
      end

      # Checks to see if the number of spaces after an lparen equals the value
      # at +@config+.
      #
      # @param [Fixnum] actual_spaces The number of spaces after the lparen.
      # @param [Fixnum] lineno Line the potential problem is on.
      # @param [Fixnum] column Column the potential problem is on.
      def measure(actual_spaces, lineno, column)
        if actual_spaces != @config
          msg = "Line has #{actual_spaces} space(s) after a (, "
          msg << "but should have #{@config}."

          @problems << Problem.new(problem_type, lineno, column + 1, msg,
            @options[:level])
        end
      end

      def check_spaces_after_lparen(lexed_line, lineno)
        unless @lparen_columns.empty?
          log "lparens found at: #{@lparen_columns}"
        end

        @lparen_columns.each do |column|
          actual_spaces = count_spaces(lexed_line, column)
          next if actual_spaces.nil?

          if !@do_measurement
            log 'Skipping measurement.'
          else
            measure(actual_spaces, lineno, column)
          end

          @do_measurement = true
        end

        @lparen_columns.clear
      end

      # Counts the number of spaces after the lparen.
      #
      # @param [LexedLine] lexed_line The LexedLine that contains the context
      #   the lparen was found in.
      # @param [Fixnum] column Column the lparen was found at.
      # @return [Fixnum] The number of spaces found after the lparen.
      def count_spaces(lexed_line, column)
        event_index = lexed_line.event_index(column)

        if event_index.nil?
          log 'No lparen in this line.  Moving on...'
          @do_measurement = false
          return
        end

        next_event = lexed_line.at(event_index + 1)
        log "Next event: #{next_event}"

        if next_event.nil?
          log 'lparen must be at the end of the line.'
          @do_measurement = false
          return 0
        end

        [:on_nl, :on_ignored_nl].each do |event|
          if next_event[1] == event
            log "lparen is followed by a '#{event}'.  Moving on."
            return 0
          end
        end

        if next_event[1] == :on_rparen
          log 'lparen is followed by an rparen.  Moving on.'
          @do_measurement = false
          return 0
        end

        second_next_event = lexed_line.at(event_index + 2)
        log "Event + 2: #{second_next_event}"

        [:on_comment, :on_lbrace, :on_lparen].each do |event|
          if second_next_event[1] == event
            log "Event + 2 is a #{event}.  Moving on."
            @do_measurement = false
            return next_event.last.size
          end
        end

        next_event[1] != :on_sp ? 0 : next_event.last.size
      end
    end
  end
end