GoBlog/micropubMedia.go

73 lines
2.0 KiB
Go
Raw Normal View History

2020-10-14 19:20:17 +00:00
package main
import (
"mime"
"net/http"
"path/filepath"
"strings"
"go.goblog.app/app/pkgs/contenttype"
2020-10-14 19:20:17 +00:00
)
const micropubMediaSubPath = "/media"
func (a *goBlog) serveMicropubMedia(w http.ResponseWriter, r *http.Request) {
2021-02-08 17:51:07 +00:00
if !strings.Contains(r.Context().Value(indieAuthScope).(string), "media") {
a.serveError(w, r, "media scope missing", http.StatusForbidden)
2020-11-22 19:30:02 +00:00
return
2020-10-14 19:20:17 +00:00
}
if ct := r.Header.Get(contentType); !strings.Contains(ct, contenttype.MultipartForm) {
a.serveError(w, r, "wrong content-type", http.StatusBadRequest)
2020-10-14 19:20:17 +00:00
return
}
err := r.ParseMultipartForm(0)
if err != nil {
a.serveError(w, r, err.Error(), http.StatusBadRequest)
2020-10-14 19:20:17 +00:00
return
}
file, header, err := r.FormFile("file")
if err != nil {
a.serveError(w, r, err.Error(), http.StatusBadRequest)
2020-10-14 19:20:17 +00:00
return
}
defer func() { _ = file.Close() }()
hashFile, _, _ := r.FormFile("file")
defer func() { _ = hashFile.Close() }()
fileName, err := getSHA256(hashFile)
if err != nil {
a.serveError(w, r, err.Error(), http.StatusInternalServerError)
2020-10-14 19:20:17 +00:00
return
}
fileExtension := filepath.Ext(header.Filename)
if len(fileExtension) == 0 {
// Find correct file extension if original filename does not contain one
mimeType := header.Header.Get(contentType)
if len(mimeType) > 0 {
allExtensions, _ := mime.ExtensionsByType(mimeType)
if len(allExtensions) > 0 {
fileExtension = allExtensions[0]
}
}
}
fileName += strings.ToLower(fileExtension)
2021-01-10 14:59:43 +00:00
// Save file
location, err := a.saveMediaFile(fileName, file)
2020-10-14 19:20:17 +00:00
if err != nil {
a.serveError(w, r, "failed to save original file: "+err.Error(), http.StatusInternalServerError)
2020-10-14 19:20:17 +00:00
return
}
// Try to compress file (only when not in private mode)
if !a.isPrivate() {
2021-06-20 13:18:02 +00:00
compressedLocation, compressionErr := a.compressMediaFile(location)
if compressionErr != nil {
a.serveError(w, r, "failed to compress file: "+compressionErr.Error(), http.StatusInternalServerError)
return
}
2021-06-20 13:18:02 +00:00
// Overwrite location
if compressedLocation != "" {
location = compressedLocation
2020-10-14 19:20:17 +00:00
}
}
2020-12-13 10:28:46 +00:00
http.Redirect(w, r, location, http.StatusCreated)
2020-10-14 19:20:17 +00:00
}