github.com/vnforks/kid@v5.11.1+incompatible/app/emoji.go (about) 1 // Copyright (c) 2017-present Mattermost, 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/mattermost/mattermost-server/mlog" 22 "github.com/mattermost/mattermost-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 result := <-a.Srv.Store.Emoji().GetByName(emoji.Name); result.Err == nil && result.Data != 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 result := <-a.Srv.Store.Emoji().Save(emoji) 71 if result.Err != nil { 72 return nil, result.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 result.Data.(*model.Emoji), nil 79 } 80 81 func (a *App) GetEmojiList(page, perPage int, sort string) ([]*model.Emoji, *model.AppError) { 82 result := <-a.Srv.Store.Emoji().GetList(page*perPage, perPage, sort) 83 if result.Err != nil { 84 return nil, result.Err 85 } 86 return result.Data.([]*model.Emoji), nil 87 } 88 89 func (a *App) UploadEmojiImage(id string, imageData *multipart.FileHeader) *model.AppError { 90 if !*a.Config().ServiceSettings.EnableCustomEmoji { 91 return model.NewAppError("UploadEmojiImage", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 92 } 93 94 if len(*a.Config().FileSettings.DriverName) == 0 { 95 return model.NewAppError("UploadEmojiImage", "api.emoji.storage.app_error", nil, "", http.StatusNotImplemented) 96 } 97 98 file, err := imageData.Open() 99 if err != nil { 100 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.open.app_error", nil, "", http.StatusBadRequest) 101 } 102 defer file.Close() 103 104 buf := bytes.NewBuffer(nil) 105 io.Copy(buf, file) 106 107 // make sure the file is an image and is within the required dimensions 108 config, _, err := image.DecodeConfig(bytes.NewReader(buf.Bytes())) 109 if err != nil { 110 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.image.app_error", nil, "", http.StatusBadRequest) 111 } 112 113 if config.Width > MaxEmojiOriginalWidth || config.Height > MaxEmojiOriginalHeight { 114 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.large_image.too_large.app_error", map[string]interface{}{ 115 "MaxWidth": MaxEmojiOriginalWidth, 116 "MaxHeight": MaxEmojiOriginalHeight, 117 }, "", http.StatusBadRequest) 118 } 119 120 if config.Width > MaxEmojiWidth || config.Height > MaxEmojiHeight { 121 data := buf.Bytes() 122 newbuf := bytes.NewBuffer(nil) 123 info, err := model.GetInfoForBytes(imageData.Filename, data) 124 if err != nil { 125 return err 126 } 127 128 if info.MimeType == "image/gif" { 129 gif_data, err := gif.DecodeAll(bytes.NewReader(data)) 130 if err != nil { 131 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.large_image.gif_decode_error", nil, "", http.StatusBadRequest) 132 } 133 134 resized_gif := resizeEmojiGif(gif_data) 135 if err := gif.EncodeAll(newbuf, resized_gif); err != nil { 136 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.large_image.gif_encode_error", nil, "", http.StatusBadRequest) 137 } 138 139 if _, err := a.WriteFile(newbuf, getEmojiImagePath(id)); err != nil { 140 return err 141 } 142 } else { 143 img, _, err := image.Decode(bytes.NewReader(data)) 144 if err != nil { 145 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.large_image.decode_error", nil, "", http.StatusBadRequest) 146 } 147 148 resized_image := resizeEmoji(img, config.Width, config.Height) 149 if err := png.Encode(newbuf, resized_image); err != nil { 150 return model.NewAppError("uploadEmojiImage", "api.emoji.upload.large_image.encode_error", nil, "", http.StatusBadRequest) 151 } 152 if _, err := a.WriteFile(newbuf, getEmojiImagePath(id)); err != nil { 153 return err 154 } 155 } 156 } 157 158 _, appErr := a.WriteFile(buf, getEmojiImagePath(id)) 159 return appErr 160 } 161 162 func (a *App) DeleteEmoji(emoji *model.Emoji) *model.AppError { 163 if err := (<-a.Srv.Store.Emoji().Delete(emoji.Id, model.GetMillis())).Err; err != nil { 164 return err 165 } 166 167 a.deleteEmojiImage(emoji.Id) 168 a.deleteReactionsForEmoji(emoji.Name) 169 return nil 170 } 171 172 func (a *App) GetEmoji(emojiId string) (*model.Emoji, *model.AppError) { 173 if !*a.Config().ServiceSettings.EnableCustomEmoji { 174 return nil, model.NewAppError("GetEmoji", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 175 } 176 177 if len(*a.Config().FileSettings.DriverName) == 0 { 178 return nil, model.NewAppError("GetEmoji", "api.emoji.storage.app_error", nil, "", http.StatusNotImplemented) 179 } 180 181 result := <-a.Srv.Store.Emoji().Get(emojiId, false) 182 if result.Err != nil { 183 return nil, result.Err 184 } 185 return result.Data.(*model.Emoji), nil 186 } 187 188 func (a *App) GetEmojiByName(emojiName string) (*model.Emoji, *model.AppError) { 189 if !*a.Config().ServiceSettings.EnableCustomEmoji { 190 return nil, model.NewAppError("GetEmoji", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 191 } 192 193 if len(*a.Config().FileSettings.DriverName) == 0 { 194 return nil, model.NewAppError("GetEmoji", "api.emoji.storage.app_error", nil, "", http.StatusNotImplemented) 195 } 196 197 result := <-a.Srv.Store.Emoji().GetByName(emojiName) 198 if result.Err != nil { 199 return nil, result.Err 200 } 201 return result.Data.(*model.Emoji), nil 202 } 203 204 func (a *App) GetMultipleEmojiByName(names []string) ([]*model.Emoji, *model.AppError) { 205 if !*a.Config().ServiceSettings.EnableCustomEmoji { 206 return nil, model.NewAppError("GetMultipleEmojiByName", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 207 } 208 209 if result := <-a.Srv.Store.Emoji().GetMultipleByName(names); result.Err != nil { 210 return nil, result.Err 211 } else { 212 return result.Data.([]*model.Emoji), nil 213 } 214 } 215 216 func (a *App) GetEmojiImage(emojiId string) ([]byte, string, *model.AppError) { 217 result := <-a.Srv.Store.Emoji().Get(emojiId, true) 218 if result.Err != nil { 219 return nil, "", result.Err 220 } 221 222 img, appErr := a.ReadFile(getEmojiImagePath(emojiId)) 223 if appErr != nil { 224 return nil, "", model.NewAppError("getEmojiImage", "api.emoji.get_image.read.app_error", nil, appErr.Error(), http.StatusNotFound) 225 } 226 227 _, imageType, err := image.DecodeConfig(bytes.NewReader(img)) 228 if err != nil { 229 return nil, "", model.NewAppError("getEmojiImage", "api.emoji.get_image.decode.app_error", nil, err.Error(), http.StatusInternalServerError) 230 } 231 232 return img, imageType, nil 233 } 234 235 func (a *App) SearchEmoji(name string, prefixOnly bool, limit int) ([]*model.Emoji, *model.AppError) { 236 if !*a.Config().ServiceSettings.EnableCustomEmoji { 237 return nil, model.NewAppError("SearchEmoji", "api.emoji.disabled.app_error", nil, "", http.StatusNotImplemented) 238 } 239 240 result := <-a.Srv.Store.Emoji().Search(name, prefixOnly, limit) 241 if result.Err != nil { 242 return nil, result.Err 243 } 244 return result.Data.([]*model.Emoji), nil 245 } 246 247 func resizeEmojiGif(gifImg *gif.GIF) *gif.GIF { 248 // Create a new RGBA image to hold the incremental frames. 249 firstFrame := gifImg.Image[0].Bounds() 250 b := image.Rect(0, 0, firstFrame.Dx(), firstFrame.Dy()) 251 img := image.NewRGBA(b) 252 253 resizedImage := image.Image(nil) 254 // Resize each frame. 255 for index, frame := range gifImg.Image { 256 bounds := frame.Bounds() 257 draw.Draw(img, bounds, frame, bounds.Min, draw.Over) 258 resizedImage = resizeEmoji(img, firstFrame.Dx(), firstFrame.Dy()) 259 gifImg.Image[index] = imageToPaletted(resizedImage) 260 } 261 // Set new gif width and height 262 gifImg.Config.Width = resizedImage.Bounds().Dx() 263 gifImg.Config.Height = resizedImage.Bounds().Dy() 264 return gifImg 265 } 266 267 func getEmojiImagePath(id string) string { 268 return "emoji/" + id + "/image" 269 } 270 271 func resizeEmoji(img image.Image, width int, height int) image.Image { 272 emojiWidth := float64(width) 273 emojiHeight := float64(height) 274 275 if emojiHeight <= MaxEmojiHeight && emojiWidth <= MaxEmojiWidth { 276 return img 277 } 278 return imaging.Fit(img, MaxEmojiWidth, MaxEmojiHeight, imaging.Lanczos) 279 } 280 281 func imageToPaletted(img image.Image) *image.Paletted { 282 b := img.Bounds() 283 pm := image.NewPaletted(b, palette.Plan9) 284 draw.FloydSteinberg.Draw(pm, b, img, image.ZP) 285 return pm 286 } 287 288 func (a *App) deleteEmojiImage(id string) { 289 if err := a.MoveFile(getEmojiImagePath(id), "emoji/"+id+"/image_deleted"); err != nil { 290 mlog.Error(fmt.Sprintf("Failed to rename image when deleting emoji %v", id)) 291 } 292 } 293 294 func (a *App) deleteReactionsForEmoji(emojiName string) { 295 if result := <-a.Srv.Store.Reaction().DeleteAllWithEmojiName(emojiName); result.Err != nil { 296 mlog.Warn(fmt.Sprintf("Unable to delete reactions when deleting emoji with emoji name %v", emojiName)) 297 mlog.Warn(fmt.Sprint(result.Err)) 298 } 299 }