alex/md2vim
1
Fork 0
Code Releases 1 Activity

Improvements to TOC generation

Browse Source
This commit is contained in:
Alex Yatskov 2015-08-15 17:15:59 +09:00
parent 517200f724
commit eab602b092
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
vimdoc.go
View File

@ -121,14 +121,14 @@ func (v *vimDoc) buildTag(text []byte) []byte {
return []byte(fmt.Sprintf("%s-%s", v.title, text)) return []byte(fmt.Sprintf("%s-%s", v.title, text))
} }
func (v *vimDoc) writeStraddle(out *bytes.Buffer, left, right []byte, trim int) { func (v *vimDoc) writeStraddle(out *bytes.Buffer, left, right []byte, repeat string, trim int) {
padding := v.cols - (len(left) + len(right)) + trim padding := v.cols - (len(left) + len(right)) + trim
if padding <= 0 { if padding <= 0 {
padding = 1 padding = 1
} }
out.Write(left) out.Write(left)
out.WriteString(strings.Repeat(" ", padding)) out.WriteString(strings.Repeat(repeat, padding))
out.Write(right) out.Write(right)
out.WriteString("\n") out.WriteString("\n")
} }
@ -141,7 +141,7 @@ func (v *vimDoc) writeRule(out *bytes.Buffer, repeat string) {
func (v *vimDoc) writeToc(out *bytes.Buffer, head *heading, depth int) { func (v *vimDoc) writeToc(out *bytes.Buffer, head *heading, depth int) {
title := fmt.Sprintf("%s%s", strings.Repeat(" ", depth*v.tabs), head.text) title := fmt.Sprintf("%s%s", strings.Repeat(" ", depth*v.tabs), head.text)
link := fmt.Sprintf("|%s|", v.buildTag(head.text)) link := fmt.Sprintf("|%s|", v.buildTag(head.text))
v.writeStraddle(out, []byte(title), []byte(link), 2) v.writeStraddle(out, []byte(title), []byte(link), ".", 2)
for _, child := range head.children { for _, child := range head.children {
v.writeToc(out, child, depth+1) v.writeToc(out, child, depth+1)
@ -232,7 +232,7 @@ func (v *vimDoc) Header(out *bytes.Buffer, text func() bool, level int, id strin
out.Truncate(headingPos) out.Truncate(headingPos)
tag := fmt.Sprintf("*%s*", v.buildTag(heading.text)) tag := fmt.Sprintf("*%s*", v.buildTag(heading.text))
v.writeStraddle(out, v.fixupHeader(heading.text), []byte(tag), 2) v.writeStraddle(out, v.fixupHeader(heading.text), []byte(tag), " ", 2)
out.WriteString("\n") out.WriteString("\n")
} }
@ -377,7 +377,7 @@ func (v *vimDoc) NormalText(out *bytes.Buffer, text []byte) {
// Header and footer // Header and footer
func (v *vimDoc) DocumentHeader(out *bytes.Buffer) { func (v *vimDoc) DocumentHeader(out *bytes.Buffer) {
if len(v.desc) > 0 { if len(v.desc) > 0 {
v.writeStraddle(out, []byte(v.filename), []byte(v.desc), 0) v.writeStraddle(out, []byte(v.filename), []byte(v.desc), " ", 0)
} else { } else {
out.WriteString(v.filename) out.WriteString(v.filename)
out.WriteString("\n") out.WriteString("\n")