public static void FixTrackNumbers(IList <IITTrack> tracks)
{
if (tracks == null)
{
throw new ArgumentNullException(nameof(tracks));
}
var trackComparer = TrackComparerFactory.GetTrackComparer(tracks);
var trackList = tracks.ToList();
trackList.Sort(trackComparer);
for (var i = 0; i < trackList.Count; i++)
{
try
{
trackList[i].TrackNumber = i + 1;
}
catch (System.Runtime.InteropServices.COMException e)
{
if (e.Message.Contains(TrackMissingError.Code))
{
throw new iTunesAssistantException(TrackMissingError.Message);
}
else
{
throw;
}
}
}
}
public void Run(IWorkflowRunnerInfo workflowRunnerInfo, ref Status status)
{
if (workflowRunnerInfo == null)
{
throw new ArgumentNullException(nameof(workflowRunnerInfo));
}
if (workflowRunnerInfo.Tracks == null)
{
throw new ArgumentNullException(nameof(workflowRunnerInfo.Tracks));
}
status = Status.Create(workflowRunnerInfo.Tracks.Count, "Generating album groups...");
var albumGroups = new Dictionary <string, IDictionary <string, IList <IITTrack> > >();
foreach (var track in workflowRunnerInfo.Tracks)
{
var newAlbumName = AlbumNameFixer.FixAlbumName(track.Album);
if (!albumGroups.ContainsKey(newAlbumName))
{
albumGroups.Add(newAlbumName, new SortedDictionary <string, IList <IITTrack> >());
albumGroups[newAlbumName].Add(track.Album, new List <IITTrack> {
track
});
}
else if (!albumGroups[newAlbumName].ContainsKey(track.Album))
{
albumGroups[newAlbumName].Add(track.Album, new List <IITTrack> {
track
});
}
else
{
albumGroups[newAlbumName][track.Album].Add(track);
}
status.ItemProcessed();
}
foreach (var album in albumGroups.SelectMany(albumGroup => albumGroup.Value.Values))
{
var comparer = TrackComparerFactory.GetTrackComparer(album);
album.ToList().Sort(comparer);
}
status = Status.Create(albumGroups.Count, "Running merge albums workflow...");
foreach (var albumGroup in albumGroups)
{
var newAlbumName = albumGroup.Key;
var trackCount = albumGroup.Value.Sum(album => album.Value.Count);
var trackNumber = 1;
foreach (var track in albumGroup.Value.SelectMany(album => album.Value))
{
track.Album = newAlbumName;
track.DiscCount = 1;
track.DiscNumber = 1;
// iTunes doesn't allow TrackCount to be set to a value higher than TrackNumber, so set number
// before count in case any old track numbers are higher than new track count.
track.TrackNumber = trackNumber++;
track.TrackCount = trackCount;
}
status.ItemProcessed();
}
}
