utils.go

package controller

import (
	"fmt"
	"os"

	"github.com/deis/builder/pkg/conf"
	deis "github.com/deis/controller-sdk-go"
	"github.com/deis/pkg/log"
)

const (
	hostEnvName = "DEIS_CONTROLLER_SERVICE_HOST"
	portEnvName = "DEIS_CONTROLLER_SERVICE_PORT"
)

// New creates a new SDK client configured as the builder.
func New() (*deis.Client, error) {
	host := os.Getenv(hostEnvName)
	port := os.Getenv(portEnvName)

	client, err := deis.New(true, fmt.Sprintf("http://%s:%s/", host, port), "")
	if err != nil {
		return client, err
	}
	client.UserAgent = "deis-builder"

	builderKey, err := conf.GetBuilderKey()
	if err != nil {
		return client, err
	}
	client.HooksToken = builderKey

	return client, nil
}

// CheckAPICompat checks for API compatibility errors and warns about them.
func CheckAPICompat(c *deis.Client, err error) error {
	if err == deis.ErrAPIMismatch {
		log.Info("WARNING: SDK and Controller API versions do not match. SDK: %s Controller: %s",
			deis.APIVersion, c.ControllerAPIVersion)

		// API mismatch isn't fatal, so after warning continue on.
		return nil
	}

	return err
}