openproject/db/migrate/migration_utils/queries.rb

#-- copyright
# OpenProject is a project management system.
#
# Copyright (C) 2012-2013 the OpenProject Foundation (OPF)
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License version 3.
#
# See doc/COPYRIGHT.rdoc for more details.
#++

require 'yaml'

require_relative 'utils'

module Migration
  module Utils

    def update_query_references_with_keys(keys)
      update_column_values('queries', COLUMNS.keys, update_query_reference(keys, COLUMNS), nil)
    end

    private

    COLUMNS = {
      'filters' => { is_text_column: false },
      'column_names' => { is_text_column: false },
      'sort_criteria' => { is_text_column: false },
      'group_by' => { is_text_column: true }
    }

    def update_query_reference(keys, columns)
      Proc.new do |row|
        columns.keys.each do |column|
          unless row[column].nil?
            if columns[column][:is_text_column]
              process_text_data(row, column, keys)
            else
              row[column] = process_yaml_data(row, column, keys)
            end
          end
        end

        UpdateResult.new(row, true)
      end
    end

    def process_yaml_data(row, column, keys)
      value = YAML.load row[column]

      if value.is_a? Array
        value.collect! do |e|
          if e.is_a? Array
            e.collect! {|v| keys.has_key?(v) ? keys[v] : v}
          else
            keys.has_key?(e.to_s) ? keys[e.to_s].to_sym : e
          end
        end
      elsif value.is_a? Hash
        keys.select {|k| value[k.to_s]}
            .each_pair {|k, v| value[v] = value.delete k}
      end

      YAML.dump value
    end

    def process_text_data(row, column, keys)
      value = row[column]

      keys.each_key do |k|
        regex = Regexp.new(k)
        replace = keys[k]

        value.gsub!(regex, replace)
      end

      value
    end
  end
end