Method GetGodoClient

doit.go:238–303

GetGodoClient returns a GodoClient.

(trace, allowRetries bool, accessToken string)

// GetGodoClient returns a GodoClient.
func (c *LiveConfig) GetGodoClient(trace, allowRetries bool, accessToken string) (*godo.Client, error) {
	if accessToken == "" {
		return nil, fmt.Errorf("access token is required. (hint: run 'doctl auth init')")
	}

	tokenSource := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: accessToken})
	oauthClient := oauth2.NewClient(context.Background(), tokenSource)

	args := []godo.ClientOpt{
		godo.SetUserAgent(userAgent()),
	}

	logger := log.New(os.Stderr, "doctl: ", log.LstdFlags)

	retryMax := viper.GetInt("http-retry-max")
	retryWaitMax := viper.GetInt("http-retry-wait-max")
	retryWaitMin := viper.GetInt("http-retry-wait-min")
	if retryMax > 0 && allowRetries {
		retryConfig := godo.RetryConfig{
			RetryMax: retryMax,
		}

		if retryWaitMax > 0 {
			retryConfig.RetryWaitMax = godo.PtrTo(float64(retryWaitMax))
		}

		if retryWaitMin > 0 {
			retryConfig.RetryWaitMin = godo.PtrTo(float64(retryWaitMin))
		}

		if trace {
			retryConfig.Logger = logger
		}

		args = append(args, godo.WithRetryAndBackoffs(retryConfig))
	}

	apiURL := viper.GetString("api-url")
	if apiURL != "" {
		args = append(args, godo.SetBaseURL(apiURL))
	}

	client, err := godo.New(oauthClient, args...)
	if err != nil {
		return nil, err
	}

	if trace {
		r := newRecorder(client.HTTPClient.Transport)

		go func() {
			for {
				select {
				case msg := <-r.req:
					logger.Println("->", strconv.Quote(msg))
				case msg := <-r.resp:
					logger.Println("<-", strconv.Quote(msg))
				}