OpenProject is the leading open source project management software.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
openproject/db/migrate/migration_utils/queries.rb

99 lines
2.8 KiB

#-- encoding: UTF-8
#-- copyright
# OpenProject is a project management system.
# Copyright (C) 2012-2015 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.
#
# OpenProject is a fork of ChiliProject, which is a fork of Redmine. The copyright follows:
# Copyright (C) 2006-2013 Jean-Philippe Lang
# Copyright (C) 2010-2013 the ChiliProject Team
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
# 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)
return row[column] if row[column].blank?
value = YAML.load row[column]
if value.is_a? Array
value.map! do |e|
if e.is_a? Array
e.map! { |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