Hello World.
\n", string(b)) 29 | }) 30 | 31 | t.Run("read error", func(t *testing.T) { 32 | in := &errorReader{} 33 | 34 | var meta struct { 35 | Title string `yml:"title"` 36 | } 37 | 38 | out := Markdown(Frontmatter(in, &meta)) 39 | 40 | _, err := ioutil.ReadAll(out) 41 | assert.EqualError(t, err, `boom`) 42 | }) 43 | } 44 | -------------------------------------------------------------------------------- /html.go: -------------------------------------------------------------------------------- 1 | package static 2 | 3 | import ( 4 | "bufio" 5 | "io" 6 | "strconv" 7 | "strings" 8 | 9 | dom "github.com/PuerkitoBio/goquery" 10 | snakecase "github.com/segmentio/go-snakecase" 11 | ) 12 | 13 | // anchorHTML is the anchor element and SVG icon. 14 | var anchorHTML = `` 17 | 18 | // anchorEl is the instance of the anchor element which is cloned. 19 | var anchorEl *dom.Selection 20 | 21 | // initialize anchor element. 22 | func init() { 23 | doc, err := dom.NewDocumentFromReader(strings.NewReader(anchorHTML)) 24 | if err != nil { 25 | panic(err) 26 | } 27 | 28 | anchorEl = doc.Find("a") 29 | } 30 | 31 | // HeadingAnchors returns a reader with 32 | // HTML heading ids derived from their text. 33 | func HeadingAnchors(r io.Reader) io.ReadCloser { 34 | pr, pw := io.Pipe() 35 | 36 | go func() { 37 | doc, err := dom.NewDocumentFromReader(r) 38 | if err != nil { 39 | pw.CloseWithError(err) 40 | return 41 | } 42 | 43 | scope := []string{""} 44 | prev := 1 45 | 46 | doc.Find("h1, h2, h3, h4, h5, h6").Each(func(i int, s *dom.Selection) { 47 | curr, _ := strconv.Atoi(string(s.Get(0).Data[1])) 48 | change := curr - prev 49 | 50 | if change <= 0 { 51 | scope = scope[:len(scope)+(change-1)] 52 | } 53 | 54 | scope = append(scope, snakecase.Snakecase(s.Text())) 55 | prev = curr 56 | 57 | id := strings.Join(scope, ".") 58 | a := anchorEl.Clone() 59 | a.SetAttr("id", id) 60 | a.SetAttr("href", "#"+id) 61 | s.SetAttr("class", "Heading") 62 | s.PrependSelection(a) 63 | }) 64 | 65 | html, err := doc.Html() 66 | if err != nil { 67 | pw.CloseWithError(err) 68 | return 69 | } 70 | 71 | pw.Write([]byte(html)) 72 | pw.Close() 73 | }() 74 | 75 | return pr 76 | } 77 | 78 | // Notes returns a reader with paragraphs starting 79 | // with "Notes:" to be considered highlighted note. 80 | func Notes(r io.Reader) io.ReadCloser { 81 | pr, pw := io.Pipe() 82 | 83 | go func() { 84 | scan := bufio.NewScanner(r) 85 | 86 | for scan.Scan() { 87 | io.WriteString(pw, note(scan.Text())) 88 | io.WriteString(pw, "\n") 89 | } 90 | 91 | pw.CloseWithError(scan.Err()) 92 | }() 93 | 94 | return pr 95 | } 96 | 97 | // note helper. 98 | func note(s string) string { 99 | if !strings.HasPrefix(s, "Note:") { 100 | return s 101 | } 102 | s = strings.TrimPrefix(s, "
Note:") 103 | s = strings.TrimSuffix(s, "
") 104 | s = strings.TrimSpace(s) 105 | return `` + s + `
.
47 | s.Parent().ReplaceWithHtml(buf.String())
48 | })
49 |
50 | html, err := doc.Html()
51 | if err != nil {
52 | pw.CloseWithError(err)
53 | return
54 | }
55 |
56 | pw.Write([]byte(html))
57 | pw.Close()
58 | }()
59 |
60 | return pr
61 | }
62 |
63 | // detectLexer returns a chroma lexer based on the classname.
64 | func detectLexer(s *dom.Selection) chroma.Lexer {
65 | var lexer chroma.Lexer
66 | var lang string
67 |
68 | if classes, ok := s.Attr("class"); ok {
69 | for _, c := range strings.Split(classes, " ") {
70 | if strings.HasPrefix(c, "language-") {
71 | lang = strings.TrimPrefix(c, "language-")
72 | }
73 | }
74 | }
75 |
76 | if lang != "" {
77 | lexer = lexers.Get(lang)
78 | }
79 |
80 | if lexer == nil {
81 | lexer = lexers.Analyse(s.Contents().Text())
82 | }
83 |
84 | if lexer == nil {
85 | lexer = lexers.Fallback
86 | }
87 |
88 | return chroma.Coalesce(lexer)
89 | }
90 |
--------------------------------------------------------------------------------
/html_highlight_test.go:
--------------------------------------------------------------------------------
1 | package static
2 |
3 | import (
4 | "io"
5 | "io/ioutil"
6 | "os"
7 | "path/filepath"
8 | "testing"
9 |
10 | "github.com/tj/assert"
11 | )
12 |
13 | func fixture(t testing.TB, name string) io.ReadCloser {
14 | path := filepath.Join("testdata", name)
15 | f, err := os.Open(path)
16 | assert.NoError(t, err, "open")
17 | return f
18 | }
19 |
20 | func TestSyntaxHighlight(t *testing.T) {
21 | in := fixture(t, "highlight_input.html")
22 | out := fixture(t, "highlight_output.html")
23 |
24 | got, _ := ioutil.ReadAll(SyntaxHighlight(in))
25 | expect, _ := ioutil.ReadAll(out)
26 |
27 | if string(got) != string(expect) {
28 | t.Errorf("\nExpected:\n\n%s\n\nGot:\n\n%s", string(expect), string(got))
29 | // ioutil.WriteFile("testdata/highlight_output.html", got, 0644)
30 | }
31 | }
32 |
--------------------------------------------------------------------------------
/html_test.go:
--------------------------------------------------------------------------------
1 | package static
2 |
3 | import (
4 | "io"
5 | "io/ioutil"
6 | "os"
7 | "strings"
8 | )
9 |
10 | func ExampleHeadingAnchors() {
11 | r := ioutil.NopCloser(strings.NewReader(`
12 | Article 1
13 | Section 1
14 | Section 2
15 | Section 3
16 | Sub Section 1
17 | Sub Sub Section 1
18 | Article 2
19 | Article 3
20 | Section 1
21 | `))
22 |
23 | r = HeadingAnchors(r)
24 | io.Copy(os.Stdout, r)
25 | // Output:
26 | // Article 1
29 | // Section 1
32 | // Section 2
35 | // Section 3
38 | // Sub Section 1
41 | // Sub Sub Section 1
44 | // Article 2
47 | // Article 3
50 | // Section 1
53 | //
54 | }
55 |
56 | func ExampleNotes() {
57 | r := ioutil.NopCloser(strings.NewReader(`
58 | Something here.
59 |
60 | Note: Some note here about whatever nothing exciting.
61 |
62 | More content.
63 | `))
64 |
65 | r = Notes(Markdown(r))
66 | io.Copy(os.Stdout, r)
67 | // Output:
68 | // Something here.
69 | //
70 | // Some note here about whatever nothing exciting.
71 | //
72 | // More content.
73 | }
74 |
--------------------------------------------------------------------------------
/inject/inject.go:
--------------------------------------------------------------------------------
1 | // Package inject provides script and style injection utilities.
2 | package inject
3 |
4 | import (
5 | "encoding/json"
6 | "html"
7 | "strings"
8 | )
9 |
10 | // Head injects a string before the closing head tag.
11 | func Head(html, s string) string {
12 | return strings.Replace(html, "", " "+s+"\n ", 1)
13 | }
14 |
15 | // Body injects a string before the closing body tag.
16 | func Body(html, s string) string {
17 | return strings.Replace(html, "