Implement single char sneak

src/actions/plugins/sneak.ts

@@ -28,6 +28,12 @@ class SneakForward extends BaseMovement {
     const editor = vscode.window.activeTextEditor!;
     const document = editor.document;
     const lineCount = document.lineCount;
+
+    if (this.keysPressed[2] === '\n') {
+      // Single key sneak
+      this.keysPressed[2] = '';
+    }
+
     const searchString = this.keysPressed[1] + this.keysPressed[2];
 
     for (let i = position.line; i < lineCount; ++i) {
@@ -82,6 +88,12 @@ class SneakBackward extends BaseMovement {
 
     const editor = vscode.window.activeTextEditor!;
     const document = editor.document;
+
+    if (this.keysPressed[2] === '\n') {
+      // Single key sneak
+      this.keysPressed[2] = '';
+    }
+
     const searchString = this.keysPressed[1] + this.keysPressed[2];
 
     for (let i = position.line; i >= 0; --i) {

test/plugins/sneak.test.ts

@@ -68,4 +68,67 @@ suite('sneak plugin', () => {
     keysPressed: 'Sab,',
     end: ['abc abc |abc'],
   });
+
+  newTest({
+    title: 'Can handle single letter s motion',
+    start: ['|abc abc'],
+    keysPressed: 'sa\n',
+    end: ['abc |abc'],
+  });
+
+  newTest({
+    title: 'Can handle single letter S motion',
+    start: ['abc |abc'],
+    keysPressed: 'Sa\n',
+    end: ['|abc abc'],
+  });
+
+  newTest({
+    title: 'Can handle single letter <operator>z motion',
+    start: ['|abc abc'],
+    keysPressed: 'dza\n',
+    end: ['|abc'],
+  });
+
+  newTest({
+    title: 'Can handle single letter <operator>Z motion',
+    start: ['abc |abc'],
+    keysPressed: 'dZa\n',
+    end: ['|abc'],
+  });
+
+  newTest({
+    title: 'Can handle single letter s; motion',
+    start: ['|abc abc abc'],
+    keysPressed: 'sa\n;',
+    end: ['abc abc |abc'],
+  });
+
+  newTest({
+    title: 'Can handle single letter s, motion',
+    start: ['abc abc| abc'],
+    keysPressed: 'sa\n,',
+    end: ['abc |abc abc'],
+  });
+
+  newTest({
+    title: 'Can handle single letter S; motion',
+    start: ['abc abc |abc'],
+    keysPressed: 'Sa\n;',
+    end: ['|abc abc abc'],
+  });
+
+  newTest({
+    title: 'Can handle single letter S, motion',
+    start: ['abc abc| abc'],
+    keysPressed: 'Sa\n,',
+    end: ['abc abc |abc'],
+  });
+
+  newTest({
+    title: 'Can handle multiline single char <number>s motion',
+    start: ['|abc', 'aac', 'abc'],
+    keysPressed: '3sa\n',
+    end: ['abc', 'aac', '|abc'],
+  });
 });