Add latest changes from gitlab-org/gitlab@master

1 files changed, 37 insertions, 0 deletions

diff --git a/lib/gitlab/database_importers/work_items/hierarchy_restrictions_importer.rb b/lib/gitlab/database_importers/work_items/hierarchy_restrictions_importer.rb
new file mode 100644
index 00000000000..1181c259a5c
--- /dev/null
+++ b/lib/gitlab/database_importers/work_items/hierarchy_restrictions_importer.rb
@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module DatabaseImporters
+    module WorkItems
+      module HierarchyRestrictionsImporter
+        def self.upsert_restrictions
+          objective = find_or_create_type(::WorkItems::Type::TYPE_NAMES[:objective])
+          key_result = find_or_create_type(::WorkItems::Type::TYPE_NAMES[:key_result])
+          issue = find_or_create_type(::WorkItems::Type::TYPE_NAMES[:issue])
+          task = find_or_create_type(::WorkItems::Type::TYPE_NAMES[:task])
+          incident = find_or_create_type(::WorkItems::Type::TYPE_NAMES[:incident])
+
+          restrictions = [
+            { parent_type_id: objective.id, child_type_id: objective.id, maximum_depth: 9 },
+            { parent_type_id: objective.id, child_type_id: key_result.id, maximum_depth: 1 },
+            { parent_type_id: issue.id, child_type_id: task.id, maximum_depth: 1 },
+            { parent_type_id: incident.id, child_type_id: task.id, maximum_depth: 1 }
+          ]
+
+          ::WorkItems::HierarchyRestriction.upsert_all(
+            restrictions,
+            unique_by: :index_work_item_hierarchy_restrictions_on_parent_and_child
+          )
+        end
+
+        def self.find_or_create_type(name)
+          type = ::WorkItems::Type.find_by_name_and_namespace_id(name, nil)
+          return type if type
+
+          Gitlab::DatabaseImporters::WorkItems::BaseTypeImporter.upsert_types
+          ::WorkItems::Type.find_by_name_and_namespace_id(name, nil)
+        end
+      end
+    end
+  end
+end