0
0
mirror of https://github.com/rls-moe/nyx synced 2024-11-14 22:12:24 +00:00
nyx/http/board/handler.go
2017-03-15 14:37:37 +01:00

178 lines
4.3 KiB
Go

package board
import (
"bytes"
"errors"
"github.com/GeertJohan/go.rice"
"github.com/pressly/chi"
"github.com/tidwall/buntdb"
"go.rls.moe/nyx/http/errw"
"go.rls.moe/nyx/http/middle"
"go.rls.moe/nyx/resources"
"html/template"
"net/http"
"strconv"
"time"
)
var (
tmpls = template.New("base")
hdlFMap = template.FuncMap{
"renderText": resources.OperateReplyText,
"dict": func(values ...interface{}) (map[string]interface{}, error) {
if len(values)%2 != 0 {
return nil, errors.New("invalid dict call")
}
dict := make(map[string]interface{}, len(values)/2)
for i := 0; i < len(values); i += 2 {
key, ok := values[i].(string)
if !ok {
return nil, errors.New("dict keys must be strings")
}
dict[key] = values[i+1]
}
return dict, nil
},
"rateSpam": resources.SpamScore,
"makeCaptcha": resources.MakeCaptcha,
"dateFromID": resources.DateFromId,
"formatDate": func(date time.Time) string {
return date.Format("02 Jan 06 15:04:05")
},
"isAdminSession": middle.IsAdminSession,
"isModSession": middle.IsModSession,
"captchaProb": resources.CaptchaProb,
"percentFloat": func(in float64) float64 { return in * 100 },
}
)
func LoadTemplates() error {
box, err := rice.FindBox("res/")
if err != nil {
return err
}
tmpls = tmpls.Funcs(hdlFMap)
tmpls, err = tmpls.New("thread/postlists").Parse(box.MustString("thread.tmpl.html"))
if err != nil {
return err
}
_, err = tmpls.New("board/dir").Parse(box.MustString("dir.html"))
if err != nil {
return err
}
_, err = tmpls.New("board/board").Parse(box.MustString("board.html"))
if err != nil {
return err
}
_, err = tmpls.New("board/thread").Parse(box.MustString("thread.html"))
if err != nil {
return err
}
return nil
}
func Router(r chi.Router) {
r.Get("/", serveDir)
r.Get("/dir.html", serveDir)
r.Get("/:board/board.html", serveBoard)
r.Post("/:board/new_thread.sh", handleNewThread)
r.Get("/:board/:thread/thread.html", serveThread)
r.Get("/:board/:thread/:reply/:unused.png", serveFullImage)
r.Get("/:board/:thread/:reply/thumb.png", serveThumb)
r.Post("/:board/:thread/reply.sh", handleNewReply)
r.Handle("/captcha/:captchaId.png", resources.ServeCaptcha)
r.Handle("/captcha/:captchaId.wav", resources.ServeCaptcha)
r.Handle("/captcha/download/:captchaId.wav", resources.ServeCaptcha)
}
func serveThumb(w http.ResponseWriter, r *http.Request) {
dat := bytes.NewBuffer([]byte{})
var date time.Time
db := middle.GetDB(r)
err := db.View(func(tx *buntdb.Tx) error {
bName := chi.URLParam(r, "board")
tid, err := strconv.Atoi(chi.URLParam(r, "thread"))
if err != nil {
return err
}
rid, err := strconv.Atoi(chi.URLParam(r, "reply"))
if err != nil {
return err
}
reply, err := resources.GetReply(tx, r.Host, bName, tid, rid)
if err != nil {
return err
}
_, err = dat.Write(reply.Thumbnail)
if err != nil {
return err
}
date = resources.DateFromId(reply.ID)
return nil
})
if err != nil {
errw.ErrorWriter(err, w, r)
return
}
http.ServeContent(w, r, "thumb.png", date, bytes.NewReader(dat.Bytes()))
}
func serveFullImage(w http.ResponseWriter, r *http.Request) {
dat := bytes.NewBuffer([]byte{})
var date time.Time
db := middle.GetDB(r)
err := db.View(func(tx *buntdb.Tx) error {
bName := chi.URLParam(r, "board")
tid, err := strconv.Atoi(chi.URLParam(r, "thread"))
if err != nil {
return err
}
rid, err := strconv.Atoi(chi.URLParam(r, "reply"))
if err != nil {
return err
}
reply, err := resources.GetReply(tx, r.Host, bName, tid, rid)
if err != nil {
return err
}
_, err = dat.Write(reply.Image)
if err != nil {
return err
}
date = resources.DateFromId(reply.ID)
return nil
})
if err != nil {
errw.ErrorWriter(err, w, r)
return
}
http.ServeContent(w, r, "image.png", date, bytes.NewReader(dat.Bytes()))
}
func serveDir(w http.ResponseWriter, r *http.Request) {
dat := bytes.NewBuffer([]byte{})
db := middle.GetDB(r)
ctx := middle.GetBaseCtx(r)
err := db.View(func(tx *buntdb.Tx) error {
bList, err := resources.ListBoards(tx, r.Host)
if err != nil {
return err
}
ctx["Boards"] = bList
return nil
})
if err != nil {
errw.ErrorWriter(err, w, r)
return
}
err = tmpls.ExecuteTemplate(dat, "board/dir", ctx)
if err != nil {
errw.ErrorWriter(err, w, r)
return
}
http.ServeContent(w, r, "dir.html", time.Now(), bytes.NewReader(dat.Bytes()))
}