github.com/xzl8028/xenia-server@v0.0.0-20190809101854-18450a97da63/app/emoji.go (about) 1 // Copyright (c) 2017-present Xenia, Inc. All Rights Reserved. 2 // See License.txt for license information. 3 4 package app 5 6 import ( 7 "bytes" 8 "fmt" 9 "image" 10 "image/draw" 11 "image/gif" 12 _ "image/jpeg" 13 "image/png" 14 "io" 15 "mime/multipart" 16 "net/http" 17 18 "image/color/palette" 19 20 "github.com/disintegration/imaging" 21 "github.com/xzl8028/xenia-server/mlog" 22 "github.com/xzl8028/xenia-server/model" 23 ) 24 25 const ( 26 MaxEmojiFileSize = 1 << 20 // 1 MB 27 MaxEmojiWidth = 128 28 MaxEmojiHeight = 128 29 MaxEmojiOriginalWidth = 1028 30 MaxEmojiOriginalHeight = 1028 31 ) 32 33 func (a *App) CreateEmoji(sessionUserId string, emoji *model.Emoji, multiPartImageData *multipart.Form) (*model.Emoji, *model.AppError) { 34 if !*a.Config().ServiceSettings.EnableCustomEmoji { 35 return nil, model.NewAppError("UploadEmojiImage", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 36 } 37 38 if len(*a.Config().FileSettings.DriverName) == 0 { 39 return nil, model.NewAppError("GetEmoji", "api.emoji.storage.app_error", nil, "", http.StatusNotImplemented) 40 } 41 42 // wipe the emoji id so that existing emojis can't get overwritten 43 emoji.Id = "" 44 45 // do our best to validate the emoji before committing anything to the DB so that we don't have to clean up 46 // orphaned files left over when validation fails later on 47 emoji.PreSave() 48 if err := emoji.IsValid(); err != nil { 49 return nil, err 50 } 51 52 if emoji.CreatorId != sessionUserId { 53 return nil, model.NewAppError("createEmoji", "api.emoji.create.other_user.app_error", nil, "", http.StatusForbidden) 54 } 55 56 if existingEmoji, err := a.Srv.Store.Emoji().GetByName(emoji.Name); err == nil && existingEmoji != nil { 57 return nil, model.NewAppError("createEmoji", "api.emoji.create.duplicate.app_error", nil, "", http.StatusBadRequest) 58 } 59 60 imageData := multiPartImageData.File["image"] 61 if len(imageData) == 0 { 62 err := model.NewAppError("Context", "api.context.invalid_body_param.app_error", map[string]interface{}{"Name": "createEmoji"}, "", http.StatusBadRequest) 63 return nil, err 64 } 65 66 if err := a.UploadEmojiImage(emoji.Id, imageData[0]); err != nil { 67 return nil, err 68 } 69 70 emoji, err := a.Srv.Store.Emoji().Save(emoji) 71 if err != nil { 72 return nil, err 73 } 74 75 message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_EMOJI_ADDED, "", "", "", nil) 76 message.Add("emoji", emoji.ToJson()) 77 a.Publish(message) 78 return emoji, nil 79 } 80 81 func (a *App) GetEmojiList(page, perPage int, sort string) ([]*model.Emoji, *model.AppError) { 82 return a.Srv.Store.Emoji().GetList(page*perPage, perPage, sort) 83 } 84 85 func (a *App) UploadEmojiImage(id string, imageData *multipart.FileHeader) *model.AppError { 86 if !*a.Config().ServiceSettings.EnableCustomEmoji { 87 return model.NewAppError("UploadEmojiImage", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 88 } 89 90 if len(*a.Config().FileSettings.DriverName) == 0 { 91 return model.NewAppError("UploadEmojiImage", "api.emoji.storage.app_error", nil, "", http.StatusNotImplemented) 92 } 93 94 file, err := imageData.Open() 95 if err != nil { 96 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.open.app_error", nil, "", http.StatusBadRequest) 97 } 98 defer file.Close() 99 100 buf := bytes.NewBuffer(nil) 101 io.Copy(buf, file) 102 103 // make sure the file is an image and is within the required dimensions 104 config, _, err := image.DecodeConfig(bytes.NewReader(buf.Bytes())) 105 if err != nil { 106 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.image.app_error", nil, "", http.StatusBadRequest) 107 } 108 109 if config.Width > MaxEmojiOriginalWidth || config.Height > MaxEmojiOriginalHeight { 110 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.large_image.too_large.app_error", map[string]interface{}{ 111 "MaxWidth": MaxEmojiOriginalWidth, 112 "MaxHeight": MaxEmojiOriginalHeight, 113 }, "", http.StatusBadRequest) 114 } 115 116 if config.Width > MaxEmojiWidth || config.Height > MaxEmojiHeight { 117 data := buf.Bytes() 118 newbuf := bytes.NewBuffer(nil) 119 info, err := model.GetInfoForBytes(imageData.Filename, data) 120 if err != nil { 121 return err 122 } 123 124 if info.MimeType == "image/gif" { 125 gif_data, err := gif.DecodeAll(bytes.NewReader(data)) 126 if err != nil { 127 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.large_image.gif_decode_error", nil, "", http.StatusBadRequest) 128 } 129 130 resized_gif := resizeEmojiGif(gif_data) 131 if err := gif.EncodeAll(newbuf, resized_gif); err != nil { 132 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.large_image.gif_encode_error", nil, "", http.StatusBadRequest) 133 } 134 135 if _, err := a.WriteFile(newbuf, getEmojiImagePath(id)); err != nil { 136 return err 137 } 138 } else { 139 img, _, err := image.Decode(bytes.NewReader(data)) 140 if err != nil { 141 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.large_image.decode_error", nil, "", http.StatusBadRequest) 142 } 143 144 resized_image := resizeEmoji(img, config.Width, config.Height) 145 if err := png.Encode(newbuf, resized_image); err != nil { 146 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.large_image.encode_error", nil, "", http.StatusBadRequest) 147 } 148 if _, err := a.WriteFile(newbuf, getEmojiImagePath(id)); err != nil { 149 return err 150 } 151 } 152 } 153 154 _, appErr := a.WriteFile(buf, getEmojiImagePath(id)) 155 return appErr 156 } 157 158 func (a *App) DeleteEmoji(emoji *model.Emoji) *model.AppError { 159 if err := a.Srv.Store.Emoji().Delete(emoji.Id, model.GetMillis()); err != nil { 160 return err 161 } 162 163 a.deleteEmojiImage(emoji.Id) 164 a.deleteReactionsForEmoji(emoji.Name) 165 return nil 166 } 167 168 func (a *App) GetEmoji(emojiId string) (*model.Emoji, *model.AppError) { 169 if !*a.Config().ServiceSettings.EnableCustomEmoji { 170 return nil, model.NewAppError("GetEmoji", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 171 } 172 173 if len(*a.Config().FileSettings.DriverName) == 0 { 174 return nil, model.NewAppError("GetEmoji", "api.emoji.storage.app_error", nil, "", http.StatusNotImplemented) 175 } 176 177 return a.Srv.Store.Emoji().Get(emojiId, false) 178 } 179 180 func (a *App) GetEmojiByName(emojiName string) (*model.Emoji, *model.AppError) { 181 if !*a.Config().ServiceSettings.EnableCustomEmoji { 182 return nil, model.NewAppError("GetEmoji", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 183 } 184 185 if len(*a.Config().FileSettings.DriverName) == 0 { 186 return nil, model.NewAppError("GetEmoji", "api.emoji.storage.app_error", nil, "", http.StatusNotImplemented) 187 } 188 189 return a.Srv.Store.Emoji().GetByName(emojiName) 190 } 191 192 func (a *App) GetMultipleEmojiByName(names []string) ([]*model.Emoji, *model.AppError) { 193 if !*a.Config().ServiceSettings.EnableCustomEmoji { 194 return nil, model.NewAppError("GetMultipleEmojiByName", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 195 } 196 197 if result := <-a.Srv.Store.Emoji().GetMultipleByName(names); result.Err != nil { 198 return nil, result.Err 199 } else { 200 return result.Data.([]*model.Emoji), nil 201 } 202 } 203 204 func (a *App) GetEmojiImage(emojiId string) ([]byte, string, *model.AppError) { 205 _, storeErr := a.Srv.Store.Emoji().Get(emojiId, true) 206 if storeErr != nil { 207 return nil, "", storeErr 208 } 209 210 img, appErr := a.ReadFile(getEmojiImagePath(emojiId)) 211 if appErr != nil { 212 return nil, "", model.NewAppError("getEmojiImage", "api.emoji.get_image.read.app_error", nil, appErr.Error(), http.StatusNotFound) 213 } 214 215 _, imageType, err := image.DecodeConfig(bytes.NewReader(img)) 216 if err != nil { 217 return nil, "", model.NewAppError("getEmojiImage", "api.emoji.get_image.decode.app_error", nil, err.Error(), http.StatusInternalServerError) 218 } 219 220 return img, imageType, nil 221 } 222 223 func (a *App) SearchEmoji(name string, prefixOnly bool, limit int) ([]*model.Emoji, *model.AppError) { 224 if !*a.Config().ServiceSettings.EnableCustomEmoji { 225 return nil, model.NewAppError("SearchEmoji", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 226 } 227 228 return a.Srv.Store.Emoji().Search(name, prefixOnly, limit) 229 } 230 231 func resizeEmojiGif(gifImg *gif.GIF) *gif.GIF { 232 // Create a new RGBA image to hold the incremental frames. 233 firstFrame := gifImg.Image[0].Bounds() 234 b := image.Rect(0, 0, firstFrame.Dx(), firstFrame.Dy()) 235 img := image.NewRGBA(b) 236 237 resizedImage := image.Image(nil) 238 // Resize each frame. 239 for index, frame := range gifImg.Image { 240 bounds := frame.Bounds() 241 draw.Draw(img, bounds, frame, bounds.Min, draw.Over) 242 resizedImage = resizeEmoji(img, firstFrame.Dx(), firstFrame.Dy()) 243 gifImg.Image[index] = imageToPaletted(resizedImage) 244 } 245 // Set new gif width and height 246 gifImg.Config.Width = resizedImage.Bounds().Dx() 247 gifImg.Config.Height = resizedImage.Bounds().Dy() 248 return gifImg 249 } 250 251 func getEmojiImagePath(id string) string { 252 return "emoji/" + id + "/image" 253 } 254 255 func resizeEmoji(img image.Image, width int, height int) image.Image { 256 emojiWidth := float64(width) 257 emojiHeight := float64(height) 258 259 if emojiHeight <= MaxEmojiHeight && emojiWidth <= MaxEmojiWidth { 260 return img 261 } 262 return imaging.Fit(img, MaxEmojiWidth, MaxEmojiHeight, imaging.Lanczos) 263 } 264 265 func imageToPaletted(img image.Image) *image.Paletted { 266 b := img.Bounds() 267 pm := image.NewPaletted(b, palette.Plan9) 268 draw.FloydSteinberg.Draw(pm, b, img, image.ZP) 269 return pm 270 } 271 272 func (a *App) deleteEmojiImage(id string) { 273 if err := a.MoveFile(getEmojiImagePath(id), "emoji/"+id+"/image_deleted"); err != nil { 274 mlog.Error(fmt.Sprintf("Failed to rename image when deleting emoji %v", id)) 275 } 276 } 277 278 func (a *App) deleteReactionsForEmoji(emojiName string) { 279 if err := a.Srv.Store.Reaction().DeleteAllWithEmojiName(emojiName); err != nil { 280 mlog.Warn(fmt.Sprintf("Unable to delete reactions when deleting emoji with emoji name %v", emojiName)) 281 mlog.Warn(fmt.Sprint(err)) 282 } 283 }