monorepo

Co-authored-by: Malte Poll <mp@edgeless.systems>
Co-authored-by: katexochen <katexochen@users.noreply.github.com>
Co-authored-by: Daniel Weiße <dw@edgeless.systems>
Co-authored-by: Thomas Tendyck <tt@edgeless.systems>
Co-authored-by: Benedict Schlueter <bs@edgeless.systems>
Co-authored-by: leongross <leon.gross@rub.de>
Co-authored-by: Moritz Eckert <m1gh7ym0@gmail.com>

cli/proto/respclients.go

@@ -0,0 +1,117 @@
+package proto
+
+import (
+	"encoding/base64"
+	"errors"
+	"fmt"
+	"io"
+
+	"github.com/edgelesssys/constellation/coordinator/pubapi/pubproto"
+	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
+)
+
+// ActivationRespClient has methods to read messages from a stream of
+// ActivateAsCoordinatorResponses. It wraps an AVPN_ActivateAsCoordinatorClient.
+type ActivationRespClient struct {
+	client            pubproto.API_ActivateAsCoordinatorClient
+	kubeconfig        string
+	coordinatorVpnKey string
+	clientVpnIp       string
+	ownerID           string
+	clusterID         string
+}
+
+// NewActivationRespClient creates a new ActivationRespClient with the handed
+// AVPN_ActivateAsCoordinatorClient.
+func NewActivationRespClient(client pubproto.API_ActivateAsCoordinatorClient) *ActivationRespClient {
+	return &ActivationRespClient{
+		client: client,
+	}
+}
+
+// NextLog reads responses from the response stream and returns the
+// first received log.
+func (a *ActivationRespClient) NextLog() (string, error) {
+	for {
+		resp, err := a.client.Recv()
+		if err != nil {
+			return "", err
+		}
+		switch x := resp.Content.(type) {
+		case *pubproto.ActivateAsCoordinatorResponse_Log:
+			return x.Log.Message, nil
+		case *pubproto.ActivateAsCoordinatorResponse_AdminConfig:
+			config := x.AdminConfig
+			a.kubeconfig = string(config.Kubeconfig)
+
+			coordinatorVpnKey, err := wgtypes.NewKey(config.CoordinatorVpnPubKey)
+			if err != nil {
+				return "", err
+			}
+			a.coordinatorVpnKey = coordinatorVpnKey.String()
+			a.clientVpnIp = config.AdminVpnIp
+			a.ownerID = base64.StdEncoding.EncodeToString(config.OwnerId)
+			a.clusterID = base64.StdEncoding.EncodeToString(config.ClusterId)
+		}
+	}
+}
+
+// WriteLogStream reads responses from the response stream and
+// writes log responses to the handed writer.
+func (a *ActivationRespClient) WriteLogStream(w io.Writer) error {
+	log, err := a.NextLog()
+	for err == nil {
+		fmt.Fprintln(w, log)
+		log, err = a.NextLog()
+	}
+	if !errors.Is(err, io.EOF) {
+		return err
+	}
+	return nil
+}
+
+// GetKubeconfig returns the kubeconfig that was received in the
+// latest AdminConfig response or an error if the field is empty.
+func (a *ActivationRespClient) GetKubeconfig() (string, error) {
+	if a.kubeconfig == "" {
+		return "", errors.New("kubeconfig is empty")
+	}
+	return a.kubeconfig, nil
+}
+
+// GetCoordinatorVpnKey returns the Coordinator's VPN key that was
+// received in the latest AdminConfig response or an error if the field
+// is empty.
+func (a *ActivationRespClient) GetCoordinatorVpnKey() (string, error) {
+	if a.coordinatorVpnKey == "" {
+		return "", errors.New("coordinator public VPN key is empty")
+	}
+	return a.coordinatorVpnKey, nil
+}
+
+// GetClientVpnIp returns the client VPN IP that was received
+// in the latest AdminConfig response or an error if the field is empty.
+func (a *ActivationRespClient) GetClientVpnIp() (string, error) {
+	if a.clientVpnIp == "" {
+		return "", errors.New("client VPN IP is empty")
+	}
+	return a.clientVpnIp, nil
+}
+
+// GetOwnerID returns the owner identifier, derived from the client's master secret
+// or an error if the field is empty.
+func (a *ActivationRespClient) GetOwnerID() (string, error) {
+	if a.ownerID == "" {
+		return "", errors.New("secret identifier is empty")
+	}
+	return a.ownerID, nil
+}
+
+// GetClusterID returns the cluster's unique identifier
+// or an error if the field is empty.
+func (a *ActivationRespClient) GetClusterID() (string, error) {
+	if a.clusterID == "" {
+		return "", errors.New("cluster identifier is empty")
+	}
+	return a.clusterID, nil
+}