opal/opal

# frozen_string_literal: true

require 'opal/nodes/base'

module Opal
  module Nodes
    class ArrayNode < Base
      handle :array

      def compile
        return push('[]') if children.empty?

        code, work = [], []

        children.each do |child|
          splat = child.type == :splat
          part  = expr(child)

          if splat
            if work.empty?
              if code.empty?
                code << fragment('[].concat(') << part << fragment(')')
              else
                code << fragment('.concat(') << part << fragment(')')
              end
            else
              if code.empty?
                code << fragment('[') << work << fragment(']')
              else
                code << fragment('.concat([') << work << fragment('])')
              end

              code << fragment('.concat(') << part << fragment(')')
            end
            work = []
          else
            work << fragment(', ') unless work.empty?
            work << part
          end
        end

        unless work.empty?
          join = [fragment('['), work, fragment(']')]

          if code.empty?
            code = join
          else
            code.push([fragment('.concat('), join, fragment(')')])
          end
        end

        push code
      end
    end
  end
end