icarus/Controllers/Managers/SongManager.cs

using System;
using System.Collections.Generic;
using System.Configuration;
using System.Linq;
using System.Data;
using System.IO;
using System.Threading.Tasks;

using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Configuration;

using NLog;

using Icarus.Controllers.Utilities;
using Icarus.Models;
using Icarus.Database.Contexts;

namespace Icarus.Controllers.Managers
{
    public class SongManager : BaseManager
    {
        #region Fields
        private string _tempDirectoryRoot;
        private string _archiveDirectoryRoot;
        private string _compressedSongFilename;
        private string _message;
        private SongContext _songContext;
        #endregion


        #region Properties
        public string ArchiveDirectoryRoot
        {
            get => _archiveDirectoryRoot;
            set => _archiveDirectoryRoot = value;
        }
        public string CompressedSongFilename
        {
            get => _compressedSongFilename;
            set => _compressedSongFilename = value;
        }
        public string Message
        {
            get => _message;
            set => _message = value;
        }
        #endregion


        #region Constructors
        public SongManager(IConfiguration config)
        {
            _config = config;
            Initialize();
        }
        public SongManager(IConfiguration config, string tempDirectoryRoot)
        {
            _config = config;
            _tempDirectoryRoot = tempDirectoryRoot;
            Initialize();
        }
        #endregion


        #region Methods
        public SongResult UpdateSong(Song song)
        {
            var result = new SongResult();

            try
            {
                var oldSongRecord = _songContext.RetrieveRecord(song);
                song.Filename = oldSongRecord.Filename;
                song.SongDirectory = oldSongRecord.SongDirectory;

                MetadataRetriever updateMetadata = new MetadataRetriever();
                updateMetadata.UpdateMetadata(song, oldSongRecord);

                var updatedSong = updateMetadata.UpdatedSongRecord;

                var updatedAlbum = UpdateAlbumInDatabase(oldSongRecord, updatedSong);
                oldSongRecord.AlbumID = updatedAlbum.AlbumID;

                var updatedArtist = UpdateArtistInDatabase(oldSongRecord, updatedSong);
                oldSongRecord.ArtistID = updatedArtist.ArtistID;

                var updatedGenre = UpdateGenreInDatabase(oldSongRecord, updatedSong);
                Console.WriteLine($"Old Genre Id {oldSongRecord.GenreID}");
                oldSongRecord.GenreID = updatedGenre.GenreID;
                Console.WriteLine($"Updated Genre ID {updatedGenre.GenreID}");

                UpdateSongInDatabase(ref oldSongRecord, ref updatedSong, ref result);

                DeleteEmptyDirectories(ref oldSongRecord, ref updatedSong);
            }
            catch (Exception ex)
            {
                var msg = ex.Message;
                _logger.Error(msg, "An error occurred");

                result.Message = $"An error occurred: {msg}";
                result.SongTitle = song.Title;
            }

            return result;
        }

        public bool DeleteSongFromFileSystem(Song songMetaData)
        {
            bool successful = false;
            try
            {
                var songPath = songMetaData.SongPath();
                System.IO.File.Delete(songPath);
                successful = true;
                DirectoryManager dirMgr = new DirectoryManager(_config, songMetaData);
                dirMgr.DeleteEmptyDirectories();
                Console.WriteLine("Song successfully deleted");
            }
            catch (Exception ex)
            {
                var exMsg = ex.Message;
            }

            return successful;
        }

        public void DeleteSong(Song song)
        {
            try
            {
                if (DeleteSongFromFilesystem(song))
                {
                    _logger.Error("Failed to delete the song");

                    throw new Exception("Failed to delete the song");
                }
                _logger.Info("Song deleted from the filesystem");

                var coverMgr = new CoverArtManager(_config);

                var coverArt = coverMgr.GetCoverArt(song);
                coverMgr.DeleteCoverArt(coverArt);

                coverMgr.DeleteCoverArtFromDatabase(coverArt);
                DeleteSongFromDatabase(song);
            }
            catch (Exception ex)
            {
                var msg = ex.Message;
                _logger.Error(msg, "An error occurred");
            }
        }


        
        
        
        public async Task SaveSongToFileSystem(IFormFile songFile)
        {
            try
            {
                _logger.Info("Starting the process of saving the song to the filesystem");

                var fileTempPath = Path.Combine(_tempDirectoryRoot, songFile.FileName);
                var song = await SaveSongTemp(songFile, fileTempPath);
                song.SongDirectory = _tempDirectoryRoot;
                song.Filename = songFile.FileName;

                DirectoryManager dirMgr = new DirectoryManager(_config, song);
                dirMgr.CreateDirectory();

                System.IO.File.Delete(fileTempPath);

                var filePath = dirMgr.SongDirectory;
                var songFilename = songFile.FileName;

                if (!songFilename.EndsWith(".mp3"))
                    filePath += $"{songFilename}.mp3";
                else
                    filePath += $"{songFilename}";

                _logger.Info($"Absolute song path: {filePath}");


                using (var fileStream = new FileStream(filePath, FileMode.Create))
                {
                    await (songFile.CopyToAsync(fileStream));
                    song.SongDirectory = dirMgr.SongDirectory;


                    _logger.Info("Song successfully saved to filesystem");
                }

                var coverMgr = new CoverArtManager(_config);
                var coverArt = coverMgr.SaveCoverArt(song);

                coverMgr.SaveCoverArtToDatabase(ref song, ref coverArt);
                SaveSongToDatabase(song);
            }
            catch (Exception ex)
            {
                var msg = ex.Message;
                _logger.Error(msg, "An error occurred");
            }
        }

        
        
        public async Task<SongData> RetrieveSong(Song songMetaData)
        {
            SongData song = new SongData();
            try
            {
                Console.WriteLine("Fetching song from filesystem");
                song = await RetrieveSongFromFileSystem(songMetaData);
            }
            catch (Exception ex)
            {
                var exMsg = ex.Message;
                Console.WriteLine($"An error occurred: {exMsg}");
            }

            return song;
        }



        private async Task<SongData> RetrieveSongFromFileSystem(Song details)
        {
            byte[] uncompressedSong = await System.IO.File.ReadAllBytesAsync(details.SongPath());
            
            return new SongData
            {
                Data = uncompressedSong
            };
        }
        private async Task<Song> SaveSongTemp(IFormFile songFile, string filePath)
        {
            var song = new Song();


            using (var filestream = new FileStream(filePath, FileMode.Create))
            {
                _logger.Info("Saving song to temporary directory");
                await songFile.CopyToAsync(filestream);
            }

            MetadataRetriever meta = new MetadataRetriever();
            song =  meta.RetrieveMetaData(filePath);

            _logger.Info("Assigning song filename");
            song.Filename = songFile.FileName;
            song.DateCreated = DateTime.Now;

            return song;
        }


        private bool SongRecordChanged(Song currentSong, Song songUpdates)
        {
            var currentTitle = currentSong.Title;
            var currentArtist = currentSong.Artist;
            var currentAlbum = currentSong.AlbumTitle;
            var currentGenre = currentSong.Genre;
            var currentYear = currentSong.Year;

            if (!currentTitle.Equals(songUpdates.Title) || !currentArtist.Equals(songUpdates.Artist) ||
                    !currentAlbum.Equals(songUpdates.AlbumTitle) ||
                    !currentGenre.Equals(songUpdates.Genre) || currentYear != songUpdates.Year)
                return true;

            return false;
        }

        private void DeleteEmptyDirectories(ref Song oldSong, ref Song updatedSong)
        {
            DirectoryManager mgr = new DirectoryManager(_config);

            _logger.Info("Checking to see if there are any directories to delete");
            mgr.DeleteEmptyDirectories(oldSong);
        }

        private void Initialize()
        {
            try
            {
                _connectionString = _config.GetConnectionString("DefaultConnection");
                _songContext = new SongContext(_connectionString);
            }
            catch (Exception ex)
            {
                Console.WriteLine($"Error Occurred: {ex.Message}");
            }
        }
        

        
        private void SaveSongToDatabase(Song song)
        {
            _logger.Info("Starting process to save the song to the database");

            SaveAlbumToDatabase(ref song);//, albumStore);
            SaveArtistToDatabase(ref song);//, artistStore);
            SaveGenreToDatabase(ref song);//, genreStore);

            var info = "Saving Song to DB";
            Console.WriteLine(info);
            _logger.Info(info);
            _songContext.Add(song);
        }
        private void SaveAlbumToDatabase(ref Song song)
        {
            _logger.Info("Starting process to save the album record of the song to the database");

            var album = new Album();

            album.Title = song.AlbumTitle;
            album.AlbumArtist = song.Artist;
            var albumTitle = song.AlbumTitle;
            AlbumContext albumContext = new AlbumContext(_connectionString);

            if (!(albumContext.Albums.FirstOrDefault(alb => alb.Title.Equals(albumTitle)) != null))
            {
                album.SongCount = 1;
                albumContext.Add(album);
                album = albumContext.Albums.FirstOrDefault(alb => alb.Title.Equals(albumTitle));
                Console.WriteLine($"Album Id {album.AlbumID}");
            }
            else
            {
                var albumRetrieved = albumContext.Albums.FirstOrDefault(alb => alb.Title.Equals(albumTitle));
                album.AlbumID = albumRetrieved.AlbumID;
                album.SongCount = albumRetrieved.SongCount + 1;

                albumContext.Update(album);
            }

            song.AlbumID = album.AlbumID;
        }
        private void SaveArtistToDatabase(ref Song song)
        {
            _logger.Info("Starting process to save the artist record of the song to the database");

            var artist = new Artist();

            artist.Name = song.Artist;
            artist.SongCount = 1;
            var artistTitle = artist.Name;
            ArtistContext artistContext = new ArtistContext(_connectionString);

            if (!(artistContext.Artists.FirstOrDefault(art => art.Name.Equals(artistTitle)) != null))
            {
                artist.SongCount = 1;
                artistContext.Add(artist);
                artist = artistContext.Artists.FirstOrDefault(art => art.Name.Equals(artistTitle));
            }
            else
            {
                var artistRetrieved = artistContext.Artists.FirstOrDefault(art => art.Name.Equals(artistTitle));
                artist.ArtistID = artistRetrieved.ArtistID;
                artist.SongCount = artistRetrieved.SongCount + 1;

                artistContext.Update(artist);
                artistContext.SaveChanges();
            }

            song.ArtistID = artist.ArtistID;
        }
        private void SaveGenreToDatabase(ref Song song)
        {
            _logger.Info("Starting process to save the genre record of the song to the database");

            var genre = new Genre
            {
                GenreName = song.Genre,
                SongCount = 1
            };

            var genreName = song.Genre;
            var genreContext = new GenreContext(_connectionString);

            if (! (genreContext.Genres.FirstOrDefault(gnr => gnr.GenreName.Equals(genreName)) != null))
            {
                genreContext.Add(genre);
                Console.WriteLine("Going to find genre");
                genre = genreContext.Genres.FirstOrDefault(gnr => gnr.GenreName.Equals(genreName));
                var genreDump = $"Genre ID {genre.GenreID} GenreName {genre.GenreName}" +
                    $" Genre song Count {genre.SongCount}";
                Console.WriteLine(genreDump);
                _logger.Info(genreDump);
            }
            else
            {
                var genreRetrieved = genreContext.Genres.FirstOrDefault(gnr => gnr.GenreName.Equals(genreName));
                genre.GenreID = genreRetrieved.GenreID;
                genre.SongCount = genreRetrieved.SongCount + 1;

                genreContext.Update(genre);
                genreContext.SaveChanges();
            }

            song.GenreID = genre.GenreID;
        }
        

        private bool DeleteSongFromFilesystem(Song song)
        {
            var songPath = song.SongPath();

            _logger.Info("Deleting song from the filesystem");

            try
            {
                System.IO.File.Delete(songPath);
            }
            catch(Exception ex)
            {
                var msg = ex.Message;
                _logger.Error(msg, "An error occurred when attempting to delete the song from the filesystem");
                return false;
            }

            return DoesSongExistOnFilesystem(song);
        }
        private bool DoesSongExistOnFilesystem(Song song)
        {
            var songPath = song.SongPath();

            if (!System.IO.File.Exists(songPath))
            {
                _logger.Info("Song does not exist on the filesystem");

                return false;
            }

            _logger.Info("Song exists on the filesystem");

            return true;
        }

        private Album UpdateAlbumInDatabase(Song oldSongRecord, Song newSongRecord)
        {
            var albumContext = new AlbumContext(_connectionString);

            var albumRecord = albumContext.Albums.FirstOrDefault(alb => alb.Title.Equals(oldSongRecord.AlbumTitle));
            var oldAlbumTitle = oldSongRecord.AlbumTitle;
            var oldAlbumArtist = oldSongRecord.Artist;
            var newAlbumTitle = newSongRecord.AlbumTitle;
            var newAlbumArtist = newSongRecord.Artist;

            var info = string.Empty;

            if (string.IsNullOrEmpty(newAlbumArtist))
                newAlbumArtist = oldAlbumArtist;
            if (string.IsNullOrEmpty(newAlbumTitle))
                newAlbumTitle = oldAlbumTitle;

            if ((string.IsNullOrEmpty(newAlbumTitle) && string.IsNullOrEmpty(newAlbumArtist) || 
                        oldAlbumTitle.Equals(newAlbumTitle) && oldAlbumArtist.Equals(newAlbumArtist)))
            {
                _logger.Info("No change to the song's album");
                return albumRecord;
            }

            info = "Change to the song's album";
            _logger.Info(info);

            if (!(albumContext.Albums.FirstOrDefault(alb => alb.Title.Equals(newSongRecord.AlbumTitle)) != null))
            {
                _logger.Info("Creating new album record");

                var newAlbumRecord = new Album
                {
                    Title = newAlbumTitle,
                    AlbumArtist = newAlbumArtist
                };

                albumContext.Add(newAlbumRecord);

                return albumContext.Albums.FirstOrDefault(alb => alb.Title.Equals(oldSongRecord.AlbumTitle));
            }
            else
            {
                _logger.Info("Updating existing album record");

                var existingAlbumRecord = albumContext.Albums.FirstOrDefault(alb => alb.Title.Equals(newSongRecord.AlbumTitle));
                existingAlbumRecord.AlbumArtist = newAlbumArtist;

                albumContext.Update(existingAlbumRecord);
                albumContext.SaveChanges();

                return existingAlbumRecord;
            }
        }
        private Artist UpdateArtistInDatabase(Song oldSongRecord, Song newSongRecord)
        {
            var artistContext = new ArtistContext(_connectionString);

            var oldArtistRecord = artistContext.Artists.FirstOrDefault(art => art.Name.Equals(oldSongRecord.AlbumTitle));
            var oldArtistName = oldArtistRecord.Name;
            var newArtistName = newSongRecord.Artist;

            if (string.IsNullOrEmpty(newArtistName) || oldArtistName.Equals(newArtistName))
            {
                _logger.Info("No change to the song's Artist");
                return oldArtistRecord;
            }

            _logger.Info("Change to the song's record found");

            if (oldArtistRecord.SongCount <= 1)
            {
                _logger.Info("Deleting artist record that no longer has any songs");

                artistContext.Remove(oldArtistRecord);
                artistContext.SaveChanges();
            }

            if (!(artistContext.Artists.FirstOrDefault(art => art.Name.Equals(oldSongRecord.AlbumTitle)) != null))
            {
                _logger.Info("Creating new artist record");

                var newArtistRecord = new Artist
                {
                    Name = newArtistName
                };

                artistContext.Add(newArtistRecord);
                
                return artistContext.Artists.FirstOrDefault(art => art.Name.Equals(newSongRecord.AlbumTitle));
            }
            else
            {
                _logger.Info("Updating existing artist record");

                var existingArtistRecord = artistContext.Artists.FirstOrDefault(art => art.Name.Equals(newSongRecord.AlbumTitle));

                artistContext.Update(existingArtistRecord);
                artistContext.SaveChanges();

                return existingArtistRecord;
            }
        }
        private Genre UpdateGenreInDatabase(Song oldSongRecord, Song newSongRecord)
        {
            var genreContext = new GenreContext(_connectionString);
            var oldGenreRecord = genreContext.Genres.FirstOrDefault(gnr => gnr.GenreName.Equals(oldSongRecord.Genre));
            var oldGenreName = oldGenreRecord.GenreName;
            var newGenreName = newSongRecord.Genre;

            if (string.IsNullOrEmpty(newGenreName) || oldGenreName.Equals(newGenreName))
            {
                _logger.Info("No change to the song's Genre");
                return oldGenreRecord;
            }

            _logger.Info("Change to the song's genre found");

            if (oldGenreRecord.SongCount <= 1)
            {
                _logger.Info("Deleting genre record");

                genreContext.Remove(oldGenreRecord);
                genreContext.SaveChanges();
            }

            if (!(genreContext.Genres.FirstOrDefault(gnr => gnr.GenreName.Equals(oldSongRecord.Genre)) != null))
            {
                _logger.Info("Creating new genre record");

                var newGenreRecord = new Genre
                {
                    GenreName = newGenreName
                };

                genreContext.Add(newGenreRecord);

                return genreContext.Genres.FirstOrDefault(gnr => gnr.GenreName.Equals(newGenreRecord.GenreName));
            }
            else
            {
                _logger.Info("Updating existing genre record");

                var existingGenreRecord = genreContext.Genres.FirstOrDefault(gnr => gnr.GenreName.Equals(oldGenreRecord.GenreName));

                genreContext.Update(existingGenreRecord);
                genreContext.SaveChanges();
                return genreContext.Genres.FirstOrDefault(gnr => gnr.GenreName.Equals(existingGenreRecord.GenreName));
            }
        }
        
        private void UpdateSongInDatabase(ref Song oldSongRecord, ref Song newSongRecord, ref SongResult result)
        {
            var updatedSongRecord = new Song
            {
                SongID = oldSongRecord.SongID,
                Title = oldSongRecord.Title,
                Artist = oldSongRecord.Artist,
                AlbumTitle = oldSongRecord.AlbumTitle,
                Genre = oldSongRecord.Genre,
                Year = oldSongRecord.Year,
                Filename = oldSongRecord.Filename,
                SongDirectory = oldSongRecord.SongDirectory,
                ArtistID = oldSongRecord.ArtistID,
                AlbumID = oldSongRecord.AlbumID,
                GenreID = oldSongRecord.GenreID

            };
            var artistOrAlbumChanged = false;

            var songContext = new SongContext(_connectionString);

            if (!SongRecordChanged(oldSongRecord, newSongRecord))
            {
                _logger.Info("No change to the song record");
                return;
            }

            _logger.Info("Changes to song record found");

            if (!string.IsNullOrEmpty(newSongRecord.Title))
                updatedSongRecord.Title = newSongRecord.Title;
            if (!string.IsNullOrEmpty(newSongRecord.AlbumTitle))
            {
                updatedSongRecord.AlbumTitle = newSongRecord.AlbumTitle;
                artistOrAlbumChanged = true;
            }
            if (!string.IsNullOrEmpty(newSongRecord.Artist))
            {
                updatedSongRecord.Artist = newSongRecord.Artist;
                artistOrAlbumChanged = true;
            }
            if (!string.IsNullOrEmpty(newSongRecord.Genre))
            {
                updatedSongRecord.Genre = newSongRecord.Genre;
                Console.WriteLine("Genre changed");
                Console.WriteLine($"{updatedSongRecord.Genre} {newSongRecord.Genre}");
            }
            if (newSongRecord.Year != null || newSongRecord.Year > 0)
                updatedSongRecord.Year = newSongRecord.Year;

            _logger.Info("Applied changes to song record");

            if (artistOrAlbumChanged)
            {
                _logger.Info("Change to song's album or artist");

                DirectoryManager dirMgr = new DirectoryManager(_config);
                var oldSongPath = updatedSongRecord.SongPath();
                var newSongPath = dirMgr.GenerateSongPath(updatedSongRecord);
                var filename = updatedSongRecord.Filename;

                Console.WriteLine($"Old song path {oldSongPath}");
                Console.WriteLine($"New song path {newSongPath}");

                _logger.Info("Copying song to the new path");

                System.IO.File.Copy(oldSongPath, newSongPath + filename, true);

                _logger.Info("Checking to see if song successfully copied");

                if (!System.IO.File.Exists(newSongPath + filename))
                {
                    _logger.Info("Song did not successfully copy");

                    Console.WriteLine("New path does not exist when it should");
                }

                updatedSongRecord.SongDirectory = newSongPath;
                updatedSongRecord.Filename = filename;
                
                _logger.Info("Deleting old song path");

                System.IO.File.Delete(oldSongPath);

                if (System.IO.File.Exists(oldSongPath))
                    Console.WriteLine("Old path exists when it should not");
            }

            _logger.Info("Saving song metadata to the database");

            if (songContext.Songs.FirstOrDefault(sng => sng.Title.Equals(updatedSongRecord.Title)) != null)
            {
                songContext.Update(updatedSongRecord);
                songContext.SaveChanges();
            }
            else
                songContext.Add(updatedSongRecord);

            newSongRecord = updatedSongRecord;

            result.Message = "Successfully updated song";
            result.SongTitle = updatedSongRecord.Title;
        }

        private void DeleteSongFromDatabase(Song song)
        {
            _logger.Info("Starting process to delete records related to the song from the database");

            DeleteAlbumFromDatabase(song);
            DeleteArtistFromDatabase(song);
            DeleteGenreFromDatabase(song);

            var sngContext = new SongContext(_connectionString);
            sngContext.Songs.Remove(song);
            sngContext.SaveChanges();
        }
        private void DeleteAlbumFromDatabase(Song song)
        {
            var albumContext = new AlbumContext(_connectionString);

            if (!(albumContext.Albums.FirstOrDefault(alb => alb.Title.Equals(song.AlbumTitle)) != null))
            {
                _logger.Info("Cannot delete the album record because it does not exist");
                return;
            }

            var album = albumContext.Albums.FirstOrDefault(alb => alb.Title.Equals(song.AlbumTitle));

            if (album.SongCount <= 1)
            {
                albumContext.Remove(album);
                albumContext.SaveChanges();
            }
        }
        private void DeleteArtistFromDatabase(Song song)
        {
            var artistContext = new ArtistContext(_connectionString);

            if (!(artistContext.Artists.FirstOrDefault(art => art.Name.Equals(song.Artist)) != null))
            {
                _logger.Info("Cannot delete the artist record because it does not exist");
                return;
            }

            var artist = artistContext.Artists.FirstOrDefault(art => art.Name.Equals(song.Artist));

            if (artist.SongCount <= 1)
            {
                artistContext.Remove(artist);
                artistContext.SaveChanges();
            }
        }
        private void DeleteGenreFromDatabase(Song song)
        {
            var genreContext = new GenreContext(_connectionString);

            if (!(genreContext.Genres.FirstOrDefault(gnr => gnr.GenreName.Equals(song.Genre)) != null))
            {
                _logger.Info("Cannot delete the genre record because it does not exist");
                return;
            }

            var genre = genreContext.Genres.FirstOrDefault(gnr => gnr.GenreName.Equals(song.Genre));

            if (genre.SongCount <= 1)
            {
                genreContext.Remove(genre);
                genreContext.SaveChanges();
            }
        }
        

        #endregion    
    }
}