hanayo/profbackground.go

117 lines
2.9 KiB
Go
Raw Normal View History

2019-02-23 13:29:15 +00:00
package main
import (
"fmt"
"image"
"image/jpeg"
"os"
"regexp"
"strings"
"time"
"github.com/gin-gonic/gin"
"github.com/nfnt/resize"
)
var hexColourRegex = regexp.MustCompile("^#([a-fA-F0-9]{6}|[a-fA-F0-9]{3})$")
func profBackground(c *gin.Context) {
ctx := getContext(c)
if ctx.User.ID == 0 {
resp403(c)
return
}
var m message = successMessage{T(c, "Your profile background has been saved.")}
defer func() {
addMessage(c, m)
getSession(c).Save()
c.Redirect(302, "/settings/profbackground")
}()
if ok, _ := CSRF.Validate(ctx.User.ID, c.PostForm("csrf")); !ok {
m = errorMessage{T(c, "Your session has expired. Please try redoing what you were trying to do.")}
return
}
t := c.Param("type")
switch t {
case "0":
db.Exec("DELETE FROM profile_backgrounds WHERE uid = ?", ctx.User.ID)
case "1":
// image
file, _, err := c.Request.FormFile("value")
if err != nil {
m = errorMessage{T(c, "An error occurred.")}
return
}
img, _, err := image.Decode(file)
if err != nil {
m = errorMessage{T(c, "An error occurred.")}
return
}
img = resize.Thumbnail(2496, 1404, img, resize.Bilinear)
f, err := os.Create(fmt.Sprintf("static/profbackgrounds/%d.png", ctx.User.ID))
2019-02-23 13:29:15 +00:00
defer f.Close()
if err != nil {
m = errorMessage{T(c, "An error occurred.")}
c.Error(err)
return
}
err = jpeg.Encode(f, img, &jpeg.Options{
Quality: 88,
})
if err != nil {
m = errorMessage{T(c, "We were not able to save your profile background.")}
c.Error(err)
return
}
saveProfileBackground(ctx, 1, fmt.Sprintf("%d.png?%d", ctx.User.ID, time.Now().Unix()))
2019-02-23 13:29:15 +00:00
case "2":
// solid colour
col := strings.ToLower(c.PostForm("value"))
// verify it's valid
if !hexColourRegex.MatchString(col) {
m = errorMessage{T(c, "Colour is invalid")}
return
}
saveProfileBackground(ctx, 2, col)
2019-02-23 14:37:47 +00:00
case "3":
// image
file, _, err := c.Request.FormFile("value")
if err != nil {
m = errorMessage{T(c, "An error occurred.")}
return
}
img, _, err := image.Decode(file)
if err != nil {
m = errorMessage{T(c, "An error occurred.")}
return
}
img = resize.Thumbnail(2496, 1404, img, resize.Bilinear)
f, err := os.Create(fmt.Sprintf("static/profbackgrounds/%d.gif", ctx.User.ID))
defer f.Close()
if err != nil {
m = errorMessage{T(c, "An error occurred.")}
c.Error(err)
return
}
err = jpeg.Encode(f, img, &jpeg.Options{
Quality: 88,
})
if err != nil {
m = errorMessage{T(c, "We were not able to save your profile background.")}
c.Error(err)
return
}
saveProfileBackground(ctx, 1, fmt.Sprintf("%d.gif?%d", ctx.User.ID, time.Now().Unix()))
2019-02-23 13:29:15 +00:00
}
}
func saveProfileBackground(ctx context, t int, val string) {
db.Exec(`INSERT INTO profile_backgrounds(uid, time, type, value)
VALUES (?, ?, ?, ?)
ON DUPLICATE KEY UPDATE
time = VALUES(time),
type = VALUES(type),
value = VALUES(value)
`, ctx.User.ID, time.Now().Unix(), t, val)
}