1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
package gmi
import (
"bytes"
"context"
"embed"
"errors"
"fmt"
"io"
"io/fs"
"mime"
"path"
"strings"
"text/template"
"dev.mediocregopher.com/mediocre-blog.git/src/post"
"dev.mediocregopher.com/mediocre-blog.git/src/render"
"dev.mediocregopher.com/mediocre-go-lib.git/mctx"
"git.sr.ht/~adnano/go-gemini"
)
type ctxKey string
const (
ctxKeyTplPath ctxKey = "tplPath"
)
func withTplPath(ctx context.Context, path string) context.Context {
return context.WithValue(ctx, ctxKeyTplPath, path)
}
//go:embed tpl
var tplFS embed.FS
func (a *api) tplHandler() (gemini.Handler, error) {
var (
postPreprocessFuncs = PostPreprocessFuncs{a.urlBuilder}
allTpls = template.New("")
)
err := fs.WalkDir(tplFS, "tpl", func(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
if d.IsDir() {
return nil
}
body, err := fs.ReadFile(tplFS, path)
if err != nil {
panic(err)
}
name := strings.TrimPrefix(path, "tpl/")
allTpls, err = allTpls.New(name).Parse(string(body))
if err != nil {
return fmt.Errorf("parsing %q as template: %w", path, err)
}
return nil
})
if err != nil {
return nil, fmt.Errorf("parsing templates: %w", err)
}
return gemini.HandlerFunc(func(
ctx context.Context,
rw gemini.ResponseWriter,
r *gemini.Request,
) {
tplPath, _ := ctx.Value(ctxKeyTplPath).(string)
if tplPath == "" {
tplPath = r.URL.Path
}
tplPath = strings.TrimPrefix(tplPath, "/")
mimeType := mime.TypeByExtension(path.Ext(tplPath))
ctx = mctx.Annotate(ctx,
"url", r.URL,
"tplPath", tplPath,
"mimeType", mimeType,
)
tpl := allTpls.Lookup(tplPath)
if tpl == nil {
rw.WriteHeader(gemini.StatusNotFound, "Page not found, sorry!")
return
}
if mimeType != "" {
rw.SetMediaType(mimeType)
}
buf := new(bytes.Buffer)
err := tpl.Execute(buf, render.NewMethods(
ctx,
r.URL,
a.urlBuilder,
a.params.HTTPGeminiGatewayURL,
a.params.PostStore,
nil, // asset.Store, not supported by gemini endpoint
nil, // post.DraftStore, not supported by gemini endpoint
postPreprocessFuncs,
))
if errors.Is(err, post.ErrPostNotFound) {
a.params.Logger.Warn(ctx, "post not found", err)
rw.WriteHeader(gemini.StatusNotFound, "Post not found")
} else if err != nil {
a.params.Logger.Error(ctx, "rendering error", err)
rw.WriteHeader(gemini.StatusTemporaryFailure, err.Error())
return
}
io.Copy(rw, buf)
}), nil
}
|
