317 lines
8.4 KiB
Go
317 lines
8.4 KiB
Go
package http
|
|
|
|
import (
|
|
"fmt"
|
|
"image/jpeg"
|
|
"math"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"git.arav.su/Arav/dwelling-home/pkg/mindflow"
|
|
"git.arav.su/Arav/dwelling-home/pkg/servicestat"
|
|
"git.arav.su/Arav/dwelling-home/pkg/util"
|
|
"git.arav.su/Arav/dwelling-home/web"
|
|
"git.arav.su/Arav/justcaptcha/pkg/captcha"
|
|
"git.arav.su/Arav/justcaptcha/pkg/captcha/inmemdb"
|
|
"git.arav.su/Arav/justcaptcha/pkg/dwcaptcha"
|
|
gbsqlite "git.arav.su/Arav/justguestbook/database/sqlite"
|
|
"git.arav.su/Arav/justguestbook/guestbook"
|
|
)
|
|
|
|
type Handlers struct {
|
|
guestbookDB *gbsqlite.SQLiteDatabase
|
|
captchaExpire time.Duration
|
|
guestbookOwner string
|
|
guestbookPageSize int64
|
|
}
|
|
|
|
func NewHandlers(captchaExpire time.Duration, dbPath, gbOwner string, gbPageSize int64) *Handlers {
|
|
gbdb, err := gbsqlite.New(dbPath + "/guestbook.sqlite")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
inmemdb.SetExpiry(captchaExpire)
|
|
|
|
return &Handlers{
|
|
guestbookDB: gbdb,
|
|
captchaExpire: captchaExpire,
|
|
guestbookOwner: gbOwner,
|
|
guestbookPageSize: gbPageSize}
|
|
}
|
|
|
|
func (h *Handlers) CloseDB() {
|
|
h.guestbookDB.Close()
|
|
}
|
|
|
|
func (h *Handlers) AssetsFS() http.FileSystem {
|
|
return web.Assets()
|
|
}
|
|
|
|
func (h *Handlers) Index(w http.ResponseWriter, r *http.Request) {
|
|
web.Index("", w)
|
|
}
|
|
|
|
func (h *Handlers) Stuff(w http.ResponseWriter, r *http.Request) {
|
|
web.Stuff("/ Stuff", util.GetServiceByHost(r.Host, util.ServiceGit),
|
|
util.GetServiceByHost(r.Host, util.ServiceFiles), w)
|
|
}
|
|
|
|
func (h *Handlers) Mindflow(w http.ResponseWriter, r *http.Request) {
|
|
posts := []mindflow.Post{
|
|
{Category: "Kek", Date: time.Now(), Title: "LMAO", Body: "Testing!"},
|
|
}
|
|
web.Mindflow("/ Mindflow", posts, r, w)
|
|
}
|
|
|
|
func (h *Handlers) About(w http.ResponseWriter, r *http.Request) {
|
|
var lst servicestat.ServiceList = make(servicestat.ServiceList)
|
|
reimu, err := servicestat.GatherStatus("http://reimu.arav.home.arpa:14882/processes")
|
|
if err == nil {
|
|
for k, v := range reimu {
|
|
lst[k] = v
|
|
}
|
|
}
|
|
sakuya, err := servicestat.GatherStatus("http://sakuya.arav.home.arpa:14882/processes")
|
|
if err == nil {
|
|
for k, v := range sakuya {
|
|
lst[k] = v
|
|
}
|
|
}
|
|
|
|
web.About("/ About", util.GetServiceByHost(r.Host, util.ServiceFiles), lst, w)
|
|
}
|
|
|
|
func (h *Handlers) Article(w http.ResponseWriter, r *http.Request) {
|
|
name := r.URL.Path[strings.LastIndex(r.URL.Path, "/")+1:]
|
|
artcl, err := web.GetArticle(name)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
web.Article(artcl.Title, artcl.Description, string(artcl.Body), artcl.Date, r.Host, name, r, w)
|
|
}
|
|
|
|
func (h *Handlers) Guestbook(w http.ResponseWriter, r *http.Request) {
|
|
var page int64
|
|
var pageSize int64
|
|
|
|
pageStr := r.URL.Query().Get("p")
|
|
if pageStr != "" {
|
|
page, _ = strconv.ParseInt(pageStr, 10, 64)
|
|
} else {
|
|
page = 1
|
|
}
|
|
|
|
pageSzStr := r.URL.Query().Get("ps")
|
|
if pageSzStr != "" {
|
|
pageSize, _ = strconv.ParseInt(pageSzStr, 10, 64)
|
|
} else {
|
|
pageSize = h.guestbookPageSize
|
|
}
|
|
|
|
entries, err := h.guestbookDB.Entries(page, pageSize)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
entry.Message = strings.ReplaceAll(entry.Message, "\\n", "\n")
|
|
}
|
|
|
|
entriesCount, _ := h.guestbookDB.Count()
|
|
pageCount := int64(math.Ceil(float64(entriesCount) / float64(h.guestbookPageSize)))
|
|
|
|
dwc := dwcaptcha.NewDwellingCaptcha(h.captchaExpire)
|
|
_, id := inmemdb.New(r.RemoteAddr, dwc)
|
|
|
|
web.Guestbook("/ Guestbook", h.guestbookOwner, string(id), pageCount, entries, r, w)
|
|
}
|
|
|
|
func (h *Handlers) GuestbookPost(w http.ResponseWriter, r *http.Request) {
|
|
var entry *guestbook.Entry
|
|
var err error
|
|
|
|
if r.Header.Get("Content-Type") == "application/x-www-form-urlencoded" {
|
|
r.ParseForm()
|
|
|
|
if !inmemdb.Solve(captcha.ID(r.FormValue("captcha_id")), captcha.Answer(r.FormValue("captcha_answer"))) {
|
|
h.ForbiddenError("Wrong answer given.", "Here's your message:"+r.FormValue("message"), w)
|
|
return
|
|
}
|
|
|
|
if r.FormValue("name") == "" {
|
|
r.Form.Set("name", "Anonymous")
|
|
}
|
|
|
|
message := strings.ReplaceAll(r.FormValue("message"), "\r\n", "\n")
|
|
message = strings.ReplaceAll(message, "\n\r", "\n")
|
|
message = strings.ReplaceAll(message, "\r", "\n")
|
|
|
|
entry, err = guestbook.NewEntry(r.FormValue("name"), message,
|
|
r.FormValue("website"), r.FormValue("hide_website") != "")
|
|
if err != nil {
|
|
h.InternalError(err.Error(), "Here's your message:"+r.FormValue("message"), w)
|
|
return
|
|
}
|
|
|
|
}
|
|
|
|
if err = h.guestbookDB.NewEntry(entry); err != nil {
|
|
h.InternalError(err.Error(), "Here's your message:"+r.FormValue("message"), w)
|
|
return
|
|
}
|
|
|
|
http.Redirect(w, r, "/guestbook", http.StatusMovedPermanently)
|
|
}
|
|
|
|
func (h *Handlers) GuestbookAdmin(w http.ResponseWriter, r *http.Request) {
|
|
entriesCount, _ := h.guestbookDB.Count()
|
|
entries, err := h.guestbookDB.Entries(1, entriesCount)
|
|
for _, entry := range entries {
|
|
entry.Message = strings.ReplaceAll(entry.Message, "\\n", "\n")
|
|
}
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
web.GuestbookAdmin("/ Guestbook Administration", h.guestbookOwner, entries, r, w)
|
|
}
|
|
|
|
func (h *Handlers) GuestbookAdminDeleteEntry(w http.ResponseWriter, r *http.Request) {
|
|
id, err := strconv.ParseInt(GetURLParam(r, "id"), 10, 64)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
_, err = h.guestbookDB.DeleteEntry(id)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
|
|
func (h *Handlers) GuestbookAdminUpdateEntry(w http.ResponseWriter, r *http.Request) {
|
|
}
|
|
|
|
func (h *Handlers) GuestbookAdminDeleteReply(w http.ResponseWriter, r *http.Request) {
|
|
id, err := strconv.ParseInt(GetURLParam(r, "id"), 10, 64)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
err = h.guestbookDB.DeleteReply(id)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
|
|
func (h *Handlers) GuestbookAdminUpdateReply(w http.ResponseWriter, r *http.Request) {
|
|
}
|
|
|
|
func (h *Handlers) GuestbookAdminReply(w http.ResponseWriter, r *http.Request) {
|
|
|
|
r.ParseForm()
|
|
|
|
id, err := strconv.ParseInt(r.FormValue("entry_id"), 10, 64)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
reply, err := guestbook.NewReply(id, r.FormValue("message"))
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
if r.URL.Query().Has("update") {
|
|
if err = h.guestbookDB.UpdateReply(reply); err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
} else {
|
|
err = h.guestbookDB.NewReply(reply)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (h *Handlers) RSS(w http.ResponseWriter, r *http.Request) {
|
|
posts := []mindflow.Post{
|
|
{Category: "Kek", Date: time.Now(), Title: "LMAO", Body: "Testing!"},
|
|
}
|
|
web.RSS(r.URL.Scheme+"://"+r.Host, "Arav", posts, r, w)
|
|
}
|
|
|
|
func (h *Handlers) Robots(w http.ResponseWriter, r *http.Request) {
|
|
data, _ := web.AssetsGetFile("robots.txt")
|
|
w.Write(data)
|
|
}
|
|
|
|
/**** Errors ******************************************************************/
|
|
|
|
func (h *Handlers) ForbiddenError(err, msg string, w http.ResponseWriter) {
|
|
w.WriteHeader(http.StatusForbidden)
|
|
web.ForbiddenError("/ Forbidden", err, msg, w)
|
|
}
|
|
|
|
func (h *Handlers) NotFound(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
web.NotFound("Not Found", w)
|
|
}
|
|
|
|
func (h *Handlers) InternalError(err, msg string, w http.ResponseWriter) {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
web.InternalError("/ Internal Error", err, msg, w)
|
|
}
|
|
|
|
/**** CAPTCHA *****************************************************************/
|
|
|
|
func (h *Handlers) CaptchaNew(w http.ResponseWriter, r *http.Request) {
|
|
dwc := dwcaptcha.NewDwellingCaptcha(h.captchaExpire)
|
|
_, id := inmemdb.New(r.RemoteAddr, dwc)
|
|
w.WriteHeader(http.StatusCreated)
|
|
fmt.Fprint(w, id)
|
|
}
|
|
|
|
func (h *Handlers) CaptchaImage(w http.ResponseWriter, r *http.Request) {
|
|
id := captcha.ID(GetURLParam(r, "id"))
|
|
|
|
image := inmemdb.Image(id, r.URL.Query().Get("style"))
|
|
if image == nil {
|
|
http.Error(w, "image not found", http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
w.Header().Add("Content-Disposition", "inline; filename=\""+string(id)+"\"")
|
|
|
|
jpeg.Encode(w, *image, &jpeg.Options{Quality: 20})
|
|
}
|
|
|
|
func (h *Handlers) CaptchaSolve(w http.ResponseWriter, r *http.Request) {
|
|
captchaID := captcha.ID(GetURLParam(r, "id"))
|
|
isJustRemove := r.URL.Query().Has("remove")
|
|
|
|
if isJustRemove {
|
|
inmemdb.Remove(captchaID)
|
|
w.WriteHeader(http.StatusNoContent)
|
|
return
|
|
}
|
|
|
|
if solved := inmemdb.IsSolved(captchaID); !solved {
|
|
http.Error(w, "wrong answer", http.StatusForbidden)
|
|
return
|
|
}
|
|
|
|
w.WriteHeader(http.StatusNoContent)
|
|
}
|