cantino/huginn

View on GitHub
lib/liquid_migrator.rb

Summary

Maintainability
B
4 hrs
Test Coverage
module LiquidMigrator
  def self.convert_all_agent_options(agent)
    agent.options = self.convert_hash(agent.options, {:merge_path_attributes => true, :leading_dollarsign_is_jsonpath => true})
    agent.save!
  end

  def self.convert_hash(hash, options={})
    options = {:merge_path_attributes => false, :leading_dollarsign_is_jsonpath => false}.merge options
    keys_to_remove = []
    hash.tap do |hash|
      hash.each_pair do |key, value|
        case value.class.to_s
        when 'String', 'FalseClass', 'TrueClass'
          path_key = "#{key}_path"
          if options[:merge_path_attributes] && !hash[path_key].nil?
            # replace the value if the path is present
            value = hash[path_key] if hash[path_key].present?
            # in any case delete the path attibute
            keys_to_remove << path_key
          end
          hash[key] = LiquidMigrator.convert_string value, options[:leading_dollarsign_is_jsonpath]
        when 'ActiveSupport::HashWithIndifferentAccess'
          hash[key] = convert_hash(hash[key], options)
        when 'Array'
          hash[key] = hash[key].collect { |k|
            if k.class == String
              convert_string(k, options[:leading_dollarsign_is_jsonpath])
            else
              convert_hash(k, options)
            end
          }
        end
      end
        # remove the unneeded *_path attributes
    end.select { |k, v| !keys_to_remove.include? k }
  end

  def self.convert_string(string, leading_dollarsign_is_jsonpath=false)
    if string == true || string == false
      # there might be empty *_path attributes for boolean defaults
      string
    elsif string[0] == '$' && leading_dollarsign_is_jsonpath
      # in most cases a *_path attribute
      convert_json_path string
    else
      # migrate the old interpolation syntax to the new liquid based
      string.gsub(/<([^>]+)>/).each do
        match = $1
        if match =~ /\Aescape /
          # convert the old escape syntax to a liquid filter
          self.convert_json_path(match.gsub(/\Aescape /, '').strip, ' | uri_escape')
        else
          self.convert_json_path(match.strip)
        end
      end
    end
  end

  def self.convert_make_message(string)
    string.gsub(/<([^>]+)>/, "{{\\1}}")
  end

  def self.convert_json_path(string, filter = "")
    check_path(string)
    if string.start_with? '$.'
      "{{#{string[2..-1]}#{filter}}}"
    else
      "{{#{string[1..-1]}#{filter}}}"
    end
  end

  def self.check_path(string)
    if string !~ /\A(\$\.?)?(\w+\.)*(\w+)\Z/
      raise "JSONPath '#{string}' is too complex, please check your migration."
    end
  end
end