Music Pipeline Refactor

internal/project/effective.go

@@ -103,11 +103,19 @@ type EffectiveScriptCompilerEnvConfig struct {
 }
 
 type EffectiveMusicConfig struct {
-	Prefixes       map[string]string `json:"prefixes" yaml:"prefixes"`
-	StageRoot      string            `json:"stage_root" yaml:"stage_root"`
-	CreditsRoot    string            `json:"credits_root" yaml:"credits_root"`
-	CreditsOverlay string            `json:"credits_overlay" yaml:"credits_overlay"`
-	MaxStemLength  int               `json:"max_stem_length" yaml:"max_stem_length"`
+	Prefixes       map[string]string                `json:"prefixes" yaml:"prefixes"`
+	StageRoot      string                           `json:"stage_root" yaml:"stage_root"`
+	CreditsRoot    string                           `json:"credits_root" yaml:"credits_root"`
+	CreditsOverlay string                           `json:"credits_overlay" yaml:"credits_overlay"`
+	MaxStemLength  int                              `json:"max_stem_length" yaml:"max_stem_length"`
+	Datasets       map[string]EffectiveMusicDataset `json:"datasets,omitempty" yaml:"datasets,omitempty"`
+}
+
+type EffectiveMusicDataset struct {
+	Source      string `json:"source" yaml:"source"`
+	Prefix      string `json:"prefix,omitempty" yaml:"prefix,omitempty"`
+	StageRoot   string `json:"stage_root,omitempty" yaml:"stage_root,omitempty"`
+	CreditsRoot string `json:"credits_root,omitempty" yaml:"credits_root,omitempty"`
 }
 
 type EffectiveTopDataConfig struct {
@@ -209,6 +217,51 @@ func (p *Project) EffectiveConfig() EffectiveConfig {
 		extract.CleanupStale = &defaultCleanup
 	}
 
+	musicStageRoot := expandPathTemplate(DefaultMusicStageRoot, paths)
+	musicCreditsRoot := expandPathTemplate(DefaultCreditsRoot, paths)
+	musicCreditsOverlay := DefaultCreditsOverlayFile
+	musicMaxStemLength := 16
+	if d := p.Config.Music.Defaults; d != nil {
+		if d.StageRoot != "" {
+			musicStageRoot = expandPathTemplate(d.StageRoot, paths)
+		}
+		if d.CreditsRoot != "" {
+			musicCreditsRoot = expandPathTemplate(d.CreditsRoot, paths)
+		}
+		if d.CreditsOverlay != "" {
+			musicCreditsOverlay = d.CreditsOverlay
+		}
+		if d.MaxStemLength != nil {
+			musicMaxStemLength = *d.MaxStemLength
+		}
+	}
+	musicPrefixes := cloneStringMap(p.Config.Music.Prefixes)
+	if len(musicPrefixes) == 0 {
+		musicPrefixes = make(map[string]string, len(p.Config.Music.Datasets))
+		for _, ds := range p.Config.Music.Datasets {
+			if ds.Source != "" && ds.Prefix != "" {
+				musicPrefixes[ds.Source] = ds.Prefix
+			}
+		}
+	}
+	effectiveDatasets := make(map[string]EffectiveMusicDataset, len(p.Config.Music.Datasets))
+	for id, ds := range p.Config.Music.Datasets {
+		dsStageRoot := ds.StageRoot
+		if dsStageRoot == "" {
+			dsStageRoot = musicStageRoot
+		}
+		dsCreditsRoot := ds.CreditsRoot
+		if dsCreditsRoot == "" {
+			dsCreditsRoot = musicCreditsRoot
+		}
+		effectiveDatasets[id] = EffectiveMusicDataset{
+			Source:      ds.Source,
+			Prefix:      ds.Prefix,
+			StageRoot:   dsStageRoot,
+			CreditsRoot: dsCreditsRoot,
+		}
+	}
+
 	effective := EffectiveConfig{
 		ConfigSource: p.ConfigSource,
 		Module:       p.Config.Module,
@@ -218,11 +271,12 @@ func (p *Project) EffectiveConfig() EffectiveConfig {
 		Inventory:    inventory,
 		Scripts:      scripts,
 		Music: EffectiveMusicConfig{
-			Prefixes:       cloneStringMap(p.Config.Music.Prefixes),
-			StageRoot:      expandPathTemplate(DefaultMusicStageRoot, paths),
-			CreditsRoot:    expandPathTemplate(DefaultCreditsRoot, paths),
-			CreditsOverlay: DefaultCreditsOverlayFile,
-			MaxStemLength:  16,
+			Prefixes:       musicPrefixes,
+			StageRoot:      musicStageRoot,
+			CreditsRoot:    musicCreditsRoot,
+			CreditsOverlay: musicCreditsOverlay,
+			MaxStemLength:  musicMaxStemLength,
+			Datasets:       effectiveDatasets,
 		},
 		TopData: top,
 		Extract: extract,