1 files changed, 35 insertions, 0 deletions
diff --git a/app/graphql/mutations/merge_requests/set_draft.rb b/app/graphql/mutations/merge_requests/set_draft.rb
new file mode 100644
index 00000000000..80006c6f70e
--- /dev/null
+++ b/app/graphql/mutations/merge_requests/set_draft.rb
@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Mutations
+  module MergeRequests
+    class SetDraft < Base
+      graphql_name 'MergeRequestSetDraft'
+
+      argument :draft,
+               GraphQL::BOOLEAN_TYPE,
+               required: true,
+               description: <<~DESC
+                 Whether or not to set the merge request as a draft.
+               DESC
+
+      def resolve(project_path:, iid:, draft: nil)
+        merge_request = authorized_find!(project_path: project_path, iid: iid)
+        project = merge_request.project
+
+        ::MergeRequests::UpdateService.new(project: project, current_user: current_user, params: { wip_event: wip_event(draft) })
+          .execute(merge_request)
+
+        {
+          merge_request: merge_request,
+          errors: errors_on_object(merge_request)
+        }
+      end
+
+      private
+
+      def wip_event(wip)
+        wip ? 'wip' : 'unwip'
+      end
+    end
+  end
+end