Set current team id after spaceship login on the spaceship client (#227)

* Set current team id after spaceship login on the spaceship client

* Extract team id from sapecship respone

* Update current team related log

Author: godrei

autocodesign/devportalclient/spaceship/spaceship.go

@@ -50,14 +50,35 @@ func NewClient(authConfig appleauth.AppleID, teamID string, cmdFactory ruby.Comm
 		cmdFactory: cmdFactory,
 	}
 
-	_, err = c.runSpaceshipCommand("login")
+	currentTeamID, err := c.login()
 	if err != nil {
-		return nil, fmt.Errorf("running command failed with error: %s", err)
+		return nil, fmt.Errorf("spaceship command failed: %s", err)
 	}
 
+	log.Debugf("current team id: %s", currentTeamID)
+
+	c.teamID = currentTeamID
+
 	return c, nil
 }
 
+func (c *Client) login() (string, error) {
+	output, err := c.runSpaceshipCommand("login")
+	if err != nil {
+		return "", fmt.Errorf("running command failed with error: %s", err)
+	}
+
+	// {"data":"72SA8V3WYL"}
+	var teamIDResponse struct {
+		Data string `json:"data"`
+	}
+	if err := json.Unmarshal([]byte(output), &teamIDResponse); err != nil {
+		return "", fmt.Errorf("failed to unmarshal response: %v", err)
+	}
+
+	return teamIDResponse.Data, nil
+}
+
 // DevPortalClient ...
 type DevPortalClient struct {
 	*CertificateSource

autocodesign/devportalclient/spaceship/spaceship/main.rb

@@ -27,17 +27,19 @@
 
   FastlaneCore::Globals.verbose = true
 
+  result = '{}'
+
   if options[:subcommand] == 'login'
     begin
-      Portal::AuthClient.login(options[:username], options[:password], options[:session], options[:team_id])
+      team_id = Portal::AuthClient.login(options[:username], options[:password], options[:session], options[:team_id])
+      result = team_id
     rescue => e
       puts "\nApple ID authentication failed: #{e}"
       exit(1)
     end
   else
     Portal::AuthClient.restore_from_session(options[:username], options[:team_id])
 
-    result = '{}'
     case options[:subcommand]
     when 'list_dev_certs'
       client = CertificateHelper.new

autocodesign/devportalclient/spaceship/spaceship/portal/auth_client.rb

@@ -17,6 +17,8 @@ def self.login(username, password, two_factor_session = nil, team_id = nil)
       end
 
       client.store_cookie
+
+      client.team_id
     end
 
     def self.restore_from_session(username, team_id)