hooks.go

// Package hooks implements the controller's builder hooks api.
//
// This is primarily intended to be consumed by the builder to communicate with the controller.
package hooks

import (
	"encoding/json"
	"fmt"
	"io"
	"io/ioutil"

	"github.com/deis/controller-sdk-go"
	"github.com/deis/controller-sdk-go/api"
)

// UserFromKey retrives a user from their SSH key fingerprint.
func UserFromKey(c *deis.Client, fingerprint string) (api.UserApps, error) {
	res, reqErr := c.Request("GET", fmt.Sprintf("/v2/hooks/key/%s", fingerprint), nil)
	if reqErr != nil && !deis.IsErrAPIMismatch(reqErr) {
		return api.UserApps{}, reqErr
	}

	// Fix json.Decoder bug in <go1.7
	defer func() {
		io.Copy(ioutil.Discard, res.Body)
		res.Body.Close()
	}()

	resUser := api.UserApps{}
	if err := json.NewDecoder(res.Body).Decode(&resUser); err != nil {
		return api.UserApps{}, err
	}

	return resUser, reqErr
}

// GetAppConfig retrives an app's configuration from the controller.
func GetAppConfig(c *deis.Client, username, app string) (api.Config, error) {
	req := api.ConfigHookRequest{User: username, App: app}
	b, err := json.Marshal(req)
	if err != nil {
		return api.Config{}, err
	}

	res, reqErr := c.Request("POST", "/v2/hooks/config/", b)
	if reqErr != nil && !deis.IsErrAPIMismatch(reqErr) {
		return api.Config{}, reqErr
	}
	// Fix json.Decoder bug in <go1.7
	defer func() {
		io.Copy(ioutil.Discard, res.Body)
		res.Body.Close()
	}()

	config := api.Config{}
	if err := json.NewDecoder(res.Body).Decode(&config); err != nil {
		return api.Config{}, err
	}

	return config, reqErr
}

// CreateBuild creates a new release of an application. It returns the version of the new release.
// gitSha should be the first 8 charecters of the git commit sha. Image is either the docker image
// location for the dockerfile app the absolute url to the tar file for a buldpack app.
func CreateBuild(c *deis.Client, username, app, image, gitSha string, procfile api.ProcessType,
	usingDockerifle bool) (int, error) {
	req := api.BuildHookRequest{
		Sha:      gitSha,
		User:     username,
		App:      app,
		Image:    image,
		Procfile: procfile,
	}

	if usingDockerifle {
		req.Dockerfile = "true"
	}

	b, err := json.Marshal(req)
	if err != nil {
		return -1, err
	}

	res, reqErr := c.Request("POST", "/v2/hooks/build/", b)
	if reqErr != nil && !deis.IsErrAPIMismatch(reqErr) {
		return -1, reqErr
	}
	// Fix json.Decoder bug in <go1.7
	defer func() {
		io.Copy(ioutil.Discard, res.Body)
		res.Body.Close()
	}()

	resMap := make(map[string]map[string]int)
	if err := json.NewDecoder(res.Body).Decode(&resMap); err != nil {
		return -1, err
	}

	return resMap["release"]["version"], reqErr
}

// CreatePush creates a new push for an application.
// gitSha should be the first 8 charecters of the git commit sha.
func CreatePush(c *deis.Client, username, app, gitSha, fingerprint, conn, cmd string) error {
	req := api.PushRequest{
		Sha:         gitSha,
		User:        username,
		App:         app,
		Fingerprint: fingerprint,
		Connection:  conn,
		Command:     cmd,
	}

	b, err := json.Marshal(req)
	if err != nil {
		return err
	}

	res, reqErr := c.Request("POST", "/v2/hooks/push/", b)
	if reqErr == nil {
		res.Body.Close()
	}
	return reqErr
}