fix(websocket): add close reasons and suppress noisy warnings

engine/sdks/typescript/runner/src/mod.ts

@@ -587,7 +587,7 @@ export class Runner {
 						msg: "error during websocket shutdown:",
 						error,
 					});
-					pegboardWebSocket.close();
+					pegboardWebSocket.close(1011, "pegboard.shutdown_error");
 				}
 			}
 		} else {
@@ -733,7 +733,7 @@ export class Runner {
 			this.log?.error(
 				"found duplicate pegboardWebSocket, closing previous",
 			);
-			this.#pegboardWebSocket.close(1000, "duplicate_websocket");
+			this.#pegboardWebSocket.close(1000, "pegboard.duplicate_websocket");
 		}
 
 		const ws = new WS(this.pegboardUrl, protocols) as any as WebSocket;

engine/sdks/typescript/runner/src/utils.ts

@@ -57,7 +57,9 @@ export function parseWebSocketCloseReason(
 	const [group, error] = mainPart.split(".");
 
 	if (!group || !error) {
-		logger()?.warn({ msg: "failed to parse close reason", reason });
+		if (reason) {
+			logger()?.warn({ msg: "failed to parse close reason", reason });
+		}
 		return undefined;
 	}
 

rivetkit-swift/Sources/RivetKitClient/ActorConnection.swift

@@ -765,7 +765,7 @@ public actor ActorConnection {
                 } else {
                     errorToThrow = ActorError(group: group, code: code, message: "Connection closed: \(reason)", metadata: nil)
                 }
-            } else {
+            } else if !reason.isEmpty {
                 logger.warning("failed to parse close reason reason=\(reason, privacy: .public)")
             }
         }

rivetkit-typescript/packages/rivetkit/src/client/utils.ts

@@ -42,7 +42,9 @@ export function parseWebSocketCloseReason(
 	const [group, code] = mainPart.split(".");
 
 	if (!group || !code) {
-		logger().warn({ msg: "failed to parse close reason", reason });
+		if (reason) {
+			logger().warn({ msg: "failed to parse close reason", reason });
+		}
 		return undefined;
 	}
 

rivetkit-typescript/packages/rivetkit/src/remote-manager-driver/ws-proxy.ts

@@ -164,9 +164,9 @@ export async function createWebSocketProxy(
 
 			if (state.targetWs) {
 				if (state.targetWs.readyState === WebSocket.OPEN) {
-					state.targetWs.close(1011, "Client WebSocket error");
+					state.targetWs.close(1011, "ws.client_error");
 				} else if (state.targetWs.readyState === WebSocket.CONNECTING) {
-					state.targetWs.close();
+					state.targetWs.close(1011, "ws.client_error");
 				}
 			}
 		},

rivetkit-typescript/packages/rivetkit/src/sandbox/client.ts

@@ -382,12 +382,12 @@ class DirectTerminalSession implements TerminalSession {
 				this.closeSignalSent = true;
 				this.sendFrame({ type: "close" });
 			}
-			this.socket.close();
+			this.socket.close(1000, "sandbox.client_closed");
 			return;
 		}
 
 		if (this.socket.readyState !== 3) {
-			this.socket.close();
+			this.socket.close(1000, "sandbox.client_closed");
 		}
 	}