e2m/dispatch
4
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Implement old storage.Path API

Browse source
This commit is contained in:
Ken-Håvard Lieng 2020-04-20 03:02:15 +02:00
parent 164e071e7f
commit 360bed00f9
11 changed files with 44 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

23
storage/directory.go
View file

@ -3,7 +3,7 @@ package storage
import (
"path/filepath"
"github.com/mitchellh/go-homedir"
homedir "github.com/mitchellh/go-homedir"
)
func DefaultDirectory() string {
@ -11,18 +11,25 @@ func DefaultDirectory() string {
return filepath.Join(home, ".dispatch")
}
type directory string
type directory struct {
dataRoot string
configRoot string
}
func (d directory) Root() string {
return string(d)
func (d directory) DataRoot() string {
return d.dataRoot
}
func (d directory) ConfigRoot() string {
return d.configRoot
}
func (d directory) LetsEncrypt() string {
return filepath.Join(d.Root(), "letsencrypt")
return filepath.Join(d.ConfigRoot(), "letsencrypt")
}
func (d directory) Users() string {
return filepath.Join(d.Root(), "users")
return filepath.Join(d.DataRoot(), "users")
}
func (d directory) User(username string) string {
@ -46,9 +53,9 @@ func (d directory) Key(username string) string {
}
func (d directory) Config() string {
return filepath.Join(d.Root(), "config.toml")
return filepath.Join(d.ConfigRoot(), "config.toml")
}
func (d directory) Database() string {
return filepath.Join(d.Root(), "dispatch.db")
return filepath.Join(d.DataRoot(), "dispatch.db")
}

20
storage/storage.go
View file

@ -5,22 +5,20 @@ import (
"os"
"github.com/khlieng/dispatch/pkg/session"
"github.com/spf13/viper"
)
var DataPath directory
var ConfigPath directory
var Path directory
func Initialize() {
if viper.GetString("dir") != DefaultDirectory() {
DataPath = directory(viper.GetString("dir"))
ConfigPath = directory(viper.GetString("dir"))
func Initialize(root, dataRoot, configRoot string) {
if root != DefaultDirectory() {
Path.dataRoot = root
Path.configRoot = root
} else {
DataPath = directory(viper.GetString("data"))
ConfigPath = directory(viper.GetString("conf"))
Path.dataRoot = dataRoot
Path.configRoot = configRoot
}
os.MkdirAll(DataPath.Root(), 0700)
os.MkdirAll(ConfigPath.Root(), 0700)
os.MkdirAll(Path.DataRoot(), 0700)
os.MkdirAll(Path.ConfigRoot(), 0700)
}
var (

4
storage/user.go
View file

@ -32,7 +32,7 @@ func NewUser(store Store) (*User, error) {
return nil, err
}
err = os.MkdirAll(DataPath.User(user.Username), 0700)
err = os.MkdirAll(Path.User(user.Username), 0700)
if err != nil {
return nil, err
}
@ -70,7 +70,7 @@ func (u *User) Remove() {
if u.messageIndex != nil {
u.messageIndex.Close()
}
os.RemoveAll(DataPath.User(u.Username))
os.RemoveAll(Path.User(u.Username))
}
func (u *User) GetLastIP() []byte {

8
storage/user_cert.go
View file

@ -28,12 +28,12 @@ func (u *User) SetCertificate(certPEM, keyPEM []byte) error {
u.certificate = &cert
u.lock.Unlock()
err = ioutil.WriteFile(ConfigPath.Certificate(u.Username), certPEM, 0600)
err = ioutil.WriteFile(Path.Certificate(u.Username), certPEM, 0600)
if err != nil {
return ErrCouldNotSaveCert
}
err = ioutil.WriteFile(ConfigPath.Key(u.Username), keyPEM, 0600)
err = ioutil.WriteFile(Path.Key(u.Username), keyPEM, 0600)
if err != nil {
return ErrCouldNotSaveCert
}
@ -42,12 +42,12 @@ func (u *User) SetCertificate(certPEM, keyPEM []byte) error {
}
func (u *User) loadCertificate() error {
certPEM, err := ioutil.ReadFile(ConfigPath.Certificate(u.Username))
certPEM, err := ioutil.ReadFile(Path.Certificate(u.Username))
if err != nil {
return err
}
keyPEM, err := ioutil.ReadFile(ConfigPath.Key(u.Username))
keyPEM, err := ioutil.ReadFile(Path.Key(u.Username))
if err != nil {
return err
}

4
storage/user_test.go
View file

@ -19,7 +19,7 @@ func tempdir() string {
}
func TestUser(t *testing.T) {
storage.Initialize(tempdir())
storage.Initialize(tempdir(), "", "")
db, err := boltdb.New(storage.Path.Database())
assert.Nil(t, err)
@ -103,7 +103,7 @@ func TestUser(t *testing.T) {
}
func TestMessages(t *testing.T) {
storage.Initialize(tempdir())
storage.Initialize(tempdir(), "", "")
db, err := boltdb.New(storage.Path.Database())
assert.Nil(t, err)