🎨 ToC

88250 · 88250 · commit 31b1cd8f40bb · 2017-11-15T18:07:47.000+08:00
diff --git a/controller/activityctl.go b/controller/activityctl.go
@@ -14,6 +14,7 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+// Package controller is the "controller" layer.
 package controller
 
 import (
diff --git a/controller/articlectl.go b/controller/articlectl.go
@@ -23,6 +23,7 @@ import (
 	"strconv"
 	"strings"
 
+	"github.com/PuerkitoBio/goquery"
 	"github.com/b3log/pipe/model"
 	"github.com/b3log/pipe/service"
 	"github.com/b3log/pipe/util"
@@ -219,7 +220,8 @@ func showArticleAction(c *gin.Context) {
 	dataModel["RelevantArticles"] = articles
 	fillPreviousArticle(c, article, &dataModel)
 	fillNextArticle(c, article, &dataModel)
-	dataModel["ToC"] = "todo"
+	logger.Info(mdResult.ContentHTML)
+	dataModel["ToC"] = ToC(mdResult.ContentHTML)
 	c.HTML(http.StatusOK, getTheme(c)+"/article.html", dataModel)
 
 	service.Article.IncArticleViewCount(article)
@@ -262,3 +264,23 @@ func fillNextArticle(c *gin.Context, article *model.Article, dataModel *DataMode
 	}
 	(*dataModel)["NextArticle"] = nextArticle
 }
+
+func ToC(content string) string {
+	doc, _ := goquery.NewDocumentFromReader(strings.NewReader(content))
+	elements := doc.Find("h1, h2, h3, h4, h5")
+	if nil == elements || 3 > elements.Length() {
+		return ""
+	}
+
+	ret := "<ul class=\"article-toc\">"
+	elements.Each(func(i int, ele *goquery.Selection) {
+		// ele.Nodes[0]
+
+		logger.Infof("%+v", ele)
+
+	})
+
+	_ = ret
+
+	return ""
+}
diff --git a/model/archive.go b/model/archive.go
@@ -14,6 +14,7 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
+// Package model is the "model" layer which defines entity structures with ORM.
 package model
 
 // Archive model.
diff --git a/model/article.go b/model/article.go
@@ -14,7 +14,6 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-// Package model is the "model" layer which defines entity structures with ORM.
 package model
 
 // Article model.
