feat: add /s5/account/pins.bin endpoint

This commit is contained in:
Derrick Hammer 2024-01-17 12:33:05 -05:00
parent cc61a090b6
commit 1cf2d9880c
Signed by: pcfreak30
GPG Key ID: C997C339BE476FF2
2 changed files with 49 additions and 0 deletions

View File

@ -35,5 +35,6 @@ func getRoutes(h *s5.HttpHandler, portal interfaces.Portal) map[string]jape.Hand
"POST /s5/account/login": h.AccountLogin, "POST /s5/account/login": h.AccountLogin,
"GET /s5/account": s5.AuthMiddleware(h.AccountInfo, portal), "GET /s5/account": s5.AuthMiddleware(h.AccountInfo, portal),
"GET /s5/account/stats": s5.AuthMiddleware(h.AccountStats, portal), "GET /s5/account/stats": s5.AuthMiddleware(h.AccountStats, portal),
"GET /s5/account/pins.bin": s5.AuthMiddleware(h.AccountPins, portal),
} }
} }

View File

@ -13,6 +13,7 @@ import (
"git.lumeweb.com/LumeWeb/portal/db/models" "git.lumeweb.com/LumeWeb/portal/db/models"
"git.lumeweb.com/LumeWeb/portal/interfaces" "git.lumeweb.com/LumeWeb/portal/interfaces"
emailverifier "github.com/AfterShip/email-verifier" emailverifier "github.com/AfterShip/email-verifier"
"github.com/vmihailenco/msgpack/v5"
"go.sia.tech/jape" "go.sia.tech/jape"
"go.uber.org/zap" "go.uber.org/zap"
"io" "io"
@ -31,6 +32,7 @@ const (
errAccountGenerateChallenge = "Error generating challenge" errAccountGenerateChallenge = "Error generating challenge"
errAccountRegister = "Error registering account" errAccountRegister = "Error registering account"
errAccountLogin = "Error logging in account" errAccountLogin = "Error logging in account"
errFailedToGetPins = "Failed to get pins"
) )
var ( var (
@ -46,6 +48,7 @@ var (
errPubkeyAlreadyExists = errors.New("Pubkey already exists") errPubkeyAlreadyExists = errors.New("Pubkey already exists")
errPubkeyNotExist = errors.New("Pubkey does not exist") errPubkeyNotExist = errors.New("Pubkey does not exist")
errAccountLoginErr = errors.New(errAccountLogin) errAccountLoginErr = errors.New(errAccountLogin)
errFailedToGetPinsErr = errors.New(errFailedToGetPins)
) )
type HttpHandler struct { type HttpHandler struct {
@ -553,6 +556,51 @@ func (h *HttpHandler) AccountStats(jc jape.Context) {
jc.Encode(info) jc.Encode(info)
} }
func (h *HttpHandler) AccountPins(jc jape.Context) {
var cursor uint64
if jc.DecodeForm("cursor", &cursor) != nil {
return
}
errored := func(err error) {
_ = jc.Error(errFailedToGetPinsErr, http.StatusInternalServerError)
h.portal.Logger().Error(errFailedToGetPins, zap.Error(err))
}
pins, err := h.portal.Accounts().AccountPins(jc.Request.Context().Value(AuthUserIDKey).(uint64), cursor)
if err != nil {
errored(err)
return
}
pinsList := make([][]byte, len(pins))
for i, pin := range pins {
hash, err := hex.DecodeString(pin.Upload.Hash)
if err != nil {
errored(err)
return
}
pinsList[i] = encoding.MultihashFromBytes(hash, types.HashTypeBlake3).FullBytes()
}
result, err := msgpack.Marshal(pinsList)
if err != nil {
errored(err)
return
}
jc.Custom(jc.Request, result)
jc.ResponseWriter.WriteHeader(http.StatusOK)
_, _ = jc.ResponseWriter.Write(result)
}
func setAuthCookie(jwt string, jc jape.Context) { func setAuthCookie(jwt string, jc jape.Context) {
authCookie := http.Cookie{ authCookie := http.Cookie{
Name: "s5-auth-token", Name: "s5-auth-token",