From d9d3aa318541092928e4e8f03f518d8444ce5811 Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Wed, 6 Mar 2024 17:25:11 -0800
Subject: [PATCH] terminal2: selectWord done

---
 src/terminal/Screen.zig  |   2 +
 src/terminal2/Screen.zig | 159 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 161 insertions(+)

diff --git a/src/terminal/Screen.zig b/src/terminal/Screen.zig
index e0f5803e4..5638bff31 100644
--- a/src/terminal/Screen.zig
+++ b/src/terminal/Screen.zig
@@ -4777,6 +4777,7 @@ test "Screen: selectWord across soft-wrap" {
     }
 }
 
+// X
 test "Screen: selectWord whitespace across soft-wrap" {
     const testing = std.testing;
     const alloc = testing.allocator;
@@ -4813,6 +4814,7 @@ test "Screen: selectWord whitespace across soft-wrap" {
     }
 }
 
+// X
 test "Screen: selectWord with character boundary" {
     const testing = std.testing;
     const alloc = testing.allocator;
diff --git a/src/terminal2/Screen.zig b/src/terminal2/Screen.zig
index eb30a04d9..aa85c0bce 100644
--- a/src/terminal2/Screen.zig
+++ b/src/terminal2/Screen.zig
@@ -4211,3 +4211,162 @@ test "Screen: selectWord across soft-wrap" {
         } }, s.pages.pointFromPin(.screen, sel.end().*).?);
     }
 }
+
+test "Screen: selectWord whitespace across soft-wrap" {
+    const testing = std.testing;
+    const alloc = testing.allocator;
+
+    var s = try init(alloc, 5, 10, 0);
+    defer s.deinit();
+    try s.testWriteString("1       1\n 123");
+
+    // Going forward
+    {
+        var sel = s.selectWord(s.pages.pin(.{ .active = .{
+            .x = 1,
+            .y = 0,
+        } }).?).?;
+        defer sel.deinit(&s);
+        try testing.expectEqual(point.Point{ .screen = .{
+            .x = 1,
+            .y = 0,
+        } }, s.pages.pointFromPin(.screen, sel.start().*).?);
+        try testing.expectEqual(point.Point{ .screen = .{
+            .x = 2,
+            .y = 1,
+        } }, s.pages.pointFromPin(.screen, sel.end().*).?);
+    }
+
+    // Going backward
+    {
+        var sel = s.selectWord(s.pages.pin(.{ .active = .{
+            .x = 1,
+            .y = 1,
+        } }).?).?;
+        defer sel.deinit(&s);
+        try testing.expectEqual(point.Point{ .screen = .{
+            .x = 1,
+            .y = 0,
+        } }, s.pages.pointFromPin(.screen, sel.start().*).?);
+        try testing.expectEqual(point.Point{ .screen = .{
+            .x = 2,
+            .y = 1,
+        } }, s.pages.pointFromPin(.screen, sel.end().*).?);
+    }
+
+    // Going forward and backward
+    {
+        var sel = s.selectWord(s.pages.pin(.{ .active = .{
+            .x = 3,
+            .y = 0,
+        } }).?).?;
+        defer sel.deinit(&s);
+        try testing.expectEqual(point.Point{ .screen = .{
+            .x = 1,
+            .y = 0,
+        } }, s.pages.pointFromPin(.screen, sel.start().*).?);
+        try testing.expectEqual(point.Point{ .screen = .{
+            .x = 2,
+            .y = 1,
+        } }, s.pages.pointFromPin(.screen, sel.end().*).?);
+    }
+}
+
+test "Screen: selectWord with character boundary" {
+    const testing = std.testing;
+    const alloc = testing.allocator;
+
+    const cases = [_][]const u8{
+        " 'abc' \n123",
+        " \"abc\" \n123",
+        " │abc│ \n123",
+        " `abc` \n123",
+        " |abc| \n123",
+        " :abc: \n123",
+        " ,abc, \n123",
+        " (abc( \n123",
+        " )abc) \n123",
+        " [abc[ \n123",
+        " ]abc] \n123",
+        " {abc{ \n123",
+        " }abc} \n123",
+        " <abc< \n123",
+        " >abc> \n123",
+    };
+
+    for (cases) |case| {
+        var s = try init(alloc, 20, 10, 0);
+        defer s.deinit();
+        try s.testWriteString(case);
+
+        // Inside character forward
+        {
+            var sel = s.selectWord(s.pages.pin(.{ .active = .{
+                .x = 2,
+                .y = 0,
+            } }).?).?;
+            defer sel.deinit(&s);
+            try testing.expectEqual(point.Point{ .screen = .{
+                .x = 2,
+                .y = 0,
+            } }, s.pages.pointFromPin(.screen, sel.start().*).?);
+            try testing.expectEqual(point.Point{ .screen = .{
+                .x = 4,
+                .y = 0,
+            } }, s.pages.pointFromPin(.screen, sel.end().*).?);
+        }
+
+        // Inside character backward
+        {
+            var sel = s.selectWord(s.pages.pin(.{ .active = .{
+                .x = 4,
+                .y = 0,
+            } }).?).?;
+            defer sel.deinit(&s);
+            try testing.expectEqual(point.Point{ .screen = .{
+                .x = 2,
+                .y = 0,
+            } }, s.pages.pointFromPin(.screen, sel.start().*).?);
+            try testing.expectEqual(point.Point{ .screen = .{
+                .x = 4,
+                .y = 0,
+            } }, s.pages.pointFromPin(.screen, sel.end().*).?);
+        }
+
+        // Inside character bidirectional
+        {
+            var sel = s.selectWord(s.pages.pin(.{ .active = .{
+                .x = 3,
+                .y = 0,
+            } }).?).?;
+            defer sel.deinit(&s);
+            try testing.expectEqual(point.Point{ .screen = .{
+                .x = 2,
+                .y = 0,
+            } }, s.pages.pointFromPin(.screen, sel.start().*).?);
+            try testing.expectEqual(point.Point{ .screen = .{
+                .x = 4,
+                .y = 0,
+            } }, s.pages.pointFromPin(.screen, sel.end().*).?);
+        }
+
+        // On quote
+        // NOTE: this behavior is not ideal, so we can change this one day,
+        // but I think its also not that important compared to the above.
+        {
+            var sel = s.selectWord(s.pages.pin(.{ .active = .{
+                .x = 1,
+                .y = 0,
+            } }).?).?;
+            defer sel.deinit(&s);
+            try testing.expectEqual(point.Point{ .screen = .{
+                .x = 0,
+                .y = 0,
+            } }, s.pages.pointFromPin(.screen, sel.start().*).?);
+            try testing.expectEqual(point.Point{ .screen = .{
+                .x = 1,
+                .y = 0,
+            } }, s.pages.pointFromPin(.screen, sel.end().*).?);
+        }
+    }
+}