package main import ( "encoding/json" "errors" "fmt" "io" "net/http" "os" "path" "regexp" "strings" "code.google.com/p/go-uuid/uuid" "github.com/zenazn/goji/web" ) type UploadRequest struct { src io.Reader filename string expiry int randomBarename bool } type Upload struct { Filename string Size int64 Expiry int } func uploadPostHandler(c web.C, w http.ResponseWriter, r *http.Request) { upReq := UploadRequest{} if r.Header.Get("Content-Type") == "application/octet-stream" { defer r.Body.Close() upReq.src = r.Body upReq.filename = r.URL.Query().Get("qqfile") } else { file, headers, err := r.FormFile("file") if err != nil { return } defer file.Close() upReq.src = file upReq.filename = headers.Filename } upload, err := processUpload(upReq) if err != nil { fmt.Fprintf(w, "Failed to upload: %v", err) return } if strings.EqualFold("application/json", r.Header.Get("Accept")) { js, _ := json.Marshal(map[string]string{ "filename": upload.Filename, "url": Config.siteURL + upload.Filename, }) w.Header().Set("Content-Type", "application/json; charset=UTF-8") w.Write(js) } else { http.Redirect(w, r, "/"+upload.Filename, 301) } } func uploadPutHandler(c web.C, w http.ResponseWriter, r *http.Request) { upReq := UploadRequest{} defer r.Body.Close() upReq.src = r.Body upload, err := processUpload(upReq) if err != nil { fmt.Fprintf(w, "Failed to upload") return } fmt.Fprintf(w, "File %s uploaded successfully.", upload.Filename) } func processUpload(upReq UploadRequest) (upload Upload, err error) { barename, extension := barePlusExt(upReq.filename) if upReq.randomBarename || len(barename) == 0 { barename = generateBarename() } if len(extension) == 0 { extension = "ext" } upload.Filename = strings.Join([]string{barename, extension}, ".") dst, err := os.Create(path.Join("files/", upload.Filename)) if err != nil { return } defer dst.Close() bytes, err := io.Copy(dst, upReq.src) if err != nil { return } else if bytes == 0 { err = errors.New("Empty file") return } upload.Size = bytes return } func generateBarename() string { return uuid.New()[:8] } func barePlusExt(filename string) (barename, extension string) { re := regexp.MustCompile(`[^A-Za-z0-9\-]`) filename = strings.TrimSpace(filename) filename = strings.ToLower(filename) extension = path.Ext(filename) barename = filename[:len(filename)-len(extension)] extension = re.ReplaceAllString(extension, "") barename = re.ReplaceAllString(barename, "") return }