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/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)
}