portal/api/s5/http.go

1394 lines
34 KiB
Go
Raw Normal View History

2024-01-15 04:54:43 +00:00
package s5
import (
"bytes"
"context"
"crypto/ed25519"
"crypto/rand"
"encoding/base64"
2024-01-16 07:01:18 +00:00
"encoding/hex"
2024-01-15 04:54:43 +00:00
"errors"
"fmt"
2024-01-15 04:54:43 +00:00
"git.lumeweb.com/LumeWeb/libs5-go/encoding"
s5interfaces "git.lumeweb.com/LumeWeb/libs5-go/interfaces"
2024-01-17 19:46:37 +00:00
"git.lumeweb.com/LumeWeb/libs5-go/metadata"
2024-01-17 21:20:51 +00:00
s5protocol "git.lumeweb.com/LumeWeb/libs5-go/protocol"
s5storage "git.lumeweb.com/LumeWeb/libs5-go/storage"
2024-01-15 19:41:24 +00:00
"git.lumeweb.com/LumeWeb/libs5-go/types"
"git.lumeweb.com/LumeWeb/portal/account"
"git.lumeweb.com/LumeWeb/portal/api/middleware"
2024-01-16 07:01:18 +00:00
"git.lumeweb.com/LumeWeb/portal/db/models"
"git.lumeweb.com/LumeWeb/portal/protocols"
"git.lumeweb.com/LumeWeb/portal/storage"
emailverifier "github.com/AfterShip/email-verifier"
"github.com/samber/lo"
"github.com/spf13/viper"
"github.com/vmihailenco/msgpack/v5"
2024-01-15 04:54:43 +00:00
"go.sia.tech/jape"
"go.uber.org/fx"
2024-01-15 04:54:43 +00:00
"go.uber.org/zap"
"gorm.io/gorm"
2024-01-15 04:54:43 +00:00
"io"
2024-01-17 17:38:52 +00:00
"math"
2024-01-15 04:54:43 +00:00
"mime/multipart"
"net/http"
"nhooyr.io/websocket"
"strconv"
"strings"
"time"
2024-01-15 04:54:43 +00:00
)
const (
errMultiformParse = "Error parsing multipart form"
errRetrievingFile = "Error retrieving the file"
errReadFile = "Error reading the file"
errClosingStream = "Error closing the stream"
errUploadingFile = "Error uploading the file"
errAccountGenerateChallenge = "Error generating challenge"
errAccountRegister = "Error registering account"
2024-01-16 18:51:03 +00:00
errAccountLogin = "Error logging in account"
errFailedToGetPins = "Failed to get pins"
2024-01-17 18:04:32 +00:00
errFailedToDelPin = "Failed to delete pin"
2024-01-17 18:13:37 +00:00
errFailedToAddPin = "Failed to add pin"
2024-01-17 19:46:37 +00:00
errorNotMultiform = "Not a multipart form"
errFetchingUrls = "Error fetching urls"
2024-01-24 06:27:05 +00:00
errDownloadingFile = "Error downloading file"
2024-01-15 04:54:43 +00:00
)
var (
errUploadingFileErr = errors.New(errUploadingFile)
errAccountGenerateChallengeErr = errors.New(errAccountGenerateChallenge)
errAccountRegisterErr = errors.New(errAccountRegister)
errInvalidChallengeErr = errors.New("Invalid challenge")
errInvalidSignatureErr = errors.New("Invalid signature")
errPubkeyNotSupported = errors.New("Only ed25519 keys are supported")
errInvalidEmail = errors.New("Invalid email")
errEmailAlreadyExists = errors.New("Email already exists")
errGeneratingPassword = errors.New("Error generating password")
errPubkeyAlreadyExists = errors.New("Pubkey already exists")
2024-01-16 21:05:28 +00:00
errPubkeyNotExist = errors.New("Pubkey does not exist")
2024-01-16 18:51:03 +00:00
errAccountLoginErr = errors.New(errAccountLogin)
errFailedToGetPinsErr = errors.New(errFailedToGetPins)
2024-01-17 18:04:32 +00:00
errFailedToDelPinErr = errors.New(errFailedToDelPin)
2024-01-17 18:13:37 +00:00
errFailedToAddPinErr = errors.New(errFailedToAddPin)
2024-01-17 19:46:37 +00:00
errNotMultiformErr = errors.New(errorNotMultiform)
errFetchingUrlsErr = errors.New(errFetchingUrls)
2024-01-24 06:27:05 +00:00
errDownloadingFileErr = errors.New(errDownloadingFile)
2024-01-15 04:54:43 +00:00
)
2024-01-16 16:42:50 +00:00
type HttpHandler struct {
verifier *emailverifier.Verifier
config *viper.Viper
logger *zap.Logger
storage *storage.StorageServiceImpl
db *gorm.DB
accounts *account.AccountServiceImpl
protocol *protocols.S5Protocol
2024-01-15 04:54:43 +00:00
}
type HttpHandlerParams struct {
fx.In
Config *viper.Viper
Logger *zap.Logger
Storage *storage.StorageServiceImpl
Db *gorm.DB
Accounts *account.AccountServiceImpl
Protocol *protocols.S5Protocol
}
2024-01-28 09:18:32 +00:00
type HttpHandlerResult struct {
fx.Out
2024-01-28 09:22:49 +00:00
HttpHandler HttpHandler
2024-01-28 09:18:32 +00:00
}
func NewHttpHandler(params HttpHandlerParams) (HttpHandlerResult, error) {
verifier := emailverifier.NewVerifier()
verifier.DisableSMTPCheck()
verifier.DisableGravatarCheck()
verifier.DisableDomainSuggest()
verifier.DisableAutoUpdateDisposable()
2024-01-28 09:18:32 +00:00
return HttpHandlerResult{
2024-01-28 09:22:49 +00:00
HttpHandler: HttpHandler{
2024-01-28 09:18:32 +00:00
verifier: verifier,
config: params.Config,
logger: params.Logger,
storage: params.Storage,
db: params.Db,
accounts: params.Accounts,
protocol: params.Protocol,
},
}, nil
2024-01-15 04:54:43 +00:00
}
2024-01-16 16:42:50 +00:00
func (h *HttpHandler) SmallFileUpload(jc jape.Context) {
2024-01-15 20:16:48 +00:00
var rs io.ReadSeeker
var bufferSize int64
2024-01-15 04:54:43 +00:00
r := jc.Request
contentType := r.Header.Get("Content-Type")
if strings.HasPrefix(contentType, "multipart/form-data") {
2024-01-15 04:54:43 +00:00
// Parse the multipart form
err := r.ParseMultipartForm(h.config.GetInt64("core.post-upload-limit"))
2024-01-15 04:54:43 +00:00
if jc.Check(errMultiformParse, err) != nil {
h.logger.Error(errMultiformParse, zap.Error(err))
2024-01-15 04:54:43 +00:00
return
}
// Retrieve the file from the form data
file, _, err := r.FormFile("file")
if jc.Check(errRetrievingFile, err) != nil {
h.logger.Error(errRetrievingFile, zap.Error(err))
2024-01-15 04:54:43 +00:00
return
}
defer func(file multipart.File) {
err := file.Close()
if err != nil {
h.logger.Error(errClosingStream, zap.Error(err))
2024-01-15 04:54:43 +00:00
}
}(file)
2024-01-15 20:16:48 +00:00
rs = file
2024-01-15 04:54:43 +00:00
} else {
2024-01-15 20:16:48 +00:00
data, err := io.ReadAll(r.Body)
2024-01-15 04:54:43 +00:00
if jc.Check(errReadFile, err) != nil {
h.logger.Error(errReadFile, zap.Error(err))
2024-01-15 04:54:43 +00:00
return
}
2024-01-15 20:16:48 +00:00
buffer := bytes.NewReader(data)
bufferSize = int64(buffer.Len())
rs = buffer
2024-01-15 04:54:43 +00:00
defer func(Body io.ReadCloser) {
err := Body.Close()
if err != nil {
h.logger.Error(errClosingStream, zap.Error(err))
2024-01-15 04:54:43 +00:00
}
}(r.Body)
}
hash, err := h.storage.GetHashSmall(rs)
2024-01-17 16:04:24 +00:00
_, err = rs.Seek(0, io.SeekStart)
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-17 16:04:24 +00:00
return
}
2024-01-16 06:05:09 +00:00
if exists, upload := h.storage.FileExists(hash); exists {
2024-01-17 16:31:48 +00:00
cid, err := encoding.CIDFromHash(hash, upload.Size, types.CIDTypeRaw, types.HashTypeBlake3)
2024-01-16 06:05:09 +00:00
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-16 06:05:09 +00:00
return
}
cidStr, err := cid.ToString()
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-16 06:05:09 +00:00
return
}
2024-01-17 22:21:15 +00:00
err = h.accounts.PinByID(upload.ID, uint(jc.Request.Context().Value(middleware.S5AuthUserIDKey).(uint64)))
2024-01-17 22:21:15 +00:00
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-17 22:21:15 +00:00
return
}
2024-01-18 18:53:04 +00:00
jc.Encode(&SmallUploadResponse{
CID: cidStr,
})
2024-01-16 06:05:09 +00:00
return
}
hash, err = h.storage.PutFileSmall(rs, "s5", false)
2024-01-15 04:54:43 +00:00
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-15 04:54:43 +00:00
return
}
h.logger.Info("Hash", zap.String("hash", hex.EncodeToString(hash)))
2024-01-18 19:28:32 +00:00
2024-01-17 16:31:48 +00:00
cid, err := encoding.CIDFromHash(hash, uint64(bufferSize), types.CIDTypeRaw, types.HashTypeBlake3)
2024-01-15 04:54:43 +00:00
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-15 04:54:43 +00:00
return
}
cidStr, err := cid.ToString()
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-15 04:54:43 +00:00
return
}
h.logger.Info("CID", zap.String("cidStr", cidStr))
2024-01-18 19:28:32 +00:00
_, err = rs.Seek(0, io.SeekStart)
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
return
}
var mimeBytes [512]byte
_, err = rs.Read(mimeBytes[:])
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
return
}
mimeType := http.DetectContentType(mimeBytes[:])
upload, err := h.storage.CreateUpload(hash, mimeType, uint(jc.Request.Context().Value(middleware.S5AuthUserIDKey).(uint64)), jc.Request.RemoteAddr, uint64(bufferSize), "s5")
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
}
err = h.accounts.PinByID(upload.ID, uint(jc.Request.Context().Value(middleware.S5AuthUserIDKey).(uint64)))
2024-01-17 19:49:35 +00:00
if err != nil {
2024-01-16 07:01:18 +00:00
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-16 07:01:18 +00:00
}
jc.Encode(&SmallUploadResponse{
CID: cidStr,
})
2024-01-15 04:54:43 +00:00
}
2024-01-16 16:42:50 +00:00
func (h *HttpHandler) AccountRegisterChallenge(jc jape.Context) {
var pubkey string
if jc.DecodeForm("pubKey", &pubkey) != nil {
return
}
challenge := make([]byte, 32)
_, err := rand.Read(challenge)
if err != nil {
_ = jc.Error(errAccountGenerateChallengeErr, http.StatusInternalServerError)
h.logger.Error(errAccountGenerateChallenge, zap.Error(err))
return
}
decodedKey, err := base64.RawURLEncoding.DecodeString(pubkey)
if err != nil {
_ = jc.Error(errAccountGenerateChallengeErr, http.StatusInternalServerError)
h.logger.Error(errAccountGenerateChallenge, zap.Error(err))
return
}
if len(decodedKey) != 33 && int(decodedKey[0]) != int(types.HashTypeEd25519) {
_ = jc.Error(errAccountGenerateChallengeErr, http.StatusInternalServerError)
h.logger.Error(errAccountGenerateChallenge, zap.Error(err))
return
}
result := h.db.Create(&models.S5Challenge{
Pubkey: pubkey,
Challenge: base64.RawURLEncoding.EncodeToString(challenge),
2024-01-16 18:51:03 +00:00
Type: "register",
})
if result.Error != nil {
_ = jc.Error(errAccountGenerateChallengeErr, http.StatusInternalServerError)
h.logger.Error(errAccountGenerateChallenge, zap.Error(err))
return
}
jc.Encode(&AccountRegisterChallengeResponse{
Challenge: base64.RawURLEncoding.EncodeToString(challenge),
})
}
func (h *HttpHandler) AccountRegister(jc jape.Context) {
var request AccountRegisterRequest
if jc.Decode(&request) != nil {
return
}
errored := func(err error) {
_ = jc.Error(errAccountRegisterErr, http.StatusInternalServerError)
h.logger.Error(errAccountRegister, zap.Error(err))
}
decodedKey, err := base64.RawURLEncoding.DecodeString(request.Pubkey)
if err != nil {
errored(err)
return
}
2024-01-16 20:26:36 +00:00
if len(decodedKey) != 33 {
errored(err)
return
}
var challenge models.S5Challenge
result := h.db.Model(&models.S5Challenge{}).Where(&models.S5Challenge{Pubkey: request.Pubkey, Type: "register"}).First(&challenge)
if result.RowsAffected == 0 || result.Error != nil {
errored(err)
return
}
decodedResponse, err := base64.RawURLEncoding.DecodeString(request.Response)
if err != nil {
2024-01-16 20:26:57 +00:00
errored(errInvalidChallengeErr)
return
}
2024-01-16 19:32:53 +00:00
if len(decodedResponse) != 65 {
2024-01-16 20:26:57 +00:00
errored(errInvalidChallengeErr)
return
}
decodedChallenge, err := base64.RawURLEncoding.DecodeString(challenge.Challenge)
if err != nil {
2024-01-16 20:26:57 +00:00
errored(errInvalidChallengeErr)
return
}
2024-01-16 19:45:07 +00:00
if !bytes.Equal(decodedResponse[1:33], decodedChallenge) {
errored(errInvalidChallengeErr)
return
}
if int(decodedKey[0]) != int(types.HashTypeEd25519) {
errored(errPubkeyNotSupported)
return
}
decodedSignature, err := base64.RawURLEncoding.DecodeString(request.Signature)
if err != nil {
errored(err)
return
}
if !ed25519.Verify(decodedKey[1:], decodedResponse, decodedSignature) {
errored(errInvalidSignatureErr)
return
}
if request.Email == "" {
request.Email = fmt.Sprintf("%s@%s", hex.EncodeToString(decodedKey[1:]), "example.com")
}
verify, _ := h.verifier.Verify(request.Email)
if !verify.Syntax.Valid {
errored(errInvalidEmail)
return
}
accountExists, _ := h.accounts.EmailExists(request.Email)
if accountExists {
errored(errEmailAlreadyExists)
return
}
pubkeyExists, _ := h.accounts.PubkeyExists(hex.EncodeToString(decodedKey[1:]))
if pubkeyExists {
errored(errPubkeyAlreadyExists)
return
}
passwd := make([]byte, 32)
_, err = rand.Read(passwd)
if accountExists {
errored(errGeneratingPassword)
return
}
newAccount, err := h.accounts.CreateAccount(request.Email, string(passwd))
if err != nil {
errored(errAccountRegisterErr)
return
}
rawPubkey := hex.EncodeToString(decodedKey[1:])
err = h.accounts.AddPubkeyToAccount(*newAccount, rawPubkey)
if err != nil {
errored(errAccountRegisterErr)
return
}
jwt, err := h.accounts.LoginPubkey(rawPubkey)
if err != nil {
errored(errAccountRegisterErr)
return
}
result = h.db.Delete(&challenge)
if result.Error != nil {
errored(errAccountRegisterErr)
return
}
2024-01-16 18:38:10 +00:00
setAuthCookie(jwt, jc)
}
func (h *HttpHandler) AccountLoginChallenge(jc jape.Context) {
2024-01-16 18:51:03 +00:00
var pubkey string
if jc.DecodeForm("pubKey", &pubkey) != nil {
return
}
errored := func(err error) {
_ = jc.Error(errAccountLoginErr, http.StatusInternalServerError)
h.logger.Error(errAccountLogin, zap.Error(err))
2024-01-16 18:51:03 +00:00
}
challenge := make([]byte, 32)
_, err := rand.Read(challenge)
if err != nil {
_ = jc.Error(errAccountGenerateChallengeErr, http.StatusInternalServerError)
h.logger.Error(errAccountGenerateChallenge, zap.Error(err))
2024-01-16 18:51:03 +00:00
return
}
decodedKey, err := base64.RawURLEncoding.DecodeString(pubkey)
if err != nil {
2024-01-16 20:23:00 +00:00
errored(errAccountGenerateChallengeErr)
2024-01-16 18:51:03 +00:00
return
}
if len(decodedKey) != 33 && int(decodedKey[0]) != int(types.HashTypeEd25519) {
errored(errPubkeyNotSupported)
2024-01-16 18:51:03 +00:00
return
}
pubkeyExists, _ := h.accounts.PubkeyExists(hex.EncodeToString(decodedKey[1:]))
2024-01-16 21:05:28 +00:00
if pubkeyExists {
errored(errPubkeyNotExist)
return
}
result := h.db.Create(&models.S5Challenge{
Challenge: base64.RawURLEncoding.EncodeToString(challenge),
2024-01-16 18:51:03 +00:00
Type: "login",
})
if result.Error != nil {
2024-01-16 20:23:00 +00:00
errored(result.Error)
2024-01-16 18:51:03 +00:00
return
}
jc.Encode(&AccountLoginChallengeResponse{
Challenge: base64.RawURLEncoding.EncodeToString(challenge),
})
}
func (h *HttpHandler) AccountLogin(jc jape.Context) {
2024-01-16 18:56:25 +00:00
var request AccountLoginRequest
if jc.Decode(&request) != nil {
return
}
errored := func(err error) {
_ = jc.Error(errAccountLoginErr, http.StatusInternalServerError)
h.logger.Error(errAccountLogin, zap.Error(err))
2024-01-16 18:56:25 +00:00
}
decodedKey, err := base64.RawURLEncoding.DecodeString(request.Pubkey)
if err != nil {
errored(err)
return
}
if len(decodedKey) != 32 {
errored(err)
return
}
var challenge models.S5Challenge
result := h.db.Model(&models.S5Challenge{}).Where(&models.S5Challenge{Pubkey: request.Pubkey, Type: "login"}).First(&challenge)
2024-01-16 18:56:25 +00:00
if result.RowsAffected == 0 || result.Error != nil {
errored(err)
return
}
decodedResponse, err := base64.RawURLEncoding.DecodeString(request.Response)
if err != nil {
errored(err)
return
}
2024-01-16 19:32:53 +00:00
if len(decodedResponse) != 65 {
2024-01-16 18:56:25 +00:00
errored(err)
return
}
decodedChallenge, err := base64.RawURLEncoding.DecodeString(challenge.Challenge)
if err != nil {
errored(err)
return
}
2024-01-16 19:45:07 +00:00
if !bytes.Equal(decodedResponse[1:33], decodedChallenge) {
2024-01-16 18:56:25 +00:00
errored(errInvalidChallengeErr)
return
}
if int(decodedKey[0]) != int(types.HashTypeEd25519) {
errored(errPubkeyNotSupported)
return
}
decodedSignature, err := base64.RawURLEncoding.DecodeString(request.Signature)
if err != nil {
errored(err)
return
}
if !ed25519.Verify(decodedKey[1:], decodedResponse, decodedSignature) {
2024-01-16 18:56:25 +00:00
errored(errInvalidSignatureErr)
return
}
jwt, err := h.accounts.LoginPubkey(request.Pubkey)
2024-01-16 18:56:25 +00:00
if err != nil {
errored(errAccountLoginErr)
return
}
result = h.db.Delete(&challenge)
if result.Error != nil {
errored(errAccountLoginErr)
return
}
2024-01-16 18:56:25 +00:00
setAuthCookie(jwt, jc)
}
2024-01-16 18:38:10 +00:00
2024-01-17 16:52:54 +00:00
func (h *HttpHandler) AccountInfo(jc jape.Context) {
_, user := h.accounts.AccountExists(jc.Request.Context().Value(middleware.S5AuthUserIDKey).(uint64))
2024-01-17 16:52:54 +00:00
info := &AccountInfoResponse{
Email: user.Email,
QuotaExceeded: false,
EmailConfirmed: false,
IsRestricted: false,
2024-01-17 17:38:52 +00:00
Tier: AccountTier{
Id: 1,
Name: "default",
2024-01-24 17:51:19 +00:00
UploadBandwidth: math.MaxUint32,
StorageLimit: math.MaxUint32,
2024-01-17 17:38:52 +00:00
Scopes: []interface{}{},
},
2024-01-17 16:52:54 +00:00
}
jc.Encode(info)
}
2024-01-17 17:03:08 +00:00
func (h *HttpHandler) AccountStats(jc jape.Context) {
_, user := h.accounts.AccountExists(jc.Request.Context().Value(middleware.S5AuthUserIDKey).(uint64))
2024-01-17 17:03:08 +00:00
info := &AccountStatsResponse{
AccountInfoResponse: AccountInfoResponse{
Email: user.Email,
QuotaExceeded: false,
EmailConfirmed: false,
IsRestricted: false,
2024-01-17 17:38:52 +00:00
Tier: AccountTier{
Id: 1,
Name: "default",
2024-01-24 17:51:19 +00:00
UploadBandwidth: math.MaxUint32,
StorageLimit: math.MaxUint32,
2024-01-17 17:38:52 +00:00
Scopes: []interface{}{},
},
2024-01-17 17:03:08 +00:00
},
Stats: AccountStats{
Total: AccountStatsTotal{
UsedStorage: 0,
},
},
}
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.logger.Error(errFailedToGetPins, zap.Error(err))
}
pins, err := h.accounts.AccountPins(jc.Request.Context().Value(middleware.S5AuthUserIDKey).(uint64), cursor)
if err != nil {
errored(err)
return
}
pinResponse := &AccountPinResponse{Cursor: cursor, Pins: pins}
result, err := msgpack.Marshal(pinResponse)
if err != nil {
errored(err)
return
}
jc.Custom(jc.Request, result)
jc.ResponseWriter.WriteHeader(http.StatusOK)
_, _ = jc.ResponseWriter.Write(result)
}
2024-01-17 18:04:32 +00:00
func (h *HttpHandler) AccountPinDelete(jc jape.Context) {
var cid string
if jc.DecodeParam("cid", &cid) != nil {
return
}
errored := func(err error) {
_ = jc.Error(errFailedToDelPinErr, http.StatusInternalServerError)
h.logger.Error(errFailedToDelPin, zap.Error(err))
2024-01-17 18:04:32 +00:00
}
decodedCid, err := encoding.CIDFromString(cid)
if err != nil {
errored(err)
return
}
hash := hex.EncodeToString(decodedCid.Hash.HashBytes())
err = h.accounts.DeletePinByHash(hash, uint(jc.Request.Context().Value(middleware.S5AuthUserIDKey).(uint64)))
2024-01-17 18:04:32 +00:00
if err != nil {
errored(err)
}
jc.ResponseWriter.WriteHeader(http.StatusNoContent)
2024-01-17 18:13:37 +00:00
}
2024-01-17 18:04:32 +00:00
2024-01-17 18:13:37 +00:00
func (h *HttpHandler) AccountPin(jc jape.Context) {
var cid string
if jc.DecodeParam("cid", &cid) != nil {
return
}
errored := func(err error) {
_ = jc.Error(errFailedToAddPinErr, http.StatusInternalServerError)
h.logger.Error(errFailedToAddPin, zap.Error(err))
2024-01-17 18:13:37 +00:00
}
decodedCid, err := encoding.CIDFromString(cid)
if err != nil {
errored(err)
return
}
h.logger.Info("CID", zap.String("cidStr", cid))
h.logger.Info("hash", zap.String("hash", hex.EncodeToString(decodedCid.Hash.HashBytes())))
2024-01-18 19:38:15 +00:00
2024-01-17 18:13:37 +00:00
hash := hex.EncodeToString(decodedCid.Hash.HashBytes())
err = h.accounts.PinByHash(hash, uint(jc.Request.Context().Value(middleware.S5AuthUserIDKey).(uint64)))
2024-01-17 18:13:37 +00:00
if err != nil {
errored(err)
return
}
jc.ResponseWriter.WriteHeader(http.StatusNoContent)
2024-01-17 18:04:32 +00:00
}
2024-01-17 19:46:37 +00:00
func (h *HttpHandler) DirectoryUpload(jc jape.Context) {
var tryFiles []string
var errorPages map[int]string
var name string
if jc.DecodeForm("tryFiles", &tryFiles) != nil {
return
}
if jc.DecodeForm("errorPages", &errorPages) != nil {
return
}
if jc.DecodeForm("name", &name) != nil {
return
}
r := jc.Request
contentType := r.Header.Get("Content-Type")
errored := func(err error) {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-17 19:46:37 +00:00
}
if !strings.HasPrefix(contentType, "multipart/form-data") {
_ = jc.Error(errNotMultiformErr, http.StatusBadRequest)
h.logger.Error(errorNotMultiform)
2024-01-17 19:46:37 +00:00
return
}
err := r.ParseMultipartForm(h.config.GetInt64("core.post-upload-limit"))
2024-01-17 19:46:37 +00:00
if jc.Check(errMultiformParse, err) != nil {
h.logger.Error(errMultiformParse, zap.Error(err))
2024-01-17 19:46:37 +00:00
return
}
uploadMap := make(map[string]models.Upload, len(r.MultipartForm.File))
mimeMap := make(map[string]string, len(r.MultipartForm.File))
for _, files := range r.MultipartForm.File {
for _, fileHeader := range files {
// Open the file.
file, err := fileHeader.Open()
if err != nil {
errored(err)
return
}
defer func(file multipart.File) {
err := file.Close()
if err != nil {
h.logger.Error(errClosingStream, zap.Error(err))
2024-01-17 19:46:37 +00:00
}
}(file)
var rs io.ReadSeeker
hash, err := h.storage.GetHashSmall(rs)
2024-01-17 19:46:37 +00:00
_, err = rs.Seek(0, io.SeekStart)
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-17 19:46:37 +00:00
return
}
if exists, upload := h.storage.FileExists(hash); exists {
2024-01-17 19:46:37 +00:00
uploadMap[fileHeader.Filename] = upload
continue
}
hash, err = h.storage.PutFileSmall(rs, "s5", false)
2024-01-17 19:46:37 +00:00
if err != nil {
errored(err)
return
}
_, err = rs.Seek(0, io.SeekStart)
if err != nil {
return
}
2024-01-17 19:46:37 +00:00
if err != nil {
errored(err)
return
}
var mimeBytes [512]byte
if _, err := file.Read(mimeBytes[:]); err != nil {
2024-01-17 19:46:37 +00:00
errored(err)
return
}
mimeType := http.DetectContentType(mimeBytes[:])
2024-01-17 19:46:37 +00:00
upload, err := h.storage.CreateUpload(hash, mimeType, uint(jc.Request.Context().Value(middleware.S5AuthUserIDKey).(uint64)), jc.Request.RemoteAddr, uint64(fileHeader.Size), "s5")
if err != nil {
2024-01-17 19:46:37 +00:00
errored(err)
return
}
// Reset the read pointer back to the start of the file.
if _, err := file.Seek(0, io.SeekStart); err != nil {
2024-01-17 19:46:37 +00:00
errored(err)
return
}
// Read a snippet of the file to determine its MIME type.
buffer := make([]byte, 512) // 512 bytes should be enough for http.DetectContentType to determine the type
if _, err := file.Read(buffer); err != nil {
errored(err)
return
}
2024-01-17 19:46:37 +00:00
uploadMap[fileHeader.Filename] = *upload
mimeMap[fileHeader.Filename] = mimeType
}
}
filesMap := make(map[string]metadata.WebAppMetadataFileReference, len(uploadMap))
for name, file := range uploadMap {
hashDecoded, err := hex.DecodeString(file.Hash)
if err != nil {
errored(err)
return
}
cid, err := encoding.CIDFromHash(hashDecoded, file.Size, types.CIDTypeRaw, types.HashTypeBlake3)
if err != nil {
errored(err)
return
}
filesMap[name] = *metadata.NewWebAppMetadataFileReference(cid, mimeMap[name])
}
app := metadata.NewWebAppMetadata(name, tryFiles, *metadata.NewExtraMetadata(map[int]interface{}{}), errorPages, filesMap)
appData, err := msgpack.Marshal(app)
if err != nil {
errored(err)
return
}
var rs = bytes.NewReader(appData)
hash, err := h.storage.GetHashSmall(rs)
2024-01-17 19:46:37 +00:00
_, err = rs.Seek(0, io.SeekStart)
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-17 19:46:37 +00:00
return
}
if exists, upload := h.storage.FileExists(hash); exists {
2024-01-17 19:46:37 +00:00
cid, err := encoding.CIDFromHash(hash, upload.Size, types.CIDTypeMetadataWebapp, types.HashTypeBlake3)
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-17 19:46:37 +00:00
return
}
cidStr, err := cid.ToString()
if err != nil {
_ = jc.Error(errUploadingFileErr, http.StatusInternalServerError)
h.logger.Error(errUploadingFile, zap.Error(err))
2024-01-17 19:46:37 +00:00
return
}
jc.Encode(map[string]string{"hash": cidStr})
return
}
hash, err = h.storage.PutFileSmall(rs, "s5", false)
2024-01-17 19:46:37 +00:00
if err != nil {
errored(err)
return
}
cid, err := encoding.CIDFromHash(hash, uint64(len(appData)), types.CIDTypeRaw, types.HashTypeBlake3)
if err != nil {
errored(err)
return
}
cidStr, err := cid.ToString()
if err != nil {
errored(err)
return
}
jc.Encode(&AppUploadResponse{CID: cidStr})
}
func (h *HttpHandler) DebugDownloadUrls(jc jape.Context) {
var cid string
if jc.DecodeParam("cid", &cid) != nil {
return
}
decodedCid, err := encoding.CIDFromString(cid)
if err != nil {
_ = jc.Error(errFetchingUrlsErr, http.StatusInternalServerError)
h.logger.Error(errFetchingUrls, zap.Error(err))
return
}
node := h.getNode()
dlUriProvider := s5storage.NewStorageLocationProvider(node, &decodedCid.Hash, types.StorageLocationTypeFull, types.StorageLocationTypeFile, types.StorageLocationTypeBridge)
err = dlUriProvider.Start()
if err != nil {
_ = jc.Error(errFetchingUrlsErr, http.StatusInternalServerError)
h.logger.Error(errFetchingUrls, zap.Error(err))
return
}
_, err = dlUriProvider.Next()
if err != nil {
_ = jc.Error(errFetchingUrlsErr, http.StatusInternalServerError)
h.logger.Error(errFetchingUrls, zap.Error(err))
return
}
locations, err := node.GetCachedStorageLocations(&decodedCid.Hash, []types.StorageLocationType{
types.StorageLocationTypeFull, types.StorageLocationTypeFile, types.StorageLocationTypeBridge,
})
if err != nil {
_ = jc.Error(errFetchingUrlsErr, http.StatusInternalServerError)
h.logger.Error(errFetchingUrls, zap.Error(err))
return
}
availableNodes := lo.Keys[string, s5interfaces.StorageLocation](locations)
availableNodesIds := make([]*encoding.NodeId, len(availableNodes))
for i, nodeIdStr := range availableNodes {
nodeId, err := encoding.DecodeNodeId(nodeIdStr)
if err != nil {
_ = jc.Error(errFetchingUrlsErr, http.StatusInternalServerError)
h.logger.Error(errFetchingUrls, zap.Error(err))
return
}
availableNodesIds[i] = nodeId
}
sorted, err := node.Services().P2P().SortNodesByScore(availableNodesIds)
if err != nil {
return
}
if err != nil {
_ = jc.Error(errFetchingUrlsErr, http.StatusInternalServerError)
h.logger.Error(errFetchingUrls, zap.Error(err))
return
}
output := make([]string, len(sorted))
for i, nodeId := range sorted {
nodeIdStr, err := nodeId.ToString()
if err != nil {
_ = jc.Error(errFetchingUrlsErr, http.StatusInternalServerError)
h.logger.Error(errFetchingUrls, zap.Error(err))
return
}
output[i] = locations[nodeIdStr].BytesURL()
}
jc.ResponseWriter.WriteHeader(http.StatusOK)
_, _ = jc.ResponseWriter.Write([]byte(strings.Join(output, "\n")))
}
2024-01-17 21:05:11 +00:00
func (h *HttpHandler) RegistryQuery(jc jape.Context) {
var pk string
if jc.DecodeForm("pk", &pk) != nil {
return
}
pkBytes, err := base64.RawURLEncoding.DecodeString(pk)
if jc.Check("error decoding pk", err) != nil {
return
}
entry, err := h.getNode().Services().Registry().Get(pkBytes)
if jc.Check("error getting registry entry", err) != nil {
return
}
if entry == nil {
jc.ResponseWriter.WriteHeader(http.StatusNotFound)
return
}
jc.Encode(&RegistryQueryResponse{
Pk: base64.RawURLEncoding.EncodeToString(entry.PK()),
Revision: entry.Revision(),
Data: base64.RawURLEncoding.EncodeToString(entry.Data()),
Signature: base64.RawURLEncoding.EncodeToString(entry.Signature()),
})
}
2024-01-17 21:20:51 +00:00
func (h *HttpHandler) RegistrySet(jc jape.Context) {
var request RegistrySetRequest
if jc.Decode(&request) != nil {
return
}
pk, err := base64.RawURLEncoding.DecodeString(request.Pk)
if jc.Check("error decoding pk", err) != nil {
return
}
data, err := base64.RawURLEncoding.DecodeString(request.Data)
if jc.Check("error decoding data", err) != nil {
return
}
signature, err := base64.RawURLEncoding.DecodeString(request.Signature)
if jc.Check("error decoding signature", err) != nil {
return
}
entry := s5protocol.NewSignedRegistryEntry(pk, request.Revision, data, signature)
err = h.getNode().Services().Registry().Set(entry, false, nil)
if jc.Check("error setting registry entry", err) != nil {
return
}
}
2024-01-17 21:05:11 +00:00
func (h *HttpHandler) RegistrySubscription(jc jape.Context) {
// Create a context for the WebSocket operations
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
var listeners []func()
// Accept the WebSocket connection
c, err := websocket.Accept(jc.ResponseWriter, jc.Request, nil)
if err != nil {
h.logger.Error("error accepting websocket connection", zap.Error(err))
return
}
defer func(c *websocket.Conn, code websocket.StatusCode, reason string) {
err := c.Close(code, reason)
if err != nil {
h.logger.Error("error closing websocket connection", zap.Error(err))
}
for _, listener := range listeners {
listener()
}
}(c, websocket.StatusNormalClosure, "connection closed")
// Main loop for reading messages
for {
// Read a message (the actual reading and unpacking is skipped here)
_, data, err := c.Read(ctx)
if err != nil {
if websocket.CloseStatus(err) == websocket.StatusNormalClosure {
// Normal closure
h.logger.Info("websocket connection closed normally")
} else {
// Handle different types of errors
h.logger.Error("error in websocket connection", zap.Error(err))
}
break
}
decoder := msgpack.NewDecoder(bytes.NewReader(data))
method, err := decoder.DecodeInt()
if err != nil {
h.logger.Error("error decoding method", zap.Error(err))
break
}
if method != 2 {
h.logger.Error("invalid method", zap.Int64("method", int64(method)))
break
}
sre, err := decoder.DecodeBytes()
if err != nil {
h.logger.Error("error decoding sre", zap.Error(err))
break
}
off, err := h.getNode().Services().Registry().Listen(sre, func(entry s5interfaces.SignedRegistryEntry) {
encoded, err := msgpack.Marshal(entry)
if err != nil {
h.logger.Error("error encoding entry", zap.Error(err))
return
}
err = c.Write(ctx, websocket.MessageBinary, encoded)
if err != nil {
h.logger.Error("error writing to websocket", zap.Error(err))
}
})
if err != nil {
h.logger.Error("error listening to registry", zap.Error(err))
break
}
listeners = append(listeners, off)
}
}
func (h *HttpHandler) getNode() s5interfaces.Node {
return h.protocol.Node()
}
2024-01-18 02:23:33 +00:00
func (h *HttpHandler) DownloadBlob(jc jape.Context) {
var cid string
if jc.DecodeParam("cid", &cid) != nil {
return
}
cid = strings.Split(cid, ".")[0]
cidDecoded, err := encoding.CIDFromString(cid)
if jc.Check("error decoding cid", err) != nil {
return
}
dlUriProvider := s5storage.NewStorageLocationProvider(h.getNode(), &cidDecoded.Hash, types.StorageLocationTypeFull, types.StorageLocationTypeFile, types.StorageLocationTypeBridge)
err = dlUriProvider.Start()
if jc.Check("error starting search", err) != nil {
return
}
next, err := dlUriProvider.Next()
if jc.Check("error fetching blob", err) != nil {
return
}
http.Redirect(jc.ResponseWriter, jc.Request, next.Location().BytesURL(), http.StatusFound)
}
func (h *HttpHandler) DebugStorageLocations(jc jape.Context) {
var hash string
if jc.DecodeParam("hash", &hash) != nil {
return
}
var kinds string
if jc.DecodeForm("kinds", &kinds) != nil {
return
}
decodedHash, err := encoding.MultihashFromBase64Url(hash)
if jc.Check("error decoding hash", err) != nil {
return
}
typeList := strings.Split(kinds, ",")
typeIntList := make([]types.StorageLocationType, 0)
for _, typeStr := range typeList {
typeInt, err := strconv.Atoi(typeStr)
if err != nil {
continue
}
typeIntList = append(typeIntList, types.StorageLocationType(typeInt))
}
if len(typeIntList) == 0 {
typeIntList = []types.StorageLocationType{
types.StorageLocationTypeFull,
types.StorageLocationTypeFile,
types.StorageLocationTypeBridge,
types.StorageLocationTypeArchive,
}
}
dlUriProvider := s5storage.NewStorageLocationProvider(h.getNode(), decodedHash, types.StorageLocationTypeFull, types.StorageLocationTypeFile, types.StorageLocationTypeBridge)
err = dlUriProvider.Start()
if jc.Check("error starting search", err) != nil {
return
}
_, err = dlUriProvider.Next()
if jc.Check("error fetching locations", err) != nil {
return
}
locations, err := h.getNode().GetCachedStorageLocations(decodedHash, typeIntList)
if jc.Check("error getting cached locations", err) != nil {
return
}
availableNodes := lo.Keys[string, s5interfaces.StorageLocation](locations)
availableNodesIds := make([]*encoding.NodeId, len(availableNodes))
for i, nodeIdStr := range availableNodes {
nodeId, err := encoding.DecodeNodeId(nodeIdStr)
if jc.Check("error decoding node id", err) != nil {
return
}
availableNodesIds[i] = nodeId
}
availableNodesIds, err = h.getNode().Services().P2P().SortNodesByScore(availableNodesIds)
if jc.Check("error sorting nodes", err) != nil {
return
}
debugLocations := make([]DebugStorageLocation, len(availableNodes))
for i, nodeId := range availableNodesIds {
nodeIdStr, err := nodeId.ToBase58()
if jc.Check("error encoding node id", err) != nil {
return
}
score, err := h.getNode().Services().P2P().GetNodeScore(nodeId)
if jc.Check("error getting node score", err) != nil {
return
}
debugLocations[i] = DebugStorageLocation{
Type: locations[nodeIdStr].Type(),
Parts: locations[nodeIdStr].Parts(),
Expiry: locations[nodeIdStr].Expiry(),
NodeId: nodeIdStr,
Score: score,
}
}
jc.Encode(&DebugStorageLocationsResponse{
Locations: debugLocations,
})
}
2024-01-18 03:16:04 +00:00
func (h *HttpHandler) DownloadMetadata(jc jape.Context) {
var cid string
if jc.DecodeParam("cid", &cid) != nil {
return
}
cidDecoded, err := encoding.CIDFromString(cid)
if jc.Check("error decoding cid", err) != nil {
h.logger.Error("error decoding cid", zap.Error(err))
2024-01-18 03:16:04 +00:00
return
}
switch cidDecoded.Type {
case types.CIDTypeRaw:
_ = jc.Error(errors.New("Raw CIDs do not have metadata"), http.StatusBadRequest)
return
case types.CIDTypeResolver:
_ = jc.Error(errors.New("Resolver CIDs not yet supported"), http.StatusBadRequest)
return
}
meta, err := h.getNode().GetMetadataByCID(cidDecoded)
if jc.Check("error getting metadata", err) != nil {
h.logger.Error("error getting metadata", zap.Error(err))
2024-01-18 03:16:04 +00:00
return
}
if cidDecoded.Type != types.CIDTypeBridge {
jc.ResponseWriter.Header().Set("Cache-Control", "public, max-age=31536000")
} else {
jc.ResponseWriter.Header().Set("Cache-Control", "public, max-age=60")
}
jc.Encode(&meta)
}
2024-01-24 06:27:05 +00:00
func (h *HttpHandler) DownloadFile(jc jape.Context) {
var cid string
if jc.DecodeParam("cid", &cid) != nil {
return
}
var hashBytes []byte
2024-01-24 06:27:05 +00:00
cidDecoded, err := encoding.CIDFromString(cid)
if err != nil {
hashDecoded, err := encoding.MultihashFromBase64Url(cid)
if jc.Check("error decoding as cid or hash", err) != nil {
return
}
hashBytes = hashDecoded.HashBytes()
2024-01-25 00:10:19 +00:00
} else {
2024-01-24 06:27:05 +00:00
2024-01-25 00:10:19 +00:00
hashBytes = cidDecoded.Hash.HashBytes()
}
file := h.storage.NewFile(hashBytes)
if !file.Exists() {
jc.ResponseWriter.WriteHeader(http.StatusNotFound)
2024-01-24 06:27:05 +00:00
return
}
2024-01-24 06:27:05 +00:00
defer func(file io.ReadCloser) {
err := file.Close()
if err != nil {
h.logger.Error("error closing file", zap.Error(err))
2024-01-24 06:27:05 +00:00
}
}(file)
jc.ResponseWriter.Header().Set("Content-Type", file.Mime())
2024-01-25 00:23:42 +00:00
http.ServeContent(jc.ResponseWriter, jc.Request, file.Name(), file.Modtime(), file)
2024-01-24 06:27:05 +00:00
}
2024-01-16 18:38:10 +00:00
func setAuthCookie(jwt string, jc jape.Context) {
authCookie := http.Cookie{
Name: "s5-auth-token",
Value: jwt,
Path: "/",
HttpOnly: true,
MaxAge: int(time.Hour.Seconds() * 24),
Secure: true,
}
http.SetCookie(jc.ResponseWriter, &authCookie)
}