1 files changed, 42 insertions, 0 deletions
diff --git a/lib/bulk_imports/groups/pipelines/milestones_pipeline.rb b/lib/bulk_imports/groups/pipelines/milestones_pipeline.rb
new file mode 100644
index 00000000000..8497162e0e7
--- /dev/null
+++ b/lib/bulk_imports/groups/pipelines/milestones_pipeline.rb
@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module BulkImports
+  module Groups
+    module Pipelines
+      class MilestonesPipeline
+        include Pipeline
+
+        extractor BulkImports::Common::Extractors::GraphqlExtractor,
+          query: BulkImports::Groups::Graphql::GetMilestonesQuery
+
+        transformer Common::Transformers::ProhibitedAttributesTransformer
+
+        def load(context, data)
+          return unless data
+
+          raise ::BulkImports::Pipeline::NotAllowedError unless authorized?
+
+          context.group.milestones.create!(data)
+        end
+
+        def after_run(extracted_data)
+          context.entity.update_tracker_for(
+            relation: :milestones,
+            has_next_page: extracted_data.has_next_page?,
+            next_page: extracted_data.next_page
+          )
+
+          if extracted_data.has_next_page?
+            run
+          end
+        end
+
+        private
+
+        def authorized?
+          context.current_user.can?(:admin_milestone, context.group)
+        end
+      end
+    end
+  end
+end