diff options
Diffstat (limited to '')
| -rw-r--r-- | routes/template.go | 35 |
1 files changed, 34 insertions, 1 deletions
diff --git a/routes/template.go b/routes/template.go index bc1dd37..3d1e1cf 100644 --- a/routes/template.go +++ b/routes/template.go @@ -1,10 +1,13 @@ package routes import ( + "bytes" "html/template" + "io" "log" "net/http" "path/filepath" + "strings" "icyphox.sh/legit/git" ) @@ -40,12 +43,42 @@ func (d *deps) listFiles(files []git.NiceTree, data map[string]any, w http.Respo } } +func countLines(r io.Reader) (int, error) { + buf := make([]byte, 32*1024) + count := 0 + nl := []byte{'\n'} + + for { + c, err := r.Read(buf) + count += bytes.Count(buf[:c], nl) + + switch { + case err == io.EOF: + return count, nil + case err != nil: + return 0, err + } + } +} + func (d *deps) showFile(content string, data map[string]any, w http.ResponseWriter) { tpath := filepath.Join(d.c.Template.Dir, "*") t := template.Must(template.ParseGlob(tpath)) - // TODO: Process content here. + lc, err := countLines(strings.NewReader(content)) + if err != nil { + // Non-fatal, we'll just skip showing line numbers in the template. + log.Printf("counting lines: %s", err) + } + + lines := make([]int, lc) + if lc > 0 { + for i := range lines { + lines[i] = i + 1 + } + } + data["linecount"] = lines data["content"] = content data["meta"] = d.c.Meta |