101 lines
2.3 KiB
Ruby
101 lines
2.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module BulkImports
|
|
module FileTransfer
|
|
class BaseConfig
|
|
include Gitlab::Utils::StrongMemoize
|
|
|
|
UPLOADS_RELATION = 'uploads'
|
|
SELF_RELATION = 'self'
|
|
|
|
def initialize(portable)
|
|
@portable = portable
|
|
end
|
|
|
|
def portable_tree
|
|
attributes_finder.find_root(portable_class_sym)
|
|
end
|
|
|
|
def top_relation_tree(relation)
|
|
portable_relations_tree[relation.to_s]
|
|
end
|
|
|
|
def relation_excluded_keys(relation)
|
|
attributes_finder.find_excluded_keys(relation)
|
|
end
|
|
|
|
def export_path
|
|
@export_path ||= Dir.mktmpdir('bulk_imports')
|
|
end
|
|
|
|
def portable_relations
|
|
tree_relations + file_relations + self_relation - skipped_relations
|
|
end
|
|
|
|
def self_relation?(relation)
|
|
relation == SELF_RELATION
|
|
end
|
|
|
|
def tree_relation?(relation)
|
|
tree_relations.include?(relation)
|
|
end
|
|
|
|
def file_relation?(relation)
|
|
file_relations.include?(relation)
|
|
end
|
|
|
|
def tree_relation_definition_for(relation)
|
|
return unless tree_relation?(relation)
|
|
|
|
portable_tree[:include].find { |include| include[relation.to_sym] }
|
|
end
|
|
|
|
def portable_relations_tree
|
|
@portable_relations_tree ||= attributes_finder.find_relations_tree(portable_class_sym).deep_stringify_keys
|
|
end
|
|
|
|
private
|
|
|
|
attr_reader :portable
|
|
|
|
def attributes_finder
|
|
strong_memoize(:attributes_finder) do
|
|
::Gitlab::ImportExport::AttributesFinder.new(config: import_export_config)
|
|
end
|
|
end
|
|
|
|
def import_export_config
|
|
@config ||= ::Gitlab::ImportExport::Config.new(config: import_export_yaml).to_h
|
|
end
|
|
|
|
def portable_class
|
|
@portable_class ||= portable.class
|
|
end
|
|
|
|
def portable_class_sym
|
|
@portable_class_sym ||= portable_class.to_s.demodulize.underscore.to_sym
|
|
end
|
|
|
|
def import_export_yaml
|
|
raise NotImplementedError
|
|
end
|
|
|
|
def tree_relations
|
|
import_export_config.dig(:tree, portable_class_sym).keys.map(&:to_s)
|
|
end
|
|
|
|
def file_relations
|
|
[UPLOADS_RELATION]
|
|
end
|
|
|
|
def skipped_relations
|
|
[]
|
|
end
|
|
|
|
def self_relation
|
|
[SELF_RELATION]
|
|
end
|
|
end
|
|
end
|
|
end
|