feat: make dance.openMenu retain count/register state

src/commands/index.ts

@@ -54,6 +54,10 @@ export class CommandDescriptor<Flags extends CommandDescriptor.Flags = CommandDe
     return (this.flags & CommandDescriptor.Flags.DoNotReplay) === 0;
   }
 
+  public get shouldResetContext() {
+    return (this.flags & CommandDescriptor.Flags.KeepContext) === 0;
+  }
+
   public constructor(
     /**
      * The unique identifier of the command.
@@ -104,8 +108,10 @@ export class CommandDescriptor<Flags extends CommandDescriptor.Flags = CommandDe
       context.doNotRecord();
     }
 
-    extension.currentCount = 0;
-    extension.currentRegister = undefined;
+    if (this.shouldResetContext) {
+      extension.currentCount = 0;
+      extension.currentRegister = undefined;
+    }
 
     let result: unknown;
 
@@ -168,6 +174,9 @@ export /* enum */ namespace CommandDescriptor {
 
     /** The command should not be replayed in macros and repeats. */
     DoNotReplay = 0b0010,
+
+    /** The command should not reset context (selected count/register). */
+    KeepContext = 0b0100,
   }
 }
 

src/commands/load-all.build.ts

@@ -226,6 +226,10 @@ function determineFunctionFlags(f: Builder.ParsedFunction) {
     flags.push("DoNotReplay");
   }
 
+  if ("keepcontext" in f.properties) {
+    flags.push("KeepContext");
+  }
+
   if (flags.length === 0) {
     return "CommandDescriptor.Flags.None";
   }

src/commands/misc.ts

@@ -318,6 +318,7 @@ const menuHistory: string[] = [];
  * like `jj`.
  *
  * @noreplay
+ * @keepcontext
  */
 export async function openMenu(
   _: Context.WithoutActiveEditor,

test/suite/utils.ts

@@ -396,6 +396,7 @@ export class ExpectedDocument {
             end = editor.document.lineAt(editor.document.lineCount - 1).rangeIncludingLineBreak.end;
 
       builder.replace(new vscode.Range(start, end), this.text);
+      builder.setEndOfLine(vscode.EndOfLine.LF);
     });
 
     if (this.selections.length > 0) {