Download functionality

1 files changed, 41 insertions, 7 deletions

diff --git a/src/http/http.go b/src/http/http.go
index 76ca62d..824c7a3 100644
--- a/src/http/http.go
+++ b/src/http/http.go
@@ -22,6 +22,8 @@ func Server() {
 	fs := http.FileServer(http.Dir("./hosted/static"))
 	static.PathPrefix("/").Handler(http.StripPrefix("/static/", fs))
 
+	r.HandleFunc("/download", downloadHandler).Methods("GET")
+
 	t := r.PathPrefix("/").Subrouter()
 	t.PathPrefix("/").HandlerFunc(pathHandler)
 
@@ -39,17 +41,49 @@ func Server() {
 	logrus.Fatal(httpServer.ListenAndServe())
 }
 
+func downloadHandler(w http.ResponseWriter, r *http.Request) {
+	query := r.URL.Query()
+
+	file := query["file"][0]
+
+	logrus.Info(file)
+
+	root := viper.GetString("server.root")
+	logrus.Info(root)
+	strippedFile := strings.Replace(file, root, "", -1)
+	strippedFile = strings.Replace(strippedFile, "..", "", -1)
+
+	logrus.Info(strippedFile)
+	logrus.Infof("stripped: %s", strippedFile)
+
+	w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s", strippedFile))
+	w.Header().Set("Content-Type", r.Header.Get("Content-Type"))
+
+	actualFile := fmt.Sprintf("%s/%s", root, strippedFile)
+	logrus.Infof("actual: %s", actualFile)
+	http.ServeFile(w, r, actualFile)
+}
+
 func pathHandler(w http.ResponseWriter, r *http.Request) {
 	var content map[string]interface{}
 	content = make(map[string]interface{})
 
-	breadcrumbsList := breadcrumbs(r)
-	content["Breadcrumbs"] = breadcrumbsList
-
 	root := viper.GetString("server.root")
 	requestURI := fmt.Sprintf("%s%s", root, r.RequestURI)
-	logrus.Infof("requestURI: %s", requestURI)
-	logrus.Infof("r.RequestURI: %s", r.RequestURI)
+
+	query := r.URL.Query()
+
+	if query["download"] != nil {
+
+		strippedFile := strings.Replace(requestURI, root, "", -1)
+		strippedFile = strings.Replace(strippedFile, "?download", "", -1)
+		path := fmt.Sprintf("/download?file=%s", strippedFile)
+		http.Redirect(w, r, path, http.StatusSeeOther)
+		return
+	}
+
+	breadcrumbsList := breadcrumbs(r)
+	content["Breadcrumbs"] = breadcrumbsList
 
 	// get all files in the request dir
 	files, err := ioutil.ReadDir(requestURI)
@@ -95,6 +129,8 @@ func pathHandler(w http.ResponseWriter, r *http.Request) {
 		if f.IsDir() == true {
 			item.IsDir = true
 			dirCount++
+		} else {
+			item.Download = true
 		}
 
 		items = append(items, item)
@@ -143,8 +179,6 @@ func pathHandler(w http.ResponseWriter, r *http.Request) {
 
 // breadcrumbs get's the breadcrumbs from the request
 func breadcrumbs(r *http.Request) structs.Breadcrumbs {
-	logrus.Debugf("----------------------------------------------------------")
-	logrus.Debugf("%s\n", r.RequestURI)
 
 	request := r.RequestURI