unparser 0.4.5 → 0.5.0

data/lib/unparser/emitter/alias.rb

@@ -11,16 +11,10 @@ module Unparser
 
     private
 
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
       def dispatch
-        write(K_ALIAS, WS)
+        write('alias ')
         visit(target)
-        write(WS)
+        ws
         visit(source)
       end
 

data/lib/unparser/emitter/args.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module Unparser
+  class Emitter
+    # Arguments emitter
+    class Args < self
+      def emit_block_arguments
+        delimited(normal_arguments)
+
+        write(',') if normal_arguments.one? && n_arg?(normal_arguments.first)
+
+        emit_shadowargs
+      end
+
+      def emit_def_arguments
+        delimited(normal_arguments)
+      end
+
+      def emit_lambda_arguments
+        delimited(normal_arguments)
+        emit_shadowargs
+      end
+
+    private
+
+      def emit_shadowargs
+        return if shadowargs.empty?
+
+        write('; ')
+        delimited(shadowargs)
+      end
+
+      def normal_arguments
+        children.reject(&method(:n_shadowarg?))
+      end
+      memoize :normal_arguments
+
+      def shadowargs
+        children.select(&method(:n_shadowarg?))
+      end
+      memoize :shadowargs
+
+    end # Arguments
+  end # Emitter
+end # Unparser

data/lib/unparser/emitter/argument.rb

@@ -2,121 +2,19 @@
 
 module Unparser
   class Emitter
-
-    # Arg expr (pattern args) emitter
-    class ArgExpr < self
-
-      handle :arg_expr
-
-      children :body
-
-    private
-
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
-      def dispatch
-        visit_parentheses(body)
-      end
-    end # ArgExpr
-
-    # Arguments emitter
-    class Arguments < self
-      include Terminated
-
-      handle :args
-
-      SHADOWARGS = ->(node) { node.type.equal?(:shadowarg) }.freeze
-      ARG        = ->(node) { node.type.equal?(:arg) }.freeze
-
-    private
-
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
-      def dispatch
-        delimited(normal_arguments)
-
-        write(', ') if procarg_disambiguator?
-
-        return if shadowargs.empty?
-
-        write('; ')
-        delimited(shadowargs)
-      end
-
-      # Test for procarg_disambiguator
-      #
-      # @return [Boolean]
-      #
-      # @api private
-      #
-      def procarg_disambiguator?
-        regular_block? && normal_arguments.all?(&ARG) && normal_arguments.one?
-      end
-
-      # Test for regular block
-      #
-      # @return [Boolean]
-      #
-      # @api private
-      #
-      def regular_block?
-        parent_type.equal?(:block) && !parent.node.children.first.type.equal?(:lambda)
-      end
-
-      # Return normal arguments
-      #
-      # @return [Enumerable<Parser::AST::Node>]
-      #
-      # @api private
-      #
-      def normal_arguments
-        children.reject(&SHADOWARGS)
-      end
-      memoize :normal_arguments
-
-      # Return shadow args
-      #
-      # @return [Enumerable<Parser::AST::Node>]
-      #
-      # @api private
-      #
-      def shadowargs
-        children.select(&SHADOWARGS)
-      end
-      memoize :shadowargs
-
-    end # Arguments
-
     # Emitter for block and kwrestarg arguments
     class Morearg < self
-      include Terminated
-
       MAP = {
-        blockarg:  T_AMP,
-        kwrestarg: T_DSPLAT
+        blockarg:  '&',
+        kwrestarg: '**'
       }.freeze
 
-      handle :blockarg
-      handle :kwrestarg
+      handle(*MAP.keys)
 
       children :name
 
     private
 
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
       def dispatch
         write(MAP.fetch(node_type), name.to_s)
       end
@@ -125,44 +23,28 @@ module Unparser
 
     # Optional argument emitter
     class Optarg < self
-      include Terminated
-
       handle :optarg
 
       children :name, :value
 
     private
 
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
       def dispatch
-        write(name.to_s, WS, T_ASN, WS)
+        write(name.to_s, ' = ')
         visit(value)
       end
     end
 
     # Optional keyword argument emitter
     class KeywordOptional < self
-      include Terminated
-
       handle :kwoptarg
 
       children :name, :value
 
     private
 
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
       def dispatch
-        write(name.to_s, T_COLON, WS)
+        write(name.to_s, ': ')
         visit(value)
       end
 
@@ -170,64 +52,40 @@ module Unparser
 
     # Keyword argument emitter
     class Kwarg < self
-      include Terminated
-
       handle :kwarg
 
       children :name
 
     private
 
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
       def dispatch
-        write(name.to_s, T_COLON)
+        write(name.to_s, ':')
       end
 
     end # Restarg
 
     # Rest argument emitter
     class Restarg < self
-      include Terminated
-
       handle :restarg
 
       children :name
 
     private
 
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
       def dispatch
-        write(T_SPLAT, name.to_s)
+        write('*', name.to_s)
       end
 
     end # Restarg
 
     # Argument emitter
     class Argument < self
-      include Terminated
-
       handle :arg, :shadowarg
 
       children :name
 
     private
 
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
       def dispatch
         write(name.to_s)
       end
@@ -236,59 +94,39 @@ module Unparser
 
     # Progarg emitter
     class Procarg < self
-      include Terminated
-
       handle :procarg0
 
-      children :first_argument
+      PARENS = %i[restarg mlhs].freeze
 
     private
 
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
       def dispatch
-        if first_argument.instance_of?(Symbol)
-          write(first_argument.to_s)
+        if needs_parens?
+          parentheses do
+            delimited(children)
+          end
         else
-          emit_multiple_children
+          delimited(children)
         end
       end
 
-      # Emit multiple children
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
-      def emit_multiple_children
-        parentheses do
-          delimited(children)
+      def needs_parens?
+        children.length > 1 || children.any? do |node|
+          PARENS.include?(node.type)
         end
       end
     end
 
     # Block pass node emitter
     class BlockPass < self
-      include Terminated
-
       handle :block_pass
 
       children :name
 
     private
 
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
       def dispatch
-        write(T_AMP)
+        write('&')
         visit(name)
       end
 

data/lib/unparser/emitter/array.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Unparser
+  class Emitter
+    # Array literal emitter
+    class Array < self
+      handle :array
+
+      def emit_heredoc_reminders
+        emitters.each(&:emit_heredoc_reminders)
+      end
+
+    private
+
+      def dispatch
+        parentheses('[', ']') do
+          delimited(emitters, &:write_to_buffer)
+        end
+      end
+
+      def emitters
+        children.map(&method(:emitter))
+      end
+      memoize :emitters
+    end # Array
+  end # Emitter
+end # Unparser

data/lib/unparser/emitter/array_pattern.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Unparser
+  class Emitter
+    # Emitter for array patterns
+    class ArrayPattern < self
+
+      handle :array_pattern
+      handle :array_pattern_with_tail
+
+    private
+
+      def dispatch
+        write('[')
+        delimited(children, &method(:emit_member))
+        write(', ') if node_type.equal?(:array_pattern_with_tail)
+        write(']')
+      end
+
+      def emit_member(node)
+        if n_match_rest?(node)
+          writer_with(MatchRest, node).emit_array_pattern
+        else
+          visit(node)
+        end
+      end
+    end # Pin
+  end # Emitter
+end # Unparser

data/lib/unparser/emitter/assignment.rb

@@ -5,163 +5,72 @@ module Unparser
 
     # Base class for assignment emitters
     class Assignment < self
+      BINARY_OPERATOR = %i[and or].freeze
+
+      def symbol_name
+        true
+      end
+
+      def emit_heredoc_reminders
+        return unless right
+
+        emitter(right).emit_heredoc_reminders
+      end
 
     private
 
-      # Perform dispatch
-      #
-      # @return [undefined]
-      #
-      # @api private
-      #
       def dispatch
         emit_left
         emit_right
       end
 
-      # Single assignment emitter
-      class Single < self
-
-        # Test for terminated emit
-        #
-        # @return [Boolean]
-        #
-        # @api private
-        #
-        def terminated?
-          right.nil?
-        end
-
-      private
+      def emit_right
+        return unless right
 
-        # Emit right
-        #
-        # @return [undefined]
-        #
-        # @api private
-        #
-        def emit_right
-          return unless right
+        write(' = ')
 
-          write(WS, T_ASN, WS)
+        if BINARY_OPERATOR.include?(right.type)
+          writer_with(Writer::Binary, right).emit_operator
+        else
           visit(right)
         end
+      end
 
-        abstract_method :emit_left
-
-        # Variable assignment emitter
-        class Variable < self
-
-          handle :lvasgn, :ivasgn, :cvasgn, :gvasgn
-
-          children :name, :right
-
-        private
-
-          # Emit left
-          #
-          # @return [undefined]
-          #
-          # @api private
-          #
-          def emit_left
-            write(name.to_s)
-          end
-
-        end # Variable
-
-        # Constant assignment emitter
-        class Constant < self
-
-          handle :casgn
-
-          children :base, :name, :right
-
-        private
-
-          # Emit left
-          #
-          # @return [undefined]
-          #
-          # @api private
-          #
-          def emit_left
-            if base
-              visit(base)
-              write(T_DCL) if base.type != :cbase
-            end
-            write(name.to_s)
-          end
+      abstract_method :emit_left
 
-        end # Constant
-      end # Single
+      # Variable assignment emitter
+      class Variable < self
 
-      # Multiple assignment
-      class Multiple < self
-        include Unterminated
+        handle :lvasgn, :ivasgn, :cvasgn, :gvasgn
 
-        handle :masgn
+        children :name, :right
 
       private
 
-        # Emit left
-        #
-        # @return [undefined]
-        #
-        # @api private
-        #
         def emit_left
-          visit_plain(first_child)
+          write(name.to_s)
         end
 
-        # Emit right
-        #
-        # @return [undefined]
-        #
-        # @api private
-        #
-        def emit_right
-          write(WS, T_ASN, WS)
-          visit(children.last)
-        end
+      end # Variable
 
-      end # Multiple
+      # Constant assignment emitter
+      class Constant < self
 
-      # Emitter for multiple assignment left hand side
-      class MLHS < Emitter
-        include Unterminated
+        handle :casgn
 
-        handle :mlhs
+        children :base, :name, :right
 
       private
 
-        NO_COMMA = %i[splat restarg].to_set.freeze
-        PARENT_MLHS = %i[mlhs masgn].freeze
-
-        # Perform dispatch
-        #
-        # @return [undefined]
-        #
-        # @api private
-        #
-        def dispatch
-          delimited(children)
-
-          write(',') if children.one? && mlhs?
-        end
-
-        # Test for mlhs context
-        #
-        # @return [undefined]
-        #
-        # @api private
-        #
-        def mlhs?
-          !NO_COMMA.include?(first_child.type) && PARENT_MLHS.include?(parent_type)
+        def emit_left
+          if base
+            visit(base)
+            write('::') unless n_cbase?(base)
+          end
+          write(name.to_s)
         end
 
-      end # MLHS
-
+      end # Constant
     end # Assignment
   end # Emitter
 end # Unparser