jlelse
/
GoBlog
mirror of https://github.com/jlelse/GoBlog
1
Fork 0
Code Packages Releases Activity

Improve micropub media handling

This commit is contained in:
Jan-Lukas Else 2021-02-17 13:35:29 +01:00
parent 91788efe67
commit 4f31d12975
1 changed files with 12 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
micropubMedia.go
View File

@ -8,7 +8,6 @@ import (
"fmt" "fmt"
"io" "io"
"mime" "mime"
"mime/multipart"
"net/http" "net/http"
"net/url" "net/url"
"os" "os"
@ -157,7 +156,7 @@ func tinify(url string, config *configMicropubMedia) (location string, err error
if err = s.ToFile(tmpFile.Name()); err != nil { if err = s.ToFile(tmpFile.Name()); err != nil {
return "", err return "", err
} }
fileName, err := hashFile(tmpFile.Name()) fileName, err := getSHA256(tmpFile)
if err != nil { if err != nil {
return "", err return "", err
} }
@ -177,8 +176,7 @@ func shortPixel(url string, config *configMicropubMedia) (location string, err e
return "", nil return "", nil
} }
// Compress // Compress
var buf bytes.Buffer j, _ := json.Marshal(map[string]interface{}{
_ = json.NewEncoder(&buf).Encode(map[string]interface{}{
"key": config.ShortPixelKey, "key": config.ShortPixelKey,
"plugin_version": "GB001", "plugin_version": "GB001",
"lossy": 1, "lossy": 1,
@ -189,7 +187,7 @@ func shortPixel(url string, config *configMicropubMedia) (location string, err e
"keep_exif": 0, "keep_exif": 0,
"url": url, "url": url,
}) })
req, err := http.NewRequest(http.MethodPut, "https://api.shortpixel.com/v2/reducer-sync.php", &buf) req, err := http.NewRequest(http.MethodPut, "https://api.shortpixel.com/v2/reducer-sync.php", bytes.NewReader(j))
if err != nil { if err != nil {
return "", err return "", err
} }
@ -203,22 +201,15 @@ func shortPixel(url string, config *configMicropubMedia) (location string, err e
if err != nil { if err != nil {
return "", err return "", err
} }
tmpFileName := tmpFile.Name()
defer func() { defer func() {
_ = resp.Body.Close() _ = resp.Body.Close()
_ = tmpFile.Close() _ = tmpFile.Close()
_ = os.Remove(tmpFileName) _ = os.Remove(tmpFile.Name())
}() }()
if _, err = io.Copy(tmpFile, resp.Body); err != nil { if _, err = io.Copy(tmpFile, resp.Body); err != nil {
return "", err return "", err
} }
fileName, err := hashFile(tmpFileName) fileName, err := getSHA256(tmpFile)
if err != nil {
return "", err
}
// Reopen tmp file
_ = tmpFile.Close()
tmpFile, err = os.Open(tmpFileName)
if err != nil { if err != nil {
return "", err return "", err
} }
@ -237,25 +228,16 @@ func compressionIsSupported(url string, allowed ...string) (string, bool) {
return ext, true return ext, true
} }
func getSHA256(file multipart.File) (filename string, err error) { func getSHA256(file io.ReadSeeker) (filename string, err error) {
if _, err = file.Seek(0, 0); err != nil {
return "", err
}
h := sha256.New() h := sha256.New()
if _, err = io.Copy(h, file); err != nil { if _, err = io.Copy(h, file); err != nil {
return "", err return "", err
} }
if _, err = file.Seek(0, 0); err != nil {
return "", err
}
return fmt.Sprintf("%x", h.Sum(nil)), nil return fmt.Sprintf("%x", h.Sum(nil)), nil
} }
func hashFile(filename string) (string, error) {
hashFile, err := os.Open(filename)
if err != nil {
return "", err
}
defer func() {
_ = hashFile.Close()
}()
fn, err := getSHA256(hashFile)
if err != nil {
return "", err
}
return fn, nil
}