148 lines
3.2 KiB
Go
148 lines
3.2 KiB
Go
package certmagic
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/gob"
|
|
"errors"
|
|
"fmt"
|
|
"path"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/mholt/certmagic"
|
|
"github.com/micro/go-micro/v2/store"
|
|
"github.com/micro/go-micro/v2/sync/lock"
|
|
)
|
|
|
|
// File represents a "File" that will be stored in store.Store - the contents and last modified time
|
|
type File struct {
|
|
// last modified time
|
|
LastModified time.Time
|
|
// Contents
|
|
Contents []byte
|
|
}
|
|
|
|
// storage is an implementation of certmagic.Storage using micro's sync.Map and store.Store interfaces.
|
|
// As certmagic storage expects a filesystem (with stat() abilities) we have to implement
|
|
// the bare minimum of metadata.
|
|
type storage struct {
|
|
lock lock.Lock
|
|
store store.Store
|
|
}
|
|
|
|
func (s *storage) Lock(key string) error {
|
|
return s.lock.Acquire(key, lock.TTL(10*time.Minute))
|
|
}
|
|
|
|
func (s *storage) Unlock(key string) error {
|
|
return s.lock.Release(key)
|
|
}
|
|
|
|
func (s *storage) Store(key string, value []byte) error {
|
|
f := File{
|
|
LastModified: time.Now(),
|
|
Contents: value,
|
|
}
|
|
buf := &bytes.Buffer{}
|
|
e := gob.NewEncoder(buf)
|
|
if err := e.Encode(f); err != nil {
|
|
return err
|
|
}
|
|
r := &store.Record{
|
|
Key: key,
|
|
Value: buf.Bytes(),
|
|
}
|
|
return s.store.Write(r)
|
|
}
|
|
|
|
func (s *storage) Load(key string) ([]byte, error) {
|
|
if !s.Exists(key) {
|
|
return nil, certmagic.ErrNotExist(errors.New(key + " doesn't exist"))
|
|
}
|
|
records, err := s.store.Read(key)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(records) != 1 {
|
|
return nil, fmt.Errorf("ACME Storage: multiple records matched key %s", key)
|
|
}
|
|
b := bytes.NewBuffer(records[0].Value)
|
|
d := gob.NewDecoder(b)
|
|
var f File
|
|
err = d.Decode(&f)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return f.Contents, nil
|
|
}
|
|
|
|
func (s *storage) Delete(key string) error {
|
|
return s.store.Delete(key)
|
|
}
|
|
|
|
func (s *storage) Exists(key string) bool {
|
|
if _, err := s.store.Read(key); err != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (s *storage) List(prefix string, recursive bool) ([]string, error) {
|
|
records, err := s.store.List()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
//nolint:prealloc
|
|
var results []string
|
|
for _, r := range records {
|
|
if strings.HasPrefix(r.Key, prefix) {
|
|
results = append(results, r.Key)
|
|
}
|
|
}
|
|
if recursive {
|
|
return results, nil
|
|
}
|
|
keysMap := make(map[string]bool)
|
|
for _, key := range results {
|
|
dir := strings.Split(strings.TrimPrefix(key, prefix+"/"), "/")
|
|
keysMap[dir[0]] = true
|
|
}
|
|
results = make([]string, 0)
|
|
for k := range keysMap {
|
|
results = append(results, path.Join(prefix, k))
|
|
}
|
|
return results, nil
|
|
}
|
|
|
|
func (s *storage) Stat(key string) (certmagic.KeyInfo, error) {
|
|
records, err := s.store.Read(key)
|
|
if err != nil {
|
|
return certmagic.KeyInfo{}, err
|
|
}
|
|
if len(records) != 1 {
|
|
return certmagic.KeyInfo{}, fmt.Errorf("ACME Storage: multiple records matched key %s", key)
|
|
}
|
|
b := bytes.NewBuffer(records[0].Value)
|
|
d := gob.NewDecoder(b)
|
|
var f File
|
|
err = d.Decode(&f)
|
|
if err != nil {
|
|
return certmagic.KeyInfo{}, err
|
|
}
|
|
return certmagic.KeyInfo{
|
|
Key: key,
|
|
Modified: f.LastModified,
|
|
Size: int64(len(f.Contents)),
|
|
IsTerminal: false,
|
|
}, nil
|
|
}
|
|
|
|
// NewStorage returns a certmagic.Storage backed by a go-micro/lock and go-micro/store
|
|
func NewStorage(lock lock.Lock, store store.Store) certmagic.Storage {
|
|
return &storage{
|
|
lock: lock,
|
|
store: store,
|
|
}
|
|
}
|