focalboard/server/app/files.go

68 lines
1.6 KiB
Go
Raw Normal View History

package app
import (
"errors"
"fmt"
"io"
2021-04-01 00:30:25 +02:00
"log"
"os"
"path/filepath"
"strings"
2021-01-26 23:13:46 +01:00
"github.com/mattermost/focalboard/server/utils"
"github.com/mattermost/mattermost-server/v5/shared/filestore"
)
func (a *App) SaveFile(reader io.Reader, workspaceID, rootID, filename string) (string, error) {
// NOTE: File extension includes the dot
fileExtension := strings.ToLower(filepath.Ext(filename))
if fileExtension == ".jpeg" {
fileExtension = ".jpg"
}
createdFilename := fmt.Sprintf(`%s%s`, utils.CreateGUID(), fileExtension)
2021-03-30 19:06:11 +02:00
filePath := filepath.Join(workspaceID, rootID, createdFilename)
_, appErr := a.filesBackend.WriteFile(reader, filePath)
if appErr != nil {
return "", errors.New("unable to store the file in the files storage")
}
return createdFilename, nil
}
func (a *App) GetFileReader(workspaceID, rootID, filename string) (filestore.ReadCloseSeeker, error) {
filePath := filepath.Join(workspaceID, rootID, filename)
exists, err := a.filesBackend.FileExists(filePath)
if err != nil {
return nil, err
}
2021-04-01 00:30:25 +02:00
// FIXUP: Check the deprecated old location
if workspaceID == "0" && !exists {
oldExists, err := a.filesBackend.FileExists(filename)
if err != nil {
return nil, err
}
if oldExists {
err := a.filesBackend.MoveFile(filename, filePath)
2021-04-01 00:30:25 +02:00
if err != nil {
log.Printf("ERROR moving old file from '%s' to '%s'", filename, filePath)
2021-04-01 00:30:25 +02:00
} else {
log.Printf("Moved old file from '%s' to '%s'", filename, filePath)
2021-04-01 00:30:25 +02:00
}
}
}
reader, err := a.filesBackend.Reader(filePath)
if err != nil {
return nil, err
}
return reader, nil
2021-04-01 00:30:25 +02:00
}
func fileExists(path string) bool {
_, err := os.Stat(path)
return !os.IsNotExist(err)
}