56 lines
1.4 KiB
Ruby
56 lines
1.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module ImportExport
|
|
class AttributeCleaner
|
|
ALLOWED_REFERENCES = [
|
|
*ProjectRelationFactory::PROJECT_REFERENCES,
|
|
*ProjectRelationFactory::USER_REFERENCES,
|
|
'group_id',
|
|
'commit_id',
|
|
'discussion_id',
|
|
'custom_attributes'
|
|
].freeze
|
|
PROHIBITED_REFERENCES = Regexp.union(
|
|
/\Acached_markdown_version\Z/,
|
|
/_id\Z/,
|
|
/_ids\Z/,
|
|
/_html\Z/,
|
|
/attributes/,
|
|
/\Aremote_\w+_(url|urls|request_header)\Z/ # carrierwave automatically creates these attribute methods for uploads
|
|
).freeze
|
|
|
|
def self.clean(*args)
|
|
new(*args).clean
|
|
end
|
|
|
|
def initialize(relation_hash:, relation_class:, excluded_keys: [])
|
|
@relation_hash = relation_hash
|
|
@relation_class = relation_class
|
|
@excluded_keys = excluded_keys
|
|
end
|
|
|
|
def clean
|
|
@relation_hash.reject do |key, _value|
|
|
prohibited_key?(key) || !@relation_class.attribute_method?(key) || excluded_key?(key)
|
|
end.except('id')
|
|
end
|
|
|
|
private
|
|
|
|
def prohibited_key?(key)
|
|
key =~ PROHIBITED_REFERENCES && !permitted_key?(key)
|
|
end
|
|
|
|
def permitted_key?(key)
|
|
ALLOWED_REFERENCES.include?(key)
|
|
end
|
|
|
|
def excluded_key?(key)
|
|
return false if @excluded_keys.empty?
|
|
|
|
@excluded_keys.include?(key)
|
|
end
|
|
end
|
|
end
|
|
end
|