fix: prevent race condition in code generation state

AshishKumar4 · AshishKumar4 · commit 188ba0d1bdf7 · 2025-11-05T22:35:16.000-05:00
- Set isGenerating flag immediately before triggering generation to avoid UI state inconsistencies
- Added flag update for both new chat and reconnection scenarios
diff --git a/src/routes/chat/hooks/use-chat.ts b/src/routes/chat/hooks/use-chat.ts
@@ -294,6 +294,7 @@ export function useChat({
 					// Request file generation for new chats only
 					if (!disableGenerate && urlChatId === 'new') {
 						logger.debug('🔄 Starting code generation for new chat');
+						setIsGenerating(true);
 						sendWebSocketMessage(ws, 'generate_all');
 					}
 				});
diff --git a/src/routes/chat/utils/handle-websocket-message.ts b/src/routes/chat/utils/handle-websocket-message.ts
@@ -254,6 +254,7 @@ export function createWebSocketMessageHandler(deps: HandleMessageDeps) {
                     
                     if (state.shouldBeGenerating && !isGenerating) {
                         logger.debug('🔄 Reconnected with shouldBeGenerating=true, auto-resuming generation');
+                        setIsGenerating(true); 
                         updateStage('code', { status: 'active' });
                         sendWebSocketMessage(websocket, 'generate_all');
                     }

Original file line number	Diff line number	Diff line change
`@@ -294,6 +294,7 @@ export function useChat({`
`294`	`294`	`// Request file generation for new chats only`
`295`	`295`	`if (!disableGenerate && urlChatId === 'new') {`
`296`	`296`	`logger.debug('🔄 Starting code generation for new chat');`
	`297`	`+ setIsGenerating(true);`
`297`	`298`	`sendWebSocketMessage(ws, 'generate_all');`
`298`	`299`	`}`
`299`	`300`	`});`
Original file line number	Diff line number	Diff line change
`@@ -254,6 +254,7 @@ export function createWebSocketMessageHandler(deps: HandleMessageDeps) {`
`254`	`254`
`255`	`255`	`if (state.shouldBeGenerating && !isGenerating) {`
`256`	`256`	`logger.debug('🔄 Reconnected with shouldBeGenerating=true, auto-resuming generation');`
	`257`	`+ setIsGenerating(true);`
`257`	`258`	`updateStage('code', { status: 'active' });`
`258`	`259`	`sendWebSocketMessage(websocket, 'generate_all');`
`259`	`260`	`}`