96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
package controller
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"be.ems/lib/file"
|
|
"be.ems/lib/log"
|
|
"be.ems/lib/services"
|
|
"be.ems/features/ue/model"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type FileExport struct {
|
|
file.FileInfo
|
|
}
|
|
|
|
func (m *FileExport) GetFileList(c *gin.Context) {
|
|
var querys model.FileExportQuery
|
|
|
|
if err := c.ShouldBindQuery(&querys); err != nil {
|
|
c.JSON(http.StatusOK, services.ErrResp(err.Error()))
|
|
return
|
|
}
|
|
|
|
files, err := file.GetFileInfo(querys.Path, querys.Suffix)
|
|
if err != nil {
|
|
log.Error("failed to GetFileInfo:", err)
|
|
c.JSON(http.StatusOK, services.ErrResp(err.Error()))
|
|
return
|
|
}
|
|
|
|
// split files list
|
|
lenNum := int64(len(files))
|
|
start := (querys.PageNum - 1) * querys.PageSize
|
|
end := start + querys.PageSize
|
|
var splitList []file.FileInfo
|
|
if start >= lenNum {
|
|
splitList = []file.FileInfo{}
|
|
} else if end >= lenNum {
|
|
splitList = files[start:]
|
|
} else {
|
|
splitList = files[start:end]
|
|
}
|
|
total := len(files)
|
|
c.JSON(http.StatusOK, services.TotalDataResp(splitList, total))
|
|
}
|
|
|
|
func (m *FileExport) Total(c *gin.Context) {
|
|
dir := c.Query("path")
|
|
|
|
fileCount, dirCount, err := file.GetFileAndDirCount(dir)
|
|
if err != nil {
|
|
log.Error("failed to GetFileAndDirCount:", err)
|
|
c.JSON(http.StatusOK, services.ErrResp(err.Error()))
|
|
return
|
|
}
|
|
total := fileCount + dirCount
|
|
c.JSON(http.StatusOK, services.TotalResp(int64(total)))
|
|
}
|
|
|
|
func (m *FileExport) DownloadHandler(c *gin.Context) {
|
|
dir := c.Query("path")
|
|
fileName := c.Param("fileName")
|
|
filePath := filepath.Join(dir, fileName)
|
|
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, services.ErrResp(err.Error()))
|
|
return
|
|
}
|
|
defer file.Close()
|
|
|
|
if _, err := os.Stat(filePath); os.IsNotExist(err) {
|
|
c.JSON(http.StatusOK, services.ErrResp(err.Error()))
|
|
return
|
|
}
|
|
|
|
c.Header("Content-Disposition", "attachment; filename="+fileName)
|
|
c.Header("Content-Type", "application/octet-stream")
|
|
c.File(filePath)
|
|
}
|
|
|
|
func (m *FileExport) Delete(c *gin.Context) {
|
|
fileName := c.Param("fileName")
|
|
dir := c.Query("path")
|
|
filePath := filepath.Join(dir, fileName)
|
|
|
|
if err := os.Remove(filePath); err != nil {
|
|
c.JSON(http.StatusOK, services.ErrResp(err.Error()))
|
|
return
|
|
}
|
|
c.JSON(http.StatusNoContent, nil) // 204 No Content
|
|
}
|