Caleb Gardner
93 lines
3.0 KiB
Go
93 lines
3.0 KiB
Go
package swassistant
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/CalebQ42/darkstorm-server/internal/backend"
|
|
"github.com/lithammer/shortuuid/v3"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
)
|
|
|
|
type UploadedProf struct {
|
|
Profile map[string]any `json:"profile" bson:"profile"`
|
|
ID string `json:"id" bson:"_id"`
|
|
Type string `json:"type" bson:"type"`
|
|
Expiration int64 `json:"expiration" bson:"expiration"`
|
|
}
|
|
|
|
func (s *SWBackend) UploadProfile(w http.ResponseWriter, r *http.Request) {
|
|
hdr, err := s.back.VerifyHeader(w, r, "profile", false)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if hdr.Key.AppID != "swassistant" {
|
|
backend.ReturnError(w, http.StatusUnauthorized, "unauthorized", "Application not authorized")
|
|
return
|
|
}
|
|
profType := r.URL.Query().Get("type")
|
|
if profType == "" || (profType != "character" && profType != "vehicle" && profType != "minion") {
|
|
backend.ReturnError(w, http.StatusBadRequest, "bad request", "Application sent a bad request")
|
|
return
|
|
}
|
|
if r.Body == nil {
|
|
backend.ReturnError(w, http.StatusBadRequest, "bad request", "Application sent a bad request")
|
|
return
|
|
}
|
|
data, err := io.ReadAll(r.Body)
|
|
r.Body.Close()
|
|
if err != nil || len(data) == 0 {
|
|
backend.ReturnError(w, http.StatusBadRequest, "bad request", "Application sent a bad request")
|
|
return
|
|
} else if len(data) > 5242880 { // 5MB
|
|
backend.ReturnError(w, http.StatusRequestEntityTooLarge, "too large", "Profile is too large")
|
|
return
|
|
}
|
|
prof := make(map[string]any)
|
|
err = json.Unmarshal(data, &prof)
|
|
if err != nil {
|
|
backend.ReturnError(w, http.StatusBadRequest, "bad request", "Application sent a bad request")
|
|
return
|
|
}
|
|
delete(prof, "uid")
|
|
toUpload := UploadedProf{
|
|
ID: shortuuid.New(),
|
|
Expiration: time.Now().Add(time.Hour * 12).Round(time.Hour).Unix(),
|
|
Type: profType,
|
|
Profile: prof,
|
|
}
|
|
_, err = s.db.Collection("profiles").InsertOne(r.Context(), toUpload)
|
|
if err != nil {
|
|
backend.ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
|
|
log.Println("error inserting profile:", err)
|
|
return
|
|
}
|
|
w.WriteHeader(http.StatusCreated)
|
|
json.NewEncoder(w).Encode(map[string]any{"id": toUpload.ID, "expiration": toUpload.Expiration})
|
|
}
|
|
|
|
func (s *SWBackend) GetProfile(w http.ResponseWriter, r *http.Request) {
|
|
res := s.db.Collection("profiles").FindOne(r.Context(), bson.M{"_id": r.PathValue("profileID")})
|
|
if res.Err() == mongo.ErrNoDocuments {
|
|
backend.ReturnError(w, 404, "not found", "Profile not found")
|
|
return
|
|
} else if res.Err() != nil {
|
|
backend.ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
|
|
log.Println("error getting profile:", res.Err())
|
|
return
|
|
}
|
|
var prof UploadedProf
|
|
err := res.Decode(&prof)
|
|
if err != nil {
|
|
backend.ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
|
|
log.Println("error decoding profile:", err)
|
|
return
|
|
}
|
|
prof.Profile["type"] = prof.Type
|
|
json.NewEncoder(w).Encode(prof.Profile)
|
|
}
|