session.go

package improvmx

import (
	"fmt"

	"occult.work/doze"
)

type Session struct {
	client      *doze.Client
	Credentials *CredentialEndpoint
	Account     *AccountEndpoint
	Domains     *DomainEndpoint
	Aliases     *AliasEndpoint
}

type SessionOption func(*Session) error

func New(token string, options ...SessionOption) (*Session, error) {
	session := &Session{}
	session.client = doze.NewClient().
		SetAuthToken(fmt.Sprintf("api:%s", token)).
		SetAuthScheme("Basic").
		SetBaseURL(BaseURLv3).
		SetError(&Error{})
	for _, option := range options {
		if error := option(session); error != nil {
			return nil, error
		}
	}
	session.Credentials = (*CredentialEndpoint)(session.client)
	session.Account = (*AccountEndpoint)(session.client)
	session.Domains = (*DomainEndpoint)(session.client)
	session.Aliases = (*AliasEndpoint)(session.client)
	return session, nil
}

// Sets the session client directly. This is provided for extreme user needs.
// However, the resty.Client.SetError type is still set to Error, and the host
// url is set toe BaseURLv3. Use WithBaseURL after calling WithClient if
// overriding the Host URL is required.
func WithClient(client *doze.Client) SessionOption {
	return func(session *Session) error {
		if client == nil {
			return fmt.Errorf("WithClient was passed a nil value")
		}
		client.SetError(&Error{}).SetBaseURL(BaseURLv3)
		session.client = client
		return nil
	}
}

// Sets the User-Agent to the given string on the session. By default, this is
// unset.
func WithUserAgent(agent string) SessionOption {
	return func(session *Session) error {
		session.client.SetUserAgent(agent)
		return nil
	}
}

// Sets the Base URL to query. This is typically BaseURLv3, however for both
// testing and possible future sandboxing support, this option is provided.
func WithBaseURL(url string) SessionOption {
	return func(session *Session) error {
		session.client.SetBaseURL(url)
		return nil
	}
}

// Enables the debug mode on the Session's internal http client.
func WithDebug() SessionOption {
	return func(session *Session) error {
		session.client.SetDebug()
		return nil
	}
}