forked from mozillazg/comic
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
75 lines (64 loc) · 2.24 KB
/
main.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
package main
import (
"bytes"
"encoding/base64"
"log"
"net/http"
"os"
"strings"
"github.com/bmizerany/pat"
"github.com/mozillazg/comic/views"
)
type ViewFunc func(http.ResponseWriter, *http.Request)
func BasicAuth(f ViewFunc, user, passwd []byte) ViewFunc {
return func(w http.ResponseWriter, r *http.Request) {
basicAuthPrefix := "Basic "
// 获取 request header
auth := r.Header.Get("Authorization")
// 如果是 http basic auth
if strings.HasPrefix(auth, basicAuthPrefix) {
// 解码认证信息
payload, err := base64.StdEncoding.DecodeString(
auth[len(basicAuthPrefix):],
)
if err == nil {
pair := bytes.SplitN(payload, []byte(":"), 2)
if len(pair) == 2 && bytes.Equal(pair[0], user) &&
bytes.Equal(pair[1], passwd) {
// 执行被装饰的函数
f(w, r)
return
}
}
}
// 认证失败,提示 401 Unauthorized
w.Header().Set("WWW-Authenticate", `Basic realm="Restricted"`)
// 401 状态码
w.WriteHeader(http.StatusUnauthorized)
}
}
func main() {
listen := os.Getenv("COMIC_LISTEN")
user := []byte(os.Getenv("COMIC_USER"))
pass := []byte(os.Getenv("COMIC_PASSWD"))
router := pat.New()
router.Get("/", http.HandlerFunc(views.LastComicView))
router.Get("/first", http.HandlerFunc(views.FirstComicView))
router.Get("/last", http.HandlerFunc(views.LastComicView))
router.Get("/random", http.HandlerFunc(views.RandomComicView))
router.Get("/admin", http.HandlerFunc(BasicAuth(views.ListView, user, pass)))
router.Post("/api/comics", http.HandlerFunc(BasicAuth(views.CreateAPIView, user, pass)))
router.Get("/api/comics", http.HandlerFunc(BasicAuth(views.ListAPIView, user, pass)))
router.Get("/api/comics/:id", http.HandlerFunc(BasicAuth(views.GetAPIView, user, pass)))
router.Del("/api/comics/:id", http.HandlerFunc(BasicAuth(views.DeleteAPIView, user, pass)))
router.Put("/api/comics/:id", http.HandlerFunc(BasicAuth(views.UpdateAPIView, user, pass)))
router.Get("/archive", http.HandlerFunc(views.ArchiveView))
router.Get("/atom", http.HandlerFunc(views.AtomView))
router.Get("/:id", http.HandlerFunc(views.GetComicView))
http.Handle("/", router)
log.Printf("listen %s\n", listen)
err := http.ListenAndServe(listen, nil)
if err != nil {
log.Fatal("ListenAndServe: ", err)
}
}