a301_server/internal/download/handler.go

package download

import (
	"log"
	"mime"
	"os"
	"path/filepath"
	"strings"

	"github.com/gofiber/fiber/v2"
)

type Handler struct {
	svc     *Service
	baseURL string
}

func NewHandler(svc *Service, baseURL string) *Handler {
	return &Handler{svc: svc, baseURL: baseURL}
}

// GetInfo godoc
// @Summary      다운로드 정보 조회
// @Description  게임 및 런처 다운로드 정보를 조회합니다
// @Tags         Download
// @Produce      json
// @Success      200  {object}  docs.DownloadInfoResponse
// @Failure      404  {object}  docs.ErrorResponse
// @Router       /api/download/info [get]
func (h *Handler) GetInfo(c *fiber.Ctx) error {
	info, err := h.svc.GetInfo()
	if err != nil {
		return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "다운로드 정보가 없습니다"})
	}
	return c.JSON(info)
}

// Upload godoc
// @Summary      게임 파일 업로드 (관리자)
// @Description  게임 zip 파일을 스트리밍 업로드합니다. Body는 raw binary입니다.
// @Tags         Download
// @Accept       application/octet-stream
// @Produce      json
// @Security     BearerAuth
// @Param        filename  query     string  false  "파일명"  default(game.zip)
// @Success      200       {object}  docs.DownloadInfoResponse
// @Failure      400       {object}  docs.ErrorResponse
// @Failure      401       {object}  docs.ErrorResponse
// @Failure      403       {object}  docs.ErrorResponse
// @Failure      500       {object}  docs.ErrorResponse
// @Router       /api/download/upload/game [post]
func (h *Handler) Upload(c *fiber.Ctx) error {
	filename := strings.TrimSpace(c.Query("filename", "game.zip"))
	// 경로 순회 방지: 디렉토리 구분자 제거, 기본 파일명만 사용
	filename = filepath.Base(filename)
	if !strings.HasSuffix(strings.ToLower(filename), ".zip") {
		return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "zip 파일만 업로드 가능합니다"})
	}
	if len(filename) > 200 {
		return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "파일명이 너무 깁니다"})
	}

	body := c.Request().BodyStream()
	info, err := h.svc.Upload(filename, body, h.baseURL)
	if err != nil {
		log.Printf("game upload failed: %v", err)
		return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "게임 파일 업로드에 실패했습니다"})
	}
	return c.JSON(info)
}

// ServeFile godoc
// @Summary      게임 파일 다운로드
// @Description  게임 zip 파일을 다운로드합니다
// @Tags         Download
// @Produce      application/octet-stream
// @Success      200  {file}    binary
// @Failure      404  {object}  docs.ErrorResponse
// @Router       /api/download/file [get]
func (h *Handler) ServeFile(c *fiber.Ctx) error {
	path := h.svc.GameFilePath()
	if _, err := os.Stat(path); err != nil {
		return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "파일이 없습니다"})
	}
	info, _ := h.svc.GetInfo()
	filename := "game.zip"
	if info != nil && info.FileName != "" {
		filename = info.FileName
	}
	c.Set("Content-Disposition", mime.FormatMediaType("attachment", map[string]string{"filename": filename}))
	return c.SendFile(path)
}

// UploadLauncher godoc
// @Summary      런처 업로드 (관리자)
// @Description  런처 실행 파일을 스트리밍 업로드합니다. Body는 raw binary입니다.
// @Tags         Download
// @Accept       application/octet-stream
// @Produce      json
// @Security     BearerAuth
// @Success      200  {object}  docs.DownloadInfoResponse
// @Failure      401  {object}  docs.ErrorResponse
// @Failure      403  {object}  docs.ErrorResponse
// @Failure      500  {object}  docs.ErrorResponse
// @Router       /api/download/upload/launcher [post]
func (h *Handler) UploadLauncher(c *fiber.Ctx) error {
	body := c.Request().BodyStream()
	info, err := h.svc.UploadLauncher(body, h.baseURL)
	if err != nil {
		log.Printf("launcher upload failed: %v", err)
		return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{"error": "런처 업로드에 실패했습니다"})
	}
	return c.JSON(info)
}

// ServeLauncher godoc
// @Summary      런처 다운로드
// @Description  런처 실행 파일을 다운로드합니다
// @Tags         Download
// @Produce      application/octet-stream
// @Success      200  {file}    binary
// @Failure      404  {object}  docs.ErrorResponse
// @Router       /api/download/launcher [get]
func (h *Handler) ServeLauncher(c *fiber.Ctx) error {
	path := h.svc.LauncherFilePath()
	if _, err := os.Stat(path); err != nil {
		return c.Status(fiber.StatusNotFound).JSON(fiber.Map{"error": "파일이 없습니다"})
	}
	c.Set("Content-Disposition", `attachment; filename="launcher.exe"`)
	return c.SendFile(path)
}