linx-server/upload.go

136 lines
2.5 KiB
Go
Raw Normal View History

2015-09-24 01:44:49 -04:00
package main
import (
"encoding/json"
2015-09-24 01:44:49 -04:00
"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")
2015-09-24 01:44:49 -04:00
} else {
file, headers, err := r.FormFile("file")
if err != nil {
oopsHandler(c, w, r)
return
}
defer file.Close()
upReq.src = file
upReq.filename = headers.Filename
}
2015-09-24 01:44:49 -04:00
upload, err := processUpload(upReq)
if err != nil {
oopsHandler(c, w, r)
2015-09-24 01:44:49 -04:00
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)
}
2015-09-24 01:44:49 -04:00
}
func uploadPutHandler(c web.C, w http.ResponseWriter, r *http.Request) {
upReq := UploadRequest{}
defer r.Body.Close()
upReq.filename = c.URLParams["name"]
2015-09-24 01:44:49 -04:00
upReq.src = r.Body
upload, err := processUpload(upReq)
if err != nil {
oopsHandler(c, w, r)
2015-09-24 01:44:49 -04:00
return
}
fmt.Fprintf(w, Config.siteURL+upload.Filename)
2015-09-24 01:44:49 -04:00
}
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}, ".")
2015-09-24 08:30:16 -04:00
dst, err := os.Create(path.Join("files/", upload.Filename))
if err != nil {
2015-09-24 01:44:49 -04:00
return
}
2015-09-24 08:30:16 -04:00
defer dst.Close()
2015-09-24 01:44:49 -04:00
2015-09-24 08:30:16 -04:00
bytes, err := io.Copy(dst, upReq.src)
if err != nil {
2015-09-24 01:44:49 -04:00
return
} else if bytes == 0 {
return
}
upload.Size = bytes
return
}
func generateBarename() string {
return uuid.New()[:8]
}
2015-09-25 09:04:06 -04:00
var barePlusRe = regexp.MustCompile(`[^A-Za-z0-9\-]`)
2015-09-24 01:44:49 -04:00
func barePlusExt(filename string) (barename, extension string) {
filename = strings.TrimSpace(filename)
filename = strings.ToLower(filename)
extension = path.Ext(filename)
barename = filename[:len(filename)-len(extension)]
2015-09-25 09:04:06 -04:00
extension = barePlusRe.ReplaceAllString(extension, "")
barename = barePlusRe.ReplaceAllString(barename, "")
2015-09-24 01:44:49 -04:00
return
}