# encoding: utf-8

module Unparser
  class Emitter
    # Emitter if nodes
    class If < self

      handle :if

      children :condition, :if_branch, :else_branch

    private

      # Perform dispatch
      #
      # @return [undefined]
      #
      # @api private
      #
      def dispatch
        if postcondition?
          emit_postcondition
        else
          emit_normal
        end
      end

      # Test for postcondition
      #
      # @return [true]
      #   if node must be emitted in postconditin style
      #
      # @return [false]
      #   otherwise
      #
      # @api private
      #
      def postcondition?
        return false unless !!if_branch ^ !!else_branch

        body = if_branch || else_branch

        AST.first_assignment_in_body_and_used_in_condition?(local_variable_root, body, condition)
      end

      # Emit in postcondition style
      #
      # @return [undefined]
      #
      # @api private
      #
      def emit_postcondition
        visit(if_branch || else_branch)
        write(WS, keyword, WS)
        emit_condition
      end

      # Emit in normal style
      #
      # @return [undefined]
      #
      # @api private
      #
      def emit_normal
        write(keyword, WS)
        emit_condition
        emit_if_branch
        emit_else_branch
        k_end
      end

      # Test for unless
      #
      # @return [true]
      #   if to emit as unless
      #
      # @return [false]
      #   otherwise
      #
      # @api private
      #
      def unless?
        !if_branch && else_branch
      end

      # Return keyword
      #
      # @return [String]
      #
      # @api private
      #
      def keyword
        unless? ? K_UNLESS : K_IF
      end

      # Emit condition
      #
      # @return [undefined]
      #
      # @api private
      #
      def emit_condition
        visit(condition)
      end

      # Emit if branch
      #
      # @return [undefined]
      #
      # @api private
      #
      def emit_if_branch
        if if_branch
          visit_indented(if_branch)
        end

        if !if_branch && !else_branch
          nl
        end
      end

      # Emit else branch
      #
      # @return [undefined]
      #
      # @api private
      #
      def emit_else_branch
        return unless else_branch
        write(K_ELSE) unless unless?
        visit_indented(else_branch)
      end

    end # If
  end # Emitter
end # Unparser