Fix cross-split wrap with snippet (#1271)

Author: pokey

src/apps/cursorless-vscode-e2e/suite/wrapWithSnippetAcrossSplit.test.ts

@@ -0,0 +1,48 @@
+import { HatStability } from "@cursorless/common";
+import { getCursorlessApi, openNewEditor } from "@cursorless/vscode-common";
+import * as assert from "assert";
+import { endToEndTestSetup } from "../endToEndTestSetup";
+import { runCursorlessCommand } from "../runCommand";
+import { setupFake } from "./setupFake";
+
+suite("Wrap with snippet across split", async function () {
+  endToEndTestSetup(this);
+
+  suiteSetup(async () => {
+    const { ide } = (await getCursorlessApi()).testHelpers!;
+    setupFake(ide, HatStability.stable);
+  });
+
+  test("Wrap with snippet across split", runTest);
+});
+
+async function runTest() {
+  const { graph } = (await getCursorlessApi()).testHelpers!;
+
+  const { document: document1 } = await openNewEditor("hello world");
+  const { document: document2 } = await openNewEditor("", { openBeside: true });
+
+  await graph.hatTokenMap.allocateHats();
+
+  await runCursorlessCommand({
+    version: 4,
+    action: { name: "wrapWithSnippet", args: ["spaghetti.foo"] },
+    targets: [
+      {
+        type: "primitive",
+        mark: {
+          type: "decoratedSymbol",
+          symbolColor: "default",
+          character: "h",
+        },
+      },
+    ],
+    usePrePhraseSnapshot: false,
+  });
+
+  assert.deepStrictEqual(
+    document1.getText(),
+    "My friend hello likes to eat spaghetti! world",
+  );
+  assert.deepStrictEqual(document2.getText(), "");
+}

src/ide/vscode/VscodeInsertSnippets.ts

@@ -11,6 +11,7 @@ export async function vscodeInsertSnippet(
     editor.selections = ranges.map((range) => range.toSelection(false));
   }
 
+  await editor.focus();
   await vscode.commands.executeCommand("editor.action.insertSnippet", {
     snippet,
   });