2018-11-16 19:37:17 -05:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2016-08-12 06:04:33 -04:00
|
|
|
module Gitlab
|
|
|
|
module ImportExport
|
|
|
|
class AttributeCleaner
|
2016-10-18 13:31:10 -04:00
|
|
|
ALLOWED_REFERENCES = RelationFactory::PROJECT_REFERENCES + RelationFactory::USER_REFERENCES + ['group_id']
|
2016-08-12 06:04:33 -04:00
|
|
|
|
2016-10-27 06:22:18 -04:00
|
|
|
def self.clean(*args)
|
|
|
|
new(*args).clean
|
|
|
|
end
|
|
|
|
|
2018-05-03 05:02:26 -04:00
|
|
|
def initialize(relation_hash:, relation_class:, excluded_keys: [])
|
2016-10-27 06:22:18 -04:00
|
|
|
@relation_hash = relation_hash
|
|
|
|
@relation_class = relation_class
|
2018-05-03 05:02:26 -04:00
|
|
|
@excluded_keys = excluded_keys
|
2016-10-27 06:22:18 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def clean
|
|
|
|
@relation_hash.reject do |key, _value|
|
2018-05-03 05:02:26 -04:00
|
|
|
prohibited_key?(key) || !@relation_class.attribute_method?(key) || excluded_key?(key)
|
2016-10-27 06:22:18 -04:00
|
|
|
end.except('id')
|
|
|
|
end
|
|
|
|
|
|
|
|
private
|
|
|
|
|
|
|
|
def prohibited_key?(key)
|
2019-04-23 22:31:20 -04:00
|
|
|
return false if allowed_reference?(key)
|
|
|
|
|
|
|
|
return true if 'cached_markdown_version'.equal?(key)
|
|
|
|
|
2019-04-24 17:17:53 -04:00
|
|
|
prohibited_suffixes = %w(_id _html)
|
2019-04-25 17:40:00 -04:00
|
|
|
prohibited_suffixes.any? do |suffix|
|
|
|
|
true if key.end_with?(suffix)
|
2019-04-23 22:31:20 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def allowed_reference?(key)
|
|
|
|
ALLOWED_REFERENCES.include?(key)
|
2016-08-12 06:04:33 -04:00
|
|
|
end
|
2018-05-03 05:02:26 -04:00
|
|
|
|
|
|
|
def excluded_key?(key)
|
|
|
|
return false if @excluded_keys.empty?
|
|
|
|
|
|
|
|
@excluded_keys.include?(key)
|
|
|
|
end
|
2016-08-12 06:04:33 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|