Browse Source

Change unknown extension from .ext to .file

pull/121/head v1.2.7
andreimarcu 8 years ago
parent
commit
37f9a0cbbc
  1. 12
      server_test.go
  2. 8
      upload.go

12
server_test.go

@ -699,7 +699,7 @@ func TestPutUpload(t *testing.T) {
mux := setup()
w := httptest.NewRecorder()
filename := generateBarename() + ".ext"
filename := generateBarename() + ".file"
req, err := http.NewRequest("PUT", "/upload/"+filename, strings.NewReader("File content"))
if err != nil {
@ -717,7 +717,7 @@ func TestPutRandomizedUpload(t *testing.T) {
mux := setup()
w := httptest.NewRecorder()
filename := generateBarename() + ".ext"
filename := generateBarename() + ".file"
req, err := http.NewRequest("PUT", "/upload/"+filename, strings.NewReader("File content"))
if err != nil {
@ -757,7 +757,7 @@ func TestPutEmptyUpload(t *testing.T) {
mux := setup()
w := httptest.NewRecorder()
filename := generateBarename() + ".ext"
filename := generateBarename() + ".file"
req, err := http.NewRequest("PUT", "/upload/"+filename, strings.NewReader(""))
if err != nil {
@ -779,7 +779,7 @@ func TestPutJSONUpload(t *testing.T) {
mux := setup()
w := httptest.NewRecorder()
filename := generateBarename() + ".ext"
filename := generateBarename() + ".file"
req, err := http.NewRequest("PUT", "/upload/"+filename, strings.NewReader("File content"))
if err != nil {
@ -806,7 +806,7 @@ func TestPutRandomizedJSONUpload(t *testing.T) {
mux := setup()
w := httptest.NewRecorder()
filename := generateBarename() + ".ext"
filename := generateBarename() + ".file"
req, err := http.NewRequest("PUT", "/upload/"+filename, strings.NewReader("File content"))
if err != nil {
@ -834,7 +834,7 @@ func TestPutExpireJSONUpload(t *testing.T) {
mux := setup()
w := httptest.NewRecorder()
filename := generateBarename() + ".ext"
filename := generateBarename() + ".file"
req, err := http.NewRequest("PUT", "/upload/"+filename, strings.NewReader("File content"))
if err != nil {

8
upload.go

@ -219,12 +219,12 @@ func processUpload(upReq UploadRequest) (upload Upload, err error) {
// Determine the type of file from header
kind, err := filetype.Match(header)
if err != nil {
extension = "ext"
}
if err != nil || kind.Extension == "unknown" {
extension = "file"
} else {
extension = kind.Extension
}
}
upload.Filename = strings.Join([]string{barename, extension}, ".")
upload.Filename = strings.Replace(upload.Filename, " ", "", -1)

Loading…
Cancel
Save