summaryrefslogtreecommitdiffstats
path: root/model/file_info.go
diff options
context:
space:
mode:
authorHarrison Healey <harrisonmhealey@gmail.com>2016-09-30 11:06:30 -0400
committerGitHub <noreply@github.com>2016-09-30 11:06:30 -0400
commit8a0e649f989a824bb3bbfd1900a5b8e5383b47e1 (patch)
tree4b424929fe13ebec438d2f41a2729e37e5160720 /model/file_info.go
parenta2deeed597dea15d9b7ca237be71988469f58cdd (diff)
downloadchat-8a0e649f989a824bb3bbfd1900a5b8e5383b47e1.tar.gz
chat-8a0e649f989a824bb3bbfd1900a5b8e5383b47e1.tar.bz2
chat-8a0e649f989a824bb3bbfd1900a5b8e5383b47e1.zip
PLT-3105 Files table migration (#4068)
* Implemented initial changes for files table * Removed *_benchmark_test.go files * Re-implemented GetPublicFile and added support for old path * Localization for files table * Moved file system code into utils package * Finished server-side changes and added initial upgrade script * Added getPostFiles api * Re-add Extension and HasPreviewImage fields to FileInfo * Removed unused translation * Fixed merge conflicts left over after permissions changes * Forced FileInfo.extension to be lower case * Changed FileUploadResponse to contain the FileInfos instead of FileIds * Fixed permissions on getFile* calls * Fixed notifications for file uploads * Added initial version of client code for files changes * Permanently added FileIds field to Post object and removed Post.HasFiles * Updated PostStore.Update to be usable in more circumstances * Re-added Filenames field and switched file migration to be entirely lazy-loaded * Increased max listener count for FileStore * Removed unused fileInfoCache * Moved file system code back into api * Removed duplicate test case * Fixed unit test running on ports other than 8065 * Renamed HasPermissionToPostContext to HasPermissionToChannelByPostContext * Refactored handleImages to make it more easily understandable * Renamed getPostFiles to getFileInfosForPost * Re-added pre-FileIds posts to analytics * Changed files to be saved as their ids as opposed to id/filename.ext * Renamed FileInfo.UserId to FileInfo.CreatorId * Fixed detection of language in CodePreview * Fixed switching between threads in the RHS not loading new files * Add serverside protection against a rare bug where the client sends the same file twice for a single post * Refactored the important parts of uploadFile api call into a function that can be called without a web context
Diffstat (limited to 'model/file_info.go')
-rw-r--r--model/file_info.go173
1 files changed, 135 insertions, 38 deletions
diff --git a/model/file_info.go b/model/file_info.go
index f785042b3..687473d4f 100644
--- a/model/file_info.go
+++ b/model/file_info.go
@@ -6,58 +6,55 @@ package model
import (
"bytes"
"encoding/json"
+ "image"
"image/gif"
"io"
"mime"
"path/filepath"
+ "strings"
)
type FileInfo struct {
- Filename string `json:"filename"`
- Size int `json:"size"`
+ Id string `json:"id"`
+ CreatorId string `json:"user_id"`
+ PostId string `json:"post_id,omitempty"`
+ CreateAt int64 `json:"create_at"`
+ UpdateAt int64 `json:"update_at"`
+ DeleteAt int64 `json:"delete_at"`
+ Path string `json:"-"` // not sent back to the client
+ ThumbnailPath string `json:"-"` // not sent back to the client
+ PreviewPath string `json:"-"` // not sent back to the client
+ Name string `json:"name"`
Extension string `json:"extension"`
+ Size int64 `json:"size"`
MimeType string `json:"mime_type"`
- HasPreviewImage bool `json:"has_preview_image"`
+ Width int `json:"width,omitempty"`
+ Height int `json:"height,omitempty"`
+ HasPreviewImage bool `json:"has_preview_image,omitempty"`
}
-func GetInfoForBytes(filename string, data []byte) (*FileInfo, *AppError) {
- size := len(data)
-
- var mimeType string
- extension := filepath.Ext(filename)
- isImage := IsFileExtImage(extension)
- if isImage {
- mimeType = GetImageMimeType(extension)
+func (info *FileInfo) ToJson() string {
+ b, err := json.Marshal(info)
+ if err != nil {
+ return ""
} else {
- mimeType = mime.TypeByExtension(extension)
+ return string(b)
}
+}
- if extension != "" && extension[0] == '.' {
- // the client expects a file extension without the leading period
- extension = extension[1:]
- }
+func FileInfoFromJson(data io.Reader) *FileInfo {
+ decoder := json.NewDecoder(data)
- hasPreviewImage := isImage
- if mimeType == "image/gif" {
- // just show the gif itself instead of a preview image for animated gifs
- if gifImage, err := gif.DecodeAll(bytes.NewReader(data)); err != nil {
- return nil, NewLocAppError("GetInfoForBytes", "model.file_info.get.gif.app_error", nil, "filename="+filename)
- } else {
- hasPreviewImage = len(gifImage.Image) == 1
- }
+ var info FileInfo
+ if err := decoder.Decode(&info); err != nil {
+ return nil
+ } else {
+ return &info
}
-
- return &FileInfo{
- Filename: filename,
- Size: size,
- Extension: extension,
- MimeType: mimeType,
- HasPreviewImage: hasPreviewImage,
- }, nil
}
-func (info *FileInfo) ToJson() string {
- b, err := json.Marshal(info)
+func FileInfosToJson(infos []*FileInfo) string {
+ b, err := json.Marshal(infos)
if err != nil {
return ""
} else {
@@ -65,13 +62,113 @@ func (info *FileInfo) ToJson() string {
}
}
-func FileInfoFromJson(data io.Reader) *FileInfo {
+func FileInfosFromJson(data io.Reader) []*FileInfo {
decoder := json.NewDecoder(data)
- var info FileInfo
- if err := decoder.Decode(&info); err != nil {
+ var infos []*FileInfo
+ if err := decoder.Decode(&infos); err != nil {
return nil
} else {
- return &info
+ return infos
+ }
+}
+
+func (o *FileInfo) PreSave() {
+ if o.Id == "" {
+ o.Id = NewId()
+ }
+
+ if o.CreateAt == 0 {
+ o.CreateAt = GetMillis()
+ o.UpdateAt = o.CreateAt
+ }
+}
+
+func (o *FileInfo) IsValid() *AppError {
+ if len(o.Id) != 26 {
+ return NewLocAppError("FileInfo.IsValid", "model.file_info.is_valid.id.app_error", nil, "")
+ }
+
+ if len(o.CreatorId) != 26 {
+ return NewLocAppError("FileInfo.IsValid", "model.file_info.is_valid.user_id.app_error", nil, "id="+o.Id)
+ }
+
+ if len(o.PostId) != 0 && len(o.PostId) != 26 {
+ return NewLocAppError("FileInfo.IsValid", "model.file_info.is_valid.post_id.app_error", nil, "id="+o.Id)
+ }
+
+ if o.CreateAt == 0 {
+ return NewLocAppError("FileInfo.IsValid", "model.file_info.is_valid.create_at.app_error", nil, "id="+o.Id)
+ }
+
+ if o.UpdateAt == 0 {
+ return NewLocAppError("FileInfo.IsValid", "model.file_info.is_valid.update_at.app_error", nil, "id="+o.Id)
+ }
+
+ if o.Path == "" {
+ return NewLocAppError("FileInfo.IsValid", "model.file_info.is_valid.path.app_error", nil, "id="+o.Id)
}
+
+ return nil
+}
+
+func (o *FileInfo) IsImage() bool {
+ return strings.HasPrefix(o.MimeType, "image")
+}
+
+func GetInfoForBytes(name string, data []byte) (*FileInfo, *AppError) {
+ info := &FileInfo{
+ Name: name,
+ Size: int64(len(data)),
+ }
+ var err *AppError
+
+ extension := strings.ToLower(filepath.Ext(name))
+ info.MimeType = mime.TypeByExtension(extension)
+
+ if extension != "" && extension[0] == '.' {
+ // The client expects a file extension without the leading period
+ info.Extension = extension[1:]
+ } else {
+ info.Extension = extension
+ }
+
+ if info.IsImage() {
+ // Only set the width and height if it's actually an image that we can understand
+ if config, _, err := image.DecodeConfig(bytes.NewReader(data)); err == nil {
+ info.Width = config.Width
+ info.Height = config.Height
+
+ if info.MimeType == "image/gif" {
+ // Just show the gif itself instead of a preview image for animated gifs
+ if gifConfig, err := gif.DecodeAll(bytes.NewReader(data)); err != nil {
+ // Still return the rest of the info even though it doesn't appear to be an actual gif
+ info.HasPreviewImage = true
+ err = NewLocAppError("GetInfoForBytes", "model.file_info.get.gif.app_error", nil, "name="+name)
+ } else {
+ info.HasPreviewImage = len(gifConfig.Image) == 1
+ }
+ } else {
+ info.HasPreviewImage = true
+ }
+ }
+ }
+
+ return info, err
+}
+
+func GetEtagForFileInfos(infos []*FileInfo) string {
+ if len(infos) == 0 {
+ return Etag()
+ }
+
+ var maxUpdateAt int64
+
+ for _, info := range infos {
+ if info.UpdateAt > maxUpdateAt {
+ maxUpdateAt = info.UpdateAt
+ }
+ }
+
+ return Etag(infos[0].PostId, maxUpdateAt)
}