Start work on new DataStore type

src/cmd/tusd/data.go

@@ -35,21 +35,6 @@ func logPath(fileId string) string {
 	return dataPath(fileId) + ".log"
 }
 
-func initFile(fileId string, size int64, contentType string) error {
-	d := dataPath(fileId)
-	file, err := os.OpenFile(d, os.O_CREATE|os.O_WRONLY, 0666)
-	if err != nil {
-		return err
-	}
-	defer file.Close()
-
-	if err := file.Truncate(size); err != nil {
-		return err
-	}
-
-	return nil
-}
-
 func putFileChunk(fileId string, start int64, end int64, r io.Reader) error {
 	d := dataPath(fileId)
 	file, err := os.OpenFile(d, os.O_WRONLY, 0666)

src/cmd/tusd/data_store.go

@@ -0,0 +1,32 @@
+package main
+
+import (
+	"os"
+	"path"
+)
+
+type DataStore struct {
+	dir string
+}
+
+func NewDataStore(dir string) *DataStore {
+	return &DataStore{dir: dir}
+}
+
+// @TODO Add support for Content-Type
+func (s *DataStore) CreateFile(id string, size int64) error {
+	file, err := os.OpenFile(s.filePath(id), os.O_CREATE|os.O_WRONLY, 0666)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	if err := file.Truncate(size); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (s *DataStore) filePath(id string) string {
+	return path.Join(s.dir, id)
+}

src/cmd/tusd/http.go

@@ -5,11 +5,27 @@ import (
 	"io"
 	"log"
 	"net/http"
+	"os"
+	"path"
 	"regexp"
 	"strconv"
 )
 
 var fileRoute = regexp.MustCompile("^/files/([^/]+)$")
+var dataStore *DataStore
+
+func init() {
+	wd, err := os.Getwd()
+	if err != nil {
+		panic(err)
+	}
+
+	dataDir = path.Join(wd, "tus_data")
+	if err := os.MkdirAll(dataDir, 0777); err != nil {
+		panic(err)
+	}
+	dataStore = NewDataStore(dataDir)
+}
 
 func serveHttp() error {
 	http.HandleFunc("/", route)
@@ -67,7 +83,7 @@ func postFiles(w http.ResponseWriter, r *http.Request) {
 	}
 
 	id := uid()
-	if err := initFile(id, contentRange.Size, contentType); err != nil {
+	if err := dataStore.CreateFile(id, contentRange.Size); err != nil {
 		reply(w, http.StatusInternalServerError, err.Error())
 		return
 	}
@@ -81,7 +97,7 @@ func postFiles(w http.ResponseWriter, r *http.Request) {
 	}
 
 	w.Header().Set("Location", "/files/"+id)
-	setFileRangeHeader(w, id)
+	//setFileRangeHeader(w, id)
 	w.WriteHeader(http.StatusCreated)
 }