mirror of https://github.com/gabehf/Koito.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.2 KiB
88 lines
2.2 KiB
package handlers
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/gabehf/koito/internal/db"
|
|
"github.com/gabehf/koito/internal/logger"
|
|
"github.com/gabehf/koito/internal/models"
|
|
"github.com/gabehf/koito/internal/utils"
|
|
)
|
|
|
|
type SearchResults struct {
|
|
Artists []*models.Artist `json:"artists"`
|
|
Albums []*models.Album `json:"albums"`
|
|
Tracks []*models.Track `json:"tracks"`
|
|
}
|
|
|
|
func SearchHandler(store db.DB) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
ctx := r.Context()
|
|
l := logger.FromContext(ctx)
|
|
q := r.URL.Query().Get("q")
|
|
|
|
l.Debug().Msgf("SearchHandler: Received search with query: %s", r.URL.Query().Encode())
|
|
|
|
var artists []*models.Artist
|
|
var albums []*models.Album
|
|
var tracks []*models.Track
|
|
|
|
if strings.HasPrefix(q, "id:") {
|
|
idStr := strings.TrimPrefix(q, "id:")
|
|
id, _ := strconv.Atoi(idStr)
|
|
|
|
artist, err := store.GetArtist(ctx, db.GetArtistOpts{ID: int32(id)})
|
|
if err != nil {
|
|
l.Debug().Msg("No artists found with id")
|
|
}
|
|
if artist != nil {
|
|
artists = append(artists, artist)
|
|
}
|
|
|
|
album, err := store.GetAlbum(ctx, db.GetAlbumOpts{ID: int32(id)})
|
|
if err != nil {
|
|
l.Debug().Msg("No albums found with id")
|
|
}
|
|
if album != nil {
|
|
albums = append(albums, album)
|
|
}
|
|
|
|
track, err := store.GetTrack(ctx, db.GetTrackOpts{ID: int32(id)})
|
|
if err != nil {
|
|
l.Debug().Msg("No tracks found with id")
|
|
}
|
|
if track != nil {
|
|
tracks = append(tracks, track)
|
|
}
|
|
} else {
|
|
var err error
|
|
artists, err = store.SearchArtists(ctx, q)
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to search for artists")
|
|
utils.WriteError(w, "failed to search in database", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
albums, err = store.SearchAlbums(ctx, q)
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to search for albums")
|
|
utils.WriteError(w, "failed to search in database", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
tracks, err = store.SearchTracks(ctx, q)
|
|
if err != nil {
|
|
l.Err(err).Msg("Failed to search for tracks")
|
|
utils.WriteError(w, "failed to search in database", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
|
|
utils.WriteJSON(w, http.StatusOK, SearchResults{
|
|
Artists: artists,
|
|
Albums: albums,
|
|
Tracks: tracks,
|
|
})
|
|
}
|
|
}
|