|
| 1 | +// Copyright 2025 Cofide Limited. |
| 2 | +// SPDX-License-Identifier: Apache-2.0 |
| 3 | + |
| 4 | +package fake |
| 5 | + |
| 6 | +import ( |
| 7 | + "context" |
| 8 | + |
| 9 | + identitysvcpb "github.com/cofide/cofide-api-sdk/gen/go/proto/connect/identity_service/v1alpha1" |
| 10 | + identitypb "github.com/cofide/cofide-api-sdk/gen/go/proto/identity/v1alpha1" |
| 11 | + fakeconnect "github.com/cofide/cofide-api-sdk/pkg/connect/client/fake/connect" |
| 12 | + identityv1alpha1 "github.com/cofide/cofide-api-sdk/pkg/connect/client/identity/v1alpha1" |
| 13 | + "google.golang.org/grpc/codes" |
| 14 | + "google.golang.org/grpc/status" |
| 15 | + "google.golang.org/protobuf/proto" |
| 16 | +) |
| 17 | + |
| 18 | +type fakeIdentityClient struct { |
| 19 | + fake *fakeconnect.FakeConnect |
| 20 | +} |
| 21 | + |
| 22 | +// New instantiates a new IdentityClient for communication with a fake Connect API. |
| 23 | +func New(fake *fakeconnect.FakeConnect) identityv1alpha1.IdentityClient { |
| 24 | + return &fakeIdentityClient{ |
| 25 | + fake: fake, |
| 26 | + } |
| 27 | +} |
| 28 | + |
| 29 | +func (c *fakeIdentityClient) GetIdentity(ctx context.Context, identityID string) (*identitypb.Identity, error) { |
| 30 | + c.fake.Mu.Lock() |
| 31 | + defer c.fake.Mu.Unlock() |
| 32 | + |
| 33 | + identity, ok := c.fake.Identities[identityID] |
| 34 | + if !ok { |
| 35 | + return nil, status.Error(codes.NotFound, "identity not found") |
| 36 | + } |
| 37 | + return clone(identity), nil |
| 38 | +} |
| 39 | + |
| 40 | +func (c *fakeIdentityClient) ListIdentities(ctx context.Context, filter *identitysvcpb.ListIdentitiesRequest_Filter) ([]*identitypb.Identity, error) { |
| 41 | + c.fake.Mu.Lock() |
| 42 | + defer c.fake.Mu.Unlock() |
| 43 | + |
| 44 | + identities := []*identitypb.Identity{} |
| 45 | + for _, identity := range c.fake.Identities { |
| 46 | + if identityMatches(identity, filter) { |
| 47 | + identities = append(identities, clone(identity)) |
| 48 | + } |
| 49 | + } |
| 50 | + return identities, nil |
| 51 | +} |
| 52 | + |
| 53 | +func identityMatches(identity *identitypb.Identity, filter *identitysvcpb.ListIdentitiesRequest_Filter) bool { |
| 54 | + if filter == nil { |
| 55 | + return true |
| 56 | + } |
| 57 | + if filter.OrgId != nil && identity.GetOrgId() != *filter.OrgId { |
| 58 | + return false |
| 59 | + } |
| 60 | + if filter.TrustZoneId != nil && identity.GetTrustZoneId() != *filter.TrustZoneId { |
| 61 | + return false |
| 62 | + } |
| 63 | + if filter.ClusterId != nil && identity.GetClusterId() != *filter.ClusterId { |
| 64 | + return false |
| 65 | + } |
| 66 | + if filter.AttestationPolicyId != nil && identity.GetAttestationPolicyId() != *filter.AttestationPolicyId { |
| 67 | + return false |
| 68 | + } |
| 69 | + if filter.ApBindingId != nil && identity.GetApBindingId() != *filter.ApBindingId { |
| 70 | + return false |
| 71 | + } |
| 72 | + if filter.WorkloadId != nil && identity.GetWorkloadId() != *filter.WorkloadId { |
| 73 | + return false |
| 74 | + } |
| 75 | + if filter.SpiffeId != nil && identity.GetSpiffeId() != *filter.SpiffeId { |
| 76 | + return false |
| 77 | + } |
| 78 | + return true |
| 79 | +} |
| 80 | + |
| 81 | +func clone(identity *identitypb.Identity) *identitypb.Identity { |
| 82 | + return proto.Clone(identity).(*identitypb.Identity) |
| 83 | +} |
0 commit comments