psu/common/client.go

package common

import (
	"bytes"
	"crypto/tls"
	"io/ioutil"
	"net/http"
	"net/url"
	"strings"

	"github.com/greenled/portainer-stack-utils/client"
	"github.com/sirupsen/logrus"

	"github.com/spf13/viper"
)

var cachedClient client.PortainerClient

// Get the cached client or a new one
func GetClient() (c client.PortainerClient, err error) {
	if cachedClient == nil {
		cachedClient, err = GetDefaultClient()
		if err != nil {
			return
		}
	}
	return cachedClient, nil
}

// Get the default client
func GetDefaultClient() (c client.PortainerClient, err error) {
	config, err := GetDefaultClientConfig()
	if err != nil {
		return
	}

	c = client.NewClient(GetDefaultHttpClient(), config)

	c.BeforeRequest(func(req *http.Request) (err error) {
		var bodyString string
		if req.Body != nil {
			bodyBytes, readErr := ioutil.ReadAll(req.Body)
			defer req.Body.Close()
			if readErr != nil {
				return readErr
			}
			bodyString = string(bodyBytes)
			req.Body = ioutil.NopCloser(bytes.NewReader(bodyBytes))
		}

		logrus.WithFields(logrus.Fields{
			"method": req.Method,
			"url":    req.URL.String(),
			"body":   string(bodyString),
		}).Trace("Request to Portainer")

		return
	})

	c.AfterResponse(func(resp *http.Response) (err error) {
		var bodyString string
		if resp.Body != nil {
			bodyBytes, readErr := ioutil.ReadAll(resp.Body)
			defer resp.Body.Close()
			if readErr != nil {
				return readErr
			}
			bodyString = string(bodyBytes)
			resp.Body = ioutil.NopCloser(bytes.NewReader(bodyBytes))
		}

		logrus.WithFields(logrus.Fields{
			"status": resp.Status,
			"body":   string(bodyString),
		}).Trace("Response from Portainer")

		return
	})

	return
}

// Get the default config for a client
func GetDefaultClientConfig() (config client.Config, err error) {
	apiUrl, err := url.Parse(strings.TrimRight(viper.GetString("url"), "/") + "/api/")
	if err != nil {
		return
	}

	config = client.Config{
		Url:           apiUrl,
		User:          viper.GetString("user"),
		Password:      viper.GetString("password"),
		Token:         viper.GetString("auth-token"),
		UserAgent:     BuildUseAgentString(),
		DoNotUseToken: false,
	}

	return
}

// Get the default http client for a Portainer client
func GetDefaultHttpClient() *http.Client {
	return &http.Client{
		Timeout: viper.GetDuration("timeout"),
		Transport: &http.Transport{
			TLSClientConfig: &tls.Config{
				InsecureSkipVerify: viper.GetBool("insecure"),
			},
		},
	}
}