fix text selection in immediate mode

modules/Element.lua

@@ -416,7 +416,15 @@ function Element.new(props)
       onTextChange = props.onTextChange,
       onEnter = props.onEnter,
     }, textEditorDeps)
-    -- Initialize will be called after self is fully constructed
+
+    -- Restore TextEditor state from StateManager in immediate mode
+    if Element._Context._immediateMode and self._stateId and self._stateId ~= "" then
+      local state = Element._StateManager.getState(self._stateId)
+      if state and state.textEditor then
+        -- Restore from nested textEditor state (saved via saveState())
+        self._textEditor:setState(state.textEditor, self)
+      end
+    end
   end
 
   -- Set parent first so it's available for size calculations
@@ -3305,6 +3313,14 @@ function Element:saveState()
     end
   end
 
+  -- Save drag tracking state for text selection
+  if self._mouseDownPosition ~= nil then
+    state._mouseDownPosition = self._mouseDownPosition
+  end
+  if self._textDragOccurred ~= nil then
+    state._textDragOccurred = self._textDragOccurred
+  end
+
   return state
 end
 
@@ -3337,6 +3353,14 @@ function Element:restoreState(state)
     self._scrollManager:setState(state.scrollManager)
   end
 
+  -- Restore drag tracking state for text selection
+  if state._mouseDownPosition ~= nil then
+    self._mouseDownPosition = state._mouseDownPosition
+  end
+  if state._textDragOccurred ~= nil then
+    self._textDragOccurred = state._textDragOccurred
+  end
+
   -- Note: Blur cache data is used for invalidation, not restoration
 end
 

modules/Renderer.lua

@@ -801,7 +801,10 @@ function Renderer:drawText(element)
     end
 
     -- Draw selection highlight for editable elements
-    if element._textEditor and element._textEditor:isFocused() and element._textEditor:hasSelection() and element.text and element.text ~= "" then
+    if element._textEditor and element._textEditor:isFocused() and element._textEditor:hasSelection() then
+      -- For editable elements, check TextEditor buffer instead of element.text
+      local textBuffer = element._textEditor:getText()
+      if textBuffer and textBuffer ~= "" then
         local selStart, selEnd = element._textEditor:getSelection()
         local selectionColor = element.selectionColor or self._Color.new(0.3, 0.5, 0.8, 0.5)
         local selectionWithOpacity = self._Color.new(selectionColor.r, selectionColor.g, selectionColor.b, selectionColor.a * self.opacity)
@@ -830,6 +833,7 @@ function Renderer:drawText(element)
           love.graphics.setScissor()
         end
       end
+    end
 
     if element.textSize then
       love.graphics.setFont(origFont)

modules/TextEditor.lua

@@ -1500,14 +1500,14 @@ end
 ---@param mouseX number
 ---@param mouseY number
 function TextEditor:handleTextDrag(element, mouseX, mouseY)
-  if not self._focused or not self._mouseDownPosition then
+  if not self._focused or not element._mouseDownPosition then
     return
   end
 
   local currentPos = self:mouseToTextPosition(element, mouseX, mouseY)
 
-  if currentPos ~= self._mouseDownPosition then
-    self:setSelection(element, self._mouseDownPosition, currentPos)
+  if currentPos ~= element._mouseDownPosition then
+    self:setSelection(element, element._mouseDownPosition, currentPos)
     self._cursorPosition = currentPos
     self._textDragOccurred = true
   else