mirror of
https://github.com/J3vb/OwnCord.git
synced 2026-09-03 03:50:00 +03:00
feat(server): add DM REST endpoints, WebSocket routing, and ready payload
Task A: REST API endpoints in api/dm_handler.go — POST /api/v1/dms
(create/get DM channel), GET /api/v1/dms (list open DMs), DELETE
/api/v1/dms/{channelId} (close DM). Routes registered in router.go.
Task B: WebSocket DM routing — handleChatSend, handleChatEdit,
handleChatDelete, and handleReaction all check channel type and use
participant-based auth for DM channels instead of role permissions.
DM messages delivered via SendToUser to each participant (bypassing
channel-subscription model). Auto-reopens DM for recipient on new
message with dm_channel_open event. New broadcastToDMParticipants
helper. New WS event types: dm_channel_open, dm_channel_close.
Task C: Ready payload includes dm_channels from GetUserDMChannels.
Also adds GetDMParticipantIDs helper to db/dm_queries.go.
This commit is contained in:
@@ -0,0 +1,207 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
|
||||
"github.com/go-chi/chi/v5"
|
||||
"github.com/owncord/server/db"
|
||||
)
|
||||
|
||||
// MountDMRoutes registers DM-related routes onto r.
|
||||
// All routes require authentication.
|
||||
func MountDMRoutes(r chi.Router, database *db.DB) {
|
||||
r.Route("/api/v1/dms", func(r chi.Router) {
|
||||
r.Use(AuthMiddleware(database))
|
||||
r.Post("/", handleCreateDM(database))
|
||||
r.Get("/", handleListDMs(database))
|
||||
r.Delete("/{channelId}", handleCloseDM(database))
|
||||
})
|
||||
}
|
||||
|
||||
// createDMRequest is the JSON body for POST /api/v1/dms.
|
||||
type createDMRequest struct {
|
||||
RecipientID int64 `json:"recipient_id"`
|
||||
}
|
||||
|
||||
// createDMResponse is the JSON response for POST /api/v1/dms.
|
||||
type createDMResponse struct {
|
||||
ChannelID int64 `json:"channel_id"`
|
||||
Recipient db.DMUser `json:"recipient"`
|
||||
Created bool `json:"created"`
|
||||
}
|
||||
|
||||
// listDMsResponse is the JSON response for GET /api/v1/dms.
|
||||
type listDMsResponse struct {
|
||||
DMChannels []db.DMChannelInfo `json:"dm_channels"`
|
||||
}
|
||||
|
||||
// handleCreateDM creates or retrieves a DM channel with a recipient.
|
||||
func handleCreateDM(database *db.DB) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
user, ok := r.Context().Value(UserKey).(*db.User)
|
||||
if !ok || user == nil {
|
||||
writeJSON(w, http.StatusUnauthorized, errorResponse{
|
||||
Error: "UNAUTHORIZED",
|
||||
Message: "authentication required",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
var req createDMRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
writeJSON(w, http.StatusBadRequest, errorResponse{
|
||||
Error: "BAD_REQUEST",
|
||||
Message: "invalid request body",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
if req.RecipientID <= 0 {
|
||||
writeJSON(w, http.StatusBadRequest, errorResponse{
|
||||
Error: "BAD_REQUEST",
|
||||
Message: "recipient_id must be a positive integer",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// Cannot DM yourself.
|
||||
if req.RecipientID == user.ID {
|
||||
writeJSON(w, http.StatusBadRequest, errorResponse{
|
||||
Error: "BAD_REQUEST",
|
||||
Message: "cannot create a DM with yourself",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// Verify recipient exists.
|
||||
recipient, err := database.GetUserByID(req.RecipientID)
|
||||
if err != nil {
|
||||
slog.Error("handleCreateDM GetUserByID", "err", err, "recipient_id", req.RecipientID)
|
||||
writeJSON(w, http.StatusInternalServerError, errorResponse{
|
||||
Error: "INTERNAL",
|
||||
Message: "failed to look up recipient",
|
||||
})
|
||||
return
|
||||
}
|
||||
if recipient == nil {
|
||||
writeJSON(w, http.StatusNotFound, errorResponse{
|
||||
Error: "NOT_FOUND",
|
||||
Message: "recipient not found",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// Get or create the DM channel.
|
||||
ch, created, err := database.GetOrCreateDMChannel(user.ID, req.RecipientID)
|
||||
if err != nil {
|
||||
slog.Error("handleCreateDM GetOrCreateDMChannel", "err", err,
|
||||
"user_id", user.ID, "recipient_id", req.RecipientID)
|
||||
writeJSON(w, http.StatusInternalServerError, errorResponse{
|
||||
Error: "INTERNAL",
|
||||
Message: "failed to create DM channel",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
// Build the recipient DMUser from the fetched user.
|
||||
avatarStr := ""
|
||||
if recipient.Avatar != nil {
|
||||
avatarStr = *recipient.Avatar
|
||||
}
|
||||
dmUser := db.DMUser{
|
||||
ID: recipient.ID,
|
||||
Username: recipient.Username,
|
||||
Avatar: avatarStr,
|
||||
Status: recipient.Status,
|
||||
}
|
||||
|
||||
status := http.StatusOK
|
||||
if created {
|
||||
status = http.StatusCreated
|
||||
}
|
||||
|
||||
writeJSON(w, status, createDMResponse{
|
||||
ChannelID: ch.ID,
|
||||
Recipient: dmUser,
|
||||
Created: created,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// handleListDMs returns all open DM channels for the authenticated user.
|
||||
func handleListDMs(database *db.DB) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
user, ok := r.Context().Value(UserKey).(*db.User)
|
||||
if !ok || user == nil {
|
||||
writeJSON(w, http.StatusUnauthorized, errorResponse{
|
||||
Error: "UNAUTHORIZED",
|
||||
Message: "authentication required",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
channels, err := database.GetUserDMChannels(user.ID)
|
||||
if err != nil {
|
||||
slog.Error("handleListDMs GetUserDMChannels", "err", err, "user_id", user.ID)
|
||||
writeJSON(w, http.StatusInternalServerError, errorResponse{
|
||||
Error: "INTERNAL",
|
||||
Message: "failed to list DM channels",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
writeJSON(w, http.StatusOK, listDMsResponse{DMChannels: channels})
|
||||
}
|
||||
}
|
||||
|
||||
// handleCloseDM removes a DM channel from the authenticated user's open list.
|
||||
func handleCloseDM(database *db.DB) http.HandlerFunc {
|
||||
return func(w http.ResponseWriter, r *http.Request) {
|
||||
user, ok := r.Context().Value(UserKey).(*db.User)
|
||||
if !ok || user == nil {
|
||||
writeJSON(w, http.StatusUnauthorized, errorResponse{
|
||||
Error: "UNAUTHORIZED",
|
||||
Message: "authentication required",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
channelID, ok := parseIDParam(w, r, "channelId")
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
|
||||
// Verify user is a participant in this DM.
|
||||
isParticipant, err := database.IsDMParticipant(user.ID, channelID)
|
||||
if err != nil {
|
||||
slog.Error("handleCloseDM IsDMParticipant", "err", err,
|
||||
"user_id", user.ID, "channel_id", channelID)
|
||||
writeJSON(w, http.StatusInternalServerError, errorResponse{
|
||||
Error: "INTERNAL",
|
||||
Message: "failed to verify DM participation",
|
||||
})
|
||||
return
|
||||
}
|
||||
if !isParticipant {
|
||||
writeJSON(w, http.StatusForbidden, errorResponse{
|
||||
Error: "FORBIDDEN",
|
||||
Message: "you are not a participant in this DM",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
if err := database.CloseDM(user.ID, channelID); err != nil {
|
||||
slog.Error("handleCloseDM CloseDM", "err", err,
|
||||
"user_id", user.ID, "channel_id", channelID)
|
||||
writeJSON(w, http.StatusInternalServerError, errorResponse{
|
||||
Error: "INTERNAL",
|
||||
Message: "failed to close DM",
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user