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

Add option --config and --data to specify where to store the configuration and the data

Browse Source
This commit is contained in:
Pierre-Alain TORET 2020-03-10 23:56:07 +01:00
parent 855f4d3e64
commit ca81475fa5
8 changed files with 33 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
commands/clear.go
View File

@ -13,14 +13,14 @@ var clearCmd = &cobra.Command{
Use: "clear",
Short: "Clear all user data",
Run: func(cmd *cobra.Command, args []string) {
err := os.Remove(storage.Path.Database())
err := os.Remove(storage.DataPath.Database())
if err == nil || os.IsNotExist(err) {
log.Println("Database cleared")
} else {
log.Println(err)
}
err = os.RemoveAll(storage.Path.Users())
err = os.RemoveAll(storage.DataPath.Users())
if err == nil {
log.Println("User data cleared")
} else {

2
commands/config.go
View File

@ -16,7 +16,7 @@ var (
Short: "Edit config file",
Run: func(cmd *cobra.Command, args []string) {
if editor := findEditor(); editor != "" {
process := exec.Command(editor, storage.Path.Config())
process := exec.Command(editor, storage.ConfigPath.Config())
process.Stdin = os.Stdin
process.Stdout = os.Stdout
process.Stderr = os.Stderr

14
commands/dispatch.go
View File

@ -46,18 +46,18 @@ var rootCmd = &cobra.Command{
fmt.Printf(logo, version.Tag, version.Commit, version.Date, runtime.Version())
}
storage.Initialize(viper.GetString("dir"))
storage.Initialize()
initConfig(storage.Path.Config(), viper.GetBool("reset-config"))
initConfig(storage.ConfigPath.Config(), viper.GetBool("reset-config"))
},
Run: func(cmd *cobra.Command, args []string) {
if viper.GetBool("dev") {
log.Println("Running in development mode, access client at http://localhost:3000")
}
log.Println("Storing data at", storage.Path.Root())
log.Println("Storing data at", storage.DataPath.Root())
db, err := boltdb.New(storage.Path.Database())
db, err := boltdb.New(storage.DataPath.Database())
if err != nil {
log.Fatal(err)
}
@ -77,11 +77,11 @@ var rootCmd = &cobra.Command{
dispatch.SessionStore = db
dispatch.GetMessageStore = func(user *storage.User) (storage.MessageStore, error) {
return boltdb.New(storage.Path.Log(user.Username))
return boltdb.New(storage.DataPath.Log(user.Username))
}
dispatch.GetMessageSearchProvider = func(user *storage.User) (storage.MessageSearchProvider, error) {
return bleve.New(storage.Path.Index(user.Username))
return bleve.New(storage.DataPath.Index(user.Username))
}
dispatch.Run()
@ -97,6 +97,8 @@ func init() {
rootCmd.AddCommand(configCmd)
rootCmd.AddCommand(versionCmd)
rootCmd.PersistentFlags().String("data", storage.DefaultDirectory(), "directory to store data in")
rootCmd.PersistentFlags().String("conf", storage.DefaultDirectory(), "directory to store configuration in")
rootCmd.PersistentFlags().String("dir", storage.DefaultDirectory(), "directory to store config and data in")
rootCmd.PersistentFlags().Bool("reset-config", false, "reset to the default configuration, overwriting the current one")
rootCmd.Flags().StringP("address", "a", "", "interface to which the server will bind")

4
config/config.go
View File

@ -3,8 +3,8 @@ package config
import (
"time"
"github.com/fsnotify/fsnotify"
"github.com/khlieng/dispatch/storage"
"github.com/fsnotify/fsnotify"
"github.com/spf13/viper"
)
@ -53,7 +53,7 @@ type LetsEncrypt struct {
func LoadConfig() (*Config, chan *Config) {
viper.SetConfigName("config")
viper.AddConfigPath(storage.Path.Root())
viper.AddConfigPath(storage.ConfigPath.Root())
viper.ReadInConfig()
config := &Config{}

2
server/server.go
View File

@ -147,7 +147,7 @@ func (d *Dispatch) startHTTP() {
PortHTTPS: cfg.HTTPS.Port,
HTTPOnly: !cfg.HTTPS.Enabled,
StoragePath: storage.Path.LetsEncrypt(),
StoragePath: storage.ConfigPath.LetsEncrypt(),
Domain: cfg.LetsEncrypt.Domain,
Email: cfg.LetsEncrypt.Email,

17
storage/storage.go
View File

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

4
storage/user.go
View File

@ -32,7 +32,7 @@ func NewUser(store Store) (*User, error) {
return nil, err
}
err = os.MkdirAll(Path.User(user.Username), 0700)
err = os.MkdirAll(DataPath.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(Path.User(u.Username))
os.RemoveAll(DataPath.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(Path.Certificate(u.Username), certPEM, 0600)
err = ioutil.WriteFile(ConfigPath.Certificate(u.Username), certPEM, 0600)
if err != nil {
return ErrCouldNotSaveCert
}
err = ioutil.WriteFile(Path.Key(u.Username), keyPEM, 0600)
err = ioutil.WriteFile(ConfigPath.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(Path.Certificate(u.Username))
certPEM, err := ioutil.ReadFile(ConfigPath.Certificate(u.Username))
if err != nil {
return err
}
keyPEM, err := ioutil.ReadFile(Path.Key(u.Username))
keyPEM, err := ioutil.ReadFile(ConfigPath.Key(u.Username))
if err != nil {
return err
}