goftp/controllers/fileopt.go

package controllers

import (
	"bufio"
	"crypto/md5"
	"fmt"
	"goftp/service"
	"goftp/utils"
	"io"
	"math/rand"
	"net/http"
	"os"
	"path"
	"path/filepath"
	"time"

	"github.com/astaxie/beego"
	"github.com/astaxie/beego/logs"
)

type FileOptUploadController struct {
	beego.Controller
}

// @Title Get
// @Description 上传文件
// @router / [get]
func (c *FileOptUploadController) Get() {
	c.TplName = "fileopt.html"
}

// @Title FileUploadPost
// @Description 上传文件
// @router /fileUploadPost [post]
func (c *FileOptUploadController) FileUploadPost() {
	f, h, _ := c.GetFile("file") //获取上传的文件
	ext := path.Ext(h.Filename)
	//创建目录
	goftpdir := beego.AppConfig.String("macgoftpdir")
	servergoftpdir := beego.AppConfig.String("servergoftpdir")
	if utils.ValiteDirExist(servergoftpdir) == true {
		goftpdir = servergoftpdir
	}
	uploadDir := utils.CreateBaseDir(goftpdir)
	//uploadDir := utils.CreateBaseDir("/Users/edao/goftpdir");
	//uploadDir := "/root/goftpdir"
	//构造文件名称
	rand.Seed(time.Now().UnixNano())
	randNum := fmt.Sprintf("%d", rand.Intn(9999)+1000)
	hashName := md5.Sum([]byte(time.Now().Format("2006_01_02_15_04_05_") + randNum))
	fileName := fmt.Sprintf("%x", hashName) + ext
	fpath := uploadDir + "/" + fileName

	defer f.Close()
	//写文件，保存excel文件
	r := bufio.NewReader(f)     //创建一个读取缓冲流
	fo, err := os.Create(fpath) //创建输出*File
	if err != nil {
		logs.Error(err)
	}
	w := bufio.NewWriter(fo) //创建输出缓冲流

	buf := make([]byte, 1024)
	for {
		n, err := r.Read(buf)
		if err != nil && err != io.EOF {
			logs.Error(err)
		}
		if n == 0 {
			break
		}

		if n2, err := w.Write(buf[:n]); err != nil {
			logs.Error(err)
		} else if n2 != n {
			logs.Debug("error in writing")
		}
	}
	if err = w.Flush(); err != nil {
		logs.Error(err)
	}
	webUrl := beego.AppConfig.String("webUrl")
	if utils.ValiteDirExist(servergoftpdir) == false {
		webUrl = "http://127.0.0.1:5000"
	}
	c.Ctx.WriteString("上传成功~！！！！！！！,文件URL:" + webUrl + "/goftp/file/getSingleFile?fileName=" + fileName)
}

// @Title PicUploadPost
// @Description 上传图片
// @router /picUploadPost [post]
func (c *FileOptUploadController) PicUploadPost() {
	f, h, err := c.GetFile("file") //获取上传的文件
	if err != nil {
		c.Ctx.WriteString("请选择上传的图片")
		return
	}
	ext := path.Ext(h.Filename)
	//验证后缀名是否符合要求
	var AllowExtMap map[string]bool = map[string]bool{
		".jpg":  true,
		".jpeg": true,
		".png":  true,
	}
	if _, ok := AllowExtMap[ext]; !ok {
		c.Ctx.WriteString("后缀名不符合上传要求")
		return
	}

	//创建目录
	//uploadDir := "static/upload/" + time.Now().Format("2006/01/02/")
	servergoftpdir := beego.AppConfig.String("servergoftpdir")
	uploadDir := utils.CreateDir("static", "pic")
	uploadPath := "static" + "/" + "pic"
	//构造文件名称
	rand.Seed(time.Now().UnixNano())
	randNum := fmt.Sprintf("%d", rand.Intn(9999)+1000)
	hashName := md5.Sum([]byte(time.Now().Format("2006_01_02_15_04_05_") + randNum))
	fileName := fmt.Sprintf("%x", hashName) + ext
	fpath := uploadDir + "/" + fileName

	defer f.Close()
	//写文件，保存excel文件
	r := bufio.NewReader(f)     //创建一个读取缓冲流
	fo, err := os.Create(fpath) //创建输出*File
	if err != nil {
		logs.Error(err)
	}
	w := bufio.NewWriter(fo) //创建输出缓冲流

	buf := make([]byte, 1024)
	for {
		n, err := r.Read(buf)
		if err != nil && err != io.EOF {
			logs.Error(err)
		}
		if n == 0 {
			break
		}

		if n2, err := w.Write(buf[:n]); err != nil {
			logs.Error(err)
		} else if n2 != n {
			logs.Debug("error in writing")
		}
	}
	if err = w.Flush(); err != nil {
		logs.Error(err)
	}
	//webUrl
	webUrl := beego.AppConfig.String("webUrl")
	if utils.ValiteDirExist(servergoftpdir) == false {
		webUrl = "http://127.0.0.1:5000"
	}
	c.Ctx.WriteString("上传成功~！！！！！！！,文件路径为:" + webUrl + "/goftp/" + uploadPath + "/" + fileName)
}

// @Title GetSingleFile
// @Description 获取单个文件
// @Param	fileName				query 	string	true		"文件名称"
// @router /getSingleFile [get]
func (c *FileOptUploadController) GetSingleFile() {
	//图片,text,pdf文件全部在浏览器中显示了，并没有完全的实现下载的功能
	fileName := c.Ctx.Input.Query("fileName")
	logs.Debug(fileName)
	goftpdir := beego.AppConfig.String("macgoftpdir")
	servergoftpdir := beego.AppConfig.String("servergoftpdir")
	if utils.ValiteDirExist(servergoftpdir) == true {
		goftpdir = servergoftpdir
	}
	logs.Debug(fileName)
	if fileName == "" {
		c.Ctx.WriteString("please input you want to download the file name!")
		return
	}
	filePath := goftpdir + "/" + fileName
	if utils.ValiteFileExist(filePath) == false {
		c.Ctx.WriteString("file not exist!")
		return
	}
	fileCountUpdateSuccess := service.UpdateDownloadCount(fileName)
	if fileCountUpdateSuccess {
		logs.Debug("fileCountUpdatesuccess")
	} else {
		logs.Debug("fileCountUpdateFail")
	}
	c.Download(filePath, fileName)
}

// @Title GetEx4File
// @Description 获取单个文件
// @Param	fileName				query 	string	true		"文件名称"
// @router /getEx4File [get]
func (c *FileOptUploadController) GetEx4File() {
	//图片,text,pdf文件全部在浏览器中显示了，并没有完全的实现下载的功能
	fileName := c.Ctx.Input.Query("fileName")
	logs.Debug(fileName)
	goftpdir := beego.AppConfig.String("macgoftpdir")
	servergoftpdir := beego.AppConfig.String("servergoftpdir")
	if utils.ValiteDirExist(servergoftpdir) == true {
		goftpdir = servergoftpdir
	}
	logs.Debug(fileName)
	if fileName == "" {
		c.Ctx.WriteString("please input you want to download the file name!")
		return
	}
	filePath := goftpdir + "/" + fileName
	if utils.ValiteFileExist(filePath) == false {
		c.Ctx.WriteString("file not exist!")
		return
	}
	fileCountUpdateSuccess := service.UpdateDownloadCount(fileName)
	if fileCountUpdateSuccess {
		logs.Debug("fileCountUpdatesuccess")
	} else {
		logs.Debug("fileCountUpdateFail")
	}
	c.Download(filePath, fileName)
}

// @Title UploadBaseFilePost
// @Description 上传原文件，文件名称不会改变
// @router /uploadBaseFileToGoFtpDir [post]
func (c *FileOptUploadController) UploadBaseFileToPost() {
	f, h, _ := c.GetFile("file") //获取上传的文件
	//ext := path.Ext(h.Filename)
	logs.Debug(h.Filename)
	//创建目录
	goftpdir := beego.AppConfig.String("macgoftpdir")
	servergoftpdir := beego.AppConfig.String("servergoftpdir")
	if utils.ValiteDirExist(servergoftpdir) == true {
		goftpdir = servergoftpdir
	}
	uploadDir := utils.CreateBaseDir(goftpdir)
	//uploadDir := utils.CreateBaseDir("/Users/edao/goftpdir");
	//uploadDir := "/root/goftpdir"
	//构造文件名称
	// rand.Seed(time.Now().UnixNano())
	// randNum := fmt.Sprintf("%d", rand.Intn(9999)+1000)
	// hashName := md5.Sum([]byte(time.Now().Format("2006_01_02_15_04_05_") + randNum))
	fileName := h.Filename
	fpath := uploadDir + "/" + fileName

	defer f.Close()
	//写文件，保存excel文件
	r := bufio.NewReader(f)     //创建一个读取缓冲流
	fo, err := os.Create(fpath) //创建输出*File
	if err != nil {
		logs.Error(err)
	}
	w := bufio.NewWriter(fo) //创建输出缓冲流

	buf := make([]byte, 1024)
	for {
		n, err := r.Read(buf)
		if err != nil && err != io.EOF {
			logs.Error(err)
		}
		if n == 0 {
			break
		}

		if n2, err := w.Write(buf[:n]); err != nil {
			logs.Error(err)
		} else if n2 != n {
			logs.Debug("error in writing")
		}
	}
	if err = w.Flush(); err != nil {
		logs.Error(err)
	}
	webUrl := beego.AppConfig.String("webUrl")
	if utils.ValiteDirExist(servergoftpdir) == false {
		webUrl = "http://127.0.0.1:5000"
	}

	//c.Ctx.WriteString("上传成功~！！！！！！！,服务器文件路径为:" + fpath)
	//c.Ctx.WriteString("\n")
	//c.Ctx.ResponseWriter.AddHeader("content-disposition", "attachment; filename=\""+fileName+"\"")
	c.Ctx.WriteString("上传成功~！！！！！！！,文件URL:" + webUrl + "/goftp/file/getSingleFile?fileName=" + fileName)
}

// 重写beego框架 Download 方法，去掉文件名url.QueryEscape(fName)
func (c *FileOptUploadController) Download(file string, filename ...string) {
	// check get file error, file not found or other error.
	if _, err := os.Stat(file); err != nil {
		http.ServeFile(c.Ctx.Output.Context.ResponseWriter, c.Ctx.Output.Context.Request, file)
		return
	}
	var fName string
	if len(filename) > 0 && filename[0] != "" {
		fName = filename[0]
	} else {
		fName = filepath.Base(file)
	}
	c.Ctx.Output.Header("Content-Disposition", "attachment; filename="+fName)
	c.Ctx.Output.Header("Content-Description", "File Transfer")
	c.Ctx.Output.Header("Content-Type", "application/octet-stream")
	c.Ctx.Output.Header("Content-Transfer-Encoding", "binary")
	c.Ctx.Output.Header("Expires", "0")
	c.Ctx.Output.Header("Cache-Control", "must-revalidate")
	c.Ctx.Output.Header("Pragma", "public")
	http.ServeFile(c.Ctx.Output.Context.ResponseWriter, c.Ctx.Output.Context.Request, file)
}