BigW Consortium Gitlab

attribute_cleaner.rb 699 Bytes
Newer Older
1 2 3
module Gitlab
  module ImportExport
    class AttributeCleaner
4
      ALLOWED_REFERENCES = RelationFactory::PROJECT_REFERENCES + RelationFactory::USER_REFERENCES + ['group_id']
5

6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
      def self.clean(*args)
        new(*args).clean
      end

      def initialize(relation_hash:, relation_class:)
        @relation_hash = relation_hash
        @relation_class = relation_class
      end

      def clean
        @relation_hash.reject do |key, _value|
          prohibited_key?(key) || !@relation_class.attribute_method?(key)
        end.except('id')
      end

      private

      def prohibited_key?(key)
        key.end_with?('_id') && !ALLOWED_REFERENCES.include?(key)
25 26 27 28
      end
    end
  end
end