feat: test

lpxxn · Jun 30, 2021 · 268401e · 268401e
1 parent 8e0761f
commit 268401e
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/config/config.go b/config/config.go
@@ -266,6 +266,7 @@ func (a *AppConfig) parseCmdConfig() error {
 	}
 	return nil
 }
+
 func configNotExist() error {
 	return NotExistErr
 }
diff --git a/config/config_file.go b/config/config_file.go
@@ -8,6 +8,7 @@ import (
 	"path"
 
 	"github.com/BurntSushi/toml"
+	"github.com/lpxxn/doraemon/internal"
 	"github.com/skratchdot/open-golang/open"
 )
 
@@ -45,10 +46,8 @@ func ConfDir() string {
 		panic(err)
 	}
 	dirPath := path.Join(rootDir, confDirName)
-	if _, err := os.Stat(dirPath); os.IsNotExist(err) {
-		if err = os.MkdirAll(dirPath, os.ModePerm); err != nil {
-			panic(err)
-		}
+	if err := internal.MakeFolder(dirPath, os.ModePerm); err != nil {
+		panic(err)
 	}
 	return dirPath
 }

diff --git a/internal/utils.go b/internal/utils.go
@@ -2,6 +2,7 @@ package internal
 
 import (
 	"fmt"
+	"os"
 )
 
 type Color string
@@ -67,3 +68,11 @@ func SendMsg(startWithNewLine bool, caption, text string, color Color, endWithNe
 		fmt.Println(startNewLine + BeautifyText(caption, color) + " " + text + endNewLine) // colorized text
 	}
 }
+
+func MakeFolder(folderName string, chmod os.FileMode) error {
+	if _, err := os.Stat(folderName); os.IsNotExist(err) {
+		return os.MkdirAll(folderName, chmod)
+	} else {
+		return err
+	}
+}