schedtxt_api/internal/handler/message.go

package handler

import (
	"fmt"
	"log"
	"net/http"

	"git.kundeng.us/phoenix/textsender-models/tx0/message"
	"github.com/google/uuid"

	"git.kundeng.us/phoenix/textsender-api/internal/app"
	"git.kundeng.us/phoenix/textsender-api/internal/store"
)

type RequestAddMessage struct {
	Content string    `json:"content"`
	UserId  uuid.UUID `json:"user_id"`
}

type AddMessageResponse struct {
	Message string            `json:"message"`
	Data    []message.Message `json:"data"`
}

type MessageHandler struct {
	App          *app.App
	MessageStore store.MessageStore
}

func NewMessageHandler(apiApp *app.App, str store.MessageStore) *MessageHandler {
	return &MessageHandler{App: apiApp, MessageStore: str}
}

const Message_Limit = 200

// AddMessage godoc
// @Summary      Add message
// @Description  Add a message record to send a text to (requires JWT)
// @Tags         messages
// @Accept       json
// @Produce      json
// @Security     BearerAuth
// @Param 	 request body 	RequestAddMessage true "Data to add message"
// @Success      201  {object}  AddMessageResponse
// @Failure      400  {object}  AddMessageResponse
// @Failure      500  {object}  AddMessageResponse
// @Router       /message/new [post]
func (m *MessageHandler) AddMessage(w http.ResponseWriter, r *http.Request) {
	if r.Method != http.MethodPost {
		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
		return
	}

	var req RequestAddMessage
	if err := ExtractFromRequest(r, &req); err != nil {
		http.Error(w, "Invalid JSON: "+err.Error(), http.StatusBadRequest)
	}
	defer r.Body.Close()

	var statusCode int
	var resp AddMessageResponse
	newMessage := message.Message{Content: req.Content, UserId: req.UserId}

	if isMessageValid(&newMessage) {
		ctx := r.Context()

		if err := m.MessageStore.CreateMessage(ctx, &newMessage); err != nil {
			fmt.Println("Error:", err)
			statusCode = http.StatusInternalServerError
			resp.Message = err.Error()
		} else {
			statusCode = http.StatusCreated
			resp.Message = "Message created"
			resp.Data = append(resp.Data, newMessage)
		}
	} else {
		statusCode = http.StatusBadRequest
		resp.Message = fmt.Sprintf("Message length exceeds maximum limit. Current size: %d Limit: %d", len(newMessage.Content), Message_Limit)
	}

	RespondWithJSON(w, statusCode, &resp)
}

func isMessageValid(msg *message.Message) bool {
	if len(msg.Content) <= Message_Limit {
		return true
	} else {
		return false
	}
}

type GetMessageResponse struct {
	Message string            `json:"message"`
	Data    []message.Message `json:"data"`
}

// GetMessage godoc
// @Summary      Get message
// @Description  Get a message record to have a recipient to send a text to (requires JWT)
// @Tags         messages
// @Accept       json
// @Produce      json
// @Param        id   path      string  true  "Message Id"
// @Param        user_id   path      string  true  "User Id"
// @Security     BearerAuth
// @Success      200  {object}  GetMessageResponse
// @Failure      400  {object}  GetMessageResponse
// @Failure      500  {object}  GetMessageResponse
// @Router       /message [get]
func (c *MessageHandler) GetMessage(w http.ResponseWriter, r *http.Request) {
	if r.Method != http.MethodGet {
		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
		return
	}

	var id, userId uuid.UUID
	queryParams := r.URL.Query()

	// Check if parameter exists
	if _, exists := queryParams["id"]; !exists {
		if _, exists := queryParams["user_id"]; !exists {
			fmt.Fprintf(w, "Name parameter not provided")
			http.Error(w, "Query params", http.StatusBadRequest)
			return
		} else {
			var err error
			userId, err = uuid.Parse(queryParams.Get("user_id"))
			if err != nil {
				fmt.Fprintf(w, "Name parameter exists: %s", userId)
			}
		}
	} else {
		var err error
		idTmp := queryParams.Get("id")
		id, err = uuid.Parse(idTmp)
		if err != nil {
			fmt.Println("Error:", err)
			http.Error(w, "Error parsing Id", http.StatusBadRequest)
			return
		}
	}

	var statusCode int
	var resp GetMessageResponse

	ctx := r.Context()

	if id != uuid.Nil {
		log.Println("Checking with Id")
		if con, err := c.MessageStore.GetMessageByID(ctx, id); err == nil {
			statusCode = http.StatusOK
			resp.Message = "Successful"
			resp.Data = append(resp.Data, *con)
		} else {
			statusCode = http.StatusInternalServerError
			resp.Message = err.Error()
		}
	} else if userId != uuid.Nil {
		log.Println("Checking with User Id")
		if messages, err := c.MessageStore.GetAllMessages(ctx); err == nil {
			log.Println("Amount of messages:", len(messages))
			for _, msg := range messages {
				if msg.UserId == userId {
					resp.Data = append(resp.Data, *msg)
				}
			}

			if len(resp.Data) > 0 {
				statusCode = http.StatusOK
				resp.Message = "Successful"
			} else {
				statusCode = http.StatusNotFound
				resp.Message = "Message not found"
			}
		} else {
			statusCode = http.StatusInternalServerError
			resp.Message = err.Error()
		}
	} else {
		statusCode = http.StatusBadRequest
		resp.Message = "Invalid query parameter"
	}

	RespondWithJSON(w, statusCode, &resp)
}