ultrasn0w/huso
1
0
Fork 0
mirror of https://github.com/ultrasn0w/huso.git synced 2025-12-15 19:49:53 +01:00
Code Issues Projects Releases Wiki Activity

Big cleanup 🧹

Browse Source
This commit is contained in:
daru
2022-07-28 00:43:35 +02:00
parent 6a2e573828
commit 1bb8536334
5 changed files with 49 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

159
ober.go
View File

@@ -14,6 +14,13 @@ import (
"github.com/xujiajun/nutsdb"
)
const (
xHusoAuth = "X-HUSO-AUTH"
htmlContentType = "text/html; charset=utf-8"
jsonContentType = "application/json; charset=utf-8"
textContentType = "text/plain; charset=utf-8"
)
func RunWebserv() {
r := router.New()
r.GET("/", Headers(Start))
@@ -62,12 +69,12 @@ func Start(ctx *fasthttp.RequestCtx) {
WriteIndex(ctx, season, oracles, charts, logOut.String())
ctx.SetContentType("text/html; charset=utf-8")
ctx.SetContentType(htmlContentType)
ctx.SetStatusCode(fasthttp.StatusOK)
}
func AuthTest(ctx *fasthttp.RequestCtx) {
auth := ctx.Request.Header.Peek("X-HUSO-AUTH")
auth := ctx.Request.Header.Peek(xHusoAuth)
if ctx.UserValue("user") == nil || auth == nil || string(auth) == "" {
ctx.SetStatusCode(fasthttp.StatusBadRequest)
return
@@ -87,15 +94,7 @@ func Season(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
err = writeResponseBody(ctx, data)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, data)
}
func SeasonNext(ctx *fasthttp.RequestCtx) {
@@ -104,15 +103,7 @@ func SeasonNext(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
err = writeResponseBody(ctx, data)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, data)
}
func SeasonNextNext(ctx *fasthttp.RequestCtx) {
@@ -121,15 +112,7 @@ func SeasonNextNext(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
err = writeResponseBody(ctx, data)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, data)
}
func SeasonLast(ctx *fasthttp.RequestCtx) {
@@ -138,15 +121,7 @@ func SeasonLast(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
err = writeResponseBody(ctx, data)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, data)
}
func AnimeGet(ctx *fasthttp.RequestCtx) {
@@ -173,14 +148,7 @@ func AnimeGet(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
_, err = ctx.Write(bytes)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, bytes)
}
func AnimeSearchGet(ctx *fasthttp.RequestCtx) {
@@ -195,14 +163,7 @@ func AnimeSearchGet(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
_, err = ctx.Write(bytes)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, bytes)
}
func AppointmentGet(ctx *fasthttp.RequestCtx) {
@@ -222,14 +183,7 @@ func AppointmentGet(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
_, err = ctx.Write(bytes)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, bytes)
}
func ChatGet(ctx *fasthttp.RequestCtx) {
@@ -268,7 +222,7 @@ func ChatGet(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("text/plain; charset=utf-8")
ctx.SetContentType(textContentType)
ctx.SetStatusCode(fasthttp.StatusOK)
}
@@ -278,7 +232,7 @@ func LogGet(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("text/plain; charset=utf-8")
ctx.SetContentType(textContentType)
ctx.SetStatusCode(fasthttp.StatusOK)
}
@@ -319,14 +273,7 @@ func UserGet(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
_, err = ctx.Write(bytes)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, bytes)
}
func UserStashGet(ctx *fasthttp.RequestCtx) {
@@ -345,14 +292,7 @@ func WatchGet(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
_, err = ctx.Write(bytes)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, bytes)
}
func WatchExtendedGet(ctx *fasthttp.RequestCtx) {
@@ -388,14 +328,7 @@ func WatchExtendedGet(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
_, err = ctx.Write(bytes)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, bytes)
}
func Register(ctx *fasthttp.RequestCtx) {
@@ -463,13 +396,11 @@ func Register(ctx *fasthttp.RequestCtx) {
return
}
ctx.SetBody(body)
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, body)
}
func RegisterUpdate(ctx *fasthttp.RequestCtx) {
auth := ctx.Request.Header.Peek("X-HUSO-AUTH")
auth := ctx.Request.Header.Peek(xHusoAuth)
if auth == nil || string(auth) == "" || !strings.Contains(string(ctx.Request.Header.ContentType()), "application/json") {
ctx.SetStatusCode(fasthttp.StatusBadRequest)
return
@@ -532,9 +463,7 @@ func RegisterUpdate(ctx *fasthttp.RequestCtx) {
return
}
ctx.SetBody(body)
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, body)
}
func AppointmentPost(ctx *fasthttp.RequestCtx) {
@@ -542,7 +471,7 @@ func AppointmentPost(ctx *fasthttp.RequestCtx) {
}
func ChatPost(ctx *fasthttp.RequestCtx) {
auth := ctx.Request.Header.Peek("X-HUSO-AUTH")
auth := ctx.Request.Header.Peek(xHusoAuth)
if ctx.UserValue("id") == nil || ctx.UserValue("user") == nil || auth == nil || string(auth) == "" || !strings.Contains(string(ctx.Request.Header.ContentType()), "text/plain") {
ctx.SetStatusCode(fasthttp.StatusBadRequest)
return
@@ -583,7 +512,7 @@ func ChatPost(ctx *fasthttp.RequestCtx) {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("text/plain; charset=utf-8")
ctx.SetContentType(textContentType)
ctx.SetStatusCode(fasthttp.StatusOK)
}
@@ -600,7 +529,7 @@ func AppointmentDelete(ctx *fasthttp.RequestCtx) {
}
func UnRegister(ctx *fasthttp.RequestCtx) {
auth := ctx.Request.Header.Peek("X-HUSO-AUTH")
auth := ctx.Request.Header.Peek(xHusoAuth)
if auth == nil || string(auth) == "" || !strings.Contains(string(ctx.Request.Header.ContentType()), "application/json") {
ctx.SetStatusCode(fasthttp.StatusBadRequest)
return
@@ -651,9 +580,7 @@ func UnRegister(ctx *fasthttp.RequestCtx) {
return
}
ctx.SetBody(body)
ctx.SetContentType("application/json; charset=utf-8")
ctx.SetStatusCode(fasthttp.StatusOK)
setResponseJson(ctx, body)
}
func WatchDelete(ctx *fasthttp.RequestCtx) {
@@ -720,7 +647,7 @@ func watchGetLogic(ctx *fasthttp.RequestCtx) ([]AnimeUser, error) {
}
func processUpdateReq(ctx *fasthttp.RequestCtx, update bool) {
auth := ctx.Request.Header.Peek("X-HUSO-AUTH")
auth := ctx.Request.Header.Peek(xHusoAuth)
if ctx.UserValue("user") == nil || auth == nil || string(auth) == "" || !strings.Contains(string(ctx.Request.Header.ContentType()), "application/json") {
ctx.SetStatusCode(fasthttp.StatusBadRequest)
return
@@ -788,16 +715,11 @@ func processUpdateReq(ctx *fasthttp.RequestCtx, update bool) {
addErrorToCtx(ctx, err)
return
}
err = writeResponseBody(ctx, data)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
setResponseJson(ctx, data)
}
func processUpdateAppointmentReq(ctx *fasthttp.RequestCtx, update bool) {
auth := ctx.Request.Header.Peek("X-HUSO-AUTH")
auth := ctx.Request.Header.Peek(xHusoAuth)
if ctx.UserValue("user") == nil || auth == nil || string(auth) == "" || !strings.Contains(string(ctx.Request.Header.ContentType()), "application/json") {
ctx.SetStatusCode(fasthttp.StatusBadRequest)
return
@@ -877,16 +799,11 @@ func processUpdateAppointmentReq(ctx *fasthttp.RequestCtx, update bool) {
addErrorToCtx(ctx, err)
return
}
err = writeResponseBody(ctx, data)
if err != nil {
addErrorToCtx(ctx, err)
return
}
ctx.SetContentType("application/json; charset=utf-8")
setResponseJson(ctx, data)
}
func processUserStashReq(ctx *fasthttp.RequestCtx, update bool) {
auth := ctx.Request.Header.Peek("X-HUSO-AUTH")
auth := ctx.Request.Header.Peek(xHusoAuth)
dataPath := fmt.Sprintf("%s", ctx.UserValue("data"))
if auth == nil || string(auth) == "" || dataPath == "" {
ctx.SetStatusCode(fasthttp.StatusBadRequest)
@@ -930,14 +847,10 @@ func processUserStashReq(ctx *fasthttp.RequestCtx, update bool) {
}
}
func writeResponseBody(ctx *fasthttp.RequestCtx, bytes []byte) error {
_, err := ctx.Write(bytes)
if err != nil {
return err
}
ctx.SetContentType("application/json; charset=utf-8")
func setResponseJson(ctx *fasthttp.RequestCtx, bytes []byte) {
ctx.SetBody(bytes)
ctx.SetContentType(jsonContentType)
ctx.SetStatusCode(fasthttp.StatusOK)
return err
}
func addErrorToCtx(ctx *fasthttp.RequestCtx, err error) {