yeray/gobyexample
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

trim

Browse Source
This commit is contained in:
Mark McGranaghan 2012-09-29 14:57:55 -07:00
parent 08026507eb
commit 6fe814d764
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tool/generate.go
View File

@ -52,7 +52,7 @@ func pipe(bin string, args []string, in string) string {
// We'll break the code into {docs, code} pairs, and // We'll break the code into {docs, code} pairs, and
// then render those text segments before including them // then render those text segments before including them
// in the HTML doc. // in the HTML doc.
type segment struct { type seg struct {
docs, code, docsRendered, codeRendered string docs, code, docsRendered, codeRendered string
} }
@ -78,19 +78,20 @@ func main() {
// Group lines into docs/code segments. // Group lines into docs/code segments.
// Special case the header to go in its own segment. // Special case the header to go in its own segment.
segments := []*segment{} headerDoc := docsPat.ReplaceAllString(lines[0], "")
segments = append(segments, &segment{code: "", docs: docsPat.ReplaceAllString(lines[0], "")}) segs := []*seg{}
segments = append(segments, &segment{code: "", docs: ""}) segs = append(segs, &seg{code: "", docs: headerDoc})
segs = append(segs, &seg{code: "", docs: ""})
last := "" last := ""
for _, line := range lines[2:] { for _, line := range lines[2:] {
head := segments[len(segments)-1] head := segs[len(segs)-1]
// Doc line - trim off the comment markers. // Doc line - trim off the comment markers.
if (line == "" && last == "docs") || docsPat.MatchString(line) { if (line == "" && last == "docs") || docsPat.MatchString(line) {
trimLine := docsPat.ReplaceAllString(line, "") trimLine := docsPat.ReplaceAllString(line, "")
if !(last == "code" && head.docs != "") { if !(last == "code" && head.docs != "") {
head.docs = head.docs + "\n" + trimLine head.docs = head.docs + "\n" + trimLine
} else { } else {
segments = append(segments, &segment{docs: trimLine, code: ""}) segs = append(segs, &seg{docs: trimLine, code: ""})
} }
last = "docs" last = "docs"
// Code line - preserve all whitespace. // Code line - preserve all whitespace.
@ -98,7 +99,7 @@ func main() {
if !(last == "docs" && head.code != "") { if !(last == "docs" && head.code != "") {
head.code = head.code + "\n" + line head.code = head.code + "\n" + line
} else { } else {
segments = append(segments, &segment{docs: "", code: line}) segs = append(segs, &seg{docs: "", code: line})
} }
last = "code" last = "code"
} }