Test: thumbnail and authn / Modify: read thumbnail config from file

pkg/filesystem/image.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/conf"
 	"github.com/HFO4/cloudreve/pkg/filesystem/response"
 	"github.com/HFO4/cloudreve/pkg/thumb"
 	"github.com/HFO4/cloudreve/pkg/util"
@@ -54,7 +55,7 @@ func (fs *FileSystem) GenerateThumbnail(ctx context.Context, file *model.File) {
 
 	image, err := thumb.NewThumbFromFile(source, file.Name)
 	if err != nil {
-		util.Log().Warning("生成缩略图时无法解析[%s]图像数据：%s", file.SourceName, err)
+		util.Log().Warning("生成缩略图时无法解析 [%s] 图像数据：%s", file.SourceName, err)
 		return
 	}
 
@@ -64,9 +65,9 @@ func (fs *FileSystem) GenerateThumbnail(ctx context.Context, file *model.File) {
 	// 生成缩略图
 	image.GetThumb(fs.GenerateThumbnailSize(w, h))
 	// 保存到文件
-	err = image.Save(file.SourceName + "._thumb")
+	err = image.Save(file.SourceName + conf.ThumbConfig.FileSuffix)
 	if err != nil {
-		util.Log().Warning("无法保存缩略图：%s", file.SourceName, err)
+		util.Log().Warning("无法保存缩略图：%s", err)
 		return
 	}
 
@@ -75,12 +76,12 @@ func (fs *FileSystem) GenerateThumbnail(ctx context.Context, file *model.File) {
 
 	// 失败时删除缩略图文件
 	if err != nil {
-		_, _ = fs.Handler.Delete(newCtx, []string{file.SourceName + "._thumb"})
+		_, _ = fs.Handler.Delete(newCtx, []string{file.SourceName + conf.ThumbConfig.FileSuffix})
 	}
 }
 
 // GenerateThumbnailSize 获取要生成的缩略图的尺寸
-// TODO 从配置文件读取
+// TODO 优先从数据库中获得
 func (fs *FileSystem) GenerateThumbnailSize(w, h int) (uint, uint) {
-	return 400, 300
+	return conf.ThumbConfig.MaxWidth, conf.ThumbConfig.MaxHeight
 }

pkg/filesystem/image_test.go

@@ -0,0 +1,156 @@
+package filesystem
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"github.com/DATA-DOG/go-sqlmock"
+	model "github.com/HFO4/cloudreve/models"
+	"github.com/HFO4/cloudreve/pkg/conf"
+	"github.com/HFO4/cloudreve/pkg/filesystem/response"
+	"github.com/HFO4/cloudreve/pkg/util"
+	"github.com/jinzhu/gorm"
+	"github.com/stretchr/testify/assert"
+	testMock "github.com/stretchr/testify/mock"
+	"image"
+	"image/jpeg"
+	"os"
+	"testing"
+)
+
+func CreateTestImage() *os.File {
+	file, err := os.Create("TestFileSystem_GenerateThumbnail.jpeg")
+	alpha := image.NewAlpha(image.Rect(0, 0, 500, 200))
+	jpeg.Encode(file, alpha, nil)
+	if err != nil {
+		fmt.Println(err)
+	}
+	_, _ = file.Seek(0, 0)
+	return file
+}
+
+func TestFileSystem_GetThumb(t *testing.T) {
+	asserts := assert.New(t)
+	fs := FileSystem{
+		User: &model.User{
+			Model: gorm.Model{ID: 1},
+		},
+	}
+	ctx := context.Background()
+
+	// 正常
+	{
+		testHandler := new(FileHeaderMock)
+		testHandler.On("Thumb", testMock.Anything, "123.jpg").Return(&response.ContentResponse{URL: "123"}, nil)
+		fs.Handler = testHandler
+		mock.ExpectQuery("SELECT(.+)").
+			WithArgs(10, 1).
+			WillReturnRows(
+				sqlmock.NewRows(
+					[]string{"id", "pic_info", "source_name"}).
+					AddRow(10, "10,10", "123.jpg"),
+			)
+
+		res, err := fs.GetThumb(ctx, 10)
+		asserts.NoError(mock.ExpectationsWereMet())
+		testHandler.AssertExpectations(t)
+		asserts.NoError(err)
+		asserts.Equal("123", res.URL)
+	}
+
+	// 文件不存在
+	{
+
+		mock.ExpectQuery("SELECT(.+)").
+			WithArgs(10, 1).
+			WillReturnRows(
+				sqlmock.NewRows(
+					[]string{"id", "pic_info", "source_name"}),
+			)
+
+		_, err := fs.GetThumb(ctx, 10)
+		asserts.NoError(mock.ExpectationsWereMet())
+		asserts.Error(err)
+	}
+}
+
+func TestFileSystem_GenerateThumbnail(t *testing.T) {
+	asserts := assert.New(t)
+	fs := FileSystem{
+		User: &model.User{
+			Model: gorm.Model{ID: 1},
+		},
+	}
+	ctx := context.Background()
+
+	// 成功
+	{
+		src := CreateTestImage()
+		testHandler := new(FileHeaderMock)
+		testHandler.On("Get", testMock.Anything, "TestFileSystem_GenerateThumbnail.jpeg").Return(src, nil)
+		fs.Handler = testHandler
+
+		mock.ExpectBegin()
+		mock.ExpectExec("UPDATE(.+)").WillReturnResult(sqlmock.NewResult(1, 1))
+		mock.ExpectCommit()
+
+		file := &model.File{
+			Name:       "123.jpg",
+			SourceName: "TestFileSystem_GenerateThumbnail.jpeg",
+		}
+
+		fs.GenerateThumbnail(ctx, file)
+		asserts.NoError(mock.ExpectationsWereMet())
+		testHandler.AssertExpectations(t)
+		asserts.True(util.Exists("TestFileSystem_GenerateThumbnail.jpeg" + conf.ThumbConfig.FileSuffix))
+
+	}
+
+	// 更新信息失败后删除文件
+	{
+		src := CreateTestImage()
+		testHandler := new(FileHeaderMock)
+		testHandler.On("Get", testMock.Anything, "TestFileSystem_GenerateThumbnail.jpeg").Return(src, nil)
+		testHandler.On("Delete", testMock.Anything, testMock.Anything).Return([]string{}, nil)
+		fs.Handler = testHandler
+
+		mock.ExpectBegin()
+		mock.ExpectExec("UPDATE(.+)").WillReturnError(errors.New("error"))
+		mock.ExpectRollback()
+
+		file := &model.File{
+			Name:       "123.jpg",
+			SourceName: "TestFileSystem_GenerateThumbnail.jpeg",
+		}
+
+		fs.GenerateThumbnail(ctx, file)
+		asserts.NoError(mock.ExpectationsWereMet())
+		testHandler.AssertExpectations(t)
+
+	}
+
+	// 不能生成缩略图
+	{
+		file := &model.File{
+			Name:       "123.123",
+			SourceName: "TestFileSystem_GenerateThumbnail.jpeg",
+		}
+
+		fs.GenerateThumbnail(ctx, file)
+		asserts.NoError(mock.ExpectationsWereMet())
+	}
+
+}
+
+func TestFileSystem_GenerateThumbnailSize(t *testing.T) {
+	asserts := assert.New(t)
+	fs := FileSystem{
+		User: &model.User{
+			Model: gorm.Model{ID: 1},
+		},
+	}
+	asserts.NotPanics(func() {
+		_, _ = fs.GenerateThumbnailSize(0, 0)
+	})
+
+}

pkg/filesystem/local/handler.go

@@ -2,6 +2,7 @@ package local
 
 import (
 	"context"
+	"github.com/HFO4/cloudreve/pkg/conf"
 	"github.com/HFO4/cloudreve/pkg/filesystem/response"
 	"github.com/HFO4/cloudreve/pkg/util"
 	"io"
@@ -78,7 +79,7 @@ func (handler Handler) Delete(ctx context.Context, files []string) ([]string, er
 		}
 
 		// 尝试删除文件的缩略图（如果有）
-		_ = os.Remove(value + "._thumb")
+		_ = os.Remove(value + conf.ThumbConfig.FileSuffix)
 	}
 
 	return deleteFailed, retErr
@@ -86,7 +87,7 @@ func (handler Handler) Delete(ctx context.Context, files []string) ([]string, er
 
 // Thumb 获取文件缩略图
 func (handler Handler) Thumb(ctx context.Context, path string) (*response.ContentResponse, error) {
-	file, err := handler.Get(ctx, path+"._thumb")
+	file, err := handler.Get(ctx, path+conf.ThumbConfig.FileSuffix)
 	if err != nil {
 		return nil, err
 	}