Move all sources to package main

This project is not so big that it needs multiple packages.

1 files changed, 31 insertions, 0 deletions

diff --git a/storage.go b/storage.go
new file mode 100644
index 0000000..02bc94d
--- /dev/null
+++ b/storage.go
@@ -0,0 +1,31 @@
+
+package main
+
+import (
+	"log"
+	"net/http"
+	"io/ioutil"
+	"path/filepath"
+)
+
+func ServeStorage(w http.ResponseWriter, r *http.Request, storage string, path string) {
+
+	// generate absolute, cleaned path of ressource
+	path = filepath.Join(storage, path)
+	path,err := filepath.Abs(path)
+	if err != nil {
+		log.Print(err)
+		http.Error(w, http.StatusText(400), 400)
+		return
+	}
+
+	// TODO check if path is still in storage folder
+
+	// serve the file if nothing has been wrong
+	http.ServeFile(w, r, path)
+}
+
+func SaveStorageFile(data *[]byte, storage string, path string) error {
+	fullpath := filepath.Join(storage, path)
+	return ioutil.WriteFile(fullpath, *data, 0644)
+}