Add redo for several actions

Shared/Sources/Document/Accessibility/DocumentPresenter.swift

@@ -64,11 +64,9 @@
                           frame: CGRect(origin: origin,
                                         size: image.size))
 
-        add(frame, into: nil, at: document.artboard.frames.count)
-
-        document.undo?.registerUndo(withTarget: self, handler: { target in
-            target.remove(frame)
-        })
+        add(frame,
+            into: nil,
+            at: document.artboard.frames.count)
     }
 
     public func append(control: any ArtboardElement) {
@@ -124,27 +122,33 @@
 
         publishControlChanges()
         select(model)
+
+        document.undo?.registerUndo(
+            withTarget: self,
+            handler: { presenter in
+                presenter.remove(model)
+            })
     }
 
     @discardableResult
     public func wrapInContainer(
         _ elements: [any ArtboardElement]
     ) -> A11yContainer? {
         controls.wrap(
             in: A11yContainer.self,
             elements.extractElements(),
             label: "Container")
     }
 
     public func unwrapContainer(_ container: A11yContainer) {
         controls.unwrapContainer(container)
     }
 }
 
 #if canImport(XCTest)
 extension DocumentPresenter {
     public func update(elements: [A11yDescription]) {
         self.controls = elements
     }
 
     public var firstFrameControls: [any ArtboardElement] {

Shared/Tests/DocumentTests/Models/DocumentPresenterTests.swift

@@ -32,8 +32,11 @@ final class DocumentPresenterTests: XCTestCase {
         sut.append(control: element1)
         XCTAssertEqual(sut.controlsWithoutFrame.count, 1)
 
-        sut.document.undo?.undo()
+        sut.undo()
         XCTAssertTrue(sut.controlsWithoutFrame.isEmpty)
+
+        sut.redo()
+        XCTAssertFalse(sut.controlsWithoutFrame.isEmpty)
     }
 
     // MARK: - Container
@@ -76,6 +79,9 @@ final class DocumentPresenterTests: XCTestCase {
 
         sut.undo()
         XCTAssertEqual(container.elements.count, 2)
+
+        sut.redo()
+        XCTAssertEqual(container.elements.count, 1)
     }
 
     func test_container_whenRemoveContainer_shouldRemoveEverything() throws {
@@ -86,6 +92,9 @@ final class DocumentPresenterTests: XCTestCase {
 
         sut.undo()
         XCTAssertEqual(sut.controlsWithoutFrame.count, 1)
+
+        sut.redo()
+        XCTAssertTrue(sut.controlsWithoutFrame.isEmpty)
     }
 
     // MARK: Artboard
@@ -112,6 +121,9 @@ final class DocumentPresenterTests: XCTestCase {
         sut.undo()
 
         XCTAssertFalse(document.artboard.frames.isEmpty)
+
+        sut.redo()
+        XCTAssertTrue(document.artboard.frames.isEmpty)
     }
 
     // MARK: DSL
@@ -133,6 +145,7 @@ final class DocumentPresenterTests: XCTestCase {
     }
 }
 
+// MARK: - Undo
 extension DocumentPresenter {
     func disableUndoRegistration() {
         document.undo?.disableUndoRegistration()
@@ -144,4 +157,8 @@ extension DocumentPresenter {
     func undo() {
         document.undo?.undo()
     }
+
+    func redo() {
+        document.undo?.redo()
+    }
 }