100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
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 {
|
|
Extension string `json:"extension" editable:"true"`
|
|
LastModified int64 `json:"lastModified" editable:"true"`
|
|
}
|
|
|
|
// SetImageBytes accepts a byte buffer that represents an image file and updates the avatar.
|
|
func (user *User) SetImageBytes(data []byte) error {
|
|
// 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
|
|
}
|