feat(config): support every path

1 files changed, 8 insertions, 4 deletions

diff --git a/main.go b/main.go
index 23c3f69..35f8e78 100644
--- a/main.go
+++ b/main.go
@@ -24,10 +24,10 @@ var (
 var (
 	domain       string
 	configPath   string
-	dev          bool = false
+	dev          = false
 	generateToml bool
 	generateJson bool
-	port         int = 80
+	port         = 80
 )
 
 func init() {
@@ -82,11 +82,15 @@ func main() {
 
 	var g *golatt.Golatt
 	if dev {
-		g = golatt.New(golatt.UsableEmbedFS("templates", templates), os.DirFS("public"), os.DirFS("dist"))
+		g = golatt.New(
+			golatt.UsableEmbedFS("templates", templates),
+			os.DirFS(getPath("public")),
+			os.DirFS("dist"),
+		)
 	} else {
 		g = golatt.New(
 			golatt.UsableEmbedFS("templates", templates),
-			os.DirFS("public"),
+			os.DirFS(getPath("public")),
 			golatt.UsableEmbedFS("dist", assets),
 		)
 	}