Mobius 38 - improve config dir flag.

1 files changed, 6 insertions, 1 deletions

diff --git a/cmd/mobius-hotline-server/main.go b/cmd/mobius-hotline-server/main.go
index 07bbc88..2f914c4 100644
--- a/cmd/mobius-hotline-server/main.go
+++ b/cmd/mobius-hotline-server/main.go
@@ -15,6 +15,7 @@ import (
 	"net/http"
 	"os"
 	"runtime"
+	"strings"
 	"time"
 )
 
@@ -55,6 +56,10 @@ func main() {
 	defer func() { _ = l.Sync() }()
 	logger := l.Sugar()
 
+	if !(strings.HasSuffix(*configDir, "/") || strings.HasSuffix(*configDir, "\\")) {
+		*configDir = *configDir + "/"
+	}
+
 	if *init {
 		if _, err := os.Stat(*configDir + "/config.yaml"); os.IsNotExist(err) {
 			if err := os.MkdirAll(*configDir, 0750); err != nil {
@@ -132,7 +137,7 @@ var zapLogLevel = map[string]zapcore.Level{
 func defaultConfigPath() (cfgPath string) {
 	switch runtime.GOOS {
 	case "windows":
-		cfgPath = "config"
+		cfgPath = "config/"
 	case "darwin":
 		if _, err := os.Stat("/usr/local/var/mobius/config/"); err == nil {
 			cfgPath = "/usr/local/var/mobius/config/"