newsbot-api/routes/articles.go

package routes

import (
	"fmt"
	"net/http"
	"strconv"

	"github.com/go-chi/chi/v5"
	"github.com/google/uuid"
	"github.com/jtom38/newsbot/collector/domain/models"
)

func (s *Server) GetArticleRouter() http.Handler {
	r := chi.NewRouter()

	r.Get("/", s.listArticles)
	r.Route("/{ID}", func(r chi.Router) {
		r.Get("/", s.getArticle)
		r.Get("/details", s.getArticleDetails)
	})
	r.Get("/by/sourceid", s.GetArticlesBySourceId)

	return r
}

type ArticlesListResults struct {
	ApiStatusModel
	Payload []models.ArticleDto `json:"payload"`
}

type ArticleGetResults struct {
	ApiStatusModel
	Payload models.ArticleDto `json:"payload"`
}

type ArticleDetailsResult struct {
	ApiStatusModel
	Payload models.ArticleDetailsDto `json:"payload"`
}

// ListArticles
// @Summary  Lists the top 50 records
// @Produce  application/json
// @Param    page  query  string  false  "page number"
// @Tags     Articles
// @Router   /articles [get]
// @Success  200  {object}  ArticlesListResults  "OK"
func (s *Server) listArticles(w http.ResponseWriter, r *http.Request) {
	p := ArticlesListResults{
		ApiStatusModel: ApiStatusModel{
			Message:    "OK",
			StatusCode: http.StatusOK,
		},
	}

	query := r.URL.Query()
	queryPage := query["page"]
	fmt.Printf("queryPage: %v\n", queryPage)
	
	// if a page number was sent, process it
	if len(queryPage) == 1 {
		page, err := strconv.Atoi(query["page"][0])
		if err != nil {
			s.WriteError(w, err.Error(), http.StatusBadRequest)
			return
		}

		res, err := s.dto.ListArticlesByPage(r.Context(), int32(page), 50)
		if err != nil {
			s.WriteError(w, err.Error(), http.StatusInternalServerError)
			return
		}
		p.Payload = res
		s.WriteJson(w, p)
	} else {
		res, err := s.dto.ListArticles(r.Context(), 50)
		if err != nil {
			s.WriteError(w, err.Error(), http.StatusInternalServerError)
			return
		}
		p.Payload = res
		s.WriteJson(w, p)
	}
}

// GetArticle
// @Summary  Returns an article based on defined ID.
// @Param    ID  path  string  true  "uuid"
// @Produce  application/json
// @Tags     Articles
// @Router   /articles/{ID} [get]
// @Success  200  {object}  ArticleGetResults  "OK"
func (s *Server) getArticle(w http.ResponseWriter, r *http.Request) {
	p := ArticleGetResults{
		ApiStatusModel: ApiStatusModel{
			Message:    "OK",
			StatusCode: http.StatusOK,
		},
	}

	id := chi.URLParam(r, "ID")
	uuid, err := uuid.Parse(id)
	if err != nil {
		s.WriteError(w, err.Error(), http.StatusBadRequest)
		return
	}

	res, err := s.dto.GetArticle(r.Context(), uuid)
	if err != nil {
		s.WriteError(w, err.Error(), http.StatusInternalServerError)
		return
	}

	p.Payload = res

	s.WriteJson(w, p)
}

// GetArticleDetails
// @Summary  Returns an article and source based on defined ID.
// @Param    ID  path  string  true  "uuid"
// @Produce  application/json
// @Tags     Articles
// @Router   /articles/{ID}/details [get]
// @Success  200  {object}  ArticleDetailsResult  "OK"
func (s *Server) getArticleDetails(w http.ResponseWriter, r *http.Request) {
	p := ArticleDetailsResult{
		ApiStatusModel: ApiStatusModel{
			Message:    "OK",
			StatusCode: http.StatusOK,
		},
	}

	id := chi.URLParam(r, "ID")
	uuid, err := uuid.Parse(id)
	if err != nil {
		s.WriteError(w, err.Error(), http.StatusBadRequest)
		return
	}

	res, err := s.dto.GetArticleDetails(r.Context(), uuid)
	if err != nil {
		s.WriteError(w, err.Error(), http.StatusInternalServerError)
		return
	}

	p.Payload = res

	s.WriteJson(w, p)
}

// TODO add page support
// GetArticlesBySourceID
// @Summary  Finds the articles based on the SourceID provided.  Returns the top 50.
// @Param    id  query  string  true  "Source ID UUID"
// @Produce  application/json
// @Tags     Articles
// @Router   /articles/by/sourceid [get]
// @Success  200  {object}  ArticlesListResults  "OK"
func (s *Server) GetArticlesBySourceId(w http.ResponseWriter, r *http.Request) {
	p := ArticlesListResults{
		ApiStatusModel: ApiStatusModel{
			Message:    "OK",
			StatusCode: http.StatusOK,
		},
	}

	r.URL.Query()
	query := r.URL.Query()
	_id := query["id"][0]

	uuid, err := uuid.Parse(_id)
	if err != nil {
		s.WriteError(w, err.Error(), http.StatusBadRequest)
		return
	}

	res, err := s.dto.GetArticlesBySourceId(r.Context(), uuid)
	if err != nil {
		s.WriteError(w, err.Error(), http.StatusInternalServerError)
		return
	}
	p.Payload = res

	s.WriteJson(w, p)
}