484 lines
13 KiB
Go
484 lines
13 KiB
Go
package rest
|
||
|
||
import (
|
||
"fmt"
|
||
"net/http"
|
||
"os"
|
||
"regexp"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
type AlienController struct {
|
||
BaseController
|
||
uploadTokenDao *UploadTokenDao
|
||
downloadTokenDao *DownloadTokenDao
|
||
matterDao *MatterDao
|
||
matterService *MatterService
|
||
imageCacheDao *ImageCacheDao
|
||
imageCacheService *ImageCacheService
|
||
}
|
||
|
||
//初始化方法
|
||
func (this *AlienController) Init(context *Context) {
|
||
this.BaseController.Init(context)
|
||
|
||
//手动装填本实例的Bean.
|
||
b := context.GetBean(this.uploadTokenDao)
|
||
if c, ok := b.(*UploadTokenDao); ok {
|
||
this.uploadTokenDao = c
|
||
}
|
||
|
||
b = context.GetBean(this.downloadTokenDao)
|
||
if c, ok := b.(*DownloadTokenDao); ok {
|
||
this.downloadTokenDao = c
|
||
}
|
||
|
||
b = context.GetBean(this.matterDao)
|
||
if c, ok := b.(*MatterDao); ok {
|
||
this.matterDao = c
|
||
}
|
||
|
||
b = context.GetBean(this.matterService)
|
||
if c, ok := b.(*MatterService); ok {
|
||
this.matterService = c
|
||
}
|
||
|
||
b = context.GetBean(this.imageCacheDao)
|
||
if c, ok := b.(*ImageCacheDao); ok {
|
||
this.imageCacheDao = c
|
||
}
|
||
|
||
b = context.GetBean(this.imageCacheService)
|
||
if c, ok := b.(*ImageCacheService); ok {
|
||
this.imageCacheService = c
|
||
}
|
||
|
||
}
|
||
|
||
//注册自己的路由。
|
||
func (this *AlienController) RegisterRoutes() map[string]func(writer http.ResponseWriter, request *http.Request) {
|
||
|
||
routeMap := make(map[string]func(writer http.ResponseWriter, request *http.Request))
|
||
|
||
//每个Controller需要主动注册自己的路由。
|
||
routeMap["/api/alien/fetch/upload/token"] = this.Wrap(this.FetchUploadToken, USER_ROLE_GUEST)
|
||
routeMap["/api/alien/fetch/download/token"] = this.Wrap(this.FetchDownloadToken, USER_ROLE_GUEST)
|
||
routeMap["/api/alien/confirm"] = this.Wrap(this.Confirm, USER_ROLE_GUEST)
|
||
routeMap["/api/alien/upload"] = this.Wrap(this.Upload, USER_ROLE_GUEST)
|
||
routeMap["/api/alien/crawl/token"] = this.Wrap(this.CrawlToken, USER_ROLE_GUEST)
|
||
routeMap["/api/alien/crawl/direct"] = this.Wrap(this.CrawlDirect, USER_ROLE_GUEST)
|
||
|
||
return routeMap
|
||
}
|
||
|
||
//处理一些特殊的接口，比如参数包含在路径中,一般情况下，controller不将参数放在url路径中
|
||
func (this *AlienController) HandleRoutes(writer http.ResponseWriter, request *http.Request) (func(writer http.ResponseWriter, request *http.Request), bool) {
|
||
|
||
path := request.URL.Path
|
||
|
||
//匹配 /api/alien/download/{uuid}/{filename}
|
||
reg := regexp.MustCompile(`^/api/alien/download/([^/]+)/([^/]+)$`)
|
||
strs := reg.FindStringSubmatch(path)
|
||
if len(strs) != 3 {
|
||
return nil, false
|
||
} else {
|
||
var f = func(writer http.ResponseWriter, request *http.Request) {
|
||
this.Download(writer, request, strs[1], strs[2])
|
||
}
|
||
return f, true
|
||
}
|
||
}
|
||
|
||
//直接使用邮箱和密码获取用户
|
||
func (this *AlienController) CheckRequestUser(email, password string) *User {
|
||
|
||
if email == "" {
|
||
panic("邮箱必填啦")
|
||
}
|
||
|
||
if password == "" {
|
||
panic("密码必填")
|
||
}
|
||
|
||
//验证用户身份合法性。
|
||
user := this.userDao.FindByEmail(email)
|
||
if user == nil {
|
||
panic(`邮箱或密码错误`)
|
||
} else {
|
||
if !MatchBcrypt(password, user.Password) {
|
||
panic(`邮箱或密码错误`)
|
||
}
|
||
}
|
||
return user
|
||
}
|
||
|
||
//系统中的用户x要获取一个UploadToken，用于提供给x信任的用户上传文件。
|
||
func (this *AlienController) FetchUploadToken(writer http.ResponseWriter, request *http.Request) *WebResult {
|
||
|
||
//文件名。
|
||
filename := request.FormValue("filename")
|
||
if filename == "" {
|
||
panic("文件名必填")
|
||
} else if m, _ := regexp.MatchString(`[<>|*?/\\]`, filename); m {
|
||
panic(fmt.Sprintf(`【%s】不符合要求，文件名中不能包含以下特殊符号：< > | * ? / \`, filename))
|
||
}
|
||
|
||
//什么时间后过期，默认24h
|
||
expireStr := request.FormValue("expire")
|
||
expire := 24 * 60 * 60
|
||
if expireStr != "" {
|
||
var err error
|
||
expire, err = strconv.Atoi(expireStr)
|
||
if err != nil {
|
||
panic(`过期时间不符合规范`)
|
||
}
|
||
if expire < 1 {
|
||
panic(`过期时间不符合规范`)
|
||
}
|
||
|
||
}
|
||
|
||
//文件公有或私有
|
||
privacyStr := request.FormValue("privacy")
|
||
var privacy bool
|
||
if privacyStr == "" {
|
||
panic(`文件公有性必填`)
|
||
} else {
|
||
if privacyStr == "true" {
|
||
privacy = true
|
||
} else if privacyStr == "false" {
|
||
privacy = false
|
||
} else {
|
||
panic(`文件公有性不符合规范`)
|
||
}
|
||
}
|
||
|
||
//文件大小
|
||
sizeStr := request.FormValue("size")
|
||
var size int64
|
||
if sizeStr == "" {
|
||
panic(`文件大小必填`)
|
||
} else {
|
||
|
||
var err error
|
||
size, err = strconv.ParseInt(sizeStr, 10, 64)
|
||
if err != nil {
|
||
panic(`文件大小不符合规范`)
|
||
}
|
||
if size < 1 {
|
||
panic(`文件大小不符合规范`)
|
||
}
|
||
}
|
||
|
||
//文件夹路径，以 / 开头。
|
||
dir := request.FormValue("dir")
|
||
|
||
user := this.CheckRequestUser(request.FormValue("email"), request.FormValue("password"))
|
||
dirUuid := this.matterService.GetDirUuid(user.Uuid, dir)
|
||
|
||
mm, _ := time.ParseDuration(fmt.Sprintf("%ds", expire))
|
||
uploadToken := &UploadToken{
|
||
UserUuid: user.Uuid,
|
||
FolderUuid: dirUuid,
|
||
MatterUuid: "",
|
||
ExpireTime: time.Now().Add(mm),
|
||
Filename: filename,
|
||
Privacy: privacy,
|
||
Size: size,
|
||
Ip: GetIpAddress(request),
|
||
}
|
||
|
||
uploadToken = this.uploadTokenDao.Create(uploadToken)
|
||
|
||
return this.Success(uploadToken)
|
||
|
||
}
|
||
|
||
//系统中的用户x 拿着某个文件的uuid来确认是否其信任的用户已经上传好了。
|
||
func (this *AlienController) Confirm(writer http.ResponseWriter, request *http.Request) *WebResult {
|
||
|
||
matterUuid := request.FormValue("matterUuid")
|
||
if matterUuid == "" {
|
||
panic("matterUuid必填")
|
||
}
|
||
|
||
user := this.CheckRequestUser(request.FormValue("email"), request.FormValue("password"))
|
||
|
||
matter := this.matterDao.CheckByUuid(matterUuid)
|
||
if matter.UserUuid != user.Uuid {
|
||
panic("文件不属于你")
|
||
}
|
||
|
||
return this.Success(matter)
|
||
}
|
||
|
||
//系统中的用户x 信任的用户上传文件。这个接口需要支持跨域。
|
||
func (this *AlienController) Upload(writer http.ResponseWriter, request *http.Request) *WebResult {
|
||
//允许跨域请求。
|
||
this.allowCORS(writer)
|
||
if request.Method == "OPTIONS" {
|
||
return this.Success("OK")
|
||
}
|
||
|
||
uploadTokenUuid := request.FormValue("uploadTokenUuid")
|
||
if uploadTokenUuid == "" {
|
||
panic("uploadTokenUuid必填")
|
||
}
|
||
|
||
uploadToken := this.uploadTokenDao.FindByUuid(uploadTokenUuid)
|
||
if uploadToken == nil {
|
||
panic("uploadTokenUuid无效")
|
||
}
|
||
|
||
if uploadToken.ExpireTime.Before(time.Now()) {
|
||
panic("uploadToken已失效")
|
||
}
|
||
|
||
user := this.userDao.CheckByUuid(uploadToken.UserUuid)
|
||
|
||
request.ParseMultipartForm(32 << 20)
|
||
file, handler, err := request.FormFile("file")
|
||
this.PanicError(err)
|
||
defer file.Close()
|
||
|
||
if handler.Filename != uploadToken.Filename {
|
||
panic("文件名称不正确")
|
||
}
|
||
|
||
if handler.Size != uploadToken.Size {
|
||
panic("文件大小不正确")
|
||
}
|
||
|
||
matter := this.matterService.Upload(file, user, uploadToken.FolderUuid, uploadToken.Filename, uploadToken.Privacy, true)
|
||
|
||
//更新这个uploadToken的信息.
|
||
uploadToken.ExpireTime = time.Now()
|
||
this.uploadTokenDao.Save(uploadToken)
|
||
|
||
return this.Success(matter)
|
||
}
|
||
|
||
//给一个指定的url，从该url中去拉取文件回来。此处采用uploadToken的模式。
|
||
func (this *AlienController) CrawlToken(writer http.ResponseWriter, request *http.Request) *WebResult {
|
||
//允许跨域请求。
|
||
this.allowCORS(writer)
|
||
if request.Method == "OPTIONS" {
|
||
return this.Success("OK")
|
||
}
|
||
|
||
uploadTokenUuid := request.FormValue("uploadTokenUuid")
|
||
if uploadTokenUuid == "" {
|
||
panic("uploadTokenUuid必填")
|
||
}
|
||
|
||
uploadToken := this.uploadTokenDao.FindByUuid(uploadTokenUuid)
|
||
if uploadToken == nil {
|
||
panic("uploadTokenUuid无效")
|
||
}
|
||
|
||
if uploadToken.ExpireTime.Before(time.Now()) {
|
||
panic("uploadToken已失效")
|
||
}
|
||
|
||
user := this.userDao.CheckByUuid(uploadToken.UserUuid)
|
||
|
||
url := request.FormValue("url")
|
||
if url == "" || (!strings.HasPrefix(url, "http://") && !strings.HasPrefix(url, "https://")) {
|
||
panic("资源url必填，并且应该以http://或者https://开头")
|
||
}
|
||
|
||
matter := this.matterService.Crawl(url, uploadToken.Filename, user, uploadToken.FolderUuid, uploadToken.Privacy)
|
||
|
||
//更新这个uploadToken的信息.
|
||
uploadToken.ExpireTime = time.Now()
|
||
this.uploadTokenDao.Save(uploadToken)
|
||
|
||
return this.Success(matter)
|
||
}
|
||
|
||
//通过一个url直接上传，无需借助uploadToken.
|
||
func (this *AlienController) CrawlDirect(writer http.ResponseWriter, request *http.Request) *WebResult {
|
||
|
||
//文件名。
|
||
filename := request.FormValue("filename")
|
||
if filename == "" {
|
||
panic("文件名必填")
|
||
} else if m, _ := regexp.MatchString(`[<>|*?/\\]`, filename); m {
|
||
panic(fmt.Sprintf(`【%s】不符合要求，文件名中不能包含以下特殊符号：< > | * ? / \`, filename))
|
||
}
|
||
|
||
url := request.FormValue("url")
|
||
if url == "" || (!strings.HasPrefix(url, "http://") && !strings.HasPrefix(url, "https://")) {
|
||
panic("资源url必填，并且应该以http://或者https://开头")
|
||
}
|
||
|
||
//文件公有或私有
|
||
privacyStr := request.FormValue("privacy")
|
||
var privacy bool
|
||
if privacyStr == "" {
|
||
panic(`文件公有性必填`)
|
||
} else {
|
||
if privacyStr == "true" {
|
||
privacy = true
|
||
} else if privacyStr == "false" {
|
||
privacy = false
|
||
} else {
|
||
panic(`文件公有性不符合规范`)
|
||
}
|
||
}
|
||
|
||
//文件夹路径，以 / 开头。
|
||
dir := request.FormValue("dir")
|
||
user := this.CheckRequestUser(request.FormValue("email"), request.FormValue("password"))
|
||
dirUuid := this.matterService.GetDirUuid(user.Uuid, dir)
|
||
|
||
matter := this.matterService.Crawl(url, filename, user, dirUuid, privacy)
|
||
|
||
return this.Success(matter)
|
||
}
|
||
|
||
//系统中的用户x要获取一个DownloadToken，用于提供给x信任的用户下载文件。
|
||
func (this *AlienController) FetchDownloadToken(writer http.ResponseWriter, request *http.Request) *WebResult {
|
||
|
||
matterUuid := request.FormValue("matterUuid")
|
||
if matterUuid == "" {
|
||
panic("matterUuid必填")
|
||
}
|
||
|
||
user := this.CheckRequestUser(request.FormValue("email"), request.FormValue("password"))
|
||
|
||
matter := this.matterDao.CheckByUuid(matterUuid)
|
||
if matter.UserUuid != user.Uuid {
|
||
panic("文件不属于你")
|
||
}
|
||
if matter.Dir {
|
||
panic("不支持下载文件夹")
|
||
}
|
||
|
||
//什么时间后过期，默认24h
|
||
expireStr := request.FormValue("expire")
|
||
expire := 24 * 60 * 60
|
||
if expireStr != "" {
|
||
var err error
|
||
expire, err = strconv.Atoi(expireStr)
|
||
if err != nil {
|
||
panic(`过期时间不符合规范`)
|
||
}
|
||
if expire < 1 {
|
||
panic(`过期时间不符合规范`)
|
||
}
|
||
|
||
}
|
||
|
||
mm, _ := time.ParseDuration(fmt.Sprintf("%ds", expire))
|
||
downloadToken := &DownloadToken{
|
||
UserUuid: user.Uuid,
|
||
MatterUuid: matterUuid,
|
||
ExpireTime: time.Now().Add(mm),
|
||
Ip: GetIpAddress(request),
|
||
}
|
||
|
||
downloadToken = this.downloadTokenDao.Create(downloadToken)
|
||
|
||
return this.Success(downloadToken)
|
||
|
||
}
|
||
|
||
//下载一个文件。既可以使用登录的方式下载，也可以使用授权的方式下载。
|
||
func (this *AlienController) Download(writer http.ResponseWriter, request *http.Request, uuid string, filename string) {
|
||
|
||
matter := this.matterDao.CheckByUuid(uuid)
|
||
|
||
//判断是否是文件夹
|
||
if matter.Dir {
|
||
panic("暂不支持下载文件夹")
|
||
}
|
||
|
||
if matter.Name != filename {
|
||
panic("文件信息错误")
|
||
}
|
||
|
||
//验证用户的权限问题。
|
||
//文件如果是私有的才需要权限
|
||
if matter.Privacy {
|
||
|
||
//1.如果带有downloadTokenUuid那么就按照token的信息去获取。
|
||
downloadTokenUuid := request.FormValue("downloadTokenUuid")
|
||
if downloadTokenUuid != "" {
|
||
downloadToken := this.downloadTokenDao.CheckByUuid(downloadTokenUuid)
|
||
if downloadToken.ExpireTime.Before(time.Now()) {
|
||
panic("downloadToken已失效")
|
||
}
|
||
|
||
if downloadToken.MatterUuid != uuid {
|
||
panic("token和文件信息不一致")
|
||
}
|
||
|
||
tokenUser := this.userDao.CheckByUuid(downloadToken.UserUuid)
|
||
if matter.UserUuid != tokenUser.Uuid {
|
||
panic(RESULT_CODE_UNAUTHORIZED)
|
||
}
|
||
|
||
//下载之后立即过期掉。
|
||
downloadToken.ExpireTime = time.Now().AddDate(0, 0, 1);
|
||
this.downloadTokenDao.Save(downloadToken)
|
||
|
||
} else {
|
||
|
||
//判断文件的所属人是否正确
|
||
user := this.checkUser(writer, request)
|
||
if user.Role != USER_ROLE_ADMINISTRATOR && matter.UserUuid != user.Uuid {
|
||
panic(RESULT_CODE_UNAUTHORIZED)
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
//对图片做缩放处理。
|
||
imageProcess := request.FormValue("imageProcess")
|
||
if imageProcess == "resize" {
|
||
|
||
//如果是图片，那么能用缓存就用缓存
|
||
uri := request.RequestURI
|
||
imageCache := this.imageCacheDao.FindByUri(uri)
|
||
if imageCache != nil {
|
||
|
||
//直接使用缓存中的信息
|
||
this.matterService.DownloadFile(writer, request, CONFIG.MatterPath+imageCache.Path, matter.Name)
|
||
|
||
} else {
|
||
|
||
//resize图片
|
||
dstImage := this.matterService.ResizeImage(request, CONFIG.MatterPath+matter.Path)
|
||
|
||
user := this.userDao.FindByUuid(matter.UserUuid)
|
||
//获取文件应该存放在的物理路径的绝对路径和相对路径。
|
||
absolutePath, relativePath := GetUserFilePath(user.Username, true)
|
||
absolutePath = absolutePath + "/" + filename
|
||
relativePath = relativePath + "/" + filename
|
||
|
||
fileWriter, err := os.Create(absolutePath)
|
||
this.PanicError(err)
|
||
defer fileWriter.Close()
|
||
|
||
//生成的图片输出到两个writer中去
|
||
this.matterService.DownloadImage(writer, fileWriter, dstImage, matter.Name)
|
||
|
||
//相关信息写到缓存中去
|
||
imageCache = &ImageCache{
|
||
UserUuid: matter.UserUuid,
|
||
MatterUuid: matter.Uuid,
|
||
Uri: uri,
|
||
Path: relativePath,
|
||
}
|
||
this.imageCacheDao.Create(imageCache)
|
||
}
|
||
|
||
} else {
|
||
this.matterService.DownloadFile(writer, request, CONFIG.MatterPath+matter.Path, matter.Name)
|
||
}
|
||
|
||
}
|