settings.go

package settings

import (
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"

	drycc "github.com/drycc/controller-sdk-go"
	"github.com/drycc/workflow-cli/version"
)

// DefaultResponseLimit is the default number of responses to return on requests that can
// be limited.
const DefaultResponseLimit = 100

// UserAgent is the user agent used by the CLI
var UserAgent = "Drycc Client " + version.Version

type settingsFile struct {
	Username   string `json:"username"`
	VerifySSL  bool   `json:"ssl_verify"`
	Controller string `json:"controller"`
	Token      string `json:"token"`
	Limit      int    `json:"response_limit"`
	Workspace  string `json:"workspace"`
}

// Settings is the settings object created from the settings file.
type Settings struct {
	Username  string
	Limit     int
	Client    *drycc.Client
	Workspace string
}

// Load loads a new client from a settings file.
func Load(cf string) (*Settings, error) {
	filename := locateSettingsFile(cf)

	if _, err := os.Stat(filename); err != nil {
		if os.IsNotExist(err) {
			return nil, fmt.Errorf(`client configuration file not found at: %s
Are you logged in? Use 'drycc login' or 'drycc register' to get started`, filename)
		}

		return nil, err
	}

	contents, err := os.ReadFile(filename)
	if err != nil {
		return nil, err
	}

	sF := settingsFile{}
	if err = json.Unmarshal(contents, &sF); err != nil {
		return nil, err
	}

	c, err := drycc.New(sF.VerifySSL, sF.Controller, sF.Token)
	if err != nil {
		return nil, err
	}

	// Set a custom user agent
	c.UserAgent = UserAgent

	settings := Settings{}
	settings.Username = sF.Username
	settings.Client = c
	settings.Workspace = sF.Workspace

	// If users have defined a custom response limit, respect it.
	if sF.Limit > 0 {
		settings.Limit = sF.Limit
	} else {
		settings.Limit = DefaultResponseLimit
	}

	return &settings, nil
}

// Save settings to a file
func (s *Settings) Save(cf string) (string, error) {
	settings := settingsFile{
		Username: s.Username, VerifySSL: s.Client.VerifySSL,
		Controller: s.Client.ControllerURL.String(), Token: s.Client.Token, Limit: s.Limit,
		Workspace: s.Workspace,
	}

	settingsContents, err := json.Marshal(settings)
	if err != nil {
		return "", err
	}

	filename := locateSettingsFile(cf)

	if err := os.MkdirAll(filepath.Dir(filename), 0o700); err != nil {
		return "", err
	}

	return filename, os.WriteFile(filename, settingsContents, 0o600)
}

// DryccHome returns the path to the user's settings path.
func DryccHome() string {
	dryccHome := filepath.Join(FindHome(), "/.drycc/")
	_, err := os.Stat(dryccHome)
	if os.IsNotExist(err) {
		if err := os.MkdirAll(dryccHome, 0o700); err != nil {
			panic(err)
		}
		return dryccHome
	}
	return dryccHome
}

// Delete user's settings file.
func Delete(cf string) error {
	filename := locateSettingsFile(cf)

	if _, err := os.Stat(filename); err != nil {
		if os.IsNotExist(err) {
			return nil
		}

		return err
	}

	return os.Remove(filename)
}