sam-coder-cli 1.0.23 → 1.0.24

package/bin/multiplayer-client.js

@@ -272,49 +272,78 @@ class MultiplayerClient extends EventEmitter {
   }
 
   async connect() {
-    try {
-      // Ensure server is running
-      this.serverUrl = await ensureServerRunning(this.port);
-      
-      this.ws = new WebSocket(this.serverUrl);
-      
-      this.ws.on('open', () => {
-        this.connected = true;
-        this.emit('connected');
-        console.log(chalk.green('✅ Connected to multiplayer server'));
-        
-        // Send client info immediately after connection
-        this.updateClientInfo({
-          name: this.name,
-          role: this.role,
-          model: this.model
-        });
-      });
-      
-      this.ws.on('message', (data) => {
+    return new Promise((resolve, reject) => {
+      const connectWithRetry = async () => {
         try {
-          const message = JSON.parse(data);
-          this.handleMessage(message);
+          // If no server URL is provided, try to start a local server
+          if (this.serverUrl === 'ws://localhost:8080') {
+            try {
+              this.serverUrl = await ensureServerRunning(8080);
+            } catch (error) {
+              console.error(chalk.red('Failed to start local server:'), error.message);
+              reject(error);
+              return;
+            }
+          }
+          
+          console.log(chalk.blue(`Connecting to ${this.serverUrl}...`));
+          this.ws = new WebSocket(this.serverUrl);
+          
+          const connectionTimeout = setTimeout(() => {
+            this.ws.terminate();
+            reject(new Error('Connection timeout'));
+          }, 10000); // 10 seconds timeout
+          
+          this.ws.on('open', () => {
+            clearTimeout(connectionTimeout);
+            this.connected = true;
+            console.log(chalk.green('✅ Connected to multiplayer server'));
+            
+            // Send client info
+            this.send({
+              type: 'client_info',
+              clientId: this.clientId,
+              name: this.name,
+              role: this.role,
+              model: this.model
+            });
+            
+            this.emit('connected');
+            resolve();
+          });
+          
+          this.ws.on('message', (data) => {
+            try {
+              const message = JSON.parse(data);
+              this.handleMessage(message);
+            } catch (error) {
+              console.error('Error parsing message:', error);
+            }
+          });
+          
+          this.ws.on('close', () => {
+            this.connected = false;
+            this.emit('disconnected');
+            console.log(chalk.yellow('\n🔌 Disconnected from multiplayer server'));
+          });
+          
+          this.ws.on('error', (error) => {
+            clearTimeout(connectionTimeout);
+            console.error('WebSocket error:', error.message);
+            if (!this.connected) {
+              reject(error);
+            }
+            this.emit('error', error);
+          });
+          
         } catch (error) {
-          console.error('Error parsing message:', error);
+          console.error('Connection error:', error.message);
+          reject(error);
         }
-      });
-      
-      this.ws.on('close', () => {
-        this.connected = false;
-        this.emit('disconnected');
-        console.log(chalk.yellow('\n🔌 Disconnected from multiplayer server'));
-      });
-      
-      this.ws.on('error', (error) => {
-        console.error('WebSocket error:', error);
-        this.emit('error', error);
-      });
+      };
       
-    } catch (error) {
-      console.error('Failed to connect to server:', error);
-      this.emit('error', error);
-    }
+      connectWithRetry();
+    });
   }
 
   handleMessage(message) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sam-coder-cli",
-  "version": "1.0.23",
+  "version": "1.0.24",
   "description": "SAM-CODER: An animated command-line AI assistant with agency capabilities.",
   "main": "bin/agi-cli.js",
   "bin": {