credentials_set.go

package goauth

import (
	"context"
	"fmt"
	"io/fs"
	"net/http"
	"sort"
	"strings"

	"github.com/grokify/mogo/encoding/jsonutil"
	"github.com/grokify/mogo/errors/errorsutil"
)

type CredentialsSet struct {
	Credentials map[string]Credentials `json:"credentials,omitempty"`
}

func ReadFileCredentialsSet(credentialsSetFilename string, inflateEndpoints bool) (*CredentialsSet, error) {
	var set *CredentialsSet
	if err := jsonutil.UnmarshalFile(credentialsSetFilename, &set); err != nil {
		return nil, err
	} else if inflateEndpoints {
		err := set.Inflate()
		if err != nil {
			return nil, err
		}
	}
	return set, nil
}

func (set *CredentialsSet) Get(key string) (Credentials, error) {
	if creds, ok := set.Credentials[key]; ok {
		return creds, nil
	}
	return Credentials{}, fmt.Errorf("credentials key not found (%s)", key)
}

func (set *CredentialsSet) Inflate() error {
	for k, v := range set.Credentials {
		err := v.Inflate()
		if err != nil {
			return err
		}
		set.Credentials[k] = v
	}
	return nil
}

/*
func (set *CredentialsSet) NewSimpleClient(accountKey string) (*httpsimple.SimpleClient, error) {
	creds, ok := set.Credentials[accountKey]
	if !ok {
		return nil, fmt.Errorf("client_not_found [%s]", accountKey)
	}
	return creds.NewSimpleClient()
}
*/

func ReadCredentialsFromSetFile(credentialsSetFilename, accountKey string, inclAccountsOnError bool) (Credentials, error) {
	set, err := ReadFileCredentialsSet(credentialsSetFilename, true)
	if err != nil {
		return Credentials{}, err
	}
	creds, err := set.Get(accountKey)
	if err != nil {
		if inclAccountsOnError {
			return creds, errorsutil.Wrap(err,
				fmt.Sprintf("validAccounts [%s]", strings.Join(set.Accounts(), ",")))
		}
		return creds, err
	}
	return creds, nil
}

/*
func ReadFileCredentialsFromCredentialsSet(credentialsSetFilename, account string) (Credentials, error) {
	set, err := ReadFileCredentialsSet(credentialsSetFilename, true)
	if err != nil {
		return Credentials{}, err
	} else {
		return set.Get(account)
	}
}
*/

func (set *CredentialsSet) NewClient(ctx context.Context, key string) (*http.Client, error) {
	creds, ok := set.Credentials[key]
	if !ok {
		return nil, fmt.Errorf("E_CREDS_KEY_NOT_FOUND [%v]", key)
	}
	return creds.NewClient(ctx)
}

func (set *CredentialsSet) Accounts() []string { return set.Keys() }

func (set *CredentialsSet) Keys() []string {
	keys := []string{}
	for key := range set.Credentials {
		keys = append(keys, key)
	}
	sort.Strings(keys)
	return keys
}

func (set *CredentialsSet) WriteFile(filename, prefix, indent string, perm fs.FileMode) error {
	return jsonutil.MarshalFile(filename, set, prefix, indent, perm)
}