1 files changed, 37 insertions, 0 deletions
diff --git a/app/services/work_items/widgets/current_user_todos_service/update_service.rb b/app/services/work_items/widgets/current_user_todos_service/update_service.rb
new file mode 100644
index 00000000000..38e2ae4de32
--- /dev/null
+++ b/app/services/work_items/widgets/current_user_todos_service/update_service.rb
@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module WorkItems
+  module Widgets
+    module CurrentUserTodosService
+      class UpdateService < WorkItems::Widgets::BaseService
+        def before_update_in_transaction(params:)
+          return unless params.present? && params.key?(:action)
+
+          case params[:action]
+          when "add"
+            add_todo
+          when "mark_as_done"
+            mark_as_done(params[:todo_id])
+          end
+        end
+
+        private
+
+        def add_todo
+          return unless has_permission?(:create_todo)
+
+          TodoService.new.mark_todo(work_item, current_user)&.first
+        end
+
+        def mark_as_done(todo_id)
+          todos = TodosFinder.new(current_user, state: :pending, target_id: work_item.id).execute
+          todos = todo_id ? todos.id_in(todo_id) : todos
+
+          return if todos.empty?
+
+          TodoService.new.resolve_todos(todos, current_user, resolved_by_action: :api_done)
+        end
+      end
+    end
+  end
+end