reveil/api/gongs.go

package api

import (
	"fmt"
	"net/http"
	"os"
	"path"
	"strings"

	"github.com/gin-gonic/gin"

	"git.nemunai.re/nemunaire/reveil/config"
	"git.nemunai.re/nemunaire/reveil/model"
)

func declareGongsRoutes(cfg *config.Config, router *gin.RouterGroup) {
	router.GET("/gongs", func(c *gin.Context) {
		gongs, err := reveil.LoadGongs(cfg)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
			return
		}

		c.JSON(http.StatusOK, gongs)
	})
	router.POST("/gongs", func(c *gin.Context) {
		fgong, err := c.FormFile("gongfile")
		if err != nil {
			c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"errmsg": "No gong found"})
			return
		}

		// Check file extension
		if path.Ext(fgong.Filename) != ".mp3" && path.Ext(fgong.Filename) != ".flac" && path.Ext(fgong.Filename) != ".wav" {
			c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"errmsg": "Bad file type. You should only upload .mp3, .wav or .flac files."})
			return
		}

		if strings.Contains(fgong.Filename, "/") {
			c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"errmsg": "Bad file name."})
			return
		}

		dst := path.Join(cfg.GongsDir, fgong.Filename)
		err = c.SaveUploadedFile(fgong, dst)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("Something goes wrong when saving the gong: %s", err.Error())})
			return
		}

		d, err := os.Stat(dst)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("Something goes wrong when saving the gong: %s", err.Error())})
			return
		}

		gong, err := reveil.LoadGong(cfg, dst, d)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("Unable to load gong: %s", err.Error())})
			return
		}

		c.JSON(http.StatusOK, gong)
	})

	gongsRoutes := router.Group("/gongs/:tid")
	gongsRoutes.Use(func(c *gin.Context) {
		gongs, err := reveil.LoadGongs(cfg)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": err.Error()})
			return
		}

		for _, g := range gongs {
			if g.Id.ToString() == c.Param("tid") {
				c.Set("gong", g)
				c.Next()
				return
			}
		}

		c.AbortWithStatusJSON(http.StatusNotFound, gin.H{"errmsg": "Gong not found"})
	})

	gongsRoutes.GET("", func(c *gin.Context) {
		c.JSON(http.StatusOK, c.MustGet("gong"))
	})
	gongsRoutes.GET("/stream", func(c *gin.Context) {
		gong := c.MustGet("gong").(*reveil.Gong)

		size, err := gong.Size()
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("Unable to open the gong: %s", err.Error())})
			return
		}

		fd, err := gong.Open()
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("Unable to open the gong: %s", err.Error())})
			return
		}
		defer fd.Close()

		c.DataFromReader(http.StatusOK, size, gong.ContentType(), fd, map[string]string{})
	})
	gongsRoutes.PUT("", func(c *gin.Context) {
		oldgong := c.MustGet("gong").(*reveil.Gong)

		var gong reveil.Gong
		if err := c.ShouldBindJSON(&gong); err != nil {
			c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"errmsg": err.Error()})
			return
		}

		if gong.Name != oldgong.Name {
			err := oldgong.Rename(gong.Name)
			if err != nil {
				c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("Unable to rename the gong: %s", err.Error())})
				return
			}
		}

		if gong.Enabled != oldgong.Enabled {
			var err error
			if gong.Enabled {
				err = oldgong.SetDefault(cfg)
			}

			if err != nil {
				c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("Unable to set the new default gong: %s", err.Error())})
				return
			}
		}

		c.JSON(http.StatusOK, oldgong)
	})
	gongsRoutes.DELETE("", func(c *gin.Context) {
		gong := c.MustGet("gong").(*reveil.Gong)

		err := gong.Remove()
		if err != nil {
			c.AbortWithStatusJSON(http.StatusInternalServerError, gin.H{"errmsg": fmt.Sprintf("Unable to remove the gong: %s", err.Error())})
			return
		}

		c.JSON(http.StatusOK, nil)
	})
}