92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
package commands
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
const mainDescription = `Taskcafé is an open soure project management
|
|
system written in Golang & React.`
|
|
|
|
var (
|
|
version = "dev"
|
|
commit = "none"
|
|
date = "unknown"
|
|
)
|
|
|
|
var versionTemplate = fmt.Sprintf(`Version: %s
|
|
Commit: %s
|
|
Built: %s`, version, commit, date+"\n")
|
|
|
|
var cfgFile string
|
|
|
|
var rootCmd = &cobra.Command{
|
|
Use: "taskcafe",
|
|
Long: mainDescription,
|
|
Version: version,
|
|
}
|
|
|
|
var migration http.FileSystem
|
|
|
|
func init() {
|
|
cobra.OnInitialize(initConfig)
|
|
|
|
rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file path")
|
|
migration = http.Dir("./migrations")
|
|
}
|
|
|
|
func initConfig() {
|
|
if cfgFile != "" {
|
|
// Use config file from the flag.
|
|
viper.SetConfigFile(cfgFile)
|
|
} else {
|
|
// Search config in home directory with name ".cobra" (without extension).
|
|
viper.AddConfigPath("./conf")
|
|
viper.AddConfigPath(".")
|
|
viper.AddConfigPath("/etc/taskcafe")
|
|
viper.SetConfigName("taskcafe")
|
|
}
|
|
|
|
viper.SetEnvPrefix("TASKCAFE")
|
|
viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
|
|
viper.AutomaticEnv()
|
|
|
|
err := viper.ReadInConfig()
|
|
if err == nil {
|
|
return
|
|
}
|
|
if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
|
|
panic(err)
|
|
}
|
|
|
|
viper.SetDefault("server.hostname", "0.0.0.0:3333")
|
|
viper.SetDefault("database.host", "127.0.0.1")
|
|
viper.SetDefault("database.name", "taskcafe")
|
|
viper.SetDefault("database.user", "taskcafe")
|
|
viper.SetDefault("database.password", "taskcafe_test")
|
|
|
|
viper.SetDefault("queue.broker", "amqp://guest:guest@localhost:5672/")
|
|
viper.SetDefault("queue.store", "memcache://localhost:11211")
|
|
|
|
}
|
|
|
|
// Execute the root cobra command
|
|
func Execute() {
|
|
viper.SetDefault("server.hostname", "0.0.0.0:3333")
|
|
viper.SetDefault("database.host", "127.0.0.1")
|
|
viper.SetDefault("database.name", "taskcafe")
|
|
viper.SetDefault("database.user", "taskcafe")
|
|
viper.SetDefault("database.password", "taskcafe_test")
|
|
|
|
viper.SetDefault("queue.broker", "amqp://guest:guest@localhost:5672/")
|
|
viper.SetDefault("queue.store", "memcache://localhost:11211")
|
|
|
|
rootCmd.SetVersionTemplate(versionTemplate)
|
|
rootCmd.AddCommand(newWebCmd(), newMigrateCmd(), newTokenCmd(), newWorkerCmd(), newResetPasswordCmd())
|
|
rootCmd.Execute()
|
|
}
|