gofmt
This commit is contained in:
parent
ebec3473f2
commit
9a3987c96a
@ -16,11 +16,3 @@ $ go run tool/number.go
|
|||||||
```console
|
```console
|
||||||
$ tool/gofmt
|
$ tool/gofmt
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
||||||
### Golit'ing
|
|
||||||
|
|
||||||
```console
|
|
||||||
$ go run tool/generate.go tool/generate > build/generate.html
|
|
||||||
$ open build/generate.html
|
|
||||||
```
|
|
||||||
|
168
tool/generate.go
168
tool/generate.go
@ -1,168 +0,0 @@
|
|||||||
// ## golit
|
|
||||||
|
|
||||||
// **golit** generates literate-programming style HTML
|
|
||||||
// documentation from a Go source files.
|
|
||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"io/ioutil"
|
|
||||||
"os"
|
|
||||||
"os/exec"
|
|
||||||
"regexp"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Recognize doc lines, extract their comment prefixes.
|
|
||||||
var docsPat = regexp.MustCompile("^\\s*\\/\\/\\s")
|
|
||||||
|
|
||||||
// Recognize title prefixes, for titling web page.
|
|
||||||
var titlePat = regexp.MustCompile("^\\/\\/\\s##\\s")
|
|
||||||
|
|
||||||
// Abort on non-nil errors.
|
|
||||||
func check(err error) {
|
|
||||||
if err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// We'll implement Markdown rendering and Pygments
|
|
||||||
// syntax highlighting by piping the source data through
|
|
||||||
// external programs. This is a general helper for
|
|
||||||
// handling both cases.
|
|
||||||
func pipe(bin string, arg []string, src string) string {
|
|
||||||
cmd := exec.Command(bin, arg...)
|
|
||||||
in, err := cmd.StdinPipe()
|
|
||||||
check(err)
|
|
||||||
out, err := cmd.StdoutPipe()
|
|
||||||
check(err)
|
|
||||||
err = cmd.Start()
|
|
||||||
check(err)
|
|
||||||
in.Write([]byte(src))
|
|
||||||
check(err)
|
|
||||||
err = in.Close()
|
|
||||||
check(err)
|
|
||||||
bytes, err := ioutil.ReadAll(out)
|
|
||||||
check(err)
|
|
||||||
err = cmd.Wait()
|
|
||||||
check(err)
|
|
||||||
return string(bytes)
|
|
||||||
}
|
|
||||||
|
|
||||||
// We'll break the code into {docs, code} pairs, and
|
|
||||||
// then render those text segments before including them
|
|
||||||
// in the HTML doc.
|
|
||||||
type seg struct {
|
|
||||||
docs, code, docsRendered, codeRendered string
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
// Accept exactly 1 argument - the input filename,
|
|
||||||
// less the .go extension.
|
|
||||||
if len(os.Args) != 2 {
|
|
||||||
fmt.Fprintln(os.Stderr, "unexpected args")
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Ensure that we have `markdown` and `pygmentize`,
|
|
||||||
// binaries, remember their paths.
|
|
||||||
markdownPath, err := exec.LookPath("markdown")
|
|
||||||
check(err)
|
|
||||||
pygmentizePath, err := exec.LookPath("pygmentize")
|
|
||||||
check(err)
|
|
||||||
|
|
||||||
// Read the source file in, split into lines.
|
|
||||||
srcBytes, err := ioutil.ReadFile(os.Args[1]+".go")
|
|
||||||
check(err)
|
|
||||||
lines := strings.Split(string(srcBytes), "\n")
|
|
||||||
|
|
||||||
// Group lines into docs/code segments. First,
|
|
||||||
// special case the header to go in its own segment.
|
|
||||||
headerDoc := docsPat.ReplaceAllString(lines[0], "")
|
|
||||||
segs := []*seg{}
|
|
||||||
segs = append(segs, &seg{code: "", docs: headerDoc})
|
|
||||||
|
|
||||||
// Then handle the remaining as code/doc pairs.
|
|
||||||
segs = append(segs, &seg{code: "", docs: ""})
|
|
||||||
last := ""
|
|
||||||
for _, line := range lines[2:] {
|
|
||||||
head := segs[len(segs)-1]
|
|
||||||
docsMatch := docsPat.MatchString(line)
|
|
||||||
emptyMatch := line == ""
|
|
||||||
lastDocs := last == "docs"
|
|
||||||
newDocs := (last == "code") && head.docs != ""
|
|
||||||
newCode := (last == "docs") && head.code != ""
|
|
||||||
// Docs line - strip out comment indicator.
|
|
||||||
if docsMatch || (emptyMatch && lastDocs) {
|
|
||||||
trimed := docsPat.ReplaceAllString(line, "")
|
|
||||||
if newDocs {
|
|
||||||
newSeg := seg{docs: trimed, code: ""}
|
|
||||||
segs = append(segs, &newSeg)
|
|
||||||
} else {
|
|
||||||
head.docs = head.docs + "\n" + trimed
|
|
||||||
}
|
|
||||||
last = "docs"
|
|
||||||
// Code line - preserve all whitespace.
|
|
||||||
} else {
|
|
||||||
if newCode {
|
|
||||||
newSeg := seg{docs: "", code: line}
|
|
||||||
segs = append(segs, &newSeg)
|
|
||||||
} else {
|
|
||||||
head.code = head.code + "\n" + line
|
|
||||||
}
|
|
||||||
last = "code"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Render docs via `markdown` and code via
|
|
||||||
// `pygmentize` in each segment.
|
|
||||||
for _, seg := range segs {
|
|
||||||
seg.docsRendered = pipe(
|
|
||||||
markdownPath,
|
|
||||||
[]string{},
|
|
||||||
seg.docs)
|
|
||||||
seg.codeRendered = pipe(
|
|
||||||
pygmentizePath,
|
|
||||||
[]string{"-l", "go", "-f", "html"},
|
|
||||||
seg.code+" ")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Print HTML header.
|
|
||||||
title := titlePat.ReplaceAllString(lines[0], "")
|
|
||||||
fmt.Printf(`
|
|
||||||
<!DOCTYPE html>
|
|
||||||
<html>
|
|
||||||
<head>
|
|
||||||
<meta http-eqiv="content-type"
|
|
||||||
content="text/html;charset=utf-8">
|
|
||||||
<title>%s</title>
|
|
||||||
<link rel=stylesheet href="../style/lit.css">
|
|
||||||
</head>
|
|
||||||
<body>
|
|
||||||
<div id="container">
|
|
||||||
<div id="background"></div>
|
|
||||||
<table cellspacing="0" cellpadding="0">
|
|
||||||
<thead>
|
|
||||||
<tr>
|
|
||||||
<td class=docs></td>
|
|
||||||
<td class=code></td>
|
|
||||||
</tr>
|
|
||||||
</thead>
|
|
||||||
<tbody>`, title)
|
|
||||||
|
|
||||||
// Print HTML docs/code segments.
|
|
||||||
for _, seg := range segs {
|
|
||||||
fmt.Printf(
|
|
||||||
`<tr>
|
|
||||||
<td class=docs>%s</td>
|
|
||||||
<td class=code>%s</td>
|
|
||||||
</tr>`, seg.docsRendered, seg.codeRendered)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Print HTML footer.
|
|
||||||
fmt.Print(`</tbody>
|
|
||||||
</table>
|
|
||||||
</div>
|
|
||||||
</body>
|
|
||||||
</html>`)
|
|
||||||
}
|
|
@ -1,3 +1,3 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
ls */*.go | xargs -n 1 gofmt -tabs=false -tabwidth=4 -w=true
|
ls src/*/*.go | xargs -n 1 gofmt -tabs=false -tabwidth=4 -w=true
|
||||||
|
Loading…
x
Reference in New Issue
Block a user