alex/md2vim
1
Fork 0
Code Releases 1 Activity

Improved TOC generation

Browse Source
This commit is contained in:
Alex Yatskov 2015-08-19 17:45:08 +09:00
parent 1964ab2dc1
commit 2ae4885064
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
vimdoc.go
View File

@ -226,17 +226,13 @@ func (v *vimDoc) Header(out *bytes.Buffer, text func() bool, level int, id strin
return return
} }
if v.tocPos == -1 && len(v.headings) > 0 {
v.tocPos = initPos
}
var temp []byte var temp []byte
temp = append(temp, out.Bytes()[headingPos:]...) temp = append(temp, out.Bytes()[headingPos:]...)
out.Truncate(headingPos)
h := &heading{temp, level} h := &heading{temp, level}
v.headings = append(v.headings, h) v.headings = append(v.headings, h)
out.Truncate(headingPos)
tag := fmt.Sprintf("*%s*", v.buildHelpTag(h.text)) tag := fmt.Sprintf("*%s*", v.buildHelpTag(h.text))
title := fmt.Sprintf("%s %s", v.buildChapters(h), bytes.ToUpper(h.text)) title := fmt.Sprintf("%s %s", v.buildChapters(h), bytes.ToUpper(h.text))
v.writeSplitText(out, []byte(title), []byte(tag), " ", 2) v.writeSplitText(out, []byte(title), []byte(tag), " ", 2)
@ -391,6 +387,7 @@ func (v *vimDoc) DocumentHeader(out *bytes.Buffer) {
} }
out.WriteString("\n") out.WriteString("\n")
v.tocPos = out.Len()
} }
func (v *vimDoc) DocumentFooter(out *bytes.Buffer) { func (v *vimDoc) DocumentFooter(out *bytes.Buffer) {
@ -398,8 +395,13 @@ func (v *vimDoc) DocumentFooter(out *bytes.Buffer) {
if v.tocPos > 0 && v.flags&flagNoToc == 0 { if v.tocPos > 0 && v.flags&flagNoToc == 0 {
temp.Write(out.Bytes()[:v.tocPos]) temp.Write(out.Bytes()[:v.tocPos])
v.writeRule(&temp, "=")
temp.WriteString("TABLE OF CONTENTS\n")
temp.WriteString("\n")
v.writeToc(&temp) v.writeToc(&temp)
temp.WriteString("\n") temp.WriteString("\n")
temp.Write(out.Bytes()[v.tocPos:]) temp.Write(out.Bytes()[v.tocPos:])
} else { } else {
temp.ReadFrom(out) temp.ReadFrom(out)