Add entity information request to system view

Makefile

@@ -127,6 +127,7 @@ proto:
 	protoc -I helper/forwarding -I vault -I ../../.. vault/*.proto --go_out=plugins=grpc:vault
 	protoc -I helper/storagepacker helper/storagepacker/types.proto --go_out=plugins=grpc:helper/storagepacker
 	protoc -I helper/forwarding -I vault -I ../../.. helper/forwarding/types.proto --go_out=plugins=grpc:helper/forwarding
+	protoc logical/*.proto --go_out=plugins=grpc:$(GOPATH)/src
 	protoc -I physical physical/types.proto --go_out=plugins=grpc:physical
 	protoc -I helper/identity -I ../../.. helper/identity/types.proto --go_out=plugins=grpc:helper/identity
 	protoc  builtin/logical/database/dbplugin/*.proto --go_out=plugins=grpc:.

logical/identity.proto

@@ -0,0 +1,29 @@
+syntax = "proto3";
+
+option go_package = "github.com/hashicorp/vault/logical";
+
+package logical;
+
+message Entity {
+	// ID is the unique identifier for the entity
+	string ID = 1;
+
+	// Name is the human-friendly unique identifier for the entity
+	string name = 2;
+
+	// Aliases contains thhe alias mappings for the given entity
+	repeated Alias aliases = 3;
+}
+
+message Alias {
+	// MountType is the backend mount's type to which this identity belongs
+	string mount_type = 1;
+
+	// MountAccessor is the identifier of the mount entry to which this
+	// identity belongs
+	string mount_accessor = 2;
+
+	// Name is the identifier of this identity in its authentication source
+	string name = 3;
+}
+

logical/plugin/grpc_system.go

@@ -132,6 +132,20 @@ func (s *gRPCSystemViewClient) LocalMount() bool {
 	return reply.Local
 }
 
+func (s *gRPCSystemViewClient) EntityInfo(entityID string) (*logical.Entity, error) {
+	reply, err := s.client.EntityInfo(context.Background(), &pb.EntityInfoArgs{
+		EntityID: entityID,
+	})
+	if err != nil {
+		return nil, err
+	}
+	if reply.Err != "" {
+		return nil, errors.New(reply.Err)
+	}
+
+	return reply.Entity, nil
+}
+
 type gRPCSystemViewServer struct {
 	impl logical.SystemView
 }
@@ -216,3 +230,15 @@ func (s *gRPCSystemViewServer) LocalMount(ctx context.Context, _ *pb.Empty) (*pb
 		Local: local,
 	}, nil
 }
+
+func (s *gRPCSystemViewServer) EntityInfo(ctx context.Context, args *pb.EntityInfoArgs) (*pb.EntityInfoReply, error) {
+	entity, err := s.impl.EntityInfo(args.EntityID)
+	if err != nil {
+		return &pb.EntityInfoReply{
+			Err: pb.ErrToString(err),
+		}, nil
+	}
+	return &pb.EntityInfoReply{
+		Entity: entity,
+	}, nil
+}

logical/plugin/grpc_system_test.go

@@ -164,3 +164,26 @@ func TestSystem_GRPC_mlockEnabled(t *testing.T) {
 		t.Fatalf("expected: %v, got: %v", expected, actual)
 	}
 }
+
+func TestSystem_GRPC_entityInfo(t *testing.T) {
+	sys := logical.TestSystemView()
+	sys.EntityVal = &logical.Entity{
+		ID:   "id",
+		Name: "name",
+	}
+	client, _ := plugin.TestGRPCConn(t, func(s *grpc.Server) {
+		pb.RegisterSystemViewServer(s, &gRPCSystemViewServer{
+			impl: sys,
+		})
+	})
+	defer client.Close()
+	testSystemView := newGRPCSystemView(client)
+
+	actual, err := testSystemView.EntityInfo("")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if sys.EntityVal.ID != actual.ID || sys.EntityVal.Name != actual.Name {
+		t.Fatalf("expected: %v, got: %v", sys.EntityVal, actual)
+	}
+}