Add Pod API to varlink.

Including: GetPod, StartPod, StopPod, RestartPod, KillPod, PausePod, UnpausePod, CreatePod, RemovePod, and InspectPod

Signed-off-by: haircommander <pehunt@redhat.com>

Closes: #1275
Approved by: mheon

2 files changed, 271 insertions, 0 deletions

diff --git a/pkg/varlinkapi/pods.go b/pkg/varlinkapi/pods.go
new file mode 100644
index 000000000..272c348fc
--- /dev/null
+++ b/pkg/varlinkapi/pods.go
@@ -0,0 +1,231 @@
+package varlinkapi
+
+import (
+	"encoding/json"
+	"syscall"
+
+	"github.com/projectatomic/libpod/cmd/podman/batchcontainer"
+	"github.com/projectatomic/libpod/cmd/podman/varlink"
+	"github.com/projectatomic/libpod/libpod"
+)
+
+// CreatePod ...
+func (i *LibpodAPI) CreatePod(call iopodman.VarlinkCall, name, cgroupParent string, labels map[string]string) error {
+	var options []libpod.PodCreateOption
+	if cgroupParent != "" {
+		options = append(options, libpod.WithPodCgroupParent(cgroupParent))
+	}
+	if len(labels) > 0 {
+		options = append(options, libpod.WithPodLabels(labels))
+	}
+	if name != "" {
+		options = append(options, libpod.WithPodName(name))
+	}
+	options = append(options, libpod.WithPodCgroups())
+
+	pod, err := i.Runtime.NewPod(options...)
+	if err != nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	return call.ReplyCreatePod(pod.ID())
+}
+
+// ListPods ...
+func (i *LibpodAPI) ListPods(call iopodman.VarlinkCall) error {
+	var (
+		listPods []iopodman.ListPodData
+	)
+
+	pods, err := i.Runtime.GetAllPods()
+	if err != nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	opts := batchcontainer.PsOptions{}
+	for _, pod := range pods {
+		listPod, err := makeListPod(pod, opts)
+		if err != nil {
+			return call.ReplyErrorOccurred(err.Error())
+		}
+		listPods = append(listPods, listPod)
+	}
+	return call.ReplyListPods(listPods)
+}
+
+// GetPod ...
+func (i *LibpodAPI) GetPod(call iopodman.VarlinkCall, name string) error {
+	pod, err := i.Runtime.LookupPod(name)
+	if err != nil {
+		return call.ReplyPodNotFound(name)
+	}
+	opts := batchcontainer.PsOptions{}
+
+	listPod, err := makeListPod(pod, opts)
+	if err != nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+
+	return call.ReplyGetPod(listPod)
+}
+
+// InspectPod ...
+func (i *LibpodAPI) InspectPod(call iopodman.VarlinkCall, name string) error {
+	pod, err := i.Runtime.LookupPod(name)
+	if err != nil {
+		return call.ReplyPodNotFound(name)
+	}
+	inspectData, err := pod.Inspect()
+	if err != nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	b, err := json.Marshal(&inspectData)
+	if err != nil {
+		return call.ReplyErrorOccurred("unable to serialize")
+	}
+	return call.ReplyInspectPod(string(b))
+}
+
+// StartPod ...
+func (i *LibpodAPI) StartPod(call iopodman.VarlinkCall, name string) error {
+	pod, err := i.Runtime.LookupPod(name)
+	if err != nil {
+		return call.ReplyPodNotFound(name)
+	}
+	ctrErrs, err := pod.Start(getContext())
+	if err != nil && ctrErrs == nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	if ctrErrs != nil {
+		containerErrs := make([]string, len(ctrErrs))
+		for ctr := range ctrErrs {
+			containerErrs = append(containerErrs, ctr)
+		}
+		return call.ReplyPodContainerError(pod.ID(), containerErrs)
+	}
+
+	return call.ReplyStartPod(pod.ID())
+}
+
+// StopPod ...
+func (i *LibpodAPI) StopPod(call iopodman.VarlinkCall, name string) error {
+	pod, err := i.Runtime.LookupPod(name)
+	if err != nil {
+		return call.ReplyPodNotFound(name)
+	}
+	ctrErrs, err := pod.Stop(true)
+	if err != nil && ctrErrs == nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	if ctrErrs != nil {
+		containerErrs := make([]string, len(ctrErrs))
+		for ctr := range ctrErrs {
+			containerErrs = append(containerErrs, ctr)
+		}
+		return call.ReplyPodContainerError(pod.ID(), containerErrs)
+	}
+
+	return call.ReplyStopPod(pod.ID())
+}
+
+// RestartPod ...
+func (i *LibpodAPI) RestartPod(call iopodman.VarlinkCall, name string) error {
+	pod, err := i.Runtime.LookupPod(name)
+	if err != nil {
+		return call.ReplyPodNotFound(name)
+	}
+	ctrErrs, err := pod.Restart(getContext())
+	if err != nil && ctrErrs == nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	if ctrErrs != nil {
+		containerErrs := make([]string, len(ctrErrs))
+		for ctr := range ctrErrs {
+			containerErrs = append(containerErrs, ctr)
+		}
+		return call.ReplyPodContainerError(pod.ID(), containerErrs)
+	}
+
+	return call.ReplyRestartPod(pod.ID())
+}
+
+// KillPod kills the running containers in a pod.  If you want to use the default SIGTERM signal,
+// just send a -1 for the signal arg.
+func (i *LibpodAPI) KillPod(call iopodman.VarlinkCall, name string, signal int64) error {
+	killSignal := uint(syscall.SIGTERM)
+	if signal != -1 {
+		killSignal = uint(signal)
+	}
+
+	pod, err := i.Runtime.LookupPod(name)
+	if err != nil {
+		return call.ReplyPodNotFound(name)
+	}
+	ctrErrs, err := pod.Kill(killSignal)
+	if err != nil && ctrErrs == nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	if ctrErrs != nil {
+		containerErrs := make([]string, len(ctrErrs))
+		for ctr := range ctrErrs {
+			containerErrs = append(containerErrs, ctr)
+		}
+		return call.ReplyPodContainerError(pod.ID(), containerErrs)
+	}
+
+	return call.ReplyKillPod(pod.ID())
+}
+
+// PausePod ...
+func (i *LibpodAPI) PausePod(call iopodman.VarlinkCall, name string) error {
+	pod, err := i.Runtime.LookupPod(name)
+	if err != nil {
+		return call.ReplyPodNotFound(name)
+	}
+	ctrErrs, err := pod.Pause()
+	if err != nil && ctrErrs == nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	if ctrErrs != nil {
+		containerErrs := make([]string, len(ctrErrs))
+		for ctr := range ctrErrs {
+			containerErrs = append(containerErrs, ctr)
+		}
+		return call.ReplyPodContainerError(pod.ID(), containerErrs)
+	}
+
+	return call.ReplyPausePod(pod.ID())
+}
+
+// UnpausePod ...
+func (i *LibpodAPI) UnpausePod(call iopodman.VarlinkCall, name string) error {
+	pod, err := i.Runtime.LookupPod(name)
+	if err != nil {
+		return call.ReplyPodNotFound(name)
+	}
+	ctrErrs, err := pod.Unpause()
+	if err != nil && ctrErrs == nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+	if ctrErrs != nil {
+		containerErrs := make([]string, len(ctrErrs))
+		for ctr := range ctrErrs {
+			containerErrs = append(containerErrs, ctr)
+		}
+		return call.ReplyPodContainerError(pod.ID(), containerErrs)
+	}
+
+	return call.ReplyUnpausePod(pod.ID())
+}
+
+// RemovePod ...
+func (i *LibpodAPI) RemovePod(call iopodman.VarlinkCall, name string, force bool) error {
+	ctx := getContext()
+	pod, err := i.Runtime.LookupPod(name)
+	if err != nil {
+		return call.ReplyPodNotFound(name)
+	}
+	if err = i.Runtime.RemovePod(ctx, pod, force, force); err != nil {
+		return call.ReplyErrorOccurred(err.Error())
+	}
+
+	return call.ReplyRemovePod(pod.ID())
+}
diff --git a/pkg/varlinkapi/util.go b/pkg/varlinkapi/util.go
index b33e43f96..74afabd0f 100644
--- a/pkg/varlinkapi/util.go
+++ b/pkg/varlinkapi/util.go
@@ -77,3 +77,43 @@ func makeListContainer(containerID string, batchInfo batchcontainer.BatchContain
 	}
 	return lc
 }
+
+func makeListPodContainers(containerID string, batchInfo batchcontainer.BatchContainerStruct) iopodman.ListPodContainerInfo {
+	lc := iopodman.ListPodContainerInfo{
+		Id:     containerID,
+		Status: batchInfo.ConState.String(),
+		Name:   batchInfo.ConConfig.Name,
+	}
+	return lc
+}
+
+func makeListPod(pod *libpod.Pod, batchInfo batchcontainer.PsOptions) (iopodman.ListPodData, error) {
+	var listPodsContainers []iopodman.ListPodContainerInfo
+	var errPodData = iopodman.ListPodData{}
+	status, err := pod.Status()
+	if err != nil {
+		return errPodData, err
+	}
+	containers, err := pod.AllContainers()
+	if err != nil {
+		return errPodData, err
+	}
+	for _, ctr := range containers {
+		batchInfo, err := batchcontainer.BatchContainerOp(ctr, batchInfo)
+		if err != nil {
+			return errPodData, err
+		}
+
+		listPodsContainers = append(listPodsContainers, makeListPodContainers(ctr.ID(), batchInfo))
+	}
+	listPod := iopodman.ListPodData{
+		Createdat:          pod.CreatedTime().String(),
+		Id:                 pod.ID(),
+		Name:               pod.Name(),
+		Status:             status,
+		Cgroup:             pod.CgroupParent(),
+		Numberofcontainers: strconv.Itoa(len(listPodsContainers)),
+		Containersinfo:     listPodsContainers,
+	}
+	return listPod, nil
+}