From 003274162bce8b99b23d5bf443173622ff400201 Mon Sep 17 00:00:00 2001
From: HFO4 <912394456@qq.com>
Date: Sat, 16 Nov 2019 13:37:42 +0800
Subject: [PATCH] Add: filesystem

---
 pkg/filesystem/filesystem.go  | 28 +++++++++++++++++++++++
 pkg/filesystem/local/file.go  | 26 +++++++++++++++++++++
 pkg/filesystem/local/local.go |  1 +
 routers/controllers/file.go   | 19 ++++++----------
 routers/controllers/main.go   |  1 +
 service/file/upload.go        | 43 +++++++++++++++++++++++++++++++++++
 6 files changed, 106 insertions(+), 12 deletions(-)
 create mode 100644 pkg/filesystem/filesystem.go
 create mode 100644 pkg/filesystem/local/file.go
 create mode 100644 pkg/filesystem/local/local.go

diff --git a/pkg/filesystem/filesystem.go b/pkg/filesystem/filesystem.go
new file mode 100644
index 0000000..16debd4
--- /dev/null
+++ b/pkg/filesystem/filesystem.go
@@ -0,0 +1,28 @@
+package filesystem
+
+import (
+	"cloudreve/models"
+	"io"
+)
+
+// FileData 上传来的文件数据处理器
+type FileData interface {
+	io.Reader
+	io.Closer
+	GetSize() uint64
+	GetMIMEType() string
+}
+
+// FileSystem 管理文件的文件系统
+type FileSystem struct {
+	// 文件系统所有者
+	User *model.User
+
+	// 文件系统处理适配器
+
+}
+
+// Upload 上传文件
+func (fs *FileSystem) Upload(File FileData) (err error) {
+	return nil
+}
diff --git a/pkg/filesystem/local/file.go b/pkg/filesystem/local/file.go
new file mode 100644
index 0000000..38698dc
--- /dev/null
+++ b/pkg/filesystem/local/file.go
@@ -0,0 +1,26 @@
+package local
+
+import "mime/multipart"
+
+// FileData 上传的文件数据
+type FileData struct {
+	File     multipart.File
+	Size     uint64
+	MIMEType string
+}
+
+func (file FileData) Read(p []byte) (n int, err error) {
+	return file.Read(p)
+}
+
+func (file FileData) GetMIMEType() string {
+	return file.MIMEType
+}
+
+func (file FileData) GetSize() uint64 {
+	return file.Size
+}
+
+func (file FileData) Close() error {
+	return file.Close()
+}
diff --git a/pkg/filesystem/local/local.go b/pkg/filesystem/local/local.go
new file mode 100644
index 0000000..469c3dc
--- /dev/null
+++ b/pkg/filesystem/local/local.go
@@ -0,0 +1 @@
+package local
diff --git a/routers/controllers/file.go b/routers/controllers/file.go
index 4dc9f7f..5fd24e3 100644
--- a/routers/controllers/file.go
+++ b/routers/controllers/file.go
@@ -3,6 +3,7 @@ package controllers
 import (
 	"cloudreve/models"
 	"cloudreve/pkg/serializer"
+	"cloudreve/service/file"
 	"github.com/gin-gonic/gin"
 )
 
@@ -15,18 +16,12 @@ func FileUpload(c *gin.Context) {
 		return
 	}
 
-	name := c.PostForm("name")
-	path := c.PostForm("path")
+	var service file.UploadService
 
-	// Source
-	_, err := c.FormFile("file")
-	if err != nil {
-		c.JSON(200, serializer.Err(serializer.CodeIOFailed, "无法获取文件数据", err))
-		return
+	if err := c.ShouldBind(&service); err == nil {
+		res := service.Upload(c)
+		c.JSON(200, res)
+	} else {
+		c.JSON(200, ErrorResponse(err))
 	}
-
-	c.JSON(200, serializer.Response{
-		Code: 0,
-		Msg:  name + path,
-	})
 }
diff --git a/routers/controllers/main.go b/routers/controllers/main.go
index 99e6614..692d25a 100644
--- a/routers/controllers/main.go
+++ b/routers/controllers/main.go
@@ -14,6 +14,7 @@ func ParamErrorMsg(filed string, tag string) string {
 	fieldMap := map[string]string{
 		"UserName": "邮箱",
 		"Password": "密码",
+		"Path":     "路径",
 	}
 	// 未通过的规则与中文对应
 	tagMap := map[string]string{
diff --git a/service/file/upload.go b/service/file/upload.go
index b691ba5..68403df 100644
--- a/service/file/upload.go
+++ b/service/file/upload.go
@@ -1 +1,44 @@
 package file
+
+import (
+	"cloudreve/models"
+	"cloudreve/pkg/filesystem"
+	"cloudreve/pkg/filesystem/local"
+	"cloudreve/pkg/serializer"
+	"github.com/gin-gonic/gin"
+	"mime/multipart"
+)
+
+// UploadService 本地策略文件上传服务
+type UploadService struct {
+	Name string                `form:"name" binding:"required,lte=255"`
+	Path string                `form:"path" binding:"lte=65536"`
+	File *multipart.FileHeader `form:"file" binding:"required"`
+}
+
+// Upload 处理本地策略小文件上传
+func (service *UploadService) Upload(c *gin.Context) serializer.Response {
+	// TODO 检查文件大小是否小于分片最大大小
+	file, err := service.File.Open()
+	if err != nil {
+		return serializer.Err(serializer.CodeIOFailed, "无法打开上传数据", err)
+	}
+
+	fileData := local.FileData{
+		MIMEType: service.File.Header.Get("Content-Type"),
+		File:     file,
+		Size:     uint64(service.File.Size),
+	}
+
+	user, _ := c.Get("user")
+
+	fs := filesystem.FileSystem{
+		User: user.(*model.User),
+	}
+	err = fs.Upload(fileData)
+
+	return serializer.Response{
+		Code: 0,
+		Msg:  "Pong",
+	}
+}