feat: refactored trim command

xiangechen · Jul 11, 2024 · d929012 · d929012
1 parent 95fa322
commit d929012
Showing 1 changed file with 21 additions and 17 deletions.
diff --git a/packages/chili/src/commands/modify/trim.ts b/packages/chili/src/commands/modify/trim.ts
@@ -35,30 +35,34 @@ export class Trim extends CancelableCommand {
         );
         transaction.start();
         try {
-            while (!this.isCompleted) {
-                this.controller = new AsyncController();
-                let handler = new PickTrimEdgeEventHandler(this.document, this.controller);
-                await this.document.selection.pickAsync(
-                    handler,
-                    "prompt.select.edges",
-                    this.controller,
-                    false,
-                    "select.default",
-                );
-                if (this.controller.result?.status !== "success") {
-                    break;
-                }
-                if (handler.selected) {
-                    this.trimEdge(handler.selected);
-                }
-            }
+            await this.trimAsync();
         } catch (e) {
             transaction.rollback();
             throw e;
         }
         transaction.commit();
     }
 
+    private async trimAsync() {
+        while (!this.isCompleted) {
+            this.controller = new AsyncController();
+            let handler = new PickTrimEdgeEventHandler(this.document, this.controller);
+            await this.document.selection.pickAsync(
+                handler,
+                "prompt.select.edges",
+                this.controller,
+                false,
+                "select.default",
+            );
+            if (this.controller.result?.status !== "success") {
+                break;
+            }
+            if (handler.selected) {
+                this.trimEdge(handler.selected);
+            }
+        }
+    }
+
     private trimEdge(selected: TrimEdge) {
         let model = this.document.visual.context.getModel(selected.edge.owner);
         selected.segments.retainSegments.forEach((segment) => {