feat: migrate preparing folder to utils

src/main.go

@@ -12,13 +12,15 @@ import (
 	"github.com/gofiber/template/html"
 	"github.com/markbates/pkger"
 	_ "github.com/mattn/go-sqlite3"
-	"os"
-	"path/filepath"
+	"log"
 )
 
 func main() {
-	// preparing folder if single binary
-	preparingFolder("statics/images/qrcode", "statics/images/senditems")
+	// preparing folder if not exist
+	err := utils.CreateFolder("statics/images/qrcode", "statics/images/senditems")
+	if err != nil {
+		log.Fatalln(err)
+	}
 
 	engine := html.NewFileSystem(pkger.Dir("/views"), ".html")
 	app := fiber.New(fiber.Config{
@@ -54,18 +56,8 @@ func main() {
 		return ctx.Render("index", fiber.Map{"AppHost": fmt.Sprintf("%s://%s", ctx.Protocol(), ctx.Hostname())})
 	})
 
-	err := app.Listen(":3000")
+	err = app.Listen(":3000")
 	if err != nil {
-		fmt.Println("Failed to start: ", err.Error())
-	}
-}
-
-func preparingFolder(folderPath ...string) {
-	for _, folder := range folderPath {
-		newpath := filepath.Join(".", folder)
-		err := os.MkdirAll(newpath, os.ModePerm)
-		if err != nil {
-			os.Exit(0)
-		}
+		log.Fatalln("Failed to start: ", err.Error())
 	}
 }

src/utils/general.go

@@ -2,9 +2,11 @@ package utils
 
 import (
 	"os"
+	"path/filepath"
 	"time"
 )
 
+// RemoveFile is removing file with delay
 func RemoveFile(delaySecond int, paths ...string) error {
 	if delaySecond > 0 {
 		time.Sleep(time.Duration(delaySecond) * time.Second)
@@ -16,3 +18,13 @@ func RemoveFile(delaySecond int, paths ...string) error {
 	}
 	return nil
 }
+
+// CreateFolder create new folder and sub folder if not exist
+func CreateFolder(folderPath ...string) error {
+	for _, folder := range folderPath {
+		newFolder := filepath.Join(".", folder)
+		err := os.MkdirAll(newFolder, os.ModePerm)
+		return err
+	}
+	return nil
+}