Broke out router into swappable

main.go

@@ -1,17 +1,39 @@
 package main
 
 import (
+	"fmt"
 	"io"
 	"log"
 	"net/http"
 	"os"
+	"strings"
+	"sync"
 
 	"github.com/gorilla/mux"
 )
 
+type routerSwapper struct {
+	mu   sync.Mutex
+	root *mux.Router
+}
+
+func (rs *routerSwapper) Swap(newRouter *mux.Router) {
+	rs.mu.Lock()
+	rs.root = newRouter
+	rs.mu.Unlock()
+}
+
+func (rs *routerSwapper) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	rs.mu.Lock()
+	root := rs.root
+	rs.mu.Unlock()
+	root.ServeHTTP(w, r)
+}
+
 func main() {
 	r := mux.NewRouter()
-
+	rs := routerSwapper{}
+	rs.Swap(r)
 	//TODO Take a look at https://github.com/gorilla/mux/issues/82 to hot-swap the routes in the config file
 
 	// parse config file here
@@ -27,11 +49,27 @@ func main() {
 	//       - css/main.css
 	//       - css/test.css
 
-	r.HandleFunc("/css/styles.css", func(w http.ResponseWriter, r *http.Request) {
+	dir, err := os.Getwd()
-		arr := [2]string{"css/main.css", "css/test.css"}
+	if err != nil {
+		log.Fatal(err)
+	}
+	fmt.Println(dir)
+
+	path := "/css/styles.css"
+	arr := []string{"css/main.css", "css/test.css"}
+	setupRoute(r, path, arr)
+	path = "/html/index.html"
+	arr = []string{"header.html", "index.html"}
+	setupRoute(r, path, arr)
+	var iface http.Handler = &rs
+	http.ListenAndServe(":8080", iface)
+}
+
+func setupRoute(r *mux.Router, path string, concats []string) {
+	r.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
 		var readers []io.Reader
 		var files []*os.File
-		for _, x := range arr {
+		for _, x := range concats {
 			file, err := os.Open(x)
 			if err != nil {
 				//TODO: Proper error handling
@@ -44,10 +82,11 @@ func main() {
 		if _, err := io.Copy(w, cat); err != nil {
 			log.Fatal(err)
 		}
+		log.Println("Request for " + path + " served: " + strings.Join(concats, ", "))
 
 		for _, x := range files {
 			x.Close()
 		}
 	})
-	http.ListenAndServe(":8080", r)
+
 }