feat: add multi folder support

closes #50

cmd/gonic/gonic.go

@@ -30,7 +30,6 @@ const (
 func main() {
 	set := flag.NewFlagSet(gonic.Name, flag.ExitOnError)
 	confListenAddr := set.String("listen-addr", "0.0.0.0:4747", "listen address (optional)")
-	confMusicPath := set.String("music-path", "", "path to music")
 	confPodcastPath := set.String("podcast-path", "", "path to podcasts")
 	confCachePath := set.String("cache-path", "", "path to cache")
 	confDBPath := set.String("db-path", "gonic.db", "path to database (optional)")
@@ -40,6 +39,10 @@ func main() {
 	confGenreSplit := set.String("genre-split", "\n", "character or string to split genre tag data on (optional)")
 	confHTTPLog := set.Bool("http-log", true, "http request logging (optional)")
 	confShowVersion := set.Bool("version", false, "show gonic version")
+
+	var confMusicPaths musicPaths
+	set.Var(&confMusicPaths, "music-path", "path to music")
+
 	_ = set.String("config-path", "", "path to config (optional)")
 
 	if err := ff.Parse(set, os.Args[1:],
@@ -62,8 +65,13 @@ func main() {
 		log.Printf("    %-15s %s\n", f.Name, value)
 	})
 
-	if _, err := os.Stat(*confMusicPath); os.IsNotExist(err) {
-		log.Fatal("please provide a valid music directory")
+	if len(confMusicPaths) == 0 {
+		log.Fatalf("please provide a music directory")
+	}
+	for _, confMusicPath := range confMusicPaths {
+		if _, err := os.Stat(confMusicPath); os.IsNotExist(err) {
+			log.Fatalf("music directory %q not found", confMusicPath)
+		}
 	}
 	if _, err := os.Stat(*confPodcastPath); os.IsNotExist(err) {
 		log.Fatal("please provide a valid podcast directory")
@@ -90,13 +98,20 @@ func main() {
 	if err != nil {
 		log.Fatalf("error opening database: %v\n", err)
 	}
-	defer db.Close()
+	defer dbc.Close()
+
+	err = dbc.Migrate(db.MigrationContext{
+		OriginalMusicPath: confMusicPaths[0],
+	})
+	if err != nil {
+		log.Panicf("error migrating database: %v\n", err)
+	}
 
 	proxyPrefixExpr := regexp.MustCompile(`^\/*(.*?)\/*$`)
 	*confProxyPrefix = proxyPrefixExpr.ReplaceAllString(*confProxyPrefix, `/$1`)
 	server, err := server.New(server.Options{
-		DB:             db,
-		MusicPath:      *confMusicPath,
+		DB:             dbc,
+		MusicPaths:     confMusicPaths,
 		CachePath:      cacheDirAudio,
 		CoverCachePath: cacheDirCovers,
 		ProxyPrefix:    *confProxyPrefix,
@@ -125,3 +140,14 @@ func main() {
 		log.Panicf("error in job: %v", err)
 	}
 }
+
+type musicPaths []string
+
+func (m musicPaths) String() string {
+	return strings.Join(m, ", ")
+}
+
+func (m *musicPaths) Set(value string) error {
+	*m = append(*m, value)
+	return nil
+}

server/ctrladmin/handlers_playlist.go

@@ -18,16 +18,16 @@ var (
 	errPlaylistNoMatch = errors.New("couldn't match track")
 )
 
-func playlistParseLine(c *Controller, path string) (int, error) {
-	if strings.HasPrefix(path, "#") || strings.TrimSpace(path) == "" {
+func playlistParseLine(c *Controller, absPath string) (int, error) {
+	if strings.HasPrefix(absPath, "#") || strings.TrimSpace(absPath) == "" {
 		return 0, nil
 	}
 	var track db.Track
 	query := c.DB.Raw(`
 		SELECT tracks.id FROM TRACKS
 		JOIN albums ON tracks.album_id=albums.id
-		WHERE (? || '/' || albums.left_path || albums.right_path || '/' || tracks.filename)=?`,
-		c.MusicPath, path)
+		WHERE (albums.root_dir || '/' || albums.left_path || albums.right_path || '/' || tracks.filename)=?`,
+		absPath)
 	err := query.First(&track).Error
 	switch {
 	case errors.Is(err, gorm.ErrRecordNotFound):

server/ctrlbase/ctrl.go

@@ -46,7 +46,6 @@ func statusToBlock(code int) string {
 
 type Controller struct {
 	DB          *db.DB
-	MusicPath   string
 	Scanner     *scanner.Scanner
 	ProxyPrefix string
 }

server/ctrlsubsonic/ctrl_test.go

@@ -55,15 +55,14 @@ func makeHTTPMock(query url.Values) (*httptest.ResponseRecorder, *http.Request)
 func runQueryCases(t *testing.T, contr *Controller, h handlerSubsonic, cases []*queryCase) {
 	t.Helper()
 	for _, qc := range cases {
-		qc := qc // pin
 		t.Run(qc.expectPath, func(t *testing.T) {
-			t.Parallel()
 			rr, req := makeHTTPMock(qc.params)
 			contr.H(h).ServeHTTP(rr, req)
 			body := rr.Body.String()
 			if status := rr.Code; status != http.StatusOK {
 				t.Fatalf("didn't give a 200\n%s", body)
 			}
+
 			goldenPath := makeGoldenPath(t.Name())
 			goldenRegen := os.Getenv("GONIC_REGEN")
 			if goldenRegen == "*" || (goldenRegen != "" && strings.HasPrefix(t.Name(), goldenRegen)) {
@@ -86,11 +85,10 @@ func runQueryCases(t *testing.T, contr *Controller, h handlerSubsonic, cases []*
 				diffOpts = append(diffOpts, jd.SET)
 			}
 			diff := expected.Diff(actual, diffOpts...)
-			// pass or fail
-			if len(diff) == 0 {
-				return
+
+			if len(diff) > 0 {
+				t.Errorf("\u001b[31;1mdiffering json\u001b[0m\n%s", diff.Render())
 			}
-			t.Errorf("\u001b[31;1mdiffering json\u001b[0m\n%s", diff.Render())
 		})
 	}
 }

server/ctrlsubsonic/handlers_by_folder.go

@@ -12,36 +12,43 @@ import (
 	"go.senan.xyz/gonic/server/db"
 )
 
-// the subsonic spec metions "artist" a lot when talking about the
+// the subsonic spec mentions "artist" a lot when talking about the
 // browse by folder endpoints. but since we're not browsing by tag
 // we can't access artists. so instead we'll consider the artist of
 // an track to be the it's respective folder that comes directly
 // under the root directory
 
 func (c *Controller) ServeGetIndexes(r *http.Request) *spec.Response {
+	params := r.Context().Value(CtxParams).(params.Params)
+	rootQ := c.DB.
+		Select("id").
+		Model(&db.Album{}).
+		Where("parent_id IS NULL")
+	if m, _ := params.Get("musicFolderId"); m != "" {
+		rootQ = rootQ.
+			Where("root_dir=?", m)
+	}
 	var folders []*db.Album
 	c.DB.
 		Select("*, count(sub.id) child_count").
 		Joins("LEFT JOIN albums sub ON albums.id=sub.parent_id").
-		Where("albums.parent_id=1").
+		Where("albums.parent_id IN ?", rootQ.SubQuery()).
 		Group("albums.id").
 		Order("albums.right_path COLLATE NOCASE").
 		Find(&folders)
 	// [a-z#] -> 27
 	indexMap := make(map[string]*spec.Index, 27)
 	resp := make([]*spec.Index, 0, 27)
 	for _, folder := range folders {
-		i := lowerUDecOrHash(folder.IndexRightPath())
-		index, ok := indexMap[i]
-		if !ok {
-			index = &spec.Index{
-				Name:    i,
+		key := lowerUDecOrHash(folder.IndexRightPath())
+		if _, ok := indexMap[key]; !ok {
+			indexMap[key] = &spec.Index{
+				Name:    key,
 				Artists: []*spec.Artist{},
 			}
-			indexMap[i] = index
-			resp = append(resp, index)
+			resp = append(resp, indexMap[key])
 		}
-		index.Artists = append(index.Artists,
+		indexMap[key].Artists = append(indexMap[key].Artists,
 			spec.NewArtistByFolder(folder))
 	}
 	sub := spec.NewResponse()
@@ -137,6 +144,10 @@ func (c *Controller) ServeGetAlbumList(r *http.Request) *spec.Response {
 	default:
 		return spec.NewError(10, "unknown value `%s` for parameter 'type'", v)
 	}
+
+	if m, _ := params.Get("musicFolderId"); m != "" {
+		q = q.Where("root_dir=?", m)
+	}
 	var folders []*db.Album
 	// TODO: think about removing this extra join to count number
 	// of children. it might make sense to store that in the db
@@ -166,50 +177,65 @@ func (c *Controller) ServeSearchTwo(r *http.Request) *spec.Response {
 		return spec.NewError(10, "please provide a `query` parameter")
 	}
 	query = fmt.Sprintf("%%%s%%", strings.TrimSuffix(query, "*"))
+
 	results := &spec.SearchResultTwo{}
+
 	// search "artists"
+	rootQ := c.DB.
+		Select("id").
+		Model(&db.Album{}).
+		Where("parent_id IS NULL")
+	if m, _ := params.Get("musicFolderId"); m != "" {
+		rootQ = rootQ.Where("root_dir=?", m)
+	}
+
 	var artists []*db.Album
-	c.DB.
-		Where(`
-			parent_id=1
-			AND (	right_path LIKE ? OR
-					right_path_u_dec LIKE ?	)`,
-			query, query).
+	q := c.DB.
+		Where(`parent_id IN ? AND (right_path LIKE ? OR right_path_u_dec LIKE ?)`, rootQ.SubQuery(), query, query).
 		Offset(params.GetOrInt("artistOffset", 0)).
-		Limit(params.GetOrInt("artistCount", 20)).
-		Find(&artists)
+		Limit(params.GetOrInt("artistCount", 20))
+	if err := q.Find(&artists).Error; err != nil {
+		return spec.NewError(0, "find artists: %v", err)
+	}
 	for _, a := range artists {
-		results.Artists = append(results.Artists,
-			spec.NewDirectoryByFolder(a, nil))
+		results.Artists = append(results.Artists, spec.NewDirectoryByFolder(a, nil))
 	}
+
 	// search "albums"
 	var albums []*db.Album
-	c.DB.
-		Where(`
-			tag_artist_id IS NOT NULL
-			AND (	right_path LIKE ? OR
-					right_path_u_dec LIKE ?	)`,
-			query, query).
+	q = c.DB.
+		Where(`tag_artist_id IS NOT NULL AND (right_path LIKE ? OR right_path_u_dec LIKE ?)`, query, query).
 		Offset(params.GetOrInt("albumOffset", 0)).
-		Limit(params.GetOrInt("albumCount", 20)).
-		Find(&albums)
+		Limit(params.GetOrInt("albumCount", 20))
+	if m, _ := params.Get("musicFolderId"); m != "" {
+		q = q.Where("root_dir=?", m)
+	}
+	if err := q.Find(&albums).Error; err != nil {
+		return spec.NewError(0, "find albums: %v", err)
+	}
 	for _, a := range albums {
 		results.Albums = append(results.Albums, spec.NewTCAlbumByFolder(a))
 	}
+
 	// search tracks
 	var tracks []*db.Track
-	c.DB.
+	q = c.DB.
 		Preload("Album").
-		Where("filename LIKE ? OR filename_u_dec LIKE ?",
-			query, query).
+		Where("filename LIKE ? OR filename_u_dec LIKE ?", query, query).
 		Offset(params.GetOrInt("songOffset", 0)).
-		Limit(params.GetOrInt("songCount", 20)).
-		Find(&tracks)
+		Limit(params.GetOrInt("songCount", 20))
+	if m, _ := params.Get("musicFolderId"); m != "" {
+		q = q.
+			Joins("JOIN albums ON albums.id=tracks.album_id").
+			Where("albums.root_dir=?", m)
+	}
+	if err := q.Find(&tracks).Error; err != nil {
+		return spec.NewError(0, "find tracks: %v", err)
+	}
 	for _, t := range tracks {
-		results.Tracks = append(results.Tracks,
-			spec.NewTCTrackByFolder(t, t.Album))
+		results.Tracks = append(results.Tracks, spec.NewTCTrackByFolder(t, t.Album))
 	}
-	//
+
 	sub := spec.NewResponse()
 	sub.SearchResultTwo = results
 	return sub

server/ctrlsubsonic/handlers_by_folder_test.go

@@ -2,6 +2,7 @@ package ctrlsubsonic
 
 import (
 	"net/url"
+	"path/filepath"
 	"testing"
 
 	_ "github.com/jinzhu/gorm/dialects/sqlite"
@@ -13,6 +14,8 @@ func TestGetIndexes(t *testing.T) {
 
 	runQueryCases(t, contr, contr.ServeGetIndexes, []*queryCase{
 		{url.Values{}, "no_args", false},
+		{url.Values{"musicFolderId": {filepath.Join(m.TmpDir(), "m-0")}}, "with_music_folder_1", false},
+		{url.Values{"musicFolderId": {filepath.Join(m.TmpDir(), "m-1")}}, "with_music_folder_2", false},
 	})
 }