98 lines
2.5 KiB
Go
98 lines
2.5 KiB
Go
package handler
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/zs/InsightReply/internal/model"
|
|
"github.com/zs/InsightReply/internal/repository"
|
|
)
|
|
|
|
type ReplyHandler struct {
|
|
repo *repository.ReplyRepository
|
|
}
|
|
|
|
func NewReplyHandler(repo *repository.ReplyRepository) *ReplyHandler {
|
|
return &ReplyHandler{repo: repo}
|
|
}
|
|
|
|
func (h *ReplyHandler) RecordReply(w http.ResponseWriter, r *http.Request) {
|
|
var body struct {
|
|
TweetID string `json:"tweet_id"`
|
|
StrategyType string `json:"strategy_type"`
|
|
Content string `json:"content"`
|
|
Language string `json:"language"`
|
|
}
|
|
|
|
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
|
|
SendError(w, http.StatusBadRequest, 4001, "Invalid request body")
|
|
return
|
|
}
|
|
|
|
ctx := r.Context()
|
|
userID, ok := ctx.Value("userID").(string)
|
|
if !ok || userID == "" {
|
|
SendError(w, http.StatusUnauthorized, 4002, "Unauthorized")
|
|
return
|
|
}
|
|
|
|
userUUID, err := uuid.Parse(userID)
|
|
if err != nil {
|
|
SendError(w, http.StatusBadRequest, 4003, "Invalid user ID format")
|
|
return
|
|
}
|
|
|
|
// Resolve the raw string X_Tweet_ID into our internal UUID
|
|
// Create a dummy tweet entry via Upsert if it doesn't exist yet so foreign keys don't panic
|
|
tweet := &model.Tweet{
|
|
XTweetID: body.TweetID,
|
|
Content: body.Content, // Temporarily store AI content as a placeholder if original is missing
|
|
IsProcessed: false,
|
|
}
|
|
|
|
err = h.repo.UpsertDummyTweet(tweet)
|
|
if err != nil {
|
|
SendError(w, http.StatusInternalServerError, 5001, "Failed to resolve tweet reference")
|
|
return
|
|
}
|
|
|
|
reply := &model.GeneratedReply{
|
|
UserID: userUUID,
|
|
TweetID: tweet.ID,
|
|
StrategyType: body.StrategyType,
|
|
Content: body.Content,
|
|
Status: "copied",
|
|
Language: body.Language,
|
|
}
|
|
|
|
if err := h.repo.CreateGeneratedReply(reply); err != nil {
|
|
SendError(w, http.StatusInternalServerError, 5002, "Failed to log generated reply")
|
|
return
|
|
}
|
|
|
|
SendSuccess(w, map[string]string{ "message": "Reply recorded successfully",
|
|
})
|
|
}
|
|
|
|
// GetReplies handles GET /api/v1/replies
|
|
func (h *ReplyHandler) GetReplies(w http.ResponseWriter, r *http.Request) {
|
|
userIDStr := r.Header.Get("X-User-ID")
|
|
userID, err := uuid.Parse(userIDStr)
|
|
if err != nil {
|
|
http.Error(w, "Unauthorized", http.StatusUnauthorized)
|
|
return
|
|
}
|
|
|
|
replies, err := h.repo.GetGeneratedRepliesByUser(userID)
|
|
if err != nil {
|
|
log.Printf("Failed to get replies: %v", err)
|
|
http.Error(w, "Failed to get replies", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
json.NewEncoder(w).Encode(replies)
|
|
}
|