add gozer new command

dannyvankooten · dannyvankooten · commit fc3dcd6f5aa5 · 2023-11-23T19:48:29.000+01:00
diff --git a/README.md b/README.md
@@ -12,7 +12,6 @@ Sample websites using Gozer:
 - [Simplest possible example](example/)
 - [My personal website](https://github.com/dannyvankooten/www.dannyvankooten.com)
 
-
 ## Directory structure
 
 Gozer expects a certain directory structure in order to build your site correctly.
@@ -36,14 +35,16 @@ Gozer - a fast & simple static site generator
 Usage: gozer [OPTIONS] <COMMAND>
 
 Commands:
-	build	Deletes the output directory if there is one and builds the site
-	serve	Builds the site and starts an HTTP server on http://localhost:8080
+    build   Deletes the output directory if there is one and builds the site
+    serve   Builds the site and starts an HTTP server on http://localhost:8080
+    new             Creates a new site structure in the given directory
 
 Options:
-	-r, --root <ROOT> Directory to use as root of project (default: .)
-	-c, --config <CONFIG> Path to confiruation file (default: config.xml)
+    -r, --root <ROOT> Directory to use as root of project (default: .)
+    -c, --config <CONFIG> Path to confiruation file (default: config.xml)
 ```
 
+Run `gozer new` in an empty directory to quickly create the directory structure for a basic site.
 
 ## Content files
 
diff --git a/main.go b/main.go
@@ -385,14 +385,15 @@ func main() {
 	flag.Parse()
 
 	command := os.Args[len(os.Args)-1]
-	if command != "build" && command != "serve" {
+	if command != "build" && command != "serve" && command != "new" {
 		fmt.Printf(`Gozer - a fast & simple static site generator
 
 Usage: gozer [OPTIONS] <COMMAND>
 
 Commands:
 	build	Deletes the output directory if there is one and builds the site
 	serve	Builds the site and starts an HTTP server on http://localhost:8080
+	new		Creates a new site structure in the given directory
 
 Options:
 	-r, --root <ROOT> Directory to use as root of project (default: .)
@@ -401,10 +402,18 @@ Options:
 		return
 	}
 
+	// ensure rootPath has a trailing slash
 	if rootPath != "" {
 		rootPath = strings.TrimSuffix(rootPath, "/") + "/"
 	}
 
+	if command == "new" {
+		if err := createDirectoryStructure(rootPath); err != nil {
+			log.Fatal("Error creating site structure: ", err)
+		}
+		return
+	}
+
 	buildSite(rootPath, configFile)
 
 	if command == "serve" {
@@ -413,6 +422,44 @@ Options:
 	}
 }
 
+func createDirectoryStructure(rootPath string) error {
+	if err := os.Mkdir(rootPath+"content", 0755); err != nil {
+		return err
+	}
+	if err := os.Mkdir(rootPath+"templates", 0755); err != nil {
+		return err
+	}
+	if err := os.Mkdir(rootPath+"public", 0755); err != nil {
+		return err
+	}
+
+	// create config.xml
+	fh, err := os.Create(rootPath + "config.xml")
+	if err != nil {
+		return err
+	}
+	_, _ = fh.WriteString("<config>\n\t<site_url>http://localhost:8080</site_url>\n</config>")
+	fh.Close()
+
+	// create default template
+	fh, err = os.Create(rootPath + "templates/default.html")
+	if err != nil {
+		return err
+	}
+	_, _ = fh.WriteString("<!DOCTYPE html>\n<head>\n\t<title>{{ .Title }}</title>\n</head>\n<body>\n{{ .Content }}\n</body>\n</html>")
+	fh.Close()
+
+	// create homepage
+	fh, err = os.Create(rootPath + "content/index.html")
+	if err != nil {
+		return err
+	}
+	_, _ = fh.WriteString("+++\ntitle = \"Gozer!\"\n+++\n\nWelcome to my website.\n")
+	fh.Close()
+
+	return nil
+}
+
 func buildSite(rootPath string, configFile string) {
 	var err error
 	timeStart := time.Now()
