feat(gateway): implement automatic device registration during OAuth2 flow

Author: Lasim

services/gateway/src/commands/login.ts

@@ -7,7 +7,6 @@ import { DeployStackAPI } from '../core/auth/api-client';
 import { MCPConfigService } from '../core/mcp';
 import { AuthenticationError } from '../types/auth';
 import { ServerStartService } from '../services/server-start-service';
-import { detectDeviceInfo } from '../utils/device-detection';
 
 export function registerLoginCommand(program: Command) {
   program
@@ -44,33 +43,13 @@ export function registerLoginCommand(program: Command) {
           timeout: 120000 // 2 minutes
         });
 
-        spinner = ora('Registering device...').start();
+        spinner = ora('Storing credentials securely...').start();
 
-        // NEW: Automatic device registration
-        try {
-          const api = new DeployStackAPI(authResult.credentials, options.url);
-          const deviceInfo = await detectDeviceInfo();
-          const device = await api.registerOrUpdateDevice(deviceInfo);
-          
-          spinner.text = 'Storing credentials securely...';
-          
-          // Store credentials with device context
-          await storage.storeCredentials({
-            ...authResult.credentials,
-            deviceId: device.id
-          });
-
-          console.log(chalk.green(`📱 Device registered: ${device.device_name}`));
-        } catch (deviceError) {
-          // If device registration fails, continue without it
-          spinner.text = 'Storing credentials securely...';
-          await storage.storeCredentials(authResult.credentials);
-          
-          console.log(chalk.yellow('⚠️  Device registration failed - continuing without device context'));
-          if (deviceError instanceof Error) {
-            console.log(chalk.gray(`   Device Error: ${deviceError.message}`));
-          }
-        }
+        // Device registration now happens automatically during OAuth2 token exchange
+        // Store credentials with device context if available from OAuth2 response
+        const credentialsToStore = authResult.credentials;
+        
+        await storage.storeCredentials(credentialsToStore);
 
         // Set default team as selected team and download MCP config
         spinner.text = 'Setting up default team...';

services/gateway/src/core/auth/oauth.ts

@@ -4,6 +4,7 @@ import { generateCodeVerifier, generateCodeChallenge, generateState } from '../.
 import { buildAuthConfig } from '../../utils/auth-config';
 import { CallbackServer } from './callback-server';
 import { BrowserManager } from './browser';
+import { detectDeviceInfo } from '../../utils/device-detection';
 import {
   AuthenticationResult,
   AuthenticationOptions,
@@ -158,19 +159,31 @@ export class OAuth2Client {
     codeVerifier: string;
   }): Promise<TokenResponse> {
     try {
+      // Detect device information for automatic registration
+      const deviceInfo = await detectDeviceInfo();
+      
+      // Add device_name field required by backend schema
+      const deviceInfoWithName = {
+        device_name: deviceInfo.hostname, // Use hostname as default device name
+        ...deviceInfo
+      };
+      
+      const requestBody = {
+        grant_type: 'authorization_code',
+        code: params.code,
+        redirect_uri: this.config.redirectUri,
+        client_id: this.config.clientId,
+        code_verifier: params.codeVerifier,
+        device_info: deviceInfoWithName
+      };
+
       const response = await fetch(this.config.tokenUrl, {
         method: 'POST',
         headers: {
           'Content-Type': 'application/json',
           'User-Agent': 'DeployStack-Gateway-CLI/0.2.0'
         },
-        body: JSON.stringify({
-          grant_type: 'authorization_code',
-          code: params.code,
-          redirect_uri: this.config.redirectUri,
-          client_id: this.config.clientId,
-          code_verifier: params.codeVerifier
-        })
+        body: JSON.stringify(requestBody)
       });
 
       if (!response.ok) {
@@ -182,7 +195,14 @@ export class OAuth2Client {
         );
       }
 
-      return await response.json() as TokenResponse;
+      const tokenResponse = await response.json() as TokenResponse;
+      
+      // Log device registration success if device info is included in response
+      if (tokenResponse.device) {
+        console.log(chalk.green(`📱 Device registered: ${tokenResponse.device.device_name}`));
+      }
+
+      return tokenResponse;
     } catch (error) {
       if (error instanceof AuthenticationError) {
         throw error;

services/gateway/src/types/auth.ts

@@ -1,3 +1,5 @@
+import { Device } from '../utils/device-detection';
+
 export interface StoredCredentials {
   accessToken: string;
   refreshToken: string;
@@ -33,6 +35,7 @@ export interface TokenResponse {
   expires_in: number;
   refresh_token: string;
   scope: string;
+  device?: Device;
 }
 
 export interface TokenInfo {