100 lines
2.2 KiB
Go
Raw Normal View History

2019-06-03 09:32:43 +00:00
package arn
import (
"bytes"
"image"
"path"
"time"
"github.com/akyoto/imageserver"
)
const (
// AvatarSmallSize is the minimum size in pixels of an avatar.
AvatarSmallSize = 100
// AvatarMaxSize is the maximum size in pixels of an avatar.
AvatarMaxSize = 560
// AvatarWebPQuality is the WebP quality of avatars.
AvatarWebPQuality = 80
// AvatarJPEGQuality is the JPEG quality of avatars.
AvatarJPEGQuality = 80
)
// Define the avatar outputs
var avatarOutputs = []imageserver.Output{
// Original - Large
&imageserver.OriginalFile{
Directory: path.Join(Root, "images/avatars/large/"),
Width: AvatarMaxSize,
Height: AvatarMaxSize,
Quality: AvatarJPEGQuality,
},
// Original - Small
&imageserver.OriginalFile{
Directory: path.Join(Root, "images/avatars/small/"),
Width: AvatarSmallSize,
Height: AvatarSmallSize,
Quality: AvatarJPEGQuality,
},
// WebP - Large
&imageserver.WebPFile{
Directory: path.Join(Root, "images/avatars/large/"),
Width: AvatarMaxSize,
Height: AvatarMaxSize,
Quality: AvatarWebPQuality,
},
// WebP - Small
&imageserver.WebPFile{
Directory: path.Join(Root, "images/avatars/small/"),
Width: AvatarSmallSize,
Height: AvatarSmallSize,
Quality: AvatarWebPQuality,
},
}
// UserAvatar ...
type UserAvatar struct {
2019-09-06 00:15:49 +00:00
Extension string `json:"extension" editable:"true"`
LastModified int64 `json:"lastModified" editable:"true"`
2019-06-03 09:32:43 +00:00
}
// SetImageBytes accepts a byte buffer that represents an image file and updates the avatar.
func (user *User) SetImageBytes(data []byte) error {
2019-06-03 09:32:43 +00:00
// Decode
img, format, err := image.Decode(bytes.NewReader(data))
if err != nil {
return err
}
return user.SetAvatar(&imageserver.MetaImage{
Image: img,
Format: format,
Data: data,
})
}
// SetAvatar sets the avatar to the given MetaImage.
func (user *User) SetAvatar(avatar *imageserver.MetaImage) error {
var lastError error
// Save the different image formats and sizes
for _, output := range avatarOutputs {
err := output.Save(avatar, user.ID)
if err != nil {
lastError = err
}
}
user.Avatar.Extension = avatar.Extension()
user.Avatar.LastModified = time.Now().Unix()
return lastError
}