1、实现网站首页

2、实现文档阅读
3、实现项目发布
4、实现用户权限

models/document_tree.go

@@ -2,6 +2,10 @@ package models
 
 import (
 	"github.com/astaxie/beego/orm"
+	"bytes"
+	"strconv"
+	"github.com/astaxie/beego"
+	"html/template"
 )
 
 type DocumentTree struct {
@@ -9,6 +13,7 @@ type DocumentTree struct {
 	DocumentName string 		`json:"text"`
 	ParentId interface{}            `json:"parent"`
 	Identify string 		`json:"identify"`
+	BookIdentify string 		`json:"-"`
 	Version int64			`json:"version"`
 	State *DocumentSelected         `json:"state,omitempty"`
 }
@@ -17,7 +22,7 @@ type DocumentSelected struct {
 	Opened bool        	`json:"opened"`
 }
 
-
+//获取项目的文档树状结构
 func (m *Document) FindDocumentTree(book_id int) ([]*DocumentTree,error){
 	o := orm.NewOrm()
 
@@ -30,6 +35,7 @@ func (m *Document) FindDocumentTree(book_id int) ([]*DocumentTree,error){
 	if err != nil {
 		return trees,err
 	}
+	book,_ := NewBook().Find(book_id)
 
 	trees = make([]*DocumentTree,count)
 
@@ -41,6 +47,7 @@ func (m *Document) FindDocumentTree(book_id int) ([]*DocumentTree,error){
 		tree.DocumentId = item.DocumentId
 		tree.Identify = item.Identify
 		tree.Version = item.Version
+		tree.BookIdentify = book.Identify
 		if item.ParentId > 0 {
 			tree.ParentId = item.ParentId
 		}else{
@@ -54,3 +61,102 @@ func (m *Document) FindDocumentTree(book_id int) ([]*DocumentTree,error){
 
 	return trees,nil
 }
+
+func (m *Document) CreateDocumentTreeForHtml(book_id, selected_id int) (string,error) {
+	trees,err := m.FindDocumentTree(book_id)
+	if err != nil {
+		return "",err
+	}
+	parent_id := getSelectedNode(trees,selected_id)
+
+	buf := bytes.NewBufferString("")
+
+	getDocumentTree(trees,0,selected_id,parent_id,buf)
+
+	return buf.String(),nil
+	
+}
+
+//使用递归的方式获取指定ID的顶级ID
+func getSelectedNode(array []*DocumentTree, parent_id int) int {
+
+	for _,item := range array {
+		if _,ok := item.ParentId.(string); ok && item.DocumentId == parent_id  {
+			return item.DocumentId
+		}else if pid,ok := item.ParentId.(int); ok  && item.DocumentId == parent_id{
+			return getSelectedNode(array,pid)
+		}
+	}
+	return  0
+}
+
+func getDocumentTree(array []*DocumentTree,parent_id int,selected_id int,selected_parent_id int,buf *bytes.Buffer)   {
+	buf.WriteString("<ul>")
+
+	for _,item := range array {
+		pid := 0
+
+		if p,ok := item.ParentId.(int);ok {
+			pid = p
+		}
+		if pid == parent_id {
+			/**
+			$selected = $item['doc_id'] == $selected_id ? ' class="jstree-clicked"' : '';
+                $selected_li = $item['doc_id'] == $selected_parent_id ? ' class="jstree-open"' : '';
+
+                $menu .= '<li id="'.$item['doc_id'].'"'.$selected_li.'><a href="'. route('document.show',['doc_id'=> $item['doc_id']]) .'" title="' . htmlspecialchars($item['doc_name']) . '"'.$selected.'>' . $item['doc_name'] .'</a>';
+
+                $key = array_search($item['doc_id'], array_column($array, 'parent_id'));
+
+                if ($key !== false) {
+                    self::createTree($item['doc_id'], $array,$selected_id,$selected_parent_id);
+                }
+                $menu .= '</li>';
+			 */
+			selected := ""
+			if item.DocumentId == selected_id {
+				selected = ` class="jstree-clicked"`
+			}
+			selected_li := ""
+			if item.DocumentId == selected_parent_id {
+				selected_li = ` class="jstree-open"`
+			}
+			buf.WriteString("<li id=\"")
+			buf.WriteString(strconv.Itoa(item.DocumentId))
+			buf.WriteString("\"")
+			buf.WriteString(selected_li)
+			buf.WriteString("><a href=\"")
+			if item.Identify != ""{
+				uri := beego.URLFor("DocumentController.Read",":key",item.BookIdentify,":id" ,item.Identify)
+				buf.WriteString(uri)
+			}else{
+				uri := beego.URLFor("DocumentController.Read",":key",item.BookIdentify,":id" ,item.DocumentId)
+				buf.WriteString(uri)
+			}
+			buf.WriteString("\" title=\"")
+			buf.WriteString(template.HTMLEscapeString(item.DocumentName) + "\"")
+			buf.WriteString(selected + ">")
+			buf.WriteString(template.HTMLEscapeString(item.DocumentName) + "</a>")
+
+			for _,sub := range array {
+				if p,ok := sub.ParentId.(int);ok && p == item.DocumentId{
+					getDocumentTree(array,p,selected_id,selected_parent_id,buf)
+				}
+			}
+			buf.WriteString("</li>")
+
+		}
+	}
+	buf.WriteString("</ul>")
+}
+
+
+
+
+
+
+
+
+
+
+