leanote/app/controllers/api/ApiBaseController.go

package api

import (
	"github.com/revel/revel"
	"gopkg.in/mgo.v2/bson"
	//	"encoding/json"
	"github.com/leanote/leanote/app/controllers"
	"github.com/leanote/leanote/app/info"
	. "github.com/leanote/leanote/app/lea"
	"os"
	//	"fmt"
	"io/ioutil"
	//	"fmt"
	//	"math"
	//	"strconv"
	"strings"
)

// 公用Controller, 其它Controller继承它
type ApiBaseContrller struct {
	controllers.BaseController // 不能用*BaseController
}

// 得到token, 这个token是在AuthInterceptor设到Session中的
func (c ApiBaseContrller) getToken() string {
	return c.Session["_token"]
}

// userId
// _userId是在AuthInterceptor设置的
func (c ApiBaseContrller) getUserId() string {
	return c.Session["_userId"]
}

// 得到用户信息
func (c ApiBaseContrller) getUserInfo() info.User {
	userId := c.Session["_userId"]
	if userId == "" {
		return info.User{}
	}
	return userService.GetUserInfo(userId)
}

// 上传附件
func (c ApiBaseContrller) uploadAttach(name string, noteId string) (ok bool, msg string, id string) {
	userId := c.getUserId()

	// 判断是否有权限为笔记添加附件
	// 如果笔记还没有添加是不是会有问题
	/*
		if !shareService.HasUpdateNotePerm(noteId, userId) {
			return
		}
	*/

	file, handel, err := c.Request.FormFile(name)
	if err != nil {
		return
	}
	defer file.Close()

	data, err := ioutil.ReadAll(file)
	if err != nil {
		return
	}
	// > 5M?
	maxFileSize := configService.GetUploadSize("uploadAttachSize")
	if maxFileSize <= 0 {
		maxFileSize = 1000
	}
	if float64(len(data)) > maxFileSize*float64(1024*1024) {
		msg = "fileIsTooLarge"
		return
	}

	// 生成上传路径
	newGuid := NewGuid()
	//	filePath :=	"files/" + Digest3(userId) + "/" + userId + "/" + Digest2(newGuid) + "/attachs"
	filePath := "files/" + GetRandomFilePath(userId, newGuid) + "/attachs"

	dir := revel.BasePath + "/" + filePath
	err = os.MkdirAll(dir, 0755)
	if err != nil {
		return
	}
	// 生成新的文件名
	filename := handel.Filename
	_, ext := SplitFilename(filename) // .doc
	filename = newGuid + ext
	toPath := dir + "/" + filename
	err = ioutil.WriteFile(toPath, data, 0777)
	if err != nil {
		return
	}

	// add File to db
	fileType := ""
	if ext != "" {
		fileType = strings.ToLower(ext[1:])
	}
	filesize := GetFilesize(toPath)
	fileInfo := info.Attach{AttachId: bson.NewObjectId(),
		Name:         filename,
		Title:        handel.Filename,
		NoteId:       bson.ObjectIdHex(noteId),
		UploadUserId: bson.ObjectIdHex(userId),
		Path:         filePath + "/" + filename,
		Type:         fileType,
		Size:         filesize}

	ok, msg = attachService.AddAttach(fileInfo, true)
	if !ok {
		return
	}

	id = fileInfo.AttachId.Hex()
	return
}

// 上传图片
func (c ApiBaseContrller) upload(name string, noteId string, isAttach bool) (ok bool, msg string, id string) {
	if isAttach {
		return c.uploadAttach(name, noteId)
	}
	file, handel, err := c.Request.FormFile(name)
	if err != nil {
		return
	}
	defer file.Close()

	newGuid := NewGuid()
	// 生成上传路径
	userId := c.getUserId()
	// fileUrlPath := "files/" + Digest3(userId) + "/" + userId + "/" + Digest2(newGuid) + "/images"
	fileUrlPath := "files/" + GetRandomFilePath(userId, newGuid) + "/images"

	dir := revel.BasePath + "/" + fileUrlPath
	err = os.MkdirAll(dir, 0755)
	if err != nil {
		return
	}
	// 生成新的文件名
	filename := handel.Filename
	_, ext := SplitFilename(filename)
	// if ext != ".gif" && ext != ".jpg" && ext != ".png" && ext != ".bmp" && ext != ".jpeg" {
	// 	msg = "notImage"
	// 	return
	// }

	filename = newGuid + ext
	data, err := ioutil.ReadAll(file)
	if err != nil {
		return
	}

	maxFileSize := configService.GetUploadSize("uploadImageSize")
	if maxFileSize <= 0 {
		maxFileSize = 1000
	}

	// > 2M?
	if float64(len(data)) > maxFileSize*float64(1024*1024) {
		msg = "fileIsTooLarge"
		return
	}

	toPath := dir + "/" + filename
	err = ioutil.WriteFile(toPath, data, 0777)
	if err != nil {
		return
	}
	// 改变成gif图片
	_, toPathGif := TransToGif(toPath, 0, true)
	filename = GetFilename(toPathGif)
	filesize := GetFilesize(toPathGif)
	fileUrlPath += "/" + filename

	// File
	fileInfo := info.File{FileId: bson.NewObjectId(),
		Name:  filename,
		Title: handel.Filename,
		Path:  fileUrlPath,
		Size:  filesize}
	ok, msg = fileService.AddImage(fileInfo, "", c.getUserId(), true)
	if ok {
		id = fileInfo.FileId.Hex()
	}
	return
}