Create ACME Provider

provider/acme/local_store.go

@@ -0,0 +1,125 @@
+package acme
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"os"
+
+	"github.com/containous/traefik/log"
+	"github.com/containous/traefik/safe"
+)
+
+var _ Store = (*LocalStore)(nil)
+
+// LocalStore Store implementation for local file
+type LocalStore struct {
+	filename     string
+	storedData   *StoredData
+	SaveDataChan chan *StoredData
+}
+
+// NewLocalStore initializes a new LocalStore with a file name
+func NewLocalStore(filename string) LocalStore {
+	store := LocalStore{filename: filename, SaveDataChan: make(chan *StoredData)}
+	store.listenSaveAction()
+	return store
+}
+
+func (s *LocalStore) get() (*StoredData, error) {
+	if s.storedData == nil {
+		s.storedData = &StoredData{HTTPChallenges: make(map[string]map[string][]byte)}
+
+		f, err := os.Open(s.filename)
+		if err != nil {
+			return nil, err
+		}
+		defer f.Close()
+
+		file, err := ioutil.ReadAll(f)
+		if err != nil {
+			return nil, err
+		}
+
+		if len(file) > 0 {
+			if err := json.Unmarshal(file, s.storedData); err != nil {
+				return nil, err
+			}
+		}
+	}
+
+	return s.storedData, nil
+}
+
+// listenSaveAction listens to a chan to store ACME data in json format into LocalStore.filename
+func (s *LocalStore) listenSaveAction() {
+	safe.Go(func() {
+		for object := range s.SaveDataChan {
+			data, err := json.MarshalIndent(object, "", "  ")
+			if err != nil {
+				log.Error(err)
+			}
+
+			err = ioutil.WriteFile(s.filename, data, 0600)
+			if err != nil {
+				log.Error(err)
+			}
+		}
+	})
+}
+
+// GetAccount returns ACME Account
+func (s *LocalStore) GetAccount() (*Account, error) {
+	storedData, err := s.get()
+	if err != nil {
+		return nil, err
+	}
+
+	return storedData.Account, nil
+}
+
+// SaveAccount stores ACME Account
+func (s *LocalStore) SaveAccount(account *Account) error {
+	storedData, err := s.get()
+	if err != nil {
+		return err
+	}
+
+	storedData.Account = account
+	s.SaveDataChan <- storedData
+
+	return nil
+}
+
+// GetCertificates returns ACME Certificates list
+func (s *LocalStore) GetCertificates() ([]*Certificate, error) {
+	storedData, err := s.get()
+	if err != nil {
+		return nil, err
+	}
+
+	return storedData.Certificates, nil
+}
+
+// SaveCertificates stores ACME Certificates list
+func (s *LocalStore) SaveCertificates(certificates []*Certificate) error {
+	storedData, err := s.get()
+	if err != nil {
+		return err
+	}
+
+	storedData.Certificates = certificates
+	s.SaveDataChan <- storedData
+
+	return nil
+}
+
+// GetHTTPChallenges returns ACME HTTP Challenges list
+func (s *LocalStore) GetHTTPChallenges() (map[string]map[string][]byte, error) {
+	return s.storedData.HTTPChallenges, nil
+}
+
+// SaveHTTPChallenges stores ACME HTTP Challenges list
+func (s *LocalStore) SaveHTTPChallenges(httpChallenges map[string]map[string][]byte) error {
+	s.storedData.HTTPChallenges = httpChallenges
+	return nil
+}