shithub: hugo

Download patch

ref: 4c804319f6db0b8459cc9b5df4a904fd2c55dedd
parent: 186a5ebfca29bffc413c5d968b3f4a69066a867c
author: Gavin D. Howard <yzena.tech@gmail.com>
date: Wed Dec 11 18:48:40 EST 2019

markup/tableofcontents: Add config option for ordered list


--- a/docs/content/en/getting-started/configuration-markup.md
+++ b/docs/content/en/getting-started/configuration-markup.md
@@ -70,4 +70,7 @@
 : The heading level, values starting at 1 (`h1`), to start render the table of contents.
 
 endLevel
-: The heading level, inclusive, to stop render the table of contents.
\ No newline at end of file
+: The heading level, inclusive, to stop render the table of contents.
+
+ordered
+: Whether or not to generate an ordered list instead of an unordered list.
--- a/docs/data/docs.json
+++ b/docs/data/docs.json
@@ -1387,7 +1387,8 @@
       },
       "tableOfContents": {
         "startLevel": 2,
-        "endLevel": 3
+        "endLevel": 3,
+        "ordered": false
       },
       "goldmark": {
         "renderer": {
--- a/hugolib/page__per_output.go
+++ b/hugolib/page__per_output.go
@@ -292,7 +292,7 @@
 	p.p.s.initInit(p.initMain, p.p)
 	if tocProvider, ok := p.convertedResult.(converter.TableOfContentsProvider); ok {
 		cfg := p.p.s.ContentSpec.Converters.GetMarkupConfig()
-		return template.HTML(tocProvider.TableOfContents().ToHTML(cfg.TableOfContents.StartLevel, cfg.TableOfContents.EndLevel))
+		return template.HTML(tocProvider.TableOfContents().ToHTML(cfg.TableOfContents.StartLevel, cfg.TableOfContents.EndLevel, cfg.TableOfContents.Ordered))
 	}
 	return p.tableOfContents
 }
--- a/markup/goldmark/toc_test.go
+++ b/markup/goldmark/toc_test.go
@@ -58,7 +58,7 @@
 	c.Assert(err, qt.IsNil)
 	b, err := conv.Convert(converter.RenderContext{Src: []byte(content), RenderTOC: true})
 	c.Assert(err, qt.IsNil)
-	got := b.(converter.TableOfContentsProvider).TableOfContents().ToHTML(2, 3)
+	got := b.(converter.TableOfContentsProvider).TableOfContents().ToHTML(2, 3, false)
 	c.Assert(got, qt.Equals, `<nav id="TableOfContents">
   <ul>
     <li><a href="#first-h2---now-with-typography">First h2&mdash;now with typography!</a>
--- a/markup/tableofcontents/tableofcontents.go
+++ b/markup/tableofcontents/tableofcontents.go
@@ -62,12 +62,13 @@
 }
 
 // ToHTML renders the ToC as HTML.
-func (toc Root) ToHTML(startLevel, stopLevel int) string {
+func (toc Root) ToHTML(startLevel, stopLevel int, ordered bool) string {
 	b := &tocBuilder{
 		s:          strings.Builder{},
 		h:          toc.Headers,
 		startLevel: startLevel,
 		stopLevel:  stopLevel,
+		ordered:    ordered,
 	}
 	b.Build()
 	return b.s.String()
@@ -79,6 +80,7 @@
 
 	startLevel int
 	stopLevel  int
+	ordered    bool
 }
 
 func (b *tocBuilder) Build() {
@@ -108,7 +110,11 @@
 	if hasChildren {
 		b.s.WriteString("\n")
 		b.indent(indent + 1)
-		b.s.WriteString("<ul>\n")
+		if b.ordered {
+			b.s.WriteString("<ol>\n")
+		} else {
+			b.s.WriteString("<ul>\n")
+		}
 	}
 
 	for _, h := range h {
@@ -117,7 +123,11 @@
 
 	if hasChildren {
 		b.indent(indent + 1)
-		b.s.WriteString("</ul>")
+		if b.ordered {
+			b.s.WriteString("</ol>")
+		} else {
+			b.s.WriteString("</ul>")
+		}
 		b.s.WriteString("\n")
 		b.indent(indent)
 	}
@@ -143,6 +153,7 @@
 var DefaultConfig = Config{
 	StartLevel: 2,
 	EndLevel:   3,
+	Ordered:    false,
 }
 
 type Config struct {
@@ -153,4 +164,7 @@
 	// Heading end level, inclusive, to include in the table of contents.
 	// Default is 3, a value of -1 will include everything.
 	EndLevel int
+
+	// Whether to produce a ordered list or not.
+	Ordered bool
 }
--- a/markup/tableofcontents/tableofcontents_test.go
+++ b/markup/tableofcontents/tableofcontents_test.go
@@ -30,7 +30,7 @@
 	toc.AddAt(Header{Text: "1-H3-1", ID: "1-h2-2"}, 0, 2)
 	toc.AddAt(Header{Text: "Header 2", ID: "h1-2"}, 1, 0)
 
-	got := toc.ToHTML(1, -1)
+	got := toc.ToHTML(1, -1, false)
 	c.Assert(got, qt.Equals, `<nav id="TableOfContents">
   <ul>
     <li><a href="#h1-1">Header 1</a>
@@ -47,7 +47,7 @@
   </ul>
 </nav>`, qt.Commentf(got))
 
-	got = toc.ToHTML(1, 1)
+	got = toc.ToHTML(1, 1, false)
 	c.Assert(got, qt.Equals, `<nav id="TableOfContents">
   <ul>
     <li><a href="#h1-1">Header 1</a></li>
@@ -55,7 +55,7 @@
   </ul>
 </nav>`, qt.Commentf(got))
 
-	got = toc.ToHTML(1, 2)
+	got = toc.ToHTML(1, 2, false)
 	c.Assert(got, qt.Equals, `<nav id="TableOfContents">
   <ul>
     <li><a href="#h1-1">Header 1</a>
@@ -68,7 +68,7 @@
   </ul>
 </nav>`, qt.Commentf(got))
 
-	got = toc.ToHTML(2, 2)
+	got = toc.ToHTML(2, 2, false)
 	c.Assert(got, qt.Equals, `<nav id="TableOfContents">
   <ul>
     <li><a href="#1-h2-1">1-H2-1</a></li>
@@ -76,6 +76,22 @@
   </ul>
 </nav>`, qt.Commentf(got))
 
+	got = toc.ToHTML(1, -1, true)
+	c.Assert(got, qt.Equals, `<nav id="TableOfContents">
+  <ol>
+    <li><a href="#h1-1">Header 1</a>
+      <ol>
+        <li><a href="#1-h2-1">1-H2-1</a></li>
+        <li><a href="#1-h2-2">1-H2-2</a>
+          <ol>
+            <li><a href="#1-h2-2">1-H3-1</a></li>
+          </ol>
+        </li>
+      </ol>
+    </li>
+    <li><a href="#h1-2">Header 2</a></li>
+  </ol>
+</nav>`, qt.Commentf(got))
 }
 
 func TestTocMissingParent(t *testing.T) {
@@ -87,7 +103,7 @@
 	toc.AddAt(Header{Text: "H3", ID: "h3"}, 1, 2)
 	toc.AddAt(Header{Text: "H3", ID: "h3"}, 1, 2)
 
-	got := toc.ToHTML(1, -1)
+	got := toc.ToHTML(1, -1, false)
 	c.Assert(got, qt.Equals, `<nav id="TableOfContents">
   <ul>
     <li>
@@ -108,12 +124,33 @@
   </ul>
 </nav>`, qt.Commentf(got))
 
-	got = toc.ToHTML(3, 3)
+	got = toc.ToHTML(3, 3, false)
 	c.Assert(got, qt.Equals, `<nav id="TableOfContents">
   <ul>
     <li><a href="#h3">H3</a></li>
     <li><a href="#h3">H3</a></li>
   </ul>
+</nav>`, qt.Commentf(got))
+
+	got = toc.ToHTML(1, -1, true)
+	c.Assert(got, qt.Equals, `<nav id="TableOfContents">
+  <ol>
+    <li>
+      <ol>
+        <li><a href="#h2">H2</a></li>
+      </ol>
+    </li>
+    <li>
+      <ol>
+        <li>
+          <ol>
+            <li><a href="#h3">H3</a></li>
+            <li><a href="#h3">H3</a></li>
+          </ol>
+        </li>
+      </ol>
+    </li>
+  </ol>
 </nav>`, qt.Commentf(got))
 
 }