1 files changed, 16 insertions, 3 deletions

diff --git a/main.go b/main.go
index 226e3d6..88d21c1 100644
--- a/main.go
+++ b/main.go
@@ -5,6 +5,7 @@ import (
         "fmt"
         "html/template"
         "log"
+        "net/http"
         "os"
         "path"
         "path/filepath"
@@ -91,6 +92,13 @@ func cleanHTMLTags(htmlString string) string {
         return cleanString
 }
 
+func simpleServer(projectRoot string) {
+        fs := http.FileServer(http.Dir(path.Join(projectRoot, "public")))
+        http.Handle("/", fs)
+        log.Println("Server started on http://localhost:6969")
+        log.Fatal(http.ListenAndServe(":6969", nil))
+}
+
 func initializeProject(projectRoot string) {
         log.Println("Initializing new project")
 
@@ -353,13 +361,14 @@ func main() {
         }
 
         var args struct {
-                Init  bool `arg:"-i,--init" help:"initialize new project"`
-                Build bool `arg:"-b,--build" help:"build the website"`
+                Init   bool `arg:"-i,--init" help:"initialize new project"`
+                Build  bool `arg:"-b,--build" help:"build the website"`
+                Server bool `arg:"-s,--server" help:"simple embedded HTTP server"`
         }
 
         arg.MustParse(&args)
 
-        if !args.Init && !args.Build {
+        if !args.Init && !args.Build && !args.Server {
                 fmt.Println("No arguments provided. Try using `jbmafp --help`")
                 os.Exit(0)
         }
@@ -371,4 +380,8 @@ func main() {
         if args.Build {
                 buildProject(projectRoot)
         }
+
+        if args.Server {
+                simpleServer(projectRoot)
+        }
 }