Contents

# frozen_string_literal: true

module Leftovers
  module Precompilers
    module YAML
      class Builder < ::Psych::TreeBuilder
        def initialize
          @constants = []

          super
        end

        def add_constant_for_tag(tag, value = nil)
          match = %r{\A!ruby/[^:]*(?::(.*))?\z}.match(tag)
          return unless match

          @constants << (match[1] || value)
        end

        def start_mapping(_anchor, tag, *rest) # leftovers:keep
          add_constant_for_tag(tag)
          tag = nil

          super
        end

        def start_sequence(_anchor, tag, *rest) # leftovers:keep
          add_constant_for_tag(tag)
          tag = nil

          super
        end

        def scalar(value, _anchor, tag, *rest) # leftovers:keep
          add_constant_for_tag(tag, value)
          tag = nil

          super
        end

        def to_ruby_file
          <<~FILE
            __leftovers_document(#{to_ruby_argument(root.to_ruby.first)})
            #{@constants.join("\n")}
          FILE
        end

        private

        def to_ruby_argument(value)
          ruby = value.inspect
          return ruby unless value.is_a?(::Array)

          ruby.delete_prefix!('[')
          ruby.delete_suffix!(']')

          ruby
        end
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
leftovers-0.12.2	lib/leftovers/precompilers/yaml/builder.rb
leftovers-0.12.1	lib/leftovers/precompilers/yaml/builder.rb
leftovers-0.12.0	lib/leftovers/precompilers/yaml/builder.rb
leftovers-0.11.2	lib/leftovers/precompilers/yaml/builder.rb