mirror of
https://github.com/woodchen-ink/webp_server_go.git
synced 2025-07-18 13:42:02 +08:00
109 lines
2.9 KiB
Go
109 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/gofiber/fiber"
|
|
)
|
|
|
|
func Convert(ImgPath string, ExhaustPath string, AllowedTypes []string, QUALITY string) func(c *fiber.Ctx) {
|
|
return func(c *fiber.Ctx) {
|
|
//basic vars
|
|
var reqURI = c.Path() // mypic/123.jpg
|
|
var RawImageAbs = path.Join(ImgPath, reqURI) // /home/xxx/mypic/123.jpg
|
|
var ImgFilename = path.Base(reqURI) // pure filename, 123.jpg
|
|
var finalFile string // We'll only need one c.sendFile()
|
|
var UA = c.Get("User-Agent")
|
|
done := goOrigin(UA)
|
|
if done {
|
|
log.Infof("A Safari/IE/whatever user has arrived...%s", UA)
|
|
// Check for Safari users. If they're Safari, just simply ignore everything.
|
|
|
|
etag := GenEtag(RawImageAbs)
|
|
c.Set("ETag", etag)
|
|
c.SendFile(RawImageAbs)
|
|
return
|
|
}
|
|
log.Debugf("Incoming connection from %s@%s with %s", UA, c.IP(), ImgFilename)
|
|
|
|
// check ext
|
|
// TODO: may remove this function. Check in Nginx.
|
|
var allowed = false
|
|
for _, ext := range AllowedTypes {
|
|
haystack := strings.ToLower(ImgFilename)
|
|
needle := strings.ToLower("." + ext)
|
|
if strings.HasSuffix(haystack, needle) {
|
|
allowed = true
|
|
break
|
|
} else {
|
|
allowed = false
|
|
}
|
|
}
|
|
if !allowed {
|
|
msg := "File extension not allowed! " + ImgFilename
|
|
log.Warn(msg)
|
|
c.Send(msg)
|
|
if ImageExists(RawImageAbs) {
|
|
etag := GenEtag(RawImageAbs)
|
|
c.Set("ETag", etag)
|
|
c.SendFile(RawImageAbs)
|
|
}
|
|
return
|
|
}
|
|
|
|
// Check the original image for existence,
|
|
if !ImageExists(RawImageAbs) {
|
|
msg := "Image not found!"
|
|
c.Send(msg)
|
|
log.Warn(msg)
|
|
c.SendStatus(404)
|
|
return
|
|
}
|
|
|
|
_, WebpAbsPath := GenWebpAbs(RawImageAbs, ExhaustPath, ImgFilename, reqURI)
|
|
|
|
if ImageExists(WebpAbsPath) {
|
|
finalFile = WebpAbsPath
|
|
} else {
|
|
// we don't have abc.jpg.png1582558990.webp
|
|
// delete the old pic and convert a new one.
|
|
// /home/webp_server/exhaust/path/to/tsuki.jpg.1582558990.webp
|
|
destHalfFile := path.Clean(path.Join(WebpAbsPath, path.Dir(reqURI), ImgFilename))
|
|
matches, err := filepath.Glob(destHalfFile + "*")
|
|
if err != nil {
|
|
log.Error(err.Error())
|
|
} else {
|
|
// /home/webp_server/exhaust/path/to/tsuki.jpg.1582558100.webp <- older ones will be removed
|
|
// /home/webp_server/exhaust/path/to/tsuki.jpg.1582558990.webp <- keep the latest one
|
|
for _, p := range matches {
|
|
if strings.Compare(destHalfFile, p) != 0 {
|
|
_ = os.Remove(p)
|
|
}
|
|
}
|
|
}
|
|
|
|
//for webp, we need to create dir first
|
|
_ = os.MkdirAll(path.Dir(WebpAbsPath), 0755)
|
|
q, _ := strconv.ParseFloat(QUALITY, 32)
|
|
err = WebpEncoder(RawImageAbs, WebpAbsPath, float32(q), true, nil)
|
|
|
|
if err != nil {
|
|
log.Error(err)
|
|
c.SendStatus(400)
|
|
c.Send("Bad file!")
|
|
return
|
|
}
|
|
finalFile = WebpAbsPath
|
|
}
|
|
etag := GenEtag(finalFile)
|
|
c.Set("ETag", etag)
|
|
c.SendFile(finalFile)
|
|
}
|
|
}
|