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

Don't assume pygmentize path (I use pythonbrew!).

Browse Source
This commit is contained in:
Adam Moore 2012-10-12 11:48:49 +09:00
parent 149fa796c5
commit 0eddaf3c2c
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tools/generate.go
View File

@ -70,13 +70,18 @@ func mustReadFile(path string) string {
func cachedPygmentize(lex string, src string) string { func cachedPygmentize(lex string, src string) string {
ensureDir(cacheDir) ensureDir(cacheDir)
arg := []string{"-l", lex, "-f", "html"} arg := []string{"-l", lex, "-f", "html"}
bin := "/usr/local/bin/pygmentize"
bin, err := exec.Command("which", "pygmentize").Output()
if err != nil {
fmt.Println("Please install Pygments and ensure pygmentize is in your path first.")
os.Exit(1)
}
cachePath := cacheDir + "/pygmentize-" + strings.Join(arg, "-") + "-" + sha1Sum(src) cachePath := cacheDir + "/pygmentize-" + strings.Join(arg, "-") + "-" + sha1Sum(src)
cacheBytes, cacheErr := ioutil.ReadFile(cachePath) cacheBytes, cacheErr := ioutil.ReadFile(cachePath)
if cacheErr == nil { if cacheErr == nil {
return string(cacheBytes) return string(cacheBytes)
} }
renderBytes := pipe(bin, arg, src) renderBytes := pipe(string(bin), arg, src)
writeErr := ioutil.WriteFile(cachePath, renderBytes, 0600) writeErr := ioutil.WriteFile(cachePath, renderBytes, 0600)
check(writeErr) check(writeErr)
return string(renderBytes) return string(renderBytes)