afsc/gcp/secretmanager/list.go at master · viant/afsc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package secretmanager

import (
	"context"
	"fmt"
	"os"
	"strings"

	"github.com/viant/afs/file"
	"github.com/viant/afs/storage"
	secretmanagerpb "google.golang.org/genproto/googleapis/cloud/secretmanager/v1"
)

// List lists location assets
func (s *Storager) List(ctx context.Context, resourceID string, options ...storage.Option) ([]os.FileInfo, error) {
	resource, err := newResource(resourceID)
	if err != nil {
		return nil, err
	}
	var info []os.FileInfo
	if resource.Secret == "" {
		if err = s.listSecret(ctx, resource, &info); err != nil {
			return nil, err
		}
		return info, nil
	}
	if resource.Version == "" {
		if err = s.listSecretVersions(ctx, resource, &info); err != nil {
			return nil, err
		}
		return info, nil
	}
	return nil, fmt.Errorf("invalid resource: %v", resourceID)
}

func (s *Storager) listSecret(ctx context.Context, resource *Resource, info *[]os.FileInfo) error {
	pageToken := ""
	for {
		request := &secretmanagerpb.ListSecretsRequest{
			Parent:    "projects/" + resource.ProjectID,
			PageToken: pageToken,
		}
		iterator := s.client.ListSecrets(ctx, request)
		for {
			secret, err := iterator.Next()
			if secret == nil {
				break
			}
			match := "/secrets/"
			index := strings.Index(secret.Name, match)
			name := secret.Name[index+len(match):]
			*info = append(*info, file.NewInfo(name, 0, file.DefaultDirOsMode, secret.CreateTime.AsTime(), true, nil))
			if err != nil {
				return err
			}
		}
		pageInfo := iterator.PageInfo()
		if pageInfo != nil {
			pageToken = pageInfo.Token
		}
		if pageToken == "" {
			break
		}
	}
	return nil
}

func (s *Storager) listSecretVersions(ctx context.Context, resource *Resource, info *[]os.FileInfo) error {
	pageToken := ""
	for {
		request := &secretmanagerpb.ListSecretVersionsRequest{
			Parent:    "projects/" + resource.ProjectID + "/secrets/" + resource.Secret,
			PageToken: pageToken,
		}
		iterator := s.client.ListSecretVersions(ctx, request)
		for {
			secret, err := iterator.Next()
			if secret == nil {
				break
			}
			match := "/secrets/" + resource.Secret
			index := strings.Index(secret.Name, match)
			name := secret.Name[index+len(match):]
			*info = append(*info, file.NewInfo(name, 0, file.DefaultDirOsMode, secret.CreateTime.AsTime(), true, nil))
			if err != nil {
				return err
			}
		}
		pageInfo := iterator.PageInfo()
		if pageInfo != nil {
			pageToken = pageInfo.Token
		}
		if pageToken == "" {
			break
		}
	}
	return nil
}