package v1

import (
	"aps_crm/model"
	"aps_crm/model/request"
	"aps_crm/model/response"
	"aps_crm/pkg/contextx"
	"aps_crm/pkg/ecode"
	"aps_crm/pkg/httpx"
	"aps_crm/service"
	"github.com/flipped-aurora/gin-vue-admin/server/utils/upload"
	"github.com/gin-gonic/gin"
	"os"
	"path/filepath"
	"strconv"
)

type FileApi struct{}

// Add
// @Tags		附件管理
// @Summary	添加附件
// @Produce	application/json
// @Param		object	body		request.AddFile	true	"查询参数"
// @Success	200		{object}	contextx.Response{}
// @Router		/api/file/add [post]
func (s *FileApi) Add(c *gin.Context) {
	var params request.AddFile
	ctx, ok := contextx.NewContext(c, &params)
	if !ok {
		return
	}

	_, header, err := c.Request.FormFile("file")
	if err != nil {
		ctx.Fail(ecode.ParamsErr)
		return
	}

	oss := upload.NewOss()
	filePath, key, uploadErr := oss.UploadFile(header)
	if uploadErr != nil {
		ctx.Fail(ecode.ParamsErr)
		return
	}

	_, filename := filepath.Split(filePath)

	fileRecord := &model.File{
		Name:          filename,
		Size:          header.Size,
		FilePath:      filePath,
		Key:           key,
		DownloadCount: 0,
		PreviewCount:  0,
		FileType:      httpx.GetMimeType(filePath),
		SourceType:    params.SourceType,
		SourceId:      params.SourceId,
	}

	errCode := service.NewFileService().AddFile(fileRecord)
	if errCode != ecode.OK {
		ctx.Fail(errCode)
		return
	}

	ctx.Ok()
}

// Delete
// @Tags		附件管理
// @Summary	删除附件
// @Produce	application/json
// @Param		id	path		int	true	"查询参数"
// @Success	200	{object}	contextx.Response{}
// @Router		/api/file/delete/{id} [delete]
func (s *FileApi) Delete(c *gin.Context) {
	ctx, ok := contextx.NewContext(c, nil)
	if !ok {
		return
	}

	id, _ := strconv.Atoi(c.Param("id"))

	file, err := model.NewFileSearch().SetId(id).First()
	if err != nil {
		ctx.FailWithMsg(ecode.ParamsErr, "查找文件失败")
		return
	}

	err = os.Remove(file.FilePath)
	if err != nil {
		ctx.FailWithMsg(ecode.ParamsErr, "删除文件失败")
		return
	}

	errCode := service.NewFileService().DeleteFile(id)
	if errCode != ecode.OK {
		ctx.Fail(errCode)
		return
	}

	ctx.Ok()
}

// Update
// @Tags		附件管理
// @Summary	更新附件
// @Produce	application/json
// @Param		object	body		request.UpdateFile	true	"查询参数"
// @Success	200		{object}	contextx.Response{}
// @Router		/api/file/update [put]
func (s *FileApi) Update(c *gin.Context) {
	var params request.UpdateFile
	ctx, ok := contextx.NewContext(c, &params)
	if !ok {
		return
	}
	if params.Id == 0 {
		ctx.Fail(ecode.ParamsErr)
	}
	params.File.Id = params.Id

	errCode := service.NewFileService().UpdateFile(&params.File)
	if errCode != ecode.OK {
		ctx.Fail(errCode)
		return
	}

	ctx.Ok()
}

// List
// @Tags		附件管理
// @Summary	获取附件列表
// @Produce	application/json
// @Param		object	query		request.GetFileList	true	"参数"
// @Success	200	{object}	response.ListResponse{data=[]model.File}
// @Router		/api/file/list [get]
func (s *FileApi) List(c *gin.Context) {
	var params request.GetFileList
	ctx, ok := contextx.NewContext(c, &params)
	if !ok {
		return
	}

	file, total, errCode := service.NewFileService().GetFileList()
	if errCode != ecode.OK {
		ctx.Fail(errCode)
		return
	}

	ctx.OkWithDetailed(response.ListResponse{
		Data:  file,
		Count: total,
	})
}