implemented getStarred (just for albums, for now)

2016-03-14 11:35:48 -04:00
parent 14934dccf7
commit e36a4201db
9 changed files with 95 additions and 13 deletions
@@ -0,0 +1,90 @@
+package api
+
+import (
+	"time"
+
+	"github.com/astaxie/beego"
+	"github.com/deluan/gosonic/api/responses"
+	"github.com/deluan/gosonic/domain"
+	"github.com/deluan/gosonic/engine"
+	"github.com/deluan/gosonic/utils"
+)
+
+type AlbumListController struct {
+	BaseAPIController
+	listGen engine.ListGenerator
+	types   map[string]strategy
+}
+
+type strategy func(offset int, size int) (*domain.Albums, error)
+
+func (c *AlbumListController) Prepare() {
+	utils.ResolveDependencies(&c.listGen)
+
+	c.types = map[string]strategy{
+		"random":   func(o int, s int) (*domain.Albums, error) { return c.listGen.GetRandom(o, s) },
+		"newest":   func(o int, s int) (*domain.Albums, error) { return c.listGen.GetNewest(o, s) },
+		"recent":   func(o int, s int) (*domain.Albums, error) { return c.listGen.GetRecent(o, s) },
+		"frequent": func(o int, s int) (*domain.Albums, error) { return c.listGen.GetFrequent(o, s) },
+		"highest":  func(o int, s int) (*domain.Albums, error) { return c.listGen.GetHighest(o, s) },
+	}
+}
+
+func (c *AlbumListController) GetAlbumList() {
+	typ := c.RequiredParamString("type", "Required string parameter 'type' is not present")
+	method, found := c.types[typ]
+
+	if !found {
+		beego.Error("albumList type", typ, "not implemented!")
+		c.SendError(responses.ERROR_GENERIC, "Not implemented!")
+	}
+
+	offset := c.ParamInt("offset", 0)
+	size := utils.MinInt(c.ParamInt("size", 0), 500)
+
+	albums, err := method(offset, size)
+	if err != nil {
+		beego.Error("Error retrieving albums:", err)
+		c.SendError(responses.ERROR_GENERIC, "Internal Error")
+	}
+
+	albumList := make([]responses.Child, len(*albums))
+
+	for i, al := range *albums {
+		albumList[i].Id = al.Id
+		albumList[i].Title = al.Name
+		albumList[i].Parent = al.ArtistId
+		albumList[i].IsDir = true
+		albumList[i].Album = al.Name
+		albumList[i].Year = al.Year
+		albumList[i].Artist = al.Artist
+		albumList[i].Genre = al.Genre
+		albumList[i].CoverArt = al.CoverArtId
+		if al.Starred {
+			t := time.Now()
+			albumList[i].Starred = &t
+		}
+	}
+
+	response := c.NewEmpty()
+	response.AlbumList = &responses.AlbumList{Album: albumList}
+	c.SendResponse(response)
+}
+
+func (c *AlbumListController) GetStarred() {
+	albums, err := c.listGen.GetStarred()
+	if err != nil {
+		beego.Error("Error retrieving starred albums:", err)
+		c.SendError(responses.ERROR_GENERIC, "Internal Error")
+	}
+
+	response := c.NewEmpty()
+	response.Starred = &responses.Starred{}
+	response.Starred.Album = make([]responses.Child, len(*albums))
+
+	for i, entry := range *albums {
+		response.Starred.Album[i] = c.ToChild(entry)
+	}
+
+	c.SendResponse(response)
+}