From d51c8d37bd1ddee1a11d6f39bb18a212862b8883 Mon Sep 17 00:00:00 2001
From: Chris Lu <chris.lu@gmail.com>
Date: Tue, 6 Aug 2013 12:13:10 -0700
Subject: [PATCH] Issue 6:	upload file without id specified

---
 go/weed/upload.go  |  6 +++---
 go/weed/version.go |  2 +-
 go/weed/volume.go  | 52 ++++++++++++++++++++++++++++++++++++++++------
 go/weed/weed.go    |  5 +++++
 4 files changed, 55 insertions(+), 10 deletions(-)

diff --git a/go/weed/upload.go b/go/weed/upload.go
index 67bb9adea..54c566ca7 100644
--- a/go/weed/upload.go
+++ b/go/weed/upload.go
@@ -58,13 +58,13 @@ type AssignResult struct {
 	Error     string `json:"error"`
 }
 
-func assign(count int) (*AssignResult, error) {
+func Assign(server string, count int) (*AssignResult, error) {
 	values := make(url.Values)
 	values.Add("count", strconv.Itoa(count))
 	if *uploadReplication != "" {
 		values.Add("replication", *uploadReplication)
 	}
-	jsonBlob, err := util.Post("http://"+*server+"/dir/assign", values)
+	jsonBlob, err := util.Post("http://"+server+"/dir/assign", values)
 	debug("assign result :", string(jsonBlob))
 	if err != nil {
 		return nil, err
@@ -118,7 +118,7 @@ func submit(files []string) ([]SubmitResult, error) {
 	for index, file := range files {
 		results[index].FileName = file
 	}
-	ret, err := assign(len(files))
+	ret, err := Assign(*server, len(files))
 	if err != nil {
 		for index, _ := range files {
 			results[index].Error = err.Error()
diff --git a/go/weed/version.go b/go/weed/version.go
index b297253dc..769149a06 100644
--- a/go/weed/version.go
+++ b/go/weed/version.go
@@ -6,7 +6,7 @@ import (
 )
 
 const (
-	VERSION = "0.37"
+	VERSION = "0.38"
 )
 
 var cmdVersion = &Command{
diff --git a/go/weed/volume.go b/go/weed/volume.go
index 01beb024a..d699fdbed 100644
--- a/go/weed/volume.go
+++ b/go/weed/volume.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"bytes"
 	"code.google.com/p/weed-fs/go/operation"
 	"code.google.com/p/weed-fs/go/replication"
 	"code.google.com/p/weed-fs/go/storage"
@@ -99,6 +100,47 @@ func freezeVolumeHandler(w http.ResponseWriter, r *http.Request) {
 	}
 	debug("freeze volume =", r.FormValue("volume"), ", error =", err)
 }
+func submitForClientHandler(w http.ResponseWriter, r *http.Request) {
+	m := make(map[string]interface{})
+	if r.Method != "POST" {
+		m["error"] = "Only submit via POST!"
+		writeJsonQuiet(w, r, m)
+		return
+	}
+
+	debug("parsing upload file...")
+	fname, data, mimeType, isGzipped, lastModified, pe := storage.ParseUpload(r)
+	if pe != nil {
+		writeJsonError(w, r, pe)
+		return
+	}
+
+	debug("assigning file id for", fname)
+	assignResult, ae := Assign(*masterNode, 1)
+	if ae != nil {
+		writeJsonError(w, r, ae)
+		return
+	}
+
+	url := "http://" + assignResult.PublicUrl + "/" + assignResult.Fid
+	if lastModified != 0 {
+		url = url + "?ts=" + strconv.FormatUint(lastModified, 10)
+	}
+
+	debug("upload file to store", url)
+	uploadResult, err := operation.Upload(url, fname, bytes.NewReader(data), isGzipped, mimeType)
+	if err != nil {
+		writeJsonError(w, r, err)
+		return
+	}
+
+	m["fileName"] = fname
+	m["fid"] = assignResult.Fid
+	m["fileUrl"] = assignResult.PublicUrl + "/" + assignResult.Fid
+	m["size"] = uploadResult.Size
+	writeJsonQuiet(w, r, m)
+	return
+}
 func storeHandler(w http.ResponseWriter, r *http.Request) {
 	switch r.Method {
 	case "GET":
@@ -199,22 +241,19 @@ func PostHandler(w http.ResponseWriter, r *http.Request) {
 	m := make(map[string]interface{})
 	if e := r.ParseForm(); e != nil {
 		debug("form parse error:", e)
-		m["error"] = e.Error()
-		writeJsonQuiet(w, r, m)
+		writeJsonError(w, r, e)
 		return
 	}
 	vid, _, _, _ := parseURLPath(r.URL.Path)
 	volumeId, ve := storage.NewVolumeId(vid)
 	if ve != nil {
 		debug("NewVolumeId error:", ve)
-		m["error"] = ve.Error()
-		writeJsonQuiet(w, r, m)
+		writeJsonError(w, r, ve)
 		return
 	}
 	needle, ne := storage.NewNeedle(r)
 	if ne != nil {
-		m["error"] = ne.Error()
-		writeJsonQuiet(w, r, m)
+		writeJsonError(w, r, ne)
 		return
 	}
 	ret, errorStatus := replication.ReplicatedWrite(*masterNode, store, volumeId, needle, r)
@@ -337,6 +376,7 @@ func runVolume(cmd *Command, args []string) bool {
 	store = storage.NewStore(*vport, *ip, *publicUrl, folders, maxCounts)
 	defer store.Close()
 	http.HandleFunc("/", storeHandler)
+	http.HandleFunc("/submit", submitForClientHandler)
 	http.HandleFunc("/status", statusHandler)
 	http.HandleFunc("/admin/assign_volume", assignVolumeHandler)
 	http.HandleFunc("/admin/vacuum_volume_check", vacuumVolumeCheckHandler)
diff --git a/go/weed/weed.go b/go/weed/weed.go
index d55c262d6..773396aa1 100644
--- a/go/weed/weed.go
+++ b/go/weed/weed.go
@@ -210,6 +210,11 @@ func writeJsonQuiet(w http.ResponseWriter, r *http.Request, obj interface{}) {
 		log.Printf("error writing JSON %s: %s", obj, err)
 	}
 }
+func writeJsonError(w http.ResponseWriter, r *http.Request, err error) {
+	m := make(map[string]interface{})
+	m["error"] = err.Error()
+	writeJsonQuiet(w, r, m)
+}
 
 func debug(params ...interface{}) {
 	if *IsDebug {