Add initial sync complete callback to ServeSession for one-shot mode handling

- Introduced `__initialSyncCompleteCallback` in `ServeSession` to manage actions after the initial sync is complete.
- Updated `App:startSession` to set the callback, ensuring proper disconnection after initial sync in one-shot mode.
- Refactored comments in `App:startSession` for clarity on sync behavior and disconnection logic.

plugin/src/App/init.lua

@@ -800,23 +800,10 @@ function App:startSession()
 
 		local result = self.confirmationEvent:Wait()
 
-		-- Track if this was the initial sync for this project
-		local isInitialSyncForProject = not self.knownProjects[serverInfo.projectName]
-
-		-- One-shot sync: disconnect immediately after confirmation
-		-- Skip the "Connected" UI state entirely to avoid flash
+		-- One-shot sync: Don't transition to Connected UI
+		-- The actual disconnect is handled by setInitialSyncCompleteCallback
+		-- to ensure any pending writes complete first
 		if Settings:get("oneShotSync") then
-			local resultType = if type(result) == "table" then result.type else result
-			if resultType == "Confirm" or resultType == "Accept" then
-				-- Use task.defer to disconnect after patch is applied (runs after current thread yields)
-				task.defer(function()
-					if self.serveSession then
-						Log.info("One-shot sync: disconnecting after sync")
-						self:endSession()
-					end
-				end)
-			end
-			-- Don't transition to Connected UI, just return result and let disconnect happen
 			return result
 		end
 
@@ -858,6 +845,15 @@ function App:startSession()
 		})
 	end)
 
+	-- One-shot sync: disconnect after initial sync is fully complete (including any writes)
+	-- This ensures that "pull" changes are sent to the server before disconnecting
+	serveSession:setInitialSyncCompleteCallback(function()
+		if Settings:get("oneShotSync") and self.serveSession then
+			Log.info("One-shot sync: disconnecting after initial sync complete")
+			self:endSession()
+		end
+	end)
+
 	serveSession:start()
 
 	self.serveSession = serveSession