leanote/app/controllers/member/MemberBlogController.go

package member

import (
	"fmt"
	"github.com/leanote/leanote/app/info"
	. "github.com/leanote/leanote/app/lea"
	"github.com/revel/revel"
	"io/ioutil"
	"os"
	"strings"
	"time"
	//	"github.com/leanote/leanote/app/lea/blog"
)

// 博客管理

type MemberBlog struct {
	MemberBaseController
}

func (c MemberBlog) common() info.UserBlog {
	userId := c.GetUserId()
	userInfo := userService.GetUserInfo(userId)
	c.ViewArgs["userInfo"] = userInfo

	// 得到博客设置信息
	c.ViewArgs["allowCustomDomain"] = configService.GetGlobalStringConfig("allowCustomDomain")

	userBlog := blogService.GetUserBlog(userId)
	c.ViewArgs["userBlog"] = userBlog

	c.SetUserInfo()
	c.SetLocale()
	return userBlog
}

// 得到sorterField 和 isAsc
// okSorter = ['email', 'username']
func (c MemberBlog) getSorter(sorterField string, isAsc bool, okSorter []string) (string, bool) {
	sorter := ""
	c.Params.Bind(&sorter, "sorter")
	if sorter == "" {
		return sorterField, isAsc
	}

	// sorter形式 email-up, email-down
	s2 := strings.Split(sorter, "-")
	if len(s2) != 2 {
		return sorterField, isAsc
	}

	// 必须是可用的sorter
	if okSorter != nil && len(okSorter) > 0 {
		if !InArray(okSorter, s2[0]) {
			return sorterField, isAsc
		}
	}

	sorterField = strings.Title(s2[0])
	if s2[1] == "up" {
		isAsc = true
	} else {
		isAsc = false
	}
	c.ViewArgs["sorter"] = sorter
	return sorterField, isAsc
}

// 博客列表
var userPageSize = 15

func (c MemberBlog) Index(sorter, keywords string) revel.Result {
	userId := c.GetUserId()
	userInfo := userService.GetUserInfo(userId)
	c.ViewArgs["userInfo"] = userInfo

	c.ViewArgs["title"] = c.Message("Posts")
	pageNumber := c.GetPage()
	sorterField, isAsc := c.getSorter("CreatedTime", false, []string{"title", "urlTitle", "updatedTime", "publicTime", "createdTime"})
	pageInfo, blogs := blogService.ListAllBlogs(c.GetUserId(), "", keywords, false, pageNumber, userPageSize, sorterField, isAsc)
	c.ViewArgs["pageInfo"] = pageInfo
	c.ViewArgs["blogs"] = blogs
	c.ViewArgs["keywords"] = keywords

	userAndBlog := userService.GetUserAndBlog(c.GetUserId())
	c.ViewArgs["userAndBlog"] = userAndBlog

	c.common()

	return c.RenderTemplate("member/blog/list.html")
}

// 修改笔记的urlTitle
func (c MemberBlog) UpdateBlogUrlTitle(noteId, urlTitle string) revel.Result {
	re := info.NewRe()
	re.Ok, re.Item = blogService.UpateBlogUrlTitle(c.GetUserId(), noteId, urlTitle)
	return c.RenderJSON(re)
}

// 修改笔记的urlTitle
func (c MemberBlog) UpdateBlogAbstract(noteId string) revel.Result {
	c.common()
	c.ViewArgs["title"] = c.Message("Update Post Abstract")
	note := noteService.GetNoteAndContent(noteId, c.GetUserId())
	if !note.Note.IsBlog {
		return c.E404()
	}
	c.ViewArgs["note"] = note
	c.ViewArgs["noteId"] = noteId
	return c.RenderTemplate("member/blog/update_abstract.html")
}
func (c MemberBlog) DoUpdateBlogAbstract(noteId, imgSrc, desc, abstract string) revel.Result {
	re := info.NewRe()
	re.Ok = blogService.UpateBlogAbstract(c.GetUserId(), noteId, imgSrc, desc, abstract)
	return c.RenderJSON(re)
}

// 基本信息设置
func (c MemberBlog) Base() revel.Result {
	c.common()
	c.ViewArgs["title"] = c.Message("Blog Base Info")
	return c.RenderTemplate("member/blog/base.html")
}
func (c MemberBlog) Comment() revel.Result {
	c.common()
	c.ViewArgs["title"] = c.Message("Comment")
	return c.RenderTemplate("member/blog/comment.html")
}

func (c MemberBlog) Paging() revel.Result {
	c.common()
	c.ViewArgs["title"] = c.Message("Paging")
	return c.RenderTemplate("member/blog/paging.html")
}

func (c MemberBlog) Cate() revel.Result {
	userBlog := c.common()
	c.ViewArgs["title"] = c.Message("Category")

	notebooks := blogService.ListBlogNotebooks(c.GetUserId())
	notebooksMap := map[string]info.Notebook{}
	for _, each := range notebooks {
		notebooksMap[each.NotebookId.Hex()] = each
	}

	var i = 0
	notebooks2 := make([]info.Notebook, len(notebooks))

	// 先要保证已有的是正确的排序
	cateIds := userBlog.CateIds
	has := map[string]bool{} // cateIds中有的
	if cateIds != nil && len(cateIds) > 0 {
		for _, cateId := range cateIds {
			if n, ok := notebooksMap[cateId]; ok {
				notebooks2[i] = n
				i++
				has[cateId] = true
			}
		}
	}
	// 之后
	for _, each := range notebooks {
		id := each.NotebookId.Hex()
		if !has[id] {
			notebooks2[i] = each
			i++
		}
	}
	c.ViewArgs["notebooks"] = notebooks2

	return c.RenderTemplate("member/blog/cate.html")
}

// 修改分类排序
func (c MemberBlog) UpateCateIds(cateIds []string) revel.Result {
	re := info.NewRe()
	re.Ok = blogService.UpateCateIds(c.GetUserId(), cateIds)
	return c.RenderJSON(re)
}

func (c MemberBlog) UpdateCateUrlTitle(cateId, urlTitle string) revel.Result {
	re := info.NewRe()
	re.Ok, re.Item = blogService.UpateCateUrlTitle(c.GetUserId(), cateId, urlTitle)
	return c.RenderJSON(re)
}

// 保存之, 包含增加与保存
func (c MemberBlog) DoAddOrUpdateSingle(singleId, title, content string) revel.Result {
	re := info.NewRe()
	re.Ok = blogService.AddOrUpdateSingle(c.GetUserId(), singleId, title, content)
	return c.RenderJSON(re)
}
func (c MemberBlog) AddOrUpdateSingle(singleId string) revel.Result {
	c.common()
	c.ViewArgs["title"] = c.Message("Add Single")
	c.ViewArgs["singleId"] = singleId
	if singleId != "" {
		c.ViewArgs["title"] = c.Message("Update Single")
		c.ViewArgs["single"] = blogService.GetSingle(singleId)
	}
	return c.RenderTemplate("member/blog/add_single.html")
}
func (c MemberBlog) SortSingles(singleIds []string) revel.Result {
	re := info.NewRe()
	re.Ok = blogService.SortSingles(c.GetUserId(), singleIds)
	return c.RenderJSON(re)
}

func (c MemberBlog) DeleteSingle(singleId string) revel.Result {
	re := info.NewRe()
	re.Ok = blogService.DeleteSingle(c.GetUserId(), singleId)
	return c.RenderJSON(re)
}

// 修改页面标题
func (c MemberBlog) UpdateSingleUrlTitle(singleId, urlTitle string) revel.Result {
	re := info.NewRe()
	re.Ok, re.Item = blogService.UpdateSingleUrlTitle(c.GetUserId(), singleId, urlTitle)
	return c.RenderJSON(re)
}

func (c MemberBlog) Single() revel.Result {
	c.common()
	c.ViewArgs["title"] = c.Message("Single")
	c.ViewArgs["singles"] = blogService.GetSingles(c.GetUserId())

	return c.RenderTemplate("member/blog/single.html")
}

// 主题
func (c MemberBlog) Theme() revel.Result {
	c.common()
	activeTheme, otherThemes := themeService.GetUserThemes(c.GetUserId())
	c.ViewArgs["activeTheme"] = activeTheme
	c.ViewArgs["otherThemes"] = otherThemes

	c.ViewArgs["optionThemes"] = themeService.GetDefaultThemes()

	c.ViewArgs["title"] = c.Message("Theme")
	return c.RenderTemplate("member/blog/theme.html")
}

// 编辑主题
var baseTpls = []string{"header.html", "footer.html", "index.html", "cate.html", "search.html", "post.html", "single.html", "tags.html", "tag_posts.html", "archive.html", "share_comment.html", "404.html", "theme.json", "style.css", "blog.js"}

func (c MemberBlog) UpdateTheme(themeId string, isNew int) revel.Result {
	// 查看用户是否有该theme, 若没有则复制default之
	// 得到主题的文件列表
	userBlog := blogService.GetUserBlog(c.GetUserId())
	if themeId == "" {
		_, themeId = themeService.NewThemeForFirst(userBlog)
		return c.Redirect("/member/blog/updateTheme?themeId=" + themeId)
	}

	c.common()
	c.ViewArgs["title"] = c.Message("Update Theme")
	c.ViewArgs["isNew"] = isNew

	// 先复制之
	c.ViewArgs["themeId"] = themeId

	// 得到脚本目录
	userId := c.GetUserId()

	theme := themeService.GetTheme(userId, themeId)
	if theme.ThemeId == "" {
		return c.E404()
	}
	c.ViewArgs["theme"] = theme

	path := revel.BasePath + "/" + theme.Path

	tpls := ListDir(path)
	myTpls := make([]string, len(baseTpls))
	tplMap := map[string]bool{}
	for i, t := range baseTpls {
		myTpls[i] = t
		tplMap[t] = true
	}
	// 得到没有的tpls
	for _, t := range tpls {
		if t == "images" {
			continue
		}
		if !tplMap[t] {
			myTpls = append(myTpls, t)
		}
	}

	c.ViewArgs["myTpls"] = myTpls

	return c.RenderTemplate("member/blog/update_theme.html")
}

// 得到文件内容
func (c MemberBlog) GetTplContent(themeId string, filename string) revel.Result {
	re := info.NewRe()
	re.Ok = true
	re.Item = themeService.GetTplContent(c.GetUserId(), themeId, filename)

	return c.RenderJSON(re)
}
func (c MemberBlog) UpdateTplContent(themeId, filename, content string) revel.Result {
	re := info.NewRe()
	re.Ok, re.Msg = themeService.UpdateTplContent(c.GetUserId(), themeId, filename, content)
	return c.RenderRe(re)
}

func (c MemberBlog) DeleteTpl(themeId, filename string) revel.Result {
	re := info.NewRe()
	re.Ok = themeService.DeleteTpl(c.GetUserId(), themeId, filename)
	return c.RenderJSON(re)
}

func (c MemberBlog) ListThemeImages(themeId string) revel.Result {
	re := info.NewRe()
	userId := c.GetUserId()
	path := themeService.GetThemeAbsolutePath(userId, themeId) + "/images"
	os.MkdirAll(path, 0755)
	images := ListDir(path)
	re.List = images
	re.Ok = true
	return c.RenderJSON(re)
}

func (c MemberBlog) DeleteThemeImage(themeId, filename string) revel.Result {
	re := info.NewRe()
	path := themeService.GetThemeAbsolutePath(c.GetUserId(), themeId) + "/images/" + filename
	re.Ok = DeleteFile(path)
	return c.RenderJSON(re)
}

// 上传主题图片
func (c MemberBlog) UploadThemeImage(themeId string) revel.Result {
	re := c.uploadImage(themeId)
	c.ViewArgs["fileUrlPath"] = re.Id
	c.ViewArgs["resultCode"] = re.Code
	c.ViewArgs["resultMsg"] = re.Msg
	return c.RenderTemplate("file/blog_logo.html")
}
func (c MemberBlog) uploadImage(themeId string) (re info.Re) {
	var fileId = ""
	var resultCode = 0     // 1表示正常
	var resultMsg = "内部错误" // 错误信息
	var Ok = false

	defer func() {
		re.Id = fileId // 只是id, 没有其它信息
		re.Code = resultCode
		re.Msg = resultMsg
		re.Ok = Ok
	}()

	file, handel, err := c.Request.FormFile("file")
	if err != nil {
		return re
	}
	defer file.Close()
	// 生成上传路径
	dir := themeService.GetThemeAbsolutePath(c.GetUserId(), themeId) + "/images"
	err = os.MkdirAll(dir, 0755)
	if err != nil {
		return re
	}
	// 生成新的文件名
	filename := handel.Filename

	var ext string

	_, ext = SplitFilename(filename)
	if ext != ".gif" && ext != ".jpg" && ext != ".png" && ext != ".bmp" && ext != ".jpeg" {
		resultMsg = "不是图片"
		return re
	}

	filename = filename
	data, err := ioutil.ReadAll(file)
	if err != nil {
		LogJ(err)
		return re
	}

	// > 2M?
	if len(data) > 5*1024*1024 {
		resultCode = 0
		resultMsg = "图片大于2M"
		return re
	}

	toPath := dir + "/" + filename
	err = ioutil.WriteFile(toPath, data, 0777)
	if err != nil {
		LogJ(err)
		return re
	}
	TransToGif(toPath, 0, true)
	resultCode = 1
	resultMsg = "上传成功!"

	return re
}

//
// 使用主题
func (c MemberBlog) ActiveTheme(themeId string) revel.Result {
	re := info.NewRe()
	re.Ok = themeService.ActiveTheme(c.GetUserId(), themeId)
	return c.RenderJSON(re)
}

// 删除主题
func (c MemberBlog) DeleteTheme(themeId string) revel.Result {
	re := info.NewRe()
	re.Ok = themeService.DeleteTheme(c.GetUserId(), themeId)
	return c.RenderJSON(re)
}

// 管理员公开主题
func (c MemberBlog) PublicTheme(themeId string) revel.Result {
	re := info.NewRe()
	re.Ok = themeService.PublicTheme(c.GetUserId(), themeId)
	return c.RenderJSON(re)
}

// 导出
func (c MemberBlog) ExportTheme(themeId string) revel.Result {
	re := info.NewRe()
	var path string
	re.Ok, path = themeService.ExportTheme(c.GetUserId(), themeId)
	if !re.Ok {
		return c.RenderText("error...")
	}
	fw, err := os.Open(path)
	if err != nil {
		return c.RenderText("error")
	}
	return c.RenderBinary(fw, GetFilename(path), revel.Attachment, time.Now()) // revel.Attachment
}

// 导入主题
func (c MemberBlog) ImportTheme() revel.Result {
	re := info.NewRe()

	file, handel, err := c.Request.FormFile("file")
	if err != nil {
		re.Msg = fmt.Sprintf("%v", err)
		return c.RenderJSON(re)
	}

	defer file.Close()
	// 生成上传路径
	userId := c.GetUserId()
	dir := revel.BasePath + "/public/upload/" + userId + "/tmp"
	err = os.MkdirAll(dir, 0755)
	if err != nil {
		re.Msg = fmt.Sprintf("%v", err)
		return c.RenderJSON(re)
	}
	// 生成新的文件名
	filename := handel.Filename

	var ext string
	_, ext = SplitFilename(filename)
	if ext != ".zip" {
		re.Msg = "Please upload zip file"
		return c.RenderJSON(re)
	}

	filename = filename
	data, err := ioutil.ReadAll(file)
	if err != nil {
		return c.RenderJSON(re)
	}

	// > 10M?
	if len(data) > 10*1024*1024 {
		re.Msg = "File is big than 10M"
		return c.RenderJSON(re)
	}

	toPath := dir + "/" + filename
	err = ioutil.WriteFile(toPath, data, 0777)
	if err != nil {
		re.Msg = fmt.Sprintf("%v", err)
		return c.RenderJSON(re)
	}

	// 上传好后, 增加之
	re.Ok, re.Msg = themeService.ImportTheme(c.GetUserId(), toPath)
	return c.RenderRe(re)
}

// 安装
func (c MemberBlog) InstallTheme(themeId string) revel.Result {
	re := info.NewRe()
	re.Ok = themeService.InstallTheme(c.GetUserId(), themeId)
	return c.RenderJSON(re)
}

// 新建主题
func (c MemberBlog) NewTheme() revel.Result {
	_, themeId := themeService.NewTheme(c.GetUserId())
	return c.Redirect("/member/blog/updateTheme?isNew=1&themeId=" + themeId)
}

//-----------
//
func (c MemberBlog) SetUserBlogBase(userBlog info.UserBlogBase) revel.Result {
	re := info.NewRe()
	re.Ok = blogService.UpdateUserBlogBase(c.GetUserId(), userBlog)
	return c.RenderJSON(re)
}
func (c MemberBlog) SetUserBlogComment(userBlog info.UserBlogComment) revel.Result {
	re := info.NewRe()
	re.Ok = blogService.UpdateUserBlogComment(c.GetUserId(), userBlog)
	return c.RenderJSON(re)
}
func (c MemberBlog) SetUserBlogStyle(userBlog info.UserBlogStyle) revel.Result {
	re := info.NewRe()
	re.Ok = blogService.UpdateUserBlogStyle(c.GetUserId(), userBlog)
	return c.RenderJSON(re)
}

func (c MemberBlog) SetUserBlogPaging(perPageSize int, sortField string, isAsc bool) revel.Result {
	re := info.NewRe()
	re.Ok, re.Msg = blogService.UpdateUserBlogPaging(c.GetUserId(), perPageSize, sortField, isAsc)
	return c.RenderRe(re)
}