mirror of
https://github.com/nprasad2077/NBA_Go.git
synced 2026-09-22 22:15:13 +00:00
97 lines
3.7 KiB
Go
97 lines
3.7 KiB
Go
package controllers
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/nprasad2077/NBA_Go/models"
|
|
"github.com/nprasad2077/NBA_Go/services"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// FetchPlayerShotChartAPI godoc
|
|
// @Summary Fetch a player's shot-chart from NBA-API and store in DB
|
|
// @Tags PlayerShotChart
|
|
// @Param playerId query string true "Player ID (e.g. hardeja01)"
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Success 200 {object} map[string]string
|
|
// @Failure 400,500 {object} map[string]string
|
|
// @Router /api/playershotchart/fetch [get]
|
|
func FetchPlayerShotChartAPI(db *gorm.DB) fiber.Handler {
|
|
return func(c *fiber.Ctx) error {
|
|
pid := c.Query("playerId")
|
|
if pid == "" {
|
|
return c.Status(400).JSON(fiber.Map{"error": "playerId query parameter is required"})
|
|
}
|
|
if err := services.FetchAndStoreShotChartForPlayer(db, pid); err != nil {
|
|
return c.Status(500).JSON(fiber.Map{"error": err.Error()})
|
|
}
|
|
return c.JSON(fiber.Map{"message": "Shot chart for " + pid + " fetched and saved."})
|
|
}
|
|
}
|
|
|
|
// ScrapePlayerShotChart godoc
|
|
// @Summary Scrape a player's shot-chart from BR website
|
|
// @Description Scrapes seasons [startSeason…endSeason] for the given playerId
|
|
// (playerName is auto-detected).
|
|
// @Tags PlayerShotChart
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param playerId query string true "Player ID (e.g. derozde01)"
|
|
// @Param startSeason query int true "Start season (e.g. 2024)"
|
|
// @Param endSeason query int true "End season (e.g. 2021)"
|
|
// @Success 200 {object} map[string]string
|
|
// @Failure 400,500 {object} map[string]string
|
|
// @Router /api/playershotchart/scrape [get]
|
|
func ScrapePlayerShotChart(db *gorm.DB) fiber.Handler {
|
|
return func(c *fiber.Ctx) error {
|
|
pid := c.Query("playerId")
|
|
start := c.QueryInt("startSeason", 0)
|
|
end := c.QueryInt("endSeason", 0)
|
|
if pid == "" || start == 0 || end == 0 {
|
|
return c.Status(400).JSON(fiber.Map{
|
|
"error": "playerId, startSeason and endSeason are required",
|
|
})
|
|
}
|
|
if start < end {
|
|
return c.Status(400).JSON(fiber.Map{
|
|
"error": "startSeason must be >= endSeason",
|
|
})
|
|
}
|
|
if err := services.FetchAndStoreShotChartScrapedForPlayer(db, pid, start, end); err != nil {
|
|
return c.Status(500).JSON(fiber.Map{"error": err.Error()})
|
|
}
|
|
return c.JSON(fiber.Map{"message": "Shot chart scraped and saved for " + pid})
|
|
}
|
|
}
|
|
|
|
// GetPlayerShotChart godoc
|
|
// @Security ApiKeyAuth
|
|
// @Summary Get shot-chart data
|
|
// @Description Returns shot-chart points, optionally filtered by playerId and/or season
|
|
// @Tags PlayerShotChart
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param playerId query string false "Player ID (e.g., hardeja01)"
|
|
// @Param season query int false "Season (e.g., 2023)"
|
|
// @Success 200 {array} models.PlayerShotChart
|
|
// @Failure 500 {object} map[string]string
|
|
// @Router /api/playershotchart [get]
|
|
func GetPlayerShotChart(db *gorm.DB) fiber.Handler {
|
|
return func(c *fiber.Ctx) error {
|
|
var shots []models.PlayerShotChart
|
|
|
|
query := db.Model(&models.PlayerShotChart{})
|
|
|
|
if pid := c.Query("playerId"); pid != "" {
|
|
query = query.Where("player_id = ?", pid)
|
|
}
|
|
if s := c.QueryInt("season", 0); s != 0 {
|
|
query = query.Where("season = ?", s)
|
|
}
|
|
|
|
if err := query.Find(&shots).Error; err != nil {
|
|
return c.Status(500).JSON(fiber.Map{"error": err.Error()})
|
|
}
|
|
return c.JSON(shots)
|
|
}
|
|
} |