Function GetTeamForToken

cli/internal/auth/team.go:35–70

GetTeamForToken returns the team for the given token If the token is a bearer token, we need to get the team from the configuration. For api keys the team name is not required as the key is bound to a team name already.

(ctx context.Context, token auth.Token)

// If the token is a bearer token, we need to get the team from the configuration.
// For api keys the team name is not required as the key is bound to a team name already.
func GetTeamForToken(ctx context.Context, token auth.Token) (string, error) {
	switch token.Type {
	case auth.BearerToken:
		team, err := config.GetValue("team")
		if configFileMissing(err) || emptyTeam(err, team) {
			teams := getAvailableUserTeams(ctx, token)
			if len(teams) > 0 {
				return "", fmt.Errorf("team is not set.\n\nAvailable teams: %s.\n\nTo set your team, run `cloudquery switch <team>`", strings.Join(teams, ", "))
			}
			return "", errors.New("team is not set.\n\n. Hint: use `cloudquery login` and/or `cloudquery switch <team>`")
		}
		if err != nil {
			return "", fmt.Errorf("failed to get team name from config: %w. Hint: use `cloudquery login` and/or `cloudquery switch <team>`", err)
		}
		return team, nil
	case auth.APIKey:
		cl, err := teamapi.NewClient(token.Value)
		if err != nil {
			return "", fmt.Errorf("failed to create API client for api key: %w", err)
		}
		teams, err := cl.ListAllTeams(ctx)
		if err != nil {
			return "", fmt.Errorf("failed to list teams for api key: %w", err)
		}
		switch l := len(teams); l {
		case 0:
			return "", errors.New("team api key has no assigned team")
		case 1:
			return teams[0].Name, nil
		default:
			return "", fmt.Errorf("team api key has more than one team: %s", teamapi.Names(teams))
		}
	default:
		return os.Getenv("_CQ_TEAM_NAME"), nil
	}
}

// IsTeamInternal checks if the team has the internal flag set
// We store this in the configuration on team switch.