1 files changed, 39 insertions, 0 deletions
diff --git a/app/services/work_items/widgets/milestone_service/base_service.rb b/app/services/work_items/widgets/milestone_service/base_service.rb
new file mode 100644
index 00000000000..f373e6daea3
--- /dev/null
+++ b/app/services/work_items/widgets/milestone_service/base_service.rb
@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+module WorkItems
+  module Widgets
+    module MilestoneService
+      class BaseService < WorkItems::Widgets::BaseService
+        private
+
+        def handle_milestone_change(params:)
+          return unless params.present? && params.key?(:milestone_id)
+
+          unless has_permission?(:set_work_item_metadata)
+            params.delete(:milestone_id)
+            return
+          end
+
+          if params[:milestone_id].nil?
+            work_item.milestone = nil
+
+            return
+          end
+
+          project = work_item.project
+          milestone = MilestonesFinder.new({
+                                             project_ids: [project.id],
+                                             group_ids: project.group&.self_and_ancestors&.select(:id),
+                                             ids: [params[:milestone_id]]
+                                           }).execute.first
+
+          if milestone
+            work_item.milestone = milestone
+          else
+            params.delete(:milestone_id)
+          end
+        end
+      end
+    end
+  end
+end