|
|
|
@@ -30,8 +30,8 @@ func (s *Handlers) requireAuth(w http.ResponseWriter, r *http.Request) (int64, s
|
|
|
|
return uid, nick, true
|
|
|
|
return uid, nick, true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// HandleRegister creates a new user and returns the auth token.
|
|
|
|
// HandleCreateSession creates a new user session and returns the auth token.
|
|
|
|
func (s *Handlers) HandleRegister() http.HandlerFunc {
|
|
|
|
func (s *Handlers) HandleCreateSession() http.HandlerFunc {
|
|
|
|
type request struct {
|
|
|
|
type request struct {
|
|
|
|
Nick string `json:"nick"`
|
|
|
|
Nick string `json:"nick"`
|
|
|
|
}
|
|
|
|
}
|
|
|
|
@@ -104,68 +104,6 @@ func (s *Handlers) HandleListAllChannels() http.HandlerFunc {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// HandleJoinChannel joins a channel (creates it if needed).
|
|
|
|
|
|
|
|
func (s *Handlers) HandleJoinChannel() http.HandlerFunc {
|
|
|
|
|
|
|
|
type request struct {
|
|
|
|
|
|
|
|
Channel string `json:"channel"`
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
|
|
|
|
uid, _, ok := s.requireAuth(w, r)
|
|
|
|
|
|
|
|
if !ok {
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
var req request
|
|
|
|
|
|
|
|
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "invalid request"}, http.StatusBadRequest)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
req.Channel = strings.TrimSpace(req.Channel)
|
|
|
|
|
|
|
|
if req.Channel == "" {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "channel name required"}, http.StatusBadRequest)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if !strings.HasPrefix(req.Channel, "#") {
|
|
|
|
|
|
|
|
req.Channel = "#" + req.Channel
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
chID, err := s.params.Database.GetOrCreateChannel(r.Context(), req.Channel)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
s.log.Error("get/create channel failed", "error", err)
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := s.params.Database.JoinChannel(r.Context(), chID, uid); err != nil {
|
|
|
|
|
|
|
|
s.log.Error("join channel failed", "error", err)
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"status": "joined", "channel": req.Channel}, http.StatusOK)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// HandlePartChannel leaves a channel.
|
|
|
|
|
|
|
|
func (s *Handlers) HandlePartChannel() http.HandlerFunc {
|
|
|
|
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
|
|
|
|
uid, _, ok := s.requireAuth(w, r)
|
|
|
|
|
|
|
|
if !ok {
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
name := "#" + chi.URLParam(r, "channel")
|
|
|
|
|
|
|
|
var chID int64
|
|
|
|
|
|
|
|
err := s.params.Database.GetDB().QueryRowContext(r.Context(),
|
|
|
|
|
|
|
|
"SELECT id FROM channels WHERE name = ?", name).Scan(&chID)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "channel not found"}, http.StatusNotFound)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := s.params.Database.PartChannel(r.Context(), chID, uid); err != nil {
|
|
|
|
|
|
|
|
s.log.Error("part channel failed", "error", err)
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"status": "parted", "channel": name}, http.StatusOK)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// HandleChannelMembers returns members of a channel.
|
|
|
|
// HandleChannelMembers returns members of a channel.
|
|
|
|
func (s *Handlers) HandleChannelMembers() http.HandlerFunc {
|
|
|
|
func (s *Handlers) HandleChannelMembers() http.HandlerFunc {
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
@@ -211,15 +149,17 @@ func (s *Handlers) HandleGetMessages() http.HandlerFunc {
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
// HandleSendMessage sends a message to a channel or user.
|
|
|
|
// HandleSendCommand handles all C2S commands via POST /messages.
|
|
|
|
// The "to" field determines the target: "#channel" for channels, "nick" for DMs.
|
|
|
|
// The "command" field dispatches to the appropriate logic.
|
|
|
|
func (s *Handlers) HandleSendMessage() http.HandlerFunc {
|
|
|
|
func (s *Handlers) HandleSendCommand() http.HandlerFunc {
|
|
|
|
type request struct {
|
|
|
|
type request struct {
|
|
|
|
To string `json:"to"`
|
|
|
|
Command string `json:"command"`
|
|
|
|
Content string `json:"content"`
|
|
|
|
To string `json:"to"`
|
|
|
|
|
|
|
|
Params []string `json:"params,omitempty"`
|
|
|
|
|
|
|
|
Body interface{} `json:"body,omitempty"`
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
uid, _, ok := s.requireAuth(w, r)
|
|
|
|
uid, nick, ok := s.requireAuth(w, r)
|
|
|
|
if !ok {
|
|
|
|
if !ok {
|
|
|
|
return
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
@@ -228,46 +168,167 @@ func (s *Handlers) HandleSendMessage() http.HandlerFunc {
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "invalid request"}, http.StatusBadRequest)
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "invalid request"}, http.StatusBadRequest)
|
|
|
|
return
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if strings.TrimSpace(req.Content) == "" {
|
|
|
|
req.Command = strings.ToUpper(strings.TrimSpace(req.Command))
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "content required"}, http.StatusBadRequest)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
req.To = strings.TrimSpace(req.To)
|
|
|
|
req.To = strings.TrimSpace(req.To)
|
|
|
|
if req.To == "" {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "to field required"}, http.StatusBadRequest)
|
|
|
|
// Helper to extract body as string lines.
|
|
|
|
return
|
|
|
|
bodyLines := func() []string {
|
|
|
|
|
|
|
|
switch v := req.Body.(type) {
|
|
|
|
|
|
|
|
case []interface{}:
|
|
|
|
|
|
|
|
lines := make([]string, 0, len(v))
|
|
|
|
|
|
|
|
for _, item := range v {
|
|
|
|
|
|
|
|
if s, ok := item.(string); ok {
|
|
|
|
|
|
|
|
lines = append(lines, s)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return lines
|
|
|
|
|
|
|
|
case []string:
|
|
|
|
|
|
|
|
return v
|
|
|
|
|
|
|
|
default:
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
if strings.HasPrefix(req.To, "#") {
|
|
|
|
switch req.Command {
|
|
|
|
// Channel message
|
|
|
|
case "PRIVMSG", "NOTICE":
|
|
|
|
|
|
|
|
if req.To == "" {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "to field required"}, http.StatusBadRequest)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
lines := bodyLines()
|
|
|
|
|
|
|
|
if len(lines) == 0 {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "body required"}, http.StatusBadRequest)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
content := strings.Join(lines, "\n")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if strings.HasPrefix(req.To, "#") {
|
|
|
|
|
|
|
|
// Channel message
|
|
|
|
|
|
|
|
var chID int64
|
|
|
|
|
|
|
|
err := s.params.Database.GetDB().QueryRowContext(r.Context(),
|
|
|
|
|
|
|
|
"SELECT id FROM channels WHERE name = ?", req.To).Scan(&chID)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "channel not found"}, http.StatusNotFound)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
msgID, err := s.params.Database.SendMessage(r.Context(), chID, uid, content)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
s.log.Error("send message failed", "error", err)
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]any{"id": msgID, "status": "sent"}, http.StatusCreated)
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
// DM
|
|
|
|
|
|
|
|
targetID, err := s.params.Database.GetUserByNick(r.Context(), req.To)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "user not found"}, http.StatusNotFound)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
msgID, err := s.params.Database.SendDM(r.Context(), uid, targetID, content)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
s.log.Error("send dm failed", "error", err)
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]any{"id": msgID, "status": "sent"}, http.StatusCreated)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
case "JOIN":
|
|
|
|
|
|
|
|
if req.To == "" {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "to field required"}, http.StatusBadRequest)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
channel := req.To
|
|
|
|
|
|
|
|
if !strings.HasPrefix(channel, "#") {
|
|
|
|
|
|
|
|
channel = "#" + channel
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
chID, err := s.params.Database.GetOrCreateChannel(r.Context(), channel)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
|
|
|
s.log.Error("get/create channel failed", "error", err)
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := s.params.Database.JoinChannel(r.Context(), chID, uid); err != nil {
|
|
|
|
|
|
|
|
s.log.Error("join channel failed", "error", err)
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"status": "joined", "channel": channel}, http.StatusOK)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
case "PART":
|
|
|
|
|
|
|
|
if req.To == "" {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "to field required"}, http.StatusBadRequest)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
channel := req.To
|
|
|
|
|
|
|
|
if !strings.HasPrefix(channel, "#") {
|
|
|
|
|
|
|
|
channel = "#" + channel
|
|
|
|
|
|
|
|
}
|
|
|
|
var chID int64
|
|
|
|
var chID int64
|
|
|
|
err := s.params.Database.GetDB().QueryRowContext(r.Context(),
|
|
|
|
err := s.params.Database.GetDB().QueryRowContext(r.Context(),
|
|
|
|
"SELECT id FROM channels WHERE name = ?", req.To).Scan(&chID)
|
|
|
|
"SELECT id FROM channels WHERE name = ?", channel).Scan(&chID)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "channel not found"}, http.StatusNotFound)
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "channel not found"}, http.StatusNotFound)
|
|
|
|
return
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
msgID, err := s.params.Database.SendMessage(r.Context(), chID, uid, req.Content)
|
|
|
|
if err := s.params.Database.PartChannel(r.Context(), chID, uid); err != nil {
|
|
|
|
if err != nil {
|
|
|
|
s.log.Error("part channel failed", "error", err)
|
|
|
|
s.log.Error("send message failed", "error", err)
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
s.respondJSON(w, r, map[string]any{"id": msgID, "status": "sent"}, http.StatusCreated)
|
|
|
|
s.respondJSON(w, r, map[string]string{"status": "parted", "channel": channel}, http.StatusOK)
|
|
|
|
} else {
|
|
|
|
|
|
|
|
// DM
|
|
|
|
case "NICK":
|
|
|
|
targetID, err := s.params.Database.GetUserByNick(r.Context(), req.To)
|
|
|
|
lines := bodyLines()
|
|
|
|
if err != nil {
|
|
|
|
if len(lines) == 0 {
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "user not found"}, http.StatusNotFound)
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "body required (new nick)"}, http.StatusBadRequest)
|
|
|
|
return
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
msgID, err := s.params.Database.SendDM(r.Context(), uid, targetID, req.Content)
|
|
|
|
newNick := strings.TrimSpace(lines[0])
|
|
|
|
if err != nil {
|
|
|
|
if newNick == "" || len(newNick) > 32 {
|
|
|
|
s.log.Error("send dm failed", "error", err)
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "nick must be 1-32 characters"}, http.StatusBadRequest)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := s.params.Database.ChangeNick(r.Context(), uid, newNick); err != nil {
|
|
|
|
|
|
|
|
if strings.Contains(err.Error(), "UNIQUE") {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "nick already in use"}, http.StatusConflict)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
s.log.Error("change nick failed", "error", err)
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
s.respondJSON(w, r, map[string]any{"id": msgID, "status": "sent"}, http.StatusCreated)
|
|
|
|
s.respondJSON(w, r, map[string]string{"status": "ok", "nick": newNick}, http.StatusOK)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
case "TOPIC":
|
|
|
|
|
|
|
|
if req.To == "" {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "to field required"}, http.StatusBadRequest)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
lines := bodyLines()
|
|
|
|
|
|
|
|
if len(lines) == 0 {
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "body required (topic text)"}, http.StatusBadRequest)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
topic := strings.Join(lines, " ")
|
|
|
|
|
|
|
|
channel := req.To
|
|
|
|
|
|
|
|
if !strings.HasPrefix(channel, "#") {
|
|
|
|
|
|
|
|
channel = "#" + channel
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := s.params.Database.SetTopic(r.Context(), channel, uid, topic); err != nil {
|
|
|
|
|
|
|
|
s.log.Error("set topic failed", "error", err)
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "internal error"}, http.StatusInternalServerError)
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"status": "ok", "topic": topic}, http.StatusOK)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
case "PING":
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"command": "PONG", "from": s.params.Config.ServerName}, http.StatusOK)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
default:
|
|
|
|
|
|
|
|
_ = nick // suppress unused warning
|
|
|
|
|
|
|
|
s.respondJSON(w, r, map[string]string{"error": "unknown command: " + req.Command}, http.StatusBadRequest)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|