480 lines
15 KiB
Go
480 lines
15 KiB
Go
package controllers
|
|
|
|
import (
|
|
"github.com/revel/revel"
|
|
// "encoding/json"
|
|
"github.com/leanote/leanote/app/info"
|
|
. "github.com/leanote/leanote/app/lea"
|
|
"gopkg.in/mgo.v2/bson"
|
|
"os"
|
|
"os/exec"
|
|
"regexp"
|
|
"strings"
|
|
"time"
|
|
// "github.com/leanote/leanote/app/types"
|
|
// "io/ioutil"
|
|
"fmt"
|
|
// "bytes"
|
|
// "os"
|
|
)
|
|
|
|
type Note struct {
|
|
BaseController
|
|
}
|
|
|
|
// 笔记首页, 判断是否已登录
|
|
// 已登录, 得到用户基本信息(notebook, shareNotebook), 跳转到index.html中
|
|
// 否则, 转向登录页面
|
|
func (c Note) Index(noteId, online string) revel.Result {
|
|
c.SetLocale()
|
|
userInfo := c.GetUserAndBlogUrl()
|
|
|
|
userId := userInfo.UserId.Hex()
|
|
|
|
// 没有登录
|
|
if userId == "" {
|
|
return c.Redirect("/login")
|
|
}
|
|
|
|
c.RenderArgs["openRegister"] = configService.IsOpenRegister()
|
|
|
|
// 已登录了, 那么得到所有信息
|
|
notebooks := notebookService.GetNotebooks(userId)
|
|
shareNotebooks, sharedUserInfos := shareService.GetShareNotebooks(userId)
|
|
|
|
// 还需要按时间排序(DESC)得到notes
|
|
notes := []info.Note{}
|
|
noteContent := info.NoteContent{}
|
|
|
|
if len(notebooks) > 0 {
|
|
// noteId是否存在
|
|
// 是否传入了正确的noteId
|
|
hasRightNoteId := false
|
|
if IsObjectId(noteId) {
|
|
note := noteService.GetNoteById(noteId)
|
|
|
|
if note.NoteId != "" {
|
|
var noteOwner = note.UserId.Hex()
|
|
noteContent = noteService.GetNoteContent(noteId, noteOwner)
|
|
|
|
hasRightNoteId = true
|
|
c.RenderArgs["curNoteId"] = noteId
|
|
c.RenderArgs["curNotebookId"] = note.NotebookId.Hex()
|
|
|
|
// 打开的是共享的笔记, 那么判断是否是共享给我的默认笔记
|
|
if noteOwner != c.GetUserId() {
|
|
if shareService.HasReadPerm(noteOwner, c.GetUserId(), noteId) {
|
|
// 不要获取notebook下的笔记
|
|
// 在前端下发请求
|
|
c.RenderArgs["curSharedNoteNotebookId"] = note.NotebookId.Hex()
|
|
c.RenderArgs["curSharedUserId"] = noteOwner
|
|
// 没有读写权限
|
|
} else {
|
|
hasRightNoteId = false
|
|
}
|
|
} else {
|
|
_, notes = noteService.ListNotes(c.GetUserId(), note.NotebookId.Hex(), false, c.GetPage(), 50, defaultSortField, false, false)
|
|
|
|
// 如果指定了某笔记, 则该笔记放在首位
|
|
lenNotes := len(notes)
|
|
if lenNotes > 1 {
|
|
notes2 := make([]info.Note, len(notes))
|
|
notes2[0] = note
|
|
i := 1
|
|
for _, note := range notes {
|
|
if note.NoteId.Hex() != noteId {
|
|
if i == lenNotes { // 防止越界
|
|
break
|
|
}
|
|
notes2[i] = note
|
|
i++
|
|
}
|
|
}
|
|
notes = notes2
|
|
}
|
|
}
|
|
}
|
|
|
|
// 得到最近的笔记
|
|
_, latestNotes := noteService.ListNotes(c.GetUserId(), "", false, c.GetPage(), 50, defaultSortField, false, false)
|
|
c.RenderArgs["latestNotes"] = latestNotes
|
|
}
|
|
|
|
// 没有传入笔记
|
|
// 那么得到最新笔记
|
|
if !hasRightNoteId {
|
|
_, notes = noteService.ListNotes(c.GetUserId(), "", false, c.GetPage(), 50, defaultSortField, false, false)
|
|
if len(notes) > 0 {
|
|
noteContent = noteService.GetNoteContent(notes[0].NoteId.Hex(), userId)
|
|
c.RenderArgs["curNoteId"] = notes[0].NoteId.Hex()
|
|
}
|
|
}
|
|
}
|
|
|
|
// 当然, 还需要得到第一个notes的content
|
|
//...
|
|
c.RenderArgs["isAdmin"] = configService.GetAdminUsername() == userInfo.Username
|
|
|
|
c.RenderArgs["userInfo"] = userInfo
|
|
c.RenderArgs["notebooks"] = notebooks
|
|
c.RenderArgs["shareNotebooks"] = shareNotebooks // note信息在notes列表中
|
|
c.RenderArgs["sharedUserInfos"] = sharedUserInfos
|
|
|
|
c.RenderArgs["notes"] = notes
|
|
c.RenderArgs["noteContentJson"] = noteContent
|
|
c.RenderArgs["noteContent"] = noteContent.Content
|
|
|
|
c.RenderArgs["tags"] = tagService.GetTags(c.GetUserId())
|
|
|
|
c.RenderArgs["globalConfigs"] = configService.GetGlobalConfigForUser()
|
|
|
|
// return c.RenderTemplate("note/note.html")
|
|
|
|
if isDev, _ := revel.Config.Bool("mode.dev"); isDev && online == "" {
|
|
return c.RenderTemplate("note/note-dev.html")
|
|
} else {
|
|
return c.RenderTemplate("note/note.html")
|
|
}
|
|
}
|
|
|
|
// 首页, 判断是否已登录
|
|
// 已登录, 得到用户基本信息(notebook, shareNotebook), 跳转到index.html中
|
|
// 否则, 转向登录页面
|
|
func (c Note) ListNotes(notebookId string) revel.Result {
|
|
_, notes := noteService.ListNotes(c.GetUserId(), notebookId, false, c.GetPage(), pageSize, defaultSortField, false, false)
|
|
return c.RenderJson(notes)
|
|
}
|
|
|
|
// 得到trash
|
|
func (c Note) ListTrashNotes() revel.Result {
|
|
_, notes := noteService.ListNotes(c.GetUserId(), "", true, c.GetPage(), pageSize, defaultSortField, false, false)
|
|
return c.RenderJson(notes)
|
|
}
|
|
|
|
// 得到note和内容
|
|
func (c Note) GetNoteAndContent(noteId string) revel.Result {
|
|
return c.RenderJson(noteService.GetNoteAndContent(noteId, c.GetUserId()))
|
|
}
|
|
|
|
// 得到内容
|
|
func (c Note) GetNoteContent(noteId string) revel.Result {
|
|
noteContent := noteService.GetNoteContent(noteId, c.GetUserId())
|
|
return c.RenderJson(noteContent)
|
|
}
|
|
|
|
// 这里不能用json, 要用post
|
|
func (c Note) UpdateNoteOrContent(noteOrContent info.NoteOrContent) revel.Result {
|
|
// 新添加note
|
|
if noteOrContent.IsNew {
|
|
userId := c.GetObjectUserId()
|
|
// myUserId := userId
|
|
// 为共享新建?
|
|
if noteOrContent.FromUserId != "" {
|
|
userId = bson.ObjectIdHex(noteOrContent.FromUserId)
|
|
}
|
|
|
|
note := info.Note{UserId: userId,
|
|
NoteId: bson.ObjectIdHex(noteOrContent.NoteId),
|
|
NotebookId: bson.ObjectIdHex(noteOrContent.NotebookId),
|
|
Title: noteOrContent.Title,
|
|
Tags: strings.Split(noteOrContent.Tags, ","),
|
|
Desc: noteOrContent.Desc,
|
|
ImgSrc: noteOrContent.ImgSrc,
|
|
IsBlog: noteOrContent.IsBlog,
|
|
IsMarkdown: noteOrContent.IsMarkdown,
|
|
}
|
|
noteContent := info.NoteContent{NoteId: note.NoteId,
|
|
UserId: userId,
|
|
IsBlog: note.IsBlog,
|
|
Content: noteOrContent.Content,
|
|
Abstract: noteOrContent.Abstract}
|
|
|
|
note = noteService.AddNoteAndContentForController(note, noteContent, c.GetUserId())
|
|
return c.RenderJson(note)
|
|
}
|
|
|
|
noteUpdate := bson.M{}
|
|
needUpdateNote := false
|
|
|
|
// Desc前台传来
|
|
if c.Has("Desc") {
|
|
needUpdateNote = true
|
|
noteUpdate["Desc"] = noteOrContent.Desc
|
|
}
|
|
if c.Has("ImgSrc") {
|
|
needUpdateNote = true
|
|
noteUpdate["ImgSrc"] = noteOrContent.ImgSrc
|
|
}
|
|
if c.Has("Title") {
|
|
needUpdateNote = true
|
|
noteUpdate["Title"] = noteOrContent.Title
|
|
}
|
|
|
|
if c.Has("Tags") {
|
|
needUpdateNote = true
|
|
noteUpdate["Tags"] = strings.Split(noteOrContent.Tags, ",")
|
|
}
|
|
|
|
// web端不控制
|
|
if needUpdateNote {
|
|
noteService.UpdateNote(c.GetUserId(),
|
|
noteOrContent.NoteId, noteUpdate, -1)
|
|
}
|
|
|
|
//-------------
|
|
afterContentUsn := 0
|
|
contentOk := false
|
|
contentMsg := ""
|
|
if c.Has("Content") {
|
|
// noteService.UpdateNoteContent(noteOrContent.UserId, c.GetUserId(),
|
|
// noteOrContent.NoteId, noteOrContent.Content, noteOrContent.Abstract)
|
|
contentOk, contentMsg, afterContentUsn = noteService.UpdateNoteContent(c.GetUserId(),
|
|
noteOrContent.NoteId, noteOrContent.Content, noteOrContent.Abstract,
|
|
needUpdateNote, -1, time.Now())
|
|
}
|
|
|
|
Log(afterContentUsn)
|
|
Log(contentOk)
|
|
Log(contentMsg)
|
|
|
|
return c.RenderJson(true)
|
|
}
|
|
|
|
// 删除note/ 删除别人共享给我的笔记
|
|
// userId 是note.UserId
|
|
func (c Note) DeleteNote(noteIds []string, isShared bool) revel.Result {
|
|
if !isShared {
|
|
for _, noteId := range noteIds {
|
|
trashService.DeleteNote(noteId, c.GetUserId())
|
|
}
|
|
return c.RenderJson(true)
|
|
}
|
|
|
|
for _, noteId := range noteIds {
|
|
trashService.DeleteSharedNote(noteId, c.GetUserId())
|
|
}
|
|
|
|
return c.RenderJson(true)
|
|
}
|
|
|
|
// 删除trash, 已弃用, 用DeleteNote
|
|
func (c Note) DeleteTrash(noteId string) revel.Result {
|
|
return c.RenderJson(trashService.DeleteTrash(noteId, c.GetUserId()))
|
|
}
|
|
|
|
// 移动note
|
|
func (c Note) MoveNote(noteIds []string, notebookId string) revel.Result {
|
|
userId := c.GetUserId()
|
|
for _, noteId := range noteIds {
|
|
noteService.MoveNote(noteId, notebookId, userId)
|
|
}
|
|
return c.RenderJson(true)
|
|
}
|
|
|
|
// 复制note
|
|
func (c Note) CopyNote(noteIds []string, notebookId string) revel.Result {
|
|
copyNotes := make([]info.Note, len(noteIds))
|
|
userId := c.GetUserId()
|
|
for i, noteId := range noteIds {
|
|
copyNotes[i] = noteService.CopyNote(noteId, notebookId, userId)
|
|
}
|
|
re := info.NewRe()
|
|
re.Ok = true
|
|
re.Item = copyNotes
|
|
return c.RenderJson(re)
|
|
}
|
|
|
|
// 复制别人共享的笔记给我
|
|
func (c Note) CopySharedNote(noteIds []string, notebookId, fromUserId string) revel.Result {
|
|
copyNotes := make([]info.Note, len(noteIds))
|
|
userId := c.GetUserId()
|
|
for i, noteId := range noteIds {
|
|
copyNotes[i] = noteService.CopySharedNote(noteId, notebookId, fromUserId, userId)
|
|
}
|
|
re := info.NewRe()
|
|
re.Ok = true
|
|
re.Item = copyNotes
|
|
return c.RenderJson(re)
|
|
}
|
|
|
|
//------------
|
|
// search
|
|
// 通过title搜索
|
|
func (c Note) SearchNote(key string) revel.Result {
|
|
_, blogs := noteService.SearchNote(key, c.GetUserId(), c.GetPage(), pageSize, "UpdatedTime", false, false)
|
|
return c.RenderJson(blogs)
|
|
}
|
|
|
|
// 通过tags搜索
|
|
func (c Note) SearchNoteByTags(tags []string) revel.Result {
|
|
_, blogs := noteService.SearchNoteByTags(tags, c.GetUserId(), c.GetPage(), pageSize, "UpdatedTime", false)
|
|
return c.RenderJson(blogs)
|
|
}
|
|
|
|
// 生成PDF
|
|
func (c Note) ToPdf(noteId, appKey string) revel.Result {
|
|
// 虽然传了cookie但是这里还是不能得到userId, 所以还是通过appKey来验证之
|
|
appKeyTrue, _ := revel.Config.String("app.secret")
|
|
if appKeyTrue != appKey {
|
|
return c.RenderText("auth error")
|
|
}
|
|
note := noteService.GetNoteById(noteId)
|
|
if note.NoteId == "" {
|
|
return c.RenderText("no note")
|
|
}
|
|
|
|
noteUserId := note.UserId.Hex()
|
|
content := noteService.GetNoteContent(noteId, noteUserId)
|
|
userInfo := userService.GetUserInfo(noteUserId)
|
|
|
|
//------------------
|
|
// 将content的图片转换为base64
|
|
contentStr := content.Content
|
|
|
|
siteUrlPattern := configService.GetSiteUrl()
|
|
if strings.Contains(siteUrlPattern, "https") {
|
|
siteUrlPattern = strings.Replace(siteUrlPattern, "https", "https*", 1)
|
|
} else {
|
|
siteUrlPattern = strings.Replace(siteUrlPattern, "http", "https*", 1)
|
|
}
|
|
|
|
regImage, _ := regexp.Compile(`<img .*?(src=('|")` + siteUrlPattern + `/(file/outputImage|api/file/getImage)\?fileId=([a-z0-9A-Z]{24})("|'))`)
|
|
|
|
findsImage := regImage.FindAllStringSubmatch(contentStr, -1) // 查找所有的
|
|
// [<img src="http://leanote.com/api/getImage?fileId=3354672e8d38f411286b000069" alt="" width="692" height="302" data-mce-src="http://leanote.com/file/outputImage?fileId=54672e8d38f411286b000069" src="http://leanote.com/file/outputImage?fileId=54672e8d38f411286b000069" " file/outputImage 54672e8d38f411286b000069 "]
|
|
for _, eachFind := range findsImage {
|
|
if len(eachFind) == 6 {
|
|
fileId := eachFind[4]
|
|
// 得到base64编码文件
|
|
fileBase64 := fileService.GetImageBase64(noteUserId, fileId)
|
|
if fileBase64 == "" {
|
|
continue
|
|
}
|
|
|
|
// 1
|
|
// src="http://leanote.com/file/outputImage?fileId=54672e8d38f411286b000069"
|
|
allFixed := strings.Replace(eachFind[0], eachFind[1], "src=\""+fileBase64+"\"", -1)
|
|
contentStr = strings.Replace(contentStr, eachFind[0], allFixed, -1)
|
|
}
|
|
}
|
|
|
|
// markdown
|
|
if note.IsMarkdown {
|
|
// 
|
|
regImageMarkdown, _ := regexp.Compile(`!\[.*?\]\(` + siteUrlPattern + `/(file/outputImage|api/file/getImage)\?fileId=([a-z0-9A-Z]{24})\)`)
|
|
findsImageMarkdown := regImageMarkdown.FindAllStringSubmatch(contentStr, -1) // 查找所有的
|
|
for _, eachFind := range findsImageMarkdown {
|
|
if len(eachFind) == 3 {
|
|
fileId := eachFind[2]
|
|
// 得到base64编码文件
|
|
fileBase64 := fileService.GetImageBase64(noteUserId, fileId)
|
|
if fileBase64 == "" {
|
|
continue
|
|
}
|
|
|
|
// 1
|
|
// src="http://leanote.com/file/outputImage?fileId=54672e8d38f411286b000069"
|
|
allFixed := "![](" + fileBase64 + ")"
|
|
contentStr = strings.Replace(contentStr, eachFind[0], allFixed, -1)
|
|
}
|
|
}
|
|
}
|
|
|
|
if note.Tags != nil && len(note.Tags) > 0 && note.Tags[0] != "" {
|
|
} else {
|
|
note.Tags = nil
|
|
}
|
|
c.RenderArgs["blog"] = note
|
|
c.RenderArgs["content"] = contentStr
|
|
c.RenderArgs["userInfo"] = userInfo
|
|
userBlog := blogService.GetUserBlog(noteUserId)
|
|
c.RenderArgs["userBlog"] = userBlog
|
|
|
|
return c.RenderTemplate("file/pdf.html")
|
|
}
|
|
|
|
// 导出成PDF
|
|
func (c Note) ExportPdf(noteId string) revel.Result {
|
|
re := info.NewRe()
|
|
userId := c.GetUserId()
|
|
note := noteService.GetNoteById(noteId)
|
|
if note.NoteId == "" {
|
|
re.Msg = "No Note"
|
|
return c.RenderText("error")
|
|
}
|
|
|
|
noteUserId := note.UserId.Hex()
|
|
// 是否有权限
|
|
if noteUserId != userId {
|
|
// 是否是有权限协作的
|
|
if !note.IsBlog && !shareService.HasReadPerm(noteUserId, userId, noteId) {
|
|
re.Msg = "No Perm"
|
|
return c.RenderText("No Perm")
|
|
}
|
|
}
|
|
|
|
// path 判断是否需要重新生成之
|
|
guid := NewGuid()
|
|
fileUrlPath := "files/export_pdf"
|
|
dir := revel.BasePath + "/" + fileUrlPath
|
|
if !MkdirAll(dir) {
|
|
return c.RenderText("error, no dir")
|
|
}
|
|
filename := guid + ".pdf"
|
|
path := dir + "/" + filename
|
|
|
|
// leanote.com的secret
|
|
appKey, _ := revel.Config.String("app.secretLeanote")
|
|
if appKey == "" {
|
|
appKey, _ = revel.Config.String("app.secret")
|
|
}
|
|
|
|
// 生成之
|
|
binPath := configService.GetGlobalStringConfig("exportPdfBinPath")
|
|
// 默认路径
|
|
if binPath == "" {
|
|
binPath = "/usr/local/bin/wkhtmltopdf"
|
|
}
|
|
|
|
url := configService.GetSiteUrl() + "/note/toPdf?noteId=" + noteId + "&appKey=" + appKey
|
|
// cc := binPath + " --no-stop-slow-scripts --javascript-delay 10000 \"" + url + "\" \"" + path + "\"" // \"" + cookieDomain + "\" \"" + cookieName + "\" \"" + cookieValue + "\""
|
|
// cc := binPath + " \"" + url + "\" \"" + path + "\"" // \"" + cookieDomain + "\" \"" + cookieName + "\" \"" + cookieValue + "\""
|
|
// 等待--window-status为done的状态
|
|
// http://madalgo.au.dk/~jakobt/wkhtmltoxdoc/wkhtmltopdf_0.10.0_rc2-doc.html
|
|
// wkhtmltopdf参数大全
|
|
var cc string
|
|
if note.IsMarkdown {
|
|
cc = binPath + " --ignore-load-errors --lowquality --window-status done \"" + url + "\" \"" + path + "\"" // \"" + cookieDomain + "\" \"" + cookieName + "\" \"" + cookieValue + "\""
|
|
} else {
|
|
cc = binPath + " --ignore-load-errors --lowquality \"" + url + "\" \"" + path + "\"" // \"" + cookieDomain + "\" \"" + cookieName + "\" \"" + cookieValue + "\""
|
|
}
|
|
|
|
cmd := exec.Command("/bin/sh", "-c", cc)
|
|
_, err := cmd.Output()
|
|
if err != nil {
|
|
return c.RenderText("export pdf error. " + fmt.Sprintf("%v", err))
|
|
}
|
|
file, err := os.Open(path)
|
|
if err != nil {
|
|
return c.RenderText("export pdf error. " + fmt.Sprintf("%v", err))
|
|
}
|
|
// http://stackoverflow.com/questions/8588818/chrome-pdf-display-duplicate-headers-received-from-the-server
|
|
// filenameReturn = strings.Replace(filenameReturn, ",", "-", -1)
|
|
filenameReturn := note.Title
|
|
filenameReturn = FixFilename(filenameReturn)
|
|
if filenameReturn == "" {
|
|
filenameReturn = "Untitled.pdf"
|
|
} else {
|
|
filenameReturn += ".pdf"
|
|
}
|
|
return c.RenderBinary(file, filenameReturn, revel.Attachment, time.Now()) // revel.Attachment
|
|
}
|
|
|
|
// 设置/取消Blog; 置顶
|
|
func (c Note) SetNote2Blog(noteIds []string, isBlog, isTop bool) revel.Result {
|
|
for _, noteId := range noteIds {
|
|
noteService.ToBlog(c.GetUserId(), noteId, isBlog, isTop)
|
|
}
|
|
return c.RenderJson(true)
|
|
}
|