181 lines
5.3 KiB
Go
181 lines
5.3 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/sha1"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"github.com/russross/blackfriday"
|
|
"io/ioutil"
|
|
"os"
|
|
"os/exec"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
func check(err error) {
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func render(bin string, arg []string, src string) []byte {
|
|
cmd := exec.Command(bin, arg...)
|
|
in, _ := cmd.StdinPipe()
|
|
out, _ := cmd.StdoutPipe()
|
|
cmd.Start()
|
|
in.Write([]byte(src))
|
|
in.Close()
|
|
bytes, _ := ioutil.ReadAll(out)
|
|
err := cmd.Wait()
|
|
check(err)
|
|
return bytes
|
|
}
|
|
|
|
func sha1Sum(s string) string {
|
|
h := sha1.New()
|
|
h.Write([]byte(s))
|
|
b := h.Sum(nil)
|
|
return hex.EncodeToString(b)
|
|
}
|
|
func cachedRender(bin string, arg []string, src string) string {
|
|
cachePath := "/tmp/gbe-book-cache/pygmentize" + "-" + strings.Join(arg, "-") + "-" + sha1Sum(src)
|
|
cacheBytes, cacheErr := ioutil.ReadFile(cachePath)
|
|
if cacheErr == nil {
|
|
return string(cacheBytes)
|
|
}
|
|
renderBytes := render(bin, arg, src)
|
|
writeErr := ioutil.WriteFile(cachePath, renderBytes, 0600)
|
|
check(writeErr)
|
|
return string(renderBytes)
|
|
}
|
|
|
|
func readLines(path string) []string {
|
|
srcBytes, err := ioutil.ReadFile(path)
|
|
check(err)
|
|
return strings.Split(string(srcBytes), "\n")
|
|
}
|
|
|
|
func whichLexer(path string) string {
|
|
if strings.HasSuffix(path, ".go") {
|
|
return "go"
|
|
} else if strings.HasSuffix(path, ".sh") {
|
|
return "console"
|
|
}
|
|
panic("No lexer for " + path)
|
|
return ""
|
|
}
|
|
|
|
func debug(msg string) {
|
|
if os.Getenv("DEBUG") == "1" {
|
|
fmt.Fprintln(os.Stderr, msg)
|
|
}
|
|
}
|
|
|
|
var docsPat = regexp.MustCompile("^\\s*(\\/\\/|#)\\s")
|
|
var headerPat = regexp.MustCompile("^\\s*(\\/\\/|#)\\s#+\\s")
|
|
var todoPat = regexp.MustCompile("\\/\\/ todo: ")
|
|
|
|
type seg struct {
|
|
docs, code, docsRendered, codeRendered string
|
|
}
|
|
|
|
func main() {
|
|
if len(os.Args) <= 1 {
|
|
fmt.Fprintln(os.Stderr, "usage: tool/build-html-inner *.{go,sh} > output.html")
|
|
os.Exit(1)
|
|
}
|
|
|
|
fmt.Print(`<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<meta http-eqiv="content-type" content="text/html;charset=utf-8">
|
|
<title>Go by Example</title>
|
|
<link rel=stylesheet href="../style/book.css">
|
|
</head>
|
|
<body>
|
|
<div id="container">
|
|
<table cellspacing="0" cellpadding="0">
|
|
<tbody>`)
|
|
|
|
for _, sourcePath := range os.Args[1:] {
|
|
lexer := whichLexer(sourcePath)
|
|
lines := readLines(sourcePath)
|
|
|
|
segs := []*seg{}
|
|
segs = append(segs, &seg{code: "", docs: ""})
|
|
lastSeen := ""
|
|
for _, line := range lines {
|
|
if todoPat.MatchString(line) {
|
|
continue
|
|
}
|
|
headerMatch := headerPat.MatchString(line)
|
|
docsMatch := docsPat.MatchString(line)
|
|
emptyMatch := line == ""
|
|
lastSeg := segs[len(segs)-1]
|
|
lastHeader := lastSeen == "header"
|
|
lastDocs := lastSeen == "docs"
|
|
newHeader := lastSeen != "header" && lastSeg.docs != ""
|
|
newDocs := lastSeen == "code" || lastSeen == "header"
|
|
newCode := (lastSeen != "code" && lastSeg.code != "") || lastSeen == "header"
|
|
if newHeader || newDocs || newCode {
|
|
debug("NEWSEG")
|
|
}
|
|
if headerMatch || (emptyMatch && lastHeader) {
|
|
trimmed := docsPat.ReplaceAllString(line, "")
|
|
if newHeader {
|
|
newSeg := seg{docs: trimmed, code: ""}
|
|
segs = append(segs, &newSeg)
|
|
} else {
|
|
lastSeg.docs = lastSeg.docs + "\n" + trimmed
|
|
}
|
|
debug("HEAD")
|
|
lastSeen = "header"
|
|
} else if docsMatch || (emptyMatch && lastDocs) {
|
|
trimmed := docsPat.ReplaceAllString(line, "")
|
|
if newDocs {
|
|
debug("NEWSEG")
|
|
newSeg := seg{docs: trimmed, code: ""}
|
|
segs = append(segs, &newSeg)
|
|
} else {
|
|
lastSeg.docs = lastSeg.docs + "\n" + trimmed
|
|
}
|
|
debug("DOCS")
|
|
lastSeen = "docs"
|
|
} else {
|
|
if newCode {
|
|
newSeg := seg{docs: "", code: line}
|
|
segs = append(segs, &newSeg)
|
|
} else {
|
|
lastSeg.code = lastSeg.code + "\n" + line
|
|
}
|
|
debug("CODE")
|
|
lastSeen = "code"
|
|
}
|
|
}
|
|
segs = append(segs, &seg{code: "", docs: ""})
|
|
|
|
for _, seg := range segs {
|
|
if seg.docs != "" {
|
|
seg.docsRendered = string(blackfriday.MarkdownCommon([]byte(seg.docs)))
|
|
}
|
|
if seg.code != "" {
|
|
seg.codeRendered = cachedRender("/usr/local/bin/pygmentize", []string{"-l", lexer, "-f", "html"}, seg.code)
|
|
}
|
|
}
|
|
|
|
for _, seg := range segs {
|
|
codeClasses := "code"
|
|
if seg.code == "" {
|
|
codeClasses = codeClasses + " empty"
|
|
}
|
|
fmt.Printf(
|
|
`<tr>
|
|
<td class=docs>%s</td>
|
|
<td class="%s">%s</td>
|
|
</tr>`, seg.docsRendered, codeClasses, seg.codeRendered)
|
|
}
|
|
}
|
|
|
|
fmt.Print(`</tbody></table></div></body></html>`)
|
|
}
|