v1.0 beta init

app/controllers/NoteController.go

@@ -5,12 +5,13 @@ import (
 //	"encoding/json"
 	"gopkg.in/mgo.v2/bson"
 	. "github.com/leanote/leanote/app/lea"
-	"github.com/leanote/leanote/app/lea/html2image"
 	"github.com/leanote/leanote/app/info"
-//	"os"
+	"os/exec"
 //	"github.com/leanote/leanote/app/types"
 //	"io/ioutil"
 //	"fmt"
+//	"bytes"
+//	"os"
 )
 
 type Note struct {
@@ -52,6 +53,8 @@ func (c Note) Index() revel.Result {
 	// 当然, 还需要得到第一个notes的content
 	//...
 	
+	adminUsername, _ := revel.Config.String("adminUsername")
+	c.RenderArgs["isAdmin"] = adminUsername == userInfo.Username
 	c.RenderArgs["userInfo"] = userInfo
 	c.RenderArgs["userInfoJson"] = c.Json(userInfo)
 	c.RenderArgs["notebooks"] = c.Json(notebooks)
@@ -221,31 +224,131 @@ func (c Note) SearchNoteByTags(tags []string) revel.Result {
 	return c.RenderJson(blogs)
 }
 
-//-----------------
+//------------------
 // html2image
+// 判断是否有权限生成
+// 博客也可以调用
+// 这是脚本调用, 没有cookie, 不执行权限控制, 通过传来的appKey判断
+func (c Note) ToImage(noteId, appKey string) revel.Result {
+	// 虽然传了cookie但是这里还是不能得到userId, 所以还是通过appKey来验证之
+	appKeyTrue, _ := revel.Config.String("app.secret")
+	if appKeyTrue != appKey {
+		 return c.RenderText("")
+	}
+	note := noteService.GetNoteById(noteId)
+	if note.NoteId == "" {
+		return c.RenderText("")
+	}
+	
+	c.SetLocale()
+	
+	noteUserId := note.UserId.Hex()
+	content := noteService.GetNoteContent(noteId, noteUserId)
+	userInfo := userService.GetUserInfo(noteUserId);
+	
+	c.RenderArgs["blog"] = note
+	c.RenderArgs["content"] = content.Content
+	c.RenderArgs["userInfo"] = userInfo
+	userBlog := blogService.GetUserBlog(noteUserId)
+	c.RenderArgs["userBlog"] = userBlog
+	
+	return c.RenderTemplate("html2Image/index.html")
+}
+
 func (c Note) Html2Image(noteId string) revel.Result {
 	re := info.NewRe()
 	userId := c.GetUserId()
-	note := noteService.GetNote(noteId, userId)
+	note := noteService.GetNoteById(noteId)
 	if note.NoteId == "" {
+		re.Msg = "No Note"
 		return c.RenderJson(re)
 	}
-	content := noteService.GetNoteContent(noteId, userId)
 	
+	noteUserId := note.UserId.Hex()
+	// 是否有权限
+	if noteUserId != userId {
+		// 是否是有权限协作的
+		if !note.IsBlog && !shareService.HasReadPerm(noteUserId, userId, noteId) {
+			re.Msg = "No Perm"
+			return c.RenderJson(re)
+		}
+	}
+
 	// path 判断是否需要重新生成之
-	fileUrlPath := "/upload/" + userId + "/images/weibo"
+	fileUrlPath := "/upload/" + noteUserId + "/images/weibo"
 	dir := revel.BasePath + "/public/" + fileUrlPath
 	if !ClearDir(dir) {
+		re.Msg = "No Dir"
 		return c.RenderJson(re)
 	}
 	
 	filename := note.NoteId.Hex() + ".png";
 	path := dir + "/" + filename
 	
-	// 生成之
-	html2image.ToImage(userId, c.GetUsername(), noteId, note.Title, content.Content, path)
+	// cookie
+	cookieName := revel.CookiePrefix + "_SESSION"
+	cookie, err := c.Request.Cookie(cookieName)
+	cookieStr := cookie.String()
+	cookieValue := ""
+	if err == nil && len(cookieStr) > len(cookieName) {
+		cookieValue = cookieStr[len(cookieName)+1:]
+	}
 	
-	re.Ok = true
-	re.Id = fileUrlPath + "/" + filename
+	appKey, _ := revel.Config.String("app.secret")
+	cookieDomain, _ := revel.Config.String("cookie.domain")
+	// 生成之
+	url := siteUrl + "/note/toImage?noteId=" + noteId + "&appKey=" + appKey;
+	// /Users/life/Documents/bin/phantomjs/bin/phantomjs /Users/life/Desktop/test/b.js
+	binPath := configService.GetGlobalStringConfig("toImageBinPath")
+	if binPath == "" {
+		return c.RenderJson(re);
+	}
+	cc := binPath + " \"" + url + "\"  \"" + path + "\" \"" + cookieDomain + "\" \"" + cookieName + "\" \"" + cookieValue + "\""
+	cmd := exec.Command("/bin/sh", "-c", cc)
+	Log(cc);
+	b, err := cmd.Output()
+    if err == nil {
+    	re.Ok = true
+		re.Id = fileUrlPath + "/" + filename
+    } else {
+    	re.Msg = string(b)
+    	Log("error:......")
+    	Log(string(b))
+    }
+    
 	return c.RenderJson(re)
+	
+    /*
+    // 这里速度慢, 生成不完全(图片和内容都不全)
+	content := noteService.GetNoteContent(noteId, noteUserId)
+	userInfo := userService.GetUserInfo(noteUserId);
+	
+	c.SetLocale()	
+    
+	c.RenderArgs["blog"] = note
+	c.RenderArgs["content"] = content.Content
+	c.RenderArgs["userInfo"] = userInfo
+	userBlog := blogService.GetUserBlog(noteUserId)
+	c.RenderArgs["userBlog"] = userBlog
+	
+	html := c.RenderTemplateStr("html2Image/index.html") // Result类型的
+	contentFile := dir + "/html";
+	fout, err := os.Create(contentFile)
+    if err != nil {
+		return c.RenderJson(re)
+    }
+    fout.WriteString(html);
+    fout.Close()
+    
+    cc := "/Users/life/Documents/bin/phantomjs/bin/phantomjs /Users/life/Desktop/test/c.js \"" + contentFile + "\"  \"" + path + "\""
+	cmd := exec.Command("/bin/sh", "-c", cc)
+	b, err := cmd.Output()
+    if err == nil {
+    	re.Ok = true
+		re.Id = fileUrlPath + "/" + filename
+    } else {
+    	Log(string(b))
+    }
+	*/
+	
 }