1 files changed, 23 insertions, 0 deletions
diff --git a/app/assets/javascripts/work_items/components/update_work_item.js b/app/assets/javascripts/work_items/components/update_work_item.js
new file mode 100644
index 00000000000..fc395fa5be3
--- /dev/null
+++ b/app/assets/javascripts/work_items/components/update_work_item.js
@@ -0,0 +1,23 @@
+import updateWorkItemMutation from '../graphql/update_work_item.mutation.graphql';
+import updateWorkItemTaskMutation from '../graphql/update_work_item_task.mutation.graphql';
+
+export function getUpdateWorkItemMutation({ input, workItemParentId }) {
+  let mutation = updateWorkItemMutation;
+
+  const variables = {
+    input,
+  };
+
+  if (workItemParentId) {
+    mutation = updateWorkItemTaskMutation;
+    variables.input = {
+      id: workItemParentId,
+      taskData: input,
+    };
+  }
+
+  return {
+    mutation,
+    variables,
+  };
+}