Add -f to support serving a file

1 files changed, 13 insertions, 0 deletions

diff --git a/main.go b/main.go
index 04a5f4f..e438f5e 100644
--- a/main.go
+++ b/main.go
@@ -4,11 +4,13 @@ import (
 	"log"
 	"net/http"
 	"io"
+	"os"
 	"flag"
 	"fmt"
 )
 
 var rawBody = false
+var returnString string = ""
 
 func handle(w http.ResponseWriter, r *http.Request) {
 	log.Println("URL:", r.URL)
@@ -43,13 +45,24 @@ func handle(w http.ResponseWriter, r *http.Request) {
 	}
 
 	log.Println("--------------------------------------")
+	if returnString != "" {
+		log.Println(returnString)
+		f, err := os.Open(returnString)
+		if err != nil {
+			log.Fatal(err)
+		}
+		defer f.Close()
+		io.Copy(w, f)
+	}
 }
 
 func main() {
 	b := flag.Bool("b", false, "Read raw body")
 	p := flag.Int("p", 8080, "Port to use")
+	f := flag.String("f", "", "File to serve (optional)")
 	flag.Parse()
 	rawBody = *b
+	returnString = *f
 	http.HandleFunc("/", handle)
 	log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", *p), nil))
 }