From dfe0579a2f2206d30ea6040c23c4343706b782dc Mon Sep 17 00:00:00 2001
From: Valentin Rothberg <rothberg@redhat.com>
Date: Tue, 19 May 2020 13:42:46 +0200
Subject: add go-bindings for `hack/podman-registry`

Signed-off-by: Valentin Rothberg <rothberg@redhat.com>
---
 hack/podman-registry-go/registry.go      | 98 ++++++++++++++++++++++++++++++++
 hack/podman-registry-go/registry_test.go | 40 +++++++++++++
 2 files changed, 138 insertions(+)
 create mode 100644 hack/podman-registry-go/registry.go
 create mode 100644 hack/podman-registry-go/registry_test.go

diff --git a/hack/podman-registry-go/registry.go b/hack/podman-registry-go/registry.go
new file mode 100644
index 000000000..a83304914
--- /dev/null
+++ b/hack/podman-registry-go/registry.go
@@ -0,0 +1,98 @@
+package registry
+
+import (
+	"strings"
+
+	"github.com/containers/libpod/utils"
+	"github.com/pkg/errors"
+	"github.com/sirupsen/logrus"
+)
+
+const (
+	imageKey = "PODMAN_REGISTRY_IMAGE"
+	userKey  = "PODMAN_REGISTRY_USER"
+	passKey  = "PODMAN_REGISTRY_PASS"
+	portKey  = "PODMAN_REGISTRY_PORT"
+)
+
+var binary = "podman-registry"
+
+// Registry is locally running registry.
+type Registry struct {
+	// Image - container image of the registry.
+	Image string
+	// User - the user to authenticate against the registry.
+	User string
+	// Password - the accompanying password for the user.
+	Password string
+	// Port - the port the registry is listening to on the host.
+	Port string
+	// running indicates if the registry is running.
+	running bool
+}
+
+// Start a new registry and return it along with it's image, user, password, and port.
+func Start() (*Registry, error) {
+	// Start a registry.
+	out, err := utils.ExecCmd(binary, "start")
+	if err != nil {
+		return nil, errors.Wrapf(err, "error running %q: %s", binary, out)
+	}
+
+	// Parse the output.
+	registry := Registry{}
+	for _, s := range strings.Split(out, "\n") {
+		if s == "" {
+			continue
+		}
+		spl := strings.Split(s, "=")
+		if len(spl) != 2 {
+			return nil, errors.Errorf("unexpected output format %q: want 'PODMAN_...=...'", s)
+		}
+		key := spl[0]
+		val := strings.TrimSuffix(strings.TrimPrefix(spl[1], "\""), "\"")
+		switch key {
+		case imageKey:
+			registry.Image = val
+		case userKey:
+			registry.User = val
+		case passKey:
+			registry.Password = val
+		case portKey:
+			registry.Port = val
+		default:
+			logrus.Errorf("unexpected podman-registry output: %q", s)
+		}
+	}
+
+	// Extra sanity check.
+	if registry.Image == "" {
+		return nil, errors.Errorf("unexpected output %q: %q missing", out, imageKey)
+	}
+	if registry.User == "" {
+		return nil, errors.Errorf("unexpected output %q: %q missing", out, userKey)
+	}
+	if registry.Password == "" {
+		return nil, errors.Errorf("unexpected output %q: %q missing", out, passKey)
+	}
+	if registry.Port == "" {
+		return nil, errors.Errorf("unexpected output %q: %q missing", out, portKey)
+	}
+
+	registry.running = true
+
+	return &registry, nil
+}
+
+// Stop the registry.
+func (r *Registry) Stop() error {
+	// Stop a registry.
+	if !r.running {
+		return nil
+	}
+	if _, err := utils.ExecCmd(binary, "-P", r.Port, "stop"); err != nil {
+		return errors.Wrapf(err, "error stopping registry (%v) with %q", *r, binary)
+	}
+	r.running = false
+	return nil
+}
diff --git a/hack/podman-registry-go/registry_test.go b/hack/podman-registry-go/registry_test.go
new file mode 100644
index 000000000..4e4bf5fe2
--- /dev/null
+++ b/hack/podman-registry-go/registry_test.go
@@ -0,0 +1,40 @@
+package registry
+
+import (
+	"testing"
+
+	"github.com/hashicorp/go-multierror"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestStartAndStopMultipleRegistries(t *testing.T) {
+	binary = "../podman-registry"
+
+	registries := []*Registry{}
+
+	// Start registries.
+	var errors *multierror.Error
+	for i := 0; i < 3; i++ {
+		reg, err := Start()
+		if err != nil {
+			errors = multierror.Append(errors, err)
+			continue
+		}
+		assert.True(t, len(reg.Image) > 0)
+		assert.True(t, len(reg.User) > 0)
+		assert.True(t, len(reg.Password) > 0)
+		assert.True(t, len(reg.Port) > 0)
+		registries = append(registries, reg)
+	}
+
+	// Stop registries.
+	for _, reg := range registries {
+		// Make sure we can stop it properly.
+		errors = multierror.Append(errors, reg.Stop())
+		// Stopping an already stopped registry is fine as well.
+		errors = multierror.Append(errors, reg.Stop())
+	}
+
+	require.NoError(t, errors.ErrorOrNil())
+}
-- 
cgit v1.2.3-54-g00ecf