compatible cookie of region/prject loss

Author: lixiaojun629

base/config.go

@@ -400,14 +400,33 @@ func NewInCloudShell() (*AggConfigManager, error) {
 	regionKey := fmt.Sprintf("c_last_region_%s", email)
 	var proj project
 	var reg region
-	err = json.Unmarshal([]byte(tokenMap[projectKey]), &proj)
-	if err != nil {
-		return nil, err
+	if _, ok := tokenMap[projectKey]; ok {
+		err = json.Unmarshal([]byte(tokenMap[projectKey]), &proj)
+		if err != nil {
+			return nil, err
+		}
+	} else {
+		id, name, err := getDefaultProject(cookie, tokenMap["CSRF_TOKEN"])
+		if err != nil {
+			return nil, fmt.Errorf("query default project error: %w", err)
+		}
+		proj.ProjectId = id
+		proj.ProjectName = name
 	}
-	err = json.Unmarshal([]byte(tokenMap[regionKey]), &reg)
-	if err != nil {
-		return nil, err
+	if _, ok := tokenMap[regionKey]; ok {
+		err = json.Unmarshal([]byte(tokenMap[regionKey]), &reg)
+		if err != nil {
+			return nil, err
+		}
+	} else {
+		region, zone, err := getDefaultRegion(cookie, tokenMap["CSRF_TOKEN"])
+		if err != nil {
+			return nil, fmt.Errorf("query default region error: %w", err)
+		}
+		reg.Region = region
+		reg.Zone = zone
 	}
+
 	ac := &AggConfig{
 		Cookie:        cookie,
 		Profile:       DefaultProfile,

base/util.go

@@ -648,3 +648,54 @@ func curGoroutineID() int64 {
 
 	return int64(id)
 }
+
+func getDefaultRegion(cookie, csrfToken string) (string, string, error) {
+	cfg := &AggConfig{
+		Cookie:        cookie,
+		BaseURL:       DefaultBaseURL,
+		CSRFToken:     csrfToken,
+		Timeout:       DefaultTimeoutSec,
+		MaxRetryTimes: sdk.Int(DefaultMaxRetryTimes),
+	}
+	bc, err := GetBizClient(cfg)
+	req := bc.NewGetRegionRequest()
+	if err != nil {
+		return "", "", err
+	}
+	resp, err := bc.GetRegion(req)
+	if err != nil {
+		return "", "", err
+	}
+	for _, r := range resp.Regions {
+		if r.IsDefault {
+			return r.Region, r.Zone, nil
+		}
+	}
+	return "", "", fmt.Errorf("default region not found")
+}
+
+func getDefaultProject(cookie, csrfToken string) (string, string, error) {
+	cfg := &AggConfig{
+		Cookie:        cookie,
+		BaseURL:       DefaultBaseURL,
+		CSRFToken:     csrfToken,
+		Timeout:       DefaultTimeoutSec,
+		MaxRetryTimes: sdk.Int(DefaultMaxRetryTimes),
+	}
+	bc, err := GetBizClient(cfg)
+	if err != nil {
+		return "", "", err
+	}
+
+	req := bc.NewGetProjectListRequest()
+	resp, err := bc.GetProjectList(req)
+	if err != nil {
+		return "", "", err
+	}
+	for _, project := range resp.ProjectSet {
+		if project.IsDefault == true {
+			return project.ProjectId, project.ProjectName, nil
+		}
+	}
+	return "", "", fmt.Errorf("default project not found")
+}