require 'reek/source/sexp_node'

module Reek
  module Source
    module SexpExtensions
      module AndNode
        def condition() self[1..2].tap {|node| node.extend SexpNode } end
      end

      module OrNode
        def condition() self[1..2].tap {|node| node.extend SexpNode } end
      end

      module AttrasgnNode
        def args() self[3] end
      end

      module CaseNode
        def condition() self[1] end
      end

      module CallNode
        def receiver() self[1] end
        def method_name() self[2] end
        def args() self[3..-1] end
        def arg_names
          args.map {|arg| arg[1]}
        end
      end

      module CvarNode
        def name() self[1] end
      end

      CvasgnNode = CvarNode
      CvdeclNode = CvarNode

      module LvarNode
        def value() self[1] end
      end

      module MethodNode
        def arg_names
          @args ||= parameter_names.reject {|param| param.to_s =~ /^&/}
        end
        def parameter_names
          @param_names ||= argslist[1..-1].map { |param| Sexp === param ?  param[1] : param }
        end

        def name_without_bang
          name.to_s.chop
        end

        def ends_with_bang?
          name[-1] == '!'
        end
      end

      module DefnNode
        def name() self[1] end
        def argslist() self[2] end
        def body()
          self[3..-1].extend SexpNode
        end
        include MethodNode
        def full_name(outer)
          prefix = outer == '' ? '' : "#{outer}#"
          "#{prefix}#{name}"
        end
      end

      module DefsNode
        def receiver() self[1] end
        def name() self[2] end
        def argslist() self[3] end
        def body()
          self[4..-1].extend SexpNode
        end
        include MethodNode
        def full_name(outer)
          prefix = outer == '' ? '' : "#{outer}#"
          "#{prefix}#{SexpNode.format(receiver)}.#{name}"
        end
      end

      module IfNode
        def condition() self[1] end
      end

      module IterNode
        def call() self[1] end
        def args() self[2] end
        def block() self[3] end
        def parameters() self[2] || [] end
        def parameter_names
          parameters[1..-1].to_a
        end
      end

      module LitNode
        def value() self[1] end
      end

      module Colon2Node
        def name
          self[2]
        end

        def simple_name
          if name.is_a? Colon2Node
            name.simple_name
          else
            name
          end
        end
      end

      module ModuleNode
        def name() self[1] end

        def simple_name
          Sexp === name ? name.simple_name : name
        end

        def full_name(outer)
          prefix = outer == '' ? '' : "#{outer}::"
          "#{prefix}#{text_name}"
        end
        def text_name
          SexpNode.format(name)
        end
      end

      module ClassNode
        include ModuleNode
        def superclass() self[2] end
      end

      module YieldNode
        def args() self[1..-1] end
        def arg_names
          args.map {|arg| arg[1]}
        end
      end
    end
  end
end