package ctrlsubsonic import ( "fmt" "log" "net/http" "sort" "strconv" "time" "unicode" "github.com/jinzhu/gorm" "senan.xyz/g/gonic/db" "senan.xyz/g/gonic/scanner" "senan.xyz/g/gonic/server/ctrlsubsonic/params" "senan.xyz/g/gonic/server/ctrlsubsonic/spec" "senan.xyz/g/gonic/server/lastfm" ) func lowerUDecOrHash(in string) string { lower := unicode.ToLower(rune(in[0])) if !unicode.IsLetter(lower) { return "#" } return string(lower) } func (c *Controller) ServeGetLicence(r *http.Request) *spec.Response { sub := spec.NewResponse() sub.Licence = &spec.Licence{ Valid: true, } return sub } func (c *Controller) ServePing(r *http.Request) *spec.Response { return spec.NewResponse() } func (c *Controller) ServeScrobble(r *http.Request) *spec.Response { params := r.Context().Value(CtxParams).(params.Params) id, err := params.GetInt("id") if err != nil { return spec.NewError(10, "please provide an `id` parameter") } // fetch user to get lastfm session user := r.Context().Value(CtxUser).(*db.User) if user.LastFMSession == "" { return spec.NewError(0, "you don't have a last.fm session") } // fetch track for getting info to send to last.fm function track := &db.Track{} c.DB. Preload("Album"). Preload("Artist"). First(track, id) // scrobble with above info opts := lastfm.ScrobbleOpts{ Track: track, // clients will provide time in miliseconds, so use that or // instead convert UnixNano to miliseconds StampMili: params.GetIntOr("time", int(time.Now().UnixNano()/1e6)), Submission: params.GetOr("submission", "true") != "false", } err = lastfm.Scrobble( c.DB.GetSetting("lastfm_api_key"), c.DB.GetSetting("lastfm_secret"), user.LastFMSession, opts, ) if err != nil { return spec.NewError(0, "error when submitting: %v", err) } return spec.NewResponse() } func (c *Controller) ServeGetMusicFolders(r *http.Request) *spec.Response { folders := &spec.MusicFolders{} folders.List = []*spec.MusicFolder{ {ID: 1, Name: "music"}, } sub := spec.NewResponse() sub.MusicFolders = folders return sub } func (c *Controller) ServeStartScan(r *http.Request) *spec.Response { go func() { if err := c.Scanner.Start(); err != nil { log.Printf("error while scanning: %v\n", err) } }() return c.ServeGetScanStatus(r) } func (c *Controller) ServeGetScanStatus(r *http.Request) *spec.Response { var trackCount int c.DB. Model(db.Track{}). Count(&trackCount) sub := spec.NewResponse() sub.ScanStatus = &spec.ScanStatus{ Scanning: scanner.IsScanning(), Count: trackCount, } return sub } func (c *Controller) ServeGetUser(r *http.Request) *spec.Response { user := r.Context().Value(CtxUser).(*db.User) sub := spec.NewResponse() sub.User = &spec.User{ Username: user.Name, AdminRole: user.IsAdmin, ScrobblingEnabled: user.LastFMSession != "", Folder: []int{1}, } return sub } func (c *Controller) ServeNotFound(r *http.Request) *spec.Response { return spec.NewError(70, "view not found") } func (c *Controller) ServeGetPlaylists(r *http.Request) *spec.Response { user := r.Context().Value(CtxUser).(*db.User) var playlists []*db.Playlist c.DB.Where("user_id=?", user.ID).Find(&playlists) sub := spec.NewResponse() sub.Playlists = &spec.Playlists{ List: make([]*spec.Playlist, len(playlists)), } fmt.Println("aaaa") for i, playlist := range playlists { sub.Playlists.List[i] = spec.NewPlaylist(playlist) sub.Playlists.List[i].Owner = user.Name sub.Playlists.List[i].SongCount = playlist.TrackCount } return sub } func (c *Controller) ServeGetPlaylist(r *http.Request) *spec.Response { params := r.Context().Value(CtxParams).(params.Params) playlistID, err := params.GetInt("id") if err != nil { return spec.NewError(10, "please provide an `id` parameter") } playlist := db.Playlist{} err = c.DB. Where("id=?", playlistID). Find(&playlist). Error if gorm.IsRecordNotFoundError(err) { return spec.NewError(70, "playlist with id `%d` not found", playlistID) } user := r.Context().Value(CtxUser).(*db.User) sub := spec.NewResponse() sub.Playlist = spec.NewPlaylist(&playlist) sub.Playlist.Owner = user.Name sub.Playlist.SongCount = playlist.TrackCount trackIDs := playlist.GetItems() sub.Playlist.List = make([]*spec.TrackChild, len(trackIDs)) for i, id := range trackIDs { track := db.Track{} c.DB. Where("id=?", id). Preload("Album"). Find(&track) sub.Playlist.List[i] = spec.NewTCTrackByFolder(&track, track.Album) } return sub } func (c *Controller) ServeUpdatePlaylist(r *http.Request) *spec.Response { user := r.Context().Value(CtxUser).(*db.User) params := r.Context().Value(CtxParams).(params.Params) var playlistID int if p := params.GetFirstList("id", "playlistId"); p != nil { playlistID, _ = strconv.Atoi(p[0]) } // playlistID may be 0 from above. in that case we get a new playlist // as intended var playlist db.Playlist c.DB. Where("id=?", playlistID). FirstOrCreate(&playlist) // ** begin update meta info playlist.UserID = user.ID if val := params.Get("name"); val != "" { playlist.Name = val } if val := params.Get("comment"); val != "" { playlist.Comment = val } trackIDs := playlist.GetItems() // ** begin delete items if p := params.GetFirstListInt("songIndexToRemove"); p != nil { sort.Sort(sort.Reverse(sort.IntSlice(p))) for _, i := range p { trackIDs = append(trackIDs[:i], trackIDs[i+1:]...) } } // ** begin add items if p := params.GetFirstListInt("songId", "songIdToAdd"); p != nil { trackIDs = append(trackIDs, p...) } // playlist.SetItems(trackIDs) c.DB.Save(playlist) return spec.NewResponse() } func (c *Controller) ServeDeletePlaylist(r *http.Request) *spec.Response { params := r.Context().Value(CtxParams).(params.Params) c.DB. Where("id=?", params.GetIntOr("id", 0)). Delete(&db.Playlist{}) return spec.NewResponse() } func (c *Controller) ServeGetPlayQueue(r *http.Request) *spec.Response { user := r.Context().Value(CtxUser).(*db.User) queue := db.PlayQueue{} err := c.DB. Where("user_id=?", user.ID). Find(&queue). Error if gorm.IsRecordNotFoundError(err) { return spec.NewResponse() } sub := spec.NewResponse() sub.PlayQueue = &spec.PlayQueue{} sub.PlayQueue.Username = user.Name sub.PlayQueue.Position = queue.Position sub.PlayQueue.Current = queue.Current sub.PlayQueue.Changed = queue.UpdatedAt sub.PlayQueue.ChangedBy = queue.ChangedBy trackIDs := queue.GetItems() sub.PlayQueue.List = make([]*spec.TrackChild, len(trackIDs)) for i, id := range trackIDs { track := db.Track{} c.DB. Where("id=?", id). Preload("Album"). Find(&track) sub.PlayQueue.List[i] = spec.NewTCTrackByFolder(&track, track.Album) } return sub } func (c *Controller) ServeSavePlayQueue(r *http.Request) *spec.Response { params := r.Context().Value(CtxParams).(params.Params) tracks := params.GetFirstListInt("id") if tracks == nil { return spec.NewError(10, "please provide some `id` parameters") } user := r.Context().Value(CtxUser).(*db.User) queue := &db.PlayQueue{UserID: user.ID} c.DB.Where(queue).First(queue) queue.Current = params.GetIntOr("current", 0) queue.Position = params.GetIntOr("position", 0) queue.ChangedBy = params.Get("c") queue.SetItems(tracks) c.DB.Save(queue) return spec.NewResponse() } func (c *Controller) ServeGetSong(r *http.Request) *spec.Response { params := r.Context().Value(CtxParams).(params.Params) id, err := params.GetInt("id") if err != nil { return spec.NewError(10, "provide an `id` parameter") } track := &db.Track{} err = c.DB. Where("id=?", id). Preload("Album"). First(track). Error if gorm.IsRecordNotFoundError(err) { return spec.NewError(10, "couldn't find a track with that id") } sub := spec.NewResponse() sub.Track = spec.NewTrackByTags(track, track.Album) return sub } func (c *Controller) ServeGetRandomSongs(r *http.Request) *spec.Response { params := r.Context().Value(CtxParams).(params.Params) var tracks []*db.Track q := c.DB.DB.Joins("JOIN albums ON tracks.album_id=albums.id"). Limit(params.GetIntOr("size", 10)). Preload("Album"). Order(gorm.Expr("random()")) if year, err := params.GetInt("fromYear"); err == nil { q = q.Where("albums.tag_year >= ?", year) } if year, err := params.GetInt("toYear"); err == nil { q = q.Where("albums.tag_year <= ?", year) } if genre := params.Get("genre"); genre != "" { q = q.Joins( "JOIN genres ON tracks.tag_genre_id=genres.id AND genres.name=?", genre, ) } q.Find(&tracks) sub := spec.NewResponse() sub.RandomTracks = &spec.RandomTracks{} sub.RandomTracks.List = make([]*spec.TrackChild, len(tracks)) for i, track := range tracks { sub.RandomTracks.List[i] = spec.NewTrackByTags(track, track.Album) } return sub }