Initial commit

vendor/github.com/hyperhq/hyper-api/client/compose.go

@@ -0,0 +1,149 @@
+package client
+
+import (
+	"context"
+	"fmt"
+	"io"
+	"net/url"
+	"strings"
+
+	"github.com/hyperhq/hyper-api/types"
+	"github.com/hyperhq/libcompose/config"
+)
+
+type composeConfigWrapper struct {
+	ServiceConfigs *config.ServiceConfigs           `json:"ServiceConfigs"`
+	VolumeConfigs  map[string]*config.VolumeConfig  `json:"VolumeConfigs"`
+	NetworkConfigs map[string]*config.NetworkConfig `json:"NetworkConfigs"`
+	AuthConfigs    map[string]types.AuthConfig      `json:"auths"`
+}
+
+func (cli *Client) ComposeUp(project string, services []string, c *config.ServiceConfigs, vc map[string]*config.VolumeConfig, nc map[string]*config.NetworkConfig, auth map[string]types.AuthConfig, forcerecreate, norecreate bool) (io.ReadCloser, error) {
+	query := url.Values{}
+	query.Set("project", project)
+	if forcerecreate {
+		query.Set("forcerecreate", "true")
+	}
+	if norecreate {
+		query.Set("norecreate", "true")
+	}
+	if len(services) > 0 {
+		query.Set("services", strings.Join(services, "}{"))
+	}
+	body := composeConfigWrapper{
+		ServiceConfigs: c,
+		VolumeConfigs:  vc,
+		NetworkConfigs: nc,
+		AuthConfigs:    auth,
+	}
+	resp, err := cli.post(context.Background(), "/compose/up", query, body, nil)
+	if err != nil {
+		return nil, err
+	}
+	return resp.body, nil
+}
+
+func (cli *Client) ComposeDown(project string, services []string, rmi string, vol, rmorphans bool) (io.ReadCloser, error) {
+
+	query := url.Values{}
+	query.Set("project", project)
+	if rmi != "" {
+		query.Set("rmi", rmi)
+	}
+	if vol {
+		query.Set("rmvol", "true")
+	}
+	if rmorphans {
+		query.Set("rmorphans", "true")
+	}
+	if len(services) > 0 {
+		query.Set("services", strings.Join(services, "}{"))
+	}
+	resp, err := cli.post(context.Background(), "/compose/down", query, nil, nil)
+	if err != nil {
+		return nil, err
+	}
+	return resp.body, nil
+}
+
+func (cli *Client) ComposeCreate(project string, services []string, c *config.ServiceConfigs, vc map[string]*config.VolumeConfig, nc map[string]*config.NetworkConfig, auth map[string]types.AuthConfig, forcerecreate, norecreate bool) (io.ReadCloser, error) {
+	query := url.Values{}
+	query.Set("project", project)
+	if forcerecreate {
+		query.Set("forcerecreate", "true")
+	}
+	if norecreate {
+		query.Set("norecreate", "true")
+	}
+	if len(services) > 0 {
+		query.Set("services", strings.Join(services, "}{"))
+	}
+	body := composeConfigWrapper{
+		ServiceConfigs: c,
+		VolumeConfigs:  vc,
+		NetworkConfigs: nc,
+		AuthConfigs:    auth,
+	}
+	resp, err := cli.post(context.Background(), "/compose/create", query, body, nil)
+	if err != nil {
+		return nil, err
+	}
+	return resp.body, nil
+}
+
+func (cli *Client) ComposeRm(project string, services []string, rmVol bool) (io.ReadCloser, error) {
+	query := url.Values{}
+	query.Set("project", project)
+	if rmVol {
+		query.Set("rmvol", "true")
+	}
+	if len(services) > 0 {
+		query.Set("services", strings.Join(services, "}{"))
+	}
+	resp, err := cli.post(context.Background(), "/compose/rm", query, nil, nil)
+	if err != nil {
+		return nil, err
+	}
+	return resp.body, nil
+}
+
+func (cli *Client) ComposeStart(project string, services []string) (io.ReadCloser, error) {
+	query := url.Values{}
+	query.Set("project", project)
+	if len(services) > 0 {
+		query.Set("services", strings.Join(services, "}{"))
+	}
+	resp, err := cli.post(context.Background(), "/compose/start", query, nil, nil)
+	if err != nil {
+		return nil, err
+	}
+	return resp.body, nil
+}
+
+func (cli *Client) ComposeStop(project string, services []string, timeout int) (io.ReadCloser, error) {
+	query := url.Values{}
+	query.Set("project", project)
+	query.Set("seconds", fmt.Sprintf("%d", timeout))
+	if len(services) > 0 {
+		query.Set("services", strings.Join(services, "}{"))
+	}
+	resp, err := cli.post(context.Background(), "/compose/stop", query, nil, nil)
+	if err != nil {
+		return nil, err
+	}
+	return resp.body, nil
+}
+
+func (cli *Client) ComposeKill(project string, services []string, signal string) (io.ReadCloser, error) {
+	query := url.Values{}
+	query.Set("project", project)
+	query.Set("signal", signal)
+	if len(services) > 0 {
+		query.Set("services", strings.Join(services, "}{"))
+	}
+	resp, err := cli.post(context.Background(), "/compose/kill", query, nil, nil)
+	if err != nil {
+		return nil, err
+	}
+	return resp.body, nil
+}