Belac/darkstorm-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Converted the last pieces to new backend

Browse Source 
Added created & updated time to blog
This commit is contained in:
Caleb Gardner
2024-08-02 05:05:56 -05:00
parent 40271fa088
commit fd89e568af
6 changed files with 41 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/cdr/backend.go
+2 -2
View File
@@ -21,7 +21,7 @@ func NewBackend(back *backend.Backend, db *mongo.Database) *CDRBackend {
go func() {
for range time.Tick(time.Hour) {
log.Println("CDR: Deleting expired dice")
res, err := db.Collection("profiles").DeleteMany(context.TODO(), bson.M{"expiration": bson.M{"$lt": time.Now().Unix()}})
res, err := db.Collection("profiles").DeleteMany(context.Background(), bson.M{"expiration": bson.M{"$lt": time.Now().Unix()}})
if err == mongo.ErrNoDocuments {
continue
}
@@ -47,7 +47,7 @@ func (b CDRBackend) CrashTable() backend.CrashTable {
}
func (s CDRBackend) AddCrash(cr backend.IndividualCrash) bool {
res := s.db.Collection("versions").FindOne(context.TODO(), bson.M{"version": cr.Version})
res := s.db.Collection("versions").FindOne(context.Background(), bson.M{"version": cr.Version})
return res.Err() != mongo.ErrNoDocuments
}
internal/cdr/die.go
+8 -4
View File
@@ -21,9 +21,13 @@ type UploadedDie struct {
}
func (b CDRBackend) UploadDie(w http.ResponseWriter, r *http.Request) {
hdr, err := b.back.VerifyHeader(w, r, "rooms", false)
hdr, err := b.back.VerifyHeader(w, r, "dice", false)
if err != nil {
return
}
if hdr.Key.AppID != "cdr" {
backend.ReturnError(w, http.StatusUnauthorized, "unauthorized", "Application not authorized")
return
}
if r.Body == nil {
backend.ReturnError(w, http.StatusBadRequest, "bad request", "Application sent a bad request")
@@ -52,7 +56,7 @@ func (b CDRBackend) UploadDie(w http.ResponseWriter, r *http.Request) {
if toUpload.Die["uuid"] != nil {
delete(toUpload.Die, "uuid")
}
_, err = b.db.Collection("dice").InsertOne(context.TODO(), toUpload)
_, err = b.db.Collection("dice").InsertOne(context.Background(), toUpload)
if err != nil {
backend.ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
log.Println("error inserting die:", err)
@@ -63,7 +67,7 @@ func (b CDRBackend) UploadDie(w http.ResponseWriter, r *http.Request) {
}
func (b CDRBackend) GetDie(w http.ResponseWriter, r *http.Request) {
res := b.db.Collection("dice").FindOne(context.TODO(), bson.M{"_id": r.PathValue("dieID")})
res := b.db.Collection("dice").FindOne(context.Background(), bson.M{"_id": r.PathValue("dieID")})
if res.Err() == mongo.ErrNoDocuments {
backend.ReturnError(w, 404, "not found", "Die with the given id is not found")
return
internal/swassistant/main.go
+2 -2
View File
@@ -21,7 +21,7 @@ func NewSWBackend(back *backend.Backend, db *mongo.Database) *SWBackend {
go func() {
for range time.Tick(time.Hour) {
log.Println("SWAssistant: Deleting expired profiles")
res, err := db.Collection("profiles").DeleteMany(context.TODO(), bson.M{"expiration": bson.M{"$lt": time.Now().Unix()}})
res, err := db.Collection("profiles").DeleteMany(context.Background(), bson.M{"expiration": bson.M{"$lt": time.Now().Unix()}})
if err == mongo.ErrNoDocuments {
continue
}
@@ -47,7 +47,7 @@ func (s *SWBackend) CrashTable() backend.CrashTable {
}
func (s *SWBackend) AddCrash(cr backend.IndividualCrash) bool {
res := s.db.Collection("versions").FindOne(context.TODO(), bson.M{"version": cr.Version})
res := s.db.Collection("versions").FindOne(context.Background(), bson.M{"version": cr.Version})
return res.Err() != mongo.ErrNoDocuments
}
internal/swassistant/profile.go
+2 -2
View File
@@ -61,7 +61,7 @@ func (s *SWBackend) UploadProfile(w http.ResponseWriter, r *http.Request) {
Type: profType,
Profile: prof,
}
_, err = s.db.Collection("profiles").InsertOne(context.TODO(), toUpload)
_, err = s.db.Collection("profiles").InsertOne(context.Background(), toUpload)
if err != nil {
backend.ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
log.Println("error inserting profile:", err)
@@ -72,7 +72,7 @@ func (s *SWBackend) UploadProfile(w http.ResponseWriter, r *http.Request) {
}
func (s *SWBackend) GetProfile(w http.ResponseWriter, r *http.Request) {
res := s.db.Collection("profiles").FindOne(context.TODO(), bson.M{"_id": r.PathValue("profileID")})
res := s.db.Collection("profiles").FindOne(context.Background(), bson.M{"_id": r.PathValue("profileID")})
if res.Err() == mongo.ErrNoDocuments {
backend.ReturnError(w, 404, "not found", "Profile not found")
return
internal/swassistant/room.go
+26 -57
View File
@@ -30,7 +30,7 @@ func (s *SWBackend) ListRooms(w http.ResponseWriter, r *http.Request) {
backend.ReturnError(w, http.StatusUnauthorized, "unauthorized", "Application not authorized")
return
}
res, err := s.db.Collection("rooms").Find(context.TODO(), bson.M{"users": hdr.User.Username}, options.Find().SetProjection(bson.M{"_id": 1, "name": 1, "owner": 1}))
res, err := s.db.Collection("rooms").Find(context.Background(), bson.M{"users": hdr.User.Username}, options.Find().SetProjection(bson.M{"_id": 1, "name": 1, "owner": 1}))
if err != nil && err != mongo.ErrNoDocuments {
log.Println("error getting room list:", err)
backend.ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
@@ -42,7 +42,7 @@ func (s *SWBackend) ListRooms(w http.ResponseWriter, r *http.Request) {
Owner string `json:"owner" bson:"owner"`
}, 0)
if err == nil {
err = res.All(context.TODO(), &out)
err = res.All(context.Background(), &out)
if err != nil {
log.Println("error decoding room list:", err)
backend.ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
@@ -61,40 +61,26 @@ func (s *SWBackend) NewRoom(w http.ResponseWriter, r *http.Request) {
backend.ReturnError(w, http.StatusUnauthorized, "unauthorized", "Application not authorized")
return
}
if req.Method != http.MethodPost || req.Query["name"] == nil || len(req.Query["name"]) != 1 || req.Query["name"][0] == "" {
req.Resp.WriteHeader(http.StatusBadRequest)
return true
} else if req.User == nil {
req.Resp.WriteHeader(http.StatusUnauthorized)
return true
name := r.URL.Query().Get("name")
if name == "" {
backend.ReturnError(w, http.StatusBadRequest, "bad request", "Application sent bad request")
return
}
//TODO: check room name for unsavory words
newRoom := Room{
ID: uuid.NewString(),
Name: req.Query["name"][0],
Owner: req.User.Username,
Name: name,
Owner: hdr.User.Username,
Users: []string{},
Profiles: []string{},
}
_, err := s.db.Collection("rooms").InsertOne(context.TODO(), newRoom)
_, err = s.db.Collection("rooms").InsertOne(context.Background(), newRoom)
if err != nil {
log.Println("SWAssistant: Error creating room:", err)
req.Resp.WriteHeader(http.StatusInternalServerError)
return true
log.Println("error creating room:", err)
backend.ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
return
}
out, err := json.Marshal(map[string]string{"id": newRoom.ID, "name": newRoom.Name})
if err != nil {
log.Println("SWAssistant: Error encoding new room:", err)
req.Resp.WriteHeader(http.StatusInternalServerError)
return true
}
req.Resp.WriteHeader(http.StatusCreated)
_, err = req.Resp.Write(out)
if err != nil {
log.Println("SWAssistant: Error writing new room:", err)
req.Resp.WriteHeader(http.StatusInternalServerError)
}
return true
json.NewEncoder(w).Encode(map[string]string{"id": newRoom.ID, "name": newRoom.Name})
}
func (s *SWBackend) GetRoom(w http.ResponseWriter, r *http.Request) {
@@ -106,39 +92,22 @@ func (s *SWBackend) GetRoom(w http.ResponseWriter, r *http.Request) {
backend.ReturnError(w, http.StatusUnauthorized, "unauthorized", "Application not authorized")
return
}
if req.Method != http.MethodGet {
req.Resp.WriteHeader(http.StatusBadRequest)
return true
} else if req.User == nil {
req.Resp.WriteHeader(http.StatusUnauthorized)
return true
}
res := s.db.Collection("rooms").FindOne(context.TODO(), bson.M{"_id": req.Path[1]})
roomID := r.PathValue("roomID")
res := s.db.Collection("rooms").FindOne(context.TODO(), bson.M{"_id": roomID})
if res.Err() == mongo.ErrNoDocuments {
req.Resp.WriteHeader(http.StatusNotFound)
return true
backend.ReturnError(w, http.StatusNotFound, "not found", "Room not found")
return
} else if res.Err() != nil {
log.Println("SWAssistant: Error getting room:", res.Err())
req.Resp.WriteHeader(http.StatusInternalServerError)
return true
log.Println("error getting room:", res.Err())
backend.ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
return
}
r := Room{}
err := res.Decode(&r)
var rm Room
err = res.Decode(&rm)
if err != nil {
log.Println("SWAssistant: Error decoding room:", err)
req.Resp.WriteHeader(http.StatusInternalServerError)
return true
log.Println("error decoding room:", err)
backend.ReturnError(w, http.StatusInternalServerError, "internal", "Server error")
return
}
out, err := json.Marshal(r)
if err != nil {
log.Println("SWAssistant: Error encoding room:", err)
req.Resp.WriteHeader(http.StatusInternalServerError)
return true
}
_, err = req.Resp.Write(out)
if err != nil {
log.Println("SWAssistant: Error writing room:", err)
req.Resp.WriteHeader(http.StatusInternalServerError)
}
return true
json.NewEncoder(w).Encode(rm)
}