-
Notifications
You must be signed in to change notification settings - Fork 9
/
gometa.go
85 lines (76 loc) · 2.13 KB
/
gometa.go
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
/*
Copyright 2017 - The TXTDirect Authors
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package txtdirect
import (
"html/template"
"net/http"
"strconv"
"strings"
)
// Gometa keeps data for "gometa" type requests
type Gometa struct {
rw http.ResponseWriter
req *http.Request
c Config
rec record
}
// NewGometa returns a fresh instance of Gometa struct
func NewGometa(w http.ResponseWriter, r *http.Request, rec record, c Config) *Gometa {
return &Gometa{
rw: w,
req: r,
rec: rec,
c: c,
}
}
var tmpl = template.Must(template.New("").Parse(`<!DOCTYPE html>
<html>
<head>
<meta name="go-import" content="{{.Host}}{{.Path}} {{.Vcs}} {{.NewURL}}">
{{if .HasGoSource}}<meta name="go-source" content="{{.Host}}{{.Path}} _ {{.NewURL}}/tree/master{/dir} {{.NewURL}}/blob/master{/dir}/{file}#L{line}">{{end}}
</head>
</html>`))
// Serve executes a template on the given ResponseWriter
// that contains go-import meta tag
func (g *Gometa) Serve() error {
if g.rec.Vcs == "" {
g.rec.Vcs = "git"
}
if g.req.URL.Path == "/" {
g.req.URL.Path = ""
}
gosource := strings.Contains(g.rec.To, "github.com")
RequestsByStatus.WithLabelValues(g.req.Host, strconv.Itoa(http.StatusFound)).Add(1)
return tmpl.Execute(g.rw, struct {
Host string
Path string
Vcs string
NewURL string
HasGoSource bool
}{
g.req.Host,
g.req.URL.Path,
g.rec.Vcs,
g.rec.To,
gosource,
})
}
// ValidQuery checks the request query to make sure the requests are
// coming from the Go tool.
func (g *Gometa) ValidQuery() bool {
if g.req.URL.Query().Get("go-get") != "1" {
fallback(g.rw, g.req, "website", http.StatusFound, g.c)
return false
}
return true
}