352 lines
9.7 KiB
Go
352 lines
9.7 KiB
Go
package rest
|
|
|
|
import (
|
|
"github.com/eyebluecn/tank/code/core"
|
|
"github.com/eyebluecn/tank/code/tool/result"
|
|
"github.com/eyebluecn/tank/code/tool/util"
|
|
"net/http"
|
|
"regexp"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type AlienController struct {
|
|
BaseController
|
|
uploadTokenDao *UploadTokenDao
|
|
downloadTokenDao *DownloadTokenDao
|
|
matterDao *MatterDao
|
|
matterService *MatterService
|
|
imageCacheDao *ImageCacheDao
|
|
imageCacheService *ImageCacheService
|
|
alienService *AlienService
|
|
shareService *ShareService
|
|
}
|
|
|
|
func (this *AlienController) Init() {
|
|
this.BaseController.Init()
|
|
|
|
b := core.CONTEXT.GetBean(this.uploadTokenDao)
|
|
if c, ok := b.(*UploadTokenDao); ok {
|
|
this.uploadTokenDao = c
|
|
}
|
|
|
|
b = core.CONTEXT.GetBean(this.downloadTokenDao)
|
|
if c, ok := b.(*DownloadTokenDao); ok {
|
|
this.downloadTokenDao = c
|
|
}
|
|
|
|
b = core.CONTEXT.GetBean(this.matterDao)
|
|
if c, ok := b.(*MatterDao); ok {
|
|
this.matterDao = c
|
|
}
|
|
|
|
b = core.CONTEXT.GetBean(this.matterService)
|
|
if c, ok := b.(*MatterService); ok {
|
|
this.matterService = c
|
|
}
|
|
|
|
b = core.CONTEXT.GetBean(this.imageCacheDao)
|
|
if c, ok := b.(*ImageCacheDao); ok {
|
|
this.imageCacheDao = c
|
|
}
|
|
|
|
b = core.CONTEXT.GetBean(this.imageCacheService)
|
|
if c, ok := b.(*ImageCacheService); ok {
|
|
this.imageCacheService = c
|
|
}
|
|
|
|
b = core.CONTEXT.GetBean(this.alienService)
|
|
if c, ok := b.(*AlienService); ok {
|
|
this.alienService = c
|
|
}
|
|
|
|
b = core.CONTEXT.GetBean(this.shareService)
|
|
if c, ok := b.(*ShareService); ok {
|
|
this.shareService = 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))
|
|
|
|
routeMap["/api/alien/fetch/upload/token"] = this.Wrap(this.FetchUploadToken, USER_ROLE_USER)
|
|
routeMap["/api/alien/fetch/download/token"] = this.Wrap(this.FetchDownloadToken, USER_ROLE_USER)
|
|
routeMap["/api/alien/confirm"] = this.Wrap(this.Confirm, USER_ROLE_USER)
|
|
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_USER)
|
|
|
|
return routeMap
|
|
}
|
|
|
|
//handle some special routes, eg. params in the url.
|
|
func (this *AlienController) HandleRoutes(writer http.ResponseWriter, request *http.Request) (func(writer http.ResponseWriter, request *http.Request), bool) {
|
|
|
|
path := request.URL.Path
|
|
|
|
//match /api/alien/preview/{uuid}/{filename} (response header not contain content-disposition)
|
|
reg := regexp.MustCompile(`^/api/alien/preview/([^/]+)/([^/]+)$`)
|
|
strs := reg.FindStringSubmatch(path)
|
|
if len(strs) == 3 {
|
|
var f = func(writer http.ResponseWriter, request *http.Request) {
|
|
this.Preview(writer, request, strs[1], strs[2])
|
|
}
|
|
return f, true
|
|
}
|
|
|
|
//match /api/alien/download/{uuid}/{filename} (response header contain content-disposition)
|
|
reg = regexp.MustCompile(`^/api/alien/download/([^/]+)/([^/]+)$`)
|
|
strs = reg.FindStringSubmatch(path)
|
|
if len(strs) == 3 {
|
|
var f = func(writer http.ResponseWriter, request *http.Request) {
|
|
this.Download(writer, request, strs[1], strs[2])
|
|
}
|
|
return f, true
|
|
}
|
|
|
|
return nil, false
|
|
}
|
|
|
|
//fetch a upload token for guest. Guest can upload file with this token.
|
|
func (this *AlienController) FetchUploadToken(writer http.ResponseWriter, request *http.Request) *result.WebResult {
|
|
|
|
filename := request.FormValue("filename")
|
|
expireTimeStr := request.FormValue("expireTime")
|
|
privacyStr := request.FormValue("privacy")
|
|
sizeStr := request.FormValue("size")
|
|
//store dir path
|
|
dirPath := request.FormValue("dirPath")
|
|
|
|
filename = CheckMatterName(request, filename)
|
|
|
|
var expireTime time.Time
|
|
if expireTimeStr == "" {
|
|
panic(result.BadRequest("time format error"))
|
|
} else {
|
|
expireTime = util.ConvertDateTimeStringToTime(expireTimeStr)
|
|
}
|
|
if expireTime.Before(time.Now()) {
|
|
panic(result.BadRequest("expire time cannot before now"))
|
|
}
|
|
|
|
var privacy = false
|
|
if privacyStr == TRUE {
|
|
privacy = true
|
|
}
|
|
|
|
var size int64
|
|
if sizeStr == "" {
|
|
panic(result.BadRequest("file size cannot be null"))
|
|
} else {
|
|
|
|
var err error
|
|
size, err = strconv.ParseInt(sizeStr, 10, 64)
|
|
if err != nil {
|
|
panic(result.BadRequest("file size error"))
|
|
}
|
|
if size < 1 {
|
|
panic(result.BadRequest("file size error"))
|
|
}
|
|
}
|
|
|
|
user := this.checkUser(request)
|
|
dirMatter := this.matterService.CreateDirectories(request, user, dirPath)
|
|
|
|
uploadToken := &UploadToken{
|
|
UserUuid: user.Uuid,
|
|
FolderUuid: dirMatter.Uuid,
|
|
MatterUuid: "",
|
|
ExpireTime: expireTime,
|
|
Filename: filename,
|
|
Privacy: privacy,
|
|
Size: size,
|
|
Ip: util.GetIpAddress(request),
|
|
}
|
|
|
|
uploadToken = this.uploadTokenDao.Create(uploadToken)
|
|
|
|
return this.Success(uploadToken)
|
|
|
|
}
|
|
|
|
//user confirm a file whether uploaded successfully.
|
|
func (this *AlienController) Confirm(writer http.ResponseWriter, request *http.Request) *result.WebResult {
|
|
|
|
matterUuid := request.FormValue("matterUuid")
|
|
if matterUuid == "" {
|
|
panic(result.BadRequest("matterUuid cannot be null"))
|
|
}
|
|
|
|
user := this.checkUser(request)
|
|
|
|
matter := this.matterDao.CheckByUuid(matterUuid)
|
|
if matter.UserUuid != user.Uuid {
|
|
panic(result.BadRequest("matter not belong to you"))
|
|
}
|
|
|
|
return this.Success(matter)
|
|
}
|
|
|
|
//a guest upload a file with a upload token.
|
|
func (this *AlienController) Upload(writer http.ResponseWriter, request *http.Request) *result.WebResult {
|
|
//allow cors.
|
|
this.allowCORS(writer)
|
|
if request.Method == "OPTIONS" {
|
|
//nil means empty response body.
|
|
return nil
|
|
}
|
|
|
|
uploadTokenUuid := request.FormValue("uploadTokenUuid")
|
|
file, handler, err := request.FormFile("file")
|
|
this.PanicError(err)
|
|
defer func() {
|
|
e := file.Close()
|
|
this.PanicError(e)
|
|
}()
|
|
|
|
if uploadTokenUuid == "" {
|
|
panic(result.BadRequest("uploadTokenUuid cannot be null"))
|
|
}
|
|
|
|
uploadToken := this.uploadTokenDao.CheckByUuid(uploadTokenUuid)
|
|
|
|
if uploadToken.ExpireTime.Before(time.Now()) {
|
|
panic(result.BadRequest("uploadToken has expired"))
|
|
}
|
|
|
|
user := this.userDao.CheckByUuid(uploadToken.UserUuid)
|
|
|
|
err = request.ParseMultipartForm(32 << 20)
|
|
this.PanicError(err)
|
|
|
|
if handler.Filename != uploadToken.Filename {
|
|
panic(result.BadRequest("filename doesn't the one in uploadToken"))
|
|
}
|
|
|
|
if handler.Size != uploadToken.Size {
|
|
panic(result.BadRequest("file size doesn't the one in uploadToken"))
|
|
}
|
|
|
|
dirMatter := this.matterDao.CheckWithRootByUuid(uploadToken.FolderUuid, user)
|
|
|
|
matter := this.matterService.Upload(request, file, user, dirMatter, uploadToken.Filename, uploadToken.Privacy)
|
|
|
|
//expire the upload token.
|
|
uploadToken.ExpireTime = time.Now()
|
|
this.uploadTokenDao.Save(uploadToken)
|
|
|
|
return this.Success(matter)
|
|
}
|
|
|
|
//crawl a url with uploadToken. guest can visit this method.
|
|
func (this *AlienController) CrawlToken(writer http.ResponseWriter, request *http.Request) *result.WebResult {
|
|
|
|
//allow cors.
|
|
this.allowCORS(writer)
|
|
if request.Method == "OPTIONS" {
|
|
//nil means empty response body.
|
|
return nil
|
|
}
|
|
|
|
uploadTokenUuid := request.FormValue("uploadTokenUuid")
|
|
url := request.FormValue("url")
|
|
|
|
if uploadTokenUuid == "" {
|
|
panic(result.BadRequest("uploadTokenUuid cannot be null"))
|
|
}
|
|
|
|
uploadToken := this.uploadTokenDao.CheckByUuid(uploadTokenUuid)
|
|
|
|
if uploadToken.ExpireTime.Before(time.Now()) {
|
|
panic(result.BadRequest("uploadToken has expired"))
|
|
}
|
|
|
|
user := this.userDao.CheckByUuid(uploadToken.UserUuid)
|
|
|
|
dirMatter := this.matterDao.CheckWithRootByUuid(uploadToken.FolderUuid, user)
|
|
|
|
matter := this.matterService.AtomicCrawl(request, url, uploadToken.Filename, user, dirMatter, uploadToken.Privacy)
|
|
|
|
//expire the upload token.
|
|
uploadToken.ExpireTime = time.Now()
|
|
this.uploadTokenDao.Save(uploadToken)
|
|
|
|
return this.Success(matter)
|
|
}
|
|
|
|
//crawl a url directly. only user can visit this method.
|
|
func (this *AlienController) CrawlDirect(writer http.ResponseWriter, request *http.Request) *result.WebResult {
|
|
|
|
filename := request.FormValue("filename")
|
|
privacyStr := request.FormValue("privacy")
|
|
dirPath := request.FormValue("dirPath")
|
|
url := request.FormValue("url")
|
|
|
|
filename = CheckMatterName(request, filename)
|
|
|
|
var privacy bool
|
|
if privacyStr == TRUE {
|
|
privacy = true
|
|
}
|
|
|
|
user := this.checkUser(request)
|
|
dirMatter := this.matterService.CreateDirectories(request, user, dirPath)
|
|
|
|
matter := this.matterService.AtomicCrawl(request, url, filename, user, dirMatter, privacy)
|
|
|
|
return this.Success(matter)
|
|
}
|
|
|
|
//fetch a download token for guest. Guest can download file with this token.
|
|
func (this *AlienController) FetchDownloadToken(writer http.ResponseWriter, request *http.Request) *result.WebResult {
|
|
|
|
matterUuid := request.FormValue("matterUuid")
|
|
expireTimeStr := request.FormValue("expireTime")
|
|
|
|
if matterUuid == "" {
|
|
panic(result.BadRequest("matterUuid cannot be null."))
|
|
}
|
|
|
|
user := this.checkUser(request)
|
|
|
|
matter := this.matterDao.CheckByUuid(matterUuid)
|
|
if matter.UserUuid != user.Uuid {
|
|
panic(result.BadRequest("matter not belong to you"))
|
|
}
|
|
|
|
var expireTime time.Time
|
|
if expireTimeStr == "" {
|
|
panic(result.BadRequest("time format error"))
|
|
} else {
|
|
expireTime = util.ConvertDateTimeStringToTime(expireTimeStr)
|
|
}
|
|
if expireTime.Before(time.Now()) {
|
|
panic(result.BadRequest("expire time cannot before now"))
|
|
}
|
|
|
|
downloadToken := &DownloadToken{
|
|
UserUuid: user.Uuid,
|
|
MatterUuid: matterUuid,
|
|
ExpireTime: expireTime,
|
|
Ip: util.GetIpAddress(request),
|
|
}
|
|
|
|
downloadToken = this.downloadTokenDao.Create(downloadToken)
|
|
|
|
return this.Success(downloadToken)
|
|
|
|
}
|
|
|
|
//preview a file.
|
|
func (this *AlienController) Preview(writer http.ResponseWriter, request *http.Request, uuid string, filename string) {
|
|
|
|
this.alienService.PreviewOrDownload(writer, request, uuid, filename, false)
|
|
}
|
|
|
|
//download a file.
|
|
func (this *AlienController) Download(writer http.ResponseWriter, request *http.Request, uuid string, filename string) {
|
|
|
|
this.alienService.PreviewOrDownload(writer, request, uuid, filename, true)
|
|
}
|