mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
96 lines
2.3 KiB
Go
96 lines
2.3 KiB
Go
package s3iam
|
|
|
|
import (
|
|
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
|
|
"github.com/chrislusf/seaweedfs/weed/pb/iam_pb"
|
|
"time"
|
|
|
|
proto "github.com/golang/protobuf/proto"
|
|
)
|
|
|
|
const (
|
|
iamConfigPrefix = "/etc/iam"
|
|
iamIdentityFile = "identity.json"
|
|
)
|
|
|
|
type IAMFilerStore struct {
|
|
client *filer_pb.SeaweedFilerClient
|
|
}
|
|
|
|
func NewIAMFilerStore(client *filer_pb.SeaweedFilerClient) *IAMFilerStore {
|
|
return &IAMFilerStore{client: client}
|
|
}
|
|
|
|
func (ifs *IAMFilerStore) getIAMConfigRequest() *filer_pb.LookupDirectoryEntryRequest {
|
|
return &filer_pb.LookupDirectoryEntryRequest{
|
|
Directory: iamConfigPrefix,
|
|
Name: iamIdentityFile,
|
|
}
|
|
}
|
|
|
|
func (ifs *IAMFilerStore) LoadIAMConfig(config *iam_pb.S3ApiConfiguration) error {
|
|
resp, err := filer_pb.LookupEntry(*ifs.client, ifs.getIAMConfigRequest())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = ifs.loadIAMConfigFromEntry(resp.Entry, config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ifs *IAMFilerStore) SaveIAMConfig(config *iam_pb.S3ApiConfiguration) error {
|
|
entry := &filer_pb.Entry{
|
|
Name: iamIdentityFile,
|
|
IsDirectory: false,
|
|
Attributes: &filer_pb.FuseAttributes{
|
|
Mtime: time.Now().Unix(),
|
|
Crtime: time.Now().Unix(),
|
|
FileMode: uint32(0644),
|
|
Collection: "",
|
|
Replication: "",
|
|
},
|
|
Content: []byte{},
|
|
}
|
|
err := ifs.saveIAMConfigToEntry(entry, config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = filer_pb.LookupEntry(*ifs.client, ifs.getIAMConfigRequest())
|
|
if err == filer_pb.ErrNotFound {
|
|
err = filer_pb.CreateEntry(*ifs.client, &filer_pb.CreateEntryRequest{
|
|
Directory: iamConfigPrefix,
|
|
Entry: entry,
|
|
IsFromOtherCluster: false,
|
|
Signatures: nil,
|
|
})
|
|
} else {
|
|
err = filer_pb.UpdateEntry(*ifs.client, &filer_pb.UpdateEntryRequest{
|
|
Directory: iamConfigPrefix,
|
|
Entry: entry,
|
|
IsFromOtherCluster: false,
|
|
Signatures: nil,
|
|
})
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ifs *IAMFilerStore) loadIAMConfigFromEntry(entry *filer_pb.Entry, config *iam_pb.S3ApiConfiguration) error {
|
|
if err := proto.Unmarshal(entry.Content, config); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ifs *IAMFilerStore) saveIAMConfigToEntry(entry *filer_pb.Entry, config *iam_pb.S3ApiConfiguration) (err error) {
|
|
entry.Content, err = proto.Marshal(config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|