fix early wrap for multi-line input elements

modules/Element.lua

@@ -4042,11 +4042,27 @@ function Element:moveCursorToPreviousWord()
     return
   end
 
+  -- Helper function to get character at position
+  local function getCharAt(p)
+    if p < 0 or p >= utf8.len(text) then
+      return nil
+    end
+    local offset1 = utf8.offset(text, p + 1)
+    local offset2 = utf8.offset(text, p + 2)
+    if not offset1 then
+      return nil
+    end
+    if not offset2 then
+      -- Last character in string
+      return text:sub(offset1)
+    end
+    return text:sub(offset1, offset2 - 1)
+  end
+
   -- Skip any whitespace/punctuation before current position
   while pos > 0 do
-    local offset = utf8.offset(text, pos + 1)
-    local char = offset and text:sub(offset, utf8.offset(text, pos + 2) - 1) or ""
-    if char:match("[%w]") then
+    local char = getCharAt(pos - 1)
+    if char and char:match("[%w]") then
       break
     end
     pos = pos - 1
@@ -4054,11 +4070,8 @@ function Element:moveCursorToPreviousWord()
 
   -- Move to start of current word
   while pos > 0 do
-    local offset = utf8.offset(text, pos + 1)
-    local char = offset and text:sub(offset, utf8.offset(text, pos + 2) - 1) or ""
-    if not char:match("[%w]") then
-      -- We've moved one position past the start, so move back
-      pos = pos + 1
+    local char = getCharAt(pos - 1)
+    if not char or not char:match("[%w]") then
       break
     end
     pos = pos - 1
@@ -4082,11 +4095,27 @@ function Element:moveCursorToNextWord()
     return
   end
 
+  -- Helper function to get character at position
+  local function getCharAt(p)
+    if p < 0 or p >= textLength then
+      return nil
+    end
+    local offset1 = utf8.offset(text, p + 1)
+    local offset2 = utf8.offset(text, p + 2)
+    if not offset1 then
+      return nil
+    end
+    if not offset2 then
+      -- Last character in string
+      return text:sub(offset1)
+    end
+    return text:sub(offset1, offset2 - 1)
+  end
+
   -- Skip current word
   while pos < textLength do
-    local offset = utf8.offset(text, pos + 1)
-    local char = offset and text:sub(offset, utf8.offset(text, pos + 2) - 1) or ""
-    if not char:match("[%w]") then
+    local char = getCharAt(pos)
+    if not char or not char:match("[%w]") then
       break
     end
     pos = pos + 1
@@ -4094,9 +4123,8 @@ function Element:moveCursorToNextWord()
 
   -- Skip any whitespace/punctuation
   while pos < textLength do
-    local offset = utf8.offset(text, pos + 1)
-    local char = offset and text:sub(offset, utf8.offset(text, pos + 2) - 1) or ""
-    if char:match("[%w]") then
+    local char = getCharAt(pos)
+    if char and char:match("[%w]") then
       break
     end
     pos = pos + 1
@@ -4597,33 +4625,68 @@ function Element:_wrapLine(line, maxWidth)
   end
 
   if self.textWrap == "word" then
-    local words = {}
-    for word in line:gmatch("%S+") do
-      table.insert(words, word)
+    -- Tokenize into words and whitespace, preserving exact spacing
+    local tokens = {}
+    local pos = 1
+    local lineLen = utf8.len(line)
+    
+    while pos <= lineLen do
+      -- Check if current position is whitespace
+      local char = getUtf8Char(line, pos)
+      if char:match("%s") then
+        -- Collect whitespace sequence
+        local wsStart = pos
+        while pos <= lineLen and getUtf8Char(line, pos):match("%s") do
+          pos = pos + 1
+        end
+        table.insert(tokens, {
+          type = "space",
+          text = line:sub(utf8.offset(line, wsStart), utf8.offset(line, pos) and utf8.offset(line, pos) - 1 or #line),
+          startPos = wsStart - 1,
+          length = pos - wsStart
+        })
+      else
+        -- Collect word (non-whitespace sequence)
+        local wordStart = pos
+        while pos <= lineLen and not getUtf8Char(line, pos):match("%s") do
+          pos = pos + 1
+        end
+        table.insert(tokens, {
+          type = "word",
+          text = line:sub(utf8.offset(line, wordStart), utf8.offset(line, pos) and utf8.offset(line, pos) - 1 or #line),
+          startPos = wordStart - 1,
+          length = pos - wordStart
+        })
+      end
     end
 
-    for i, word in ipairs(words) do
-      local testLine = currentLine == "" and word or (currentLine .. " " .. word)
+    -- Process tokens and wrap
+    local charPos = 0  -- Track our position in the original line
+    for i, token in ipairs(tokens) do
+      if token.type == "word" then
+        local testLine = currentLine .. token.text
         local width = font:getWidth(testLine)
 
         if width > maxWidth and currentLine ~= "" then
+          -- Current line is full, wrap before this word
           local currentLineLen = utf8.len(currentLine)
           table.insert(wrappedParts, {
             text = currentLine,
             startIdx = startIdx,
             endIdx = startIdx + currentLineLen,
           })
-        startIdx = startIdx + currentLineLen + 1 -- +1 for the space
-        currentLine = word
+          startIdx = charPos
+          currentLine = token.text
+          charPos = charPos + token.length
 
           -- Check if the word itself is too long - if so, break it with character wrapping
-        if font:getWidth(word) > maxWidth then
-          local wordLen = utf8.len(word)
+          if font:getWidth(token.text) > maxWidth then
+            local wordLen = utf8.len(token.text)
             local charLine = ""
             local charStartIdx = startIdx
 
             for j = 1, wordLen do
-            local char = getUtf8Char(word, j)
+              local char = getUtf8Char(token.text, j)
               local testCharLine = charLine .. char
               local charWidth = font:getWidth(testCharLine)
 
@@ -4645,12 +4708,12 @@ function Element:_wrapLine(line, maxWidth)
           end
         elseif width > maxWidth and currentLine == "" then
           -- Word is too long to fit on a line by itself - use character wrapping
-        local wordLen = utf8.len(word)
+          local wordLen = utf8.len(token.text)
           local charLine = ""
           local charStartIdx = startIdx
 
           for j = 1, wordLen do
-          local char = getUtf8Char(word, j)
+            local char = getUtf8Char(token.text, j)
             local testCharLine = charLine .. char
             local charWidth = font:getWidth(testCharLine)
 
@@ -4669,8 +4732,15 @@ function Element:_wrapLine(line, maxWidth)
 
           currentLine = charLine
           startIdx = charStartIdx
+          charPos = charPos + token.length
         else
           currentLine = testLine
+          charPos = charPos + token.length
+        end
+      else
+        -- It's whitespace - add to current line
+        currentLine = currentLine .. token.text
+        charPos = charPos + token.length
       end
     end
   else