Initial publication of ZFSSA CSI driver

pkg/service/identity.go

@@ -0,0 +1,83 @@
+/*
+ * Copyright (c) 2021, Oracle and/or its affiliates.
+ * Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
+ */
+
+package service
+
+import (
+	"github.com/oracle/zfssa-csi-driver/pkg/utils"
+	"github.com/oracle/zfssa-csi-driver/pkg/zfssarest"
+	"github.com/container-storage-interface/spec/lib/go/csi"
+	"github.com/golang/protobuf/ptypes/wrappers"
+	"golang.org/x/net/context"
+	"google.golang.org/grpc/codes"
+	grpcStatus "google.golang.org/grpc/status"
+)
+
+func newZFSSAIdentityServer(zd *ZFSSADriver) *csi.IdentityServer {
+	var id csi.IdentityServer = zd
+	return &id
+}
+
+func (zd *ZFSSADriver) GetPluginInfo(ctx context.Context, req *csi.GetPluginInfoRequest) (
+	*csi.GetPluginInfoResponse, error) {
+
+	utils.GetLogIDTY(ctx, 5).Println("GetPluginInfo")
+
+	return &csi.GetPluginInfoResponse{
+		Name:          zd.name,
+		VendorVersion: zd.version,
+	}, nil
+}
+
+func (zd *ZFSSADriver) GetPluginCapabilities(ctx context.Context, req *csi.GetPluginCapabilitiesRequest) (
+	*csi.GetPluginCapabilitiesResponse, error) {
+
+	utils.GetLogIDTY(ctx, 5).Println("GetPluginCapabilities")
+
+	return &csi.GetPluginCapabilitiesResponse{
+		Capabilities: []*csi.PluginCapability{
+			{
+				Type: &csi.PluginCapability_Service_{
+					Service: &csi.PluginCapability_Service{
+						Type: csi.PluginCapability_Service_CONTROLLER_SERVICE,
+					},
+				},
+			},
+			{
+				Type: &csi.PluginCapability_VolumeExpansion_{
+					VolumeExpansion: &csi.PluginCapability_VolumeExpansion{
+						Type: csi.PluginCapability_VolumeExpansion_ONLINE,
+					},
+				},
+			},
+		},
+	}, nil
+}
+
+// This is a readiness probe for the driver, it is for checking if proper drivers are
+// loaded. Typical response to failure is a driver restart.
+//
+func (zd *ZFSSADriver) Probe(ctx context.Context, req *csi.ProbeRequest) (
+	*csi.ProbeResponse, error) {
+
+	utils.GetLogIDTY(ctx, 5).Println("Probe")
+
+	// Check that the appliance is responsive, if it is not, we are on hold
+	user, password, err := zd.getUserLogin(ctx, nil)
+	if err != nil {
+		return nil, grpcStatus.Error(codes.Unauthenticated, "Invalid credentials")
+	}
+	token := zfssarest.LookUpToken(user, password)
+	_, err = zfssarest.GetServices(ctx, token)
+	if err != nil {
+		return &csi.ProbeResponse{
+			Ready: &wrappers.BoolValue{Value: false},
+		}, grpcStatus.Error(codes.FailedPrecondition, "Failure creating token")
+	}
+
+	return &csi.ProbeResponse{
+		Ready: &wrappers.BoolValue{Value: true},
+	}, nil
+}