auth.go

package hubspot

import (
	"fmt"
	"net/http"
)

type Authenticator interface {
	SetAuthentication(r *http.Request) error
}

type AuthMethod func(c *Client)

func SetOAuth(config *OAuthConfig) AuthMethod {
	return func(c *Client) {
		c.authenticator = &OAuth{
			retriever: &OAuthTokenManager{
				oauthPath:  fmt.Sprintf("%s/%s", c.baseURL.String(), oauthTokenPath),
				HTTPClient: c.HTTPClient,
				Config:     config,
			},
		}
	}
}

// Deprecated: Use hubspot.SetPrivateAppToken.
func SetAPIKey(key string) AuthMethod {
	return func(c *Client) {
		c.authenticator = &APIKey{
			apikey: key,
		}
	}
}

func SetPrivateAppToken(token string) AuthMethod {
	return func(c *Client) {
		c.authenticator = &PrivateAppToken{
			accessToken: token,
		}
	}
}

type OAuth struct {
	retriever OAuthTokenRetriever
}

func (o *OAuth) SetAuthentication(r *http.Request) error {
	t, err := o.retriever.RetrieveToken()
	if err != nil {
		return err
	}
	r.Header.Set("Authorization", "Bearer "+t.AccessToken)
	return nil
}

type APIKey struct {
	apikey string
}

func (a *APIKey) SetAuthentication(r *http.Request) error {
	q := r.URL.Query()
	q.Set("hapikey", a.apikey)
	r.URL.RawQuery = q.Encode()
	return nil
}

type PrivateAppToken struct {
	accessToken string
}

func (p *PrivateAppToken) SetAuthentication(r *http.Request) error {
	h := r.Header
	h.Set("Authorization", "Bearer "+p.accessToken)
	return nil
}