Merge pull request #119 from Quaver/download-mapset-head

Support head method on download mapset

Author: Swan

cmd/api/server.go

@@ -171,9 +171,9 @@ func initializeRoutes(engine *gin.Engine) {
 	// Download
 	engine.GET("/v2/download/map/:id", handlers.CreateHandler(handlers.DownloadQua))
 	engine.GET("/v2/download/replay/:id", handlers.CreateHandler(handlers.DownloadReplay))
-	engine.GET("/v2/download/mapset/:id", middleware.RequireAuth, handlers.CreateHandler(handlers.DownloadMapset))
+	engine.Match([]string{"GET", "HEAD"}, "/v2/download/mapset/:id", middleware.RequireAuth, handlers.CreateHandler(handlers.DownloadMapset))
 	engine.POST("/v2/download/multiplayer/:id/upload", middleware.RequireAuth, handlers.CreateHandler(handlers.UploadMultiplayerMapset))
-	engine.GET("/v2/download/multiplayer/:id", middleware.RequireAuth, handlers.CreateHandler(handlers.DownloadMultiplayerMapset))
+	engine.Match([]string{"GET", "HEAD"}, "/v2/download/multiplayer/:id", middleware.RequireAuth, handlers.CreateHandler(handlers.DownloadMultiplayerMapset))
 
 	// Logs
 	engine.POST("/v2/logs/crash", middleware.RequireAuth, handlers.CreateHandler(handlers.AddCrashLog))

handlers/download.go

@@ -75,6 +75,10 @@ func DownloadMapset(c *gin.Context) *APIError {
 		return APIErrorNotFound("Mapset")
 	}
 
+	if isHeadRequest(c) {
+		return setFileContentLength(c, path)
+	}
+
 	if err := db.InsertMapsetDownload(&db.MapsetDownload{
 		UserId:    user.Id,
 		MapsetId:  mapset.Id,
@@ -154,6 +158,29 @@ func DownloadMultiplayerMapset(c *gin.Context) *APIError {
 		return APIErrorNotFound("Multiplayer Mapset")
 	}
 
+	if isHeadRequest(c) {
+		return setFileContentLength(c, path)
+	}
+
 	c.FileAttachment(path, fmt.Sprintf("multiplayer_%v.qp", id))
 	return nil
 }
+
+// Check if request is HEAD
+func isHeadRequest(c *gin.Context) bool {
+	return c.Request.Method == "HEAD"
+}
+
+func setFileContentLength(c *gin.Context, path string) *APIError {
+	fileInfo, err := os.Stat(path)
+
+	if err != nil {
+		return APIErrorServerError("Error getting file information", err)
+	}
+
+	// Set Content-Length header
+	c.Header("Content-Length", strconv.FormatInt(fileInfo.Size(), 10))
+	return nil
+}
+
+