TaikoSoundEditor/MainForm.RequestFiles.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Windows.Forms;
using TaikoSoundEditor.Collections;
using TaikoSoundEditor.Commons.Controls;
using TaikoSoundEditor.Commons.IO;
using TaikoSoundEditor.Commons.Utils;
using TaikoSoundEditor.Data;

namespace TaikoSoundEditor
{
    partial class MainForm
    {
        #region Requesting Files

        private void WordListPathSelector_PathChanged(object sender, EventArgs args)
        {
            Logger.Info($"WordListPathSelector_PathChanged : {WordListPathSelector.Path}");
            WordListPath = WordListPathSelector.Path;
        }

        private void MusicInfoPathSelector_PathChanged(object sender, EventArgs args)
        {
            Logger.Info($"MusicInfoPathSelector_PathChanged : {MusicInfoPathSelector.Path}");
            MusicInfoPath = MusicInfoPathSelector.Path;
        }

        private void MusicOrderPathSelector_PathChanged(object sender, EventArgs args)
        {
            Logger.Info($"MusicOrderPathSelector_PathChanged : {MusicOrderPathSelector.Path}");
            MusicOrderPath = MusicOrderPathSelector.Path;
        }

        private void MusicAttributePathSelector_PathChanged(object sender, EventArgs args)
        {
            Logger.Info($"MusicAttributePathSelector_PathChanged : {MusicAttributePathSelector.Path}");
            MusicAttributePath = MusicAttributePathSelector.Path;
        }

        private void DirSelector_PathChanged(object sender, EventArgs args) => ExceptionGuard.Run(() =>
        {
            Logger.Info($"MusicAttributePathSelector_PathChanged : {DirSelector.Path}");
            var dir = DirSelector.Path;
            var files = new string[] { "music_attribute.bin", "music_order.bin", "musicinfo.bin", "wordlist.bin" };
            var sels = new PathSelector[] { MusicAttributePathSelector, MusicOrderPathSelector, MusicInfoPathSelector, WordListPathSelector };

            Config.DatatablesPath = dir;

            List<string> NotFoundFiles = new List<string>();

            for (int i = 0; i < files.Length; i++)
            {
                var path = Path.Combine(dir, files[i]);
                if (!File.Exists(path))
                {
                    NotFoundFiles.Add(files[i]);
                    continue;
                }
                sels[i].Path = path;
            }

            if (NotFoundFiles.Count > 0)
            {
                Logger.Warning("The following files could not be found:\n\n" + String.Join("\n", NotFoundFiles));
                MessageBox.Show("The following files could not be found:\n\n" + String.Join("\n", NotFoundFiles));
            }
        });

        private void OkButton_Click(object sender, EventArgs e) => ExceptionGuard.Run(() =>
        {
            Logger.Info($"Clicked 'Looks good' ");

            Config.DatatableIO.IsEncrypted = UseEncryptionBox.Checked;

            SSL.LoadKeys();

            try
            {                                
                MusicAttributes = Config.DatatableIO.DeserializeCollection<MusicAttributes, IMusicAttribute>(MusicAttributePath, DatatableTypes.MusicAttribute);
            }
            catch (Exception ex)
            {
                throw new Exception($"Failed to parse\n{MusicAttributePath}\nReason:\n{ex.InnerException}");
            }
            try
            {                
                MusicOrders = Config.DatatableIO.DeserializeCollection<MusicOrders, IMusicOrder>(MusicOrderPath, DatatableTypes.MusicOrder);                
            }
            catch (Exception ex)
            {
                throw;
                throw new Exception($"Failed to parse\n{MusicOrderPath}\nReason:\n{ex.InnerException}");
            }
            try
            {                                
                MusicInfos = Config.DatatableIO.DeserializeCollection<MusicInfos, IMusicInfo>(MusicInfoPath, DatatableTypes.MusicInfo);
            }
            catch (Exception ex)
            {
                throw new Exception($"Failed to parse\n{MusicInfoPath}\nReason:\n{ex.InnerException}");
            }
            try
            {
                WordList = Config.DatatableIO.DeserializeCollection<WordList, IWord>(WordListPath, DatatableTypes.Word);
            }
            catch (Exception ex)
            {
                throw;
                throw new Exception($"Failed to parse\n{WordListPath}\nReason:\n{ex.InnerException}");
            }

            Logger.Info($"Fixing Wordlist missing data");

            foreach (var mi in MusicInfos.Items)
            {
                if (mi.UniqueId == 0) continue;
                var songId = mi.Id;

                if(MusicAttributes.GetByUniqueId(mi.UniqueId)==null)
                {
                    Logger.Info($"Added missing music_attribute entry for {mi.UniqueId}.{songId}");

                    MusicAttributes.Items.Add(DatatableTypes.CreateMusicAttribute(songId, mi.UniqueId, false));
                }

                if(MusicOrders.GetByUniqueId(mi.UniqueId)==null)
                {
                    Logger.Info($"Added missing music_order entry for {mi.UniqueId}.{songId}");
                    MusicOrders.Items.Add(DatatableTypes.CreateMusicOrder(mi.Genre, songId, mi.UniqueId));
                }

                if (WordList.GetBySong(songId) == null)
                {
                    Logger.Info($"Added missing word title entry for {mi.UniqueId}.{songId}");
                    WordList.Items.Add(DatatableTypes.CreateWord($"song_{songId}", ""));
                }
                if (WordList.GetBySongSub(songId) == null)
                {
                    Logger.Info($"Added missing word subtitle entry for {mi.UniqueId}.{songId}");
                    WordList.Items.Add(DatatableTypes.CreateWord($"song_sub_{songId}", ""));                    
                }
                if (WordList.GetBySongDetail(songId) == null)
                {
                    Logger.Info($"Added missing word detail entry for {mi.UniqueId}.{songId}");
                    WordList.Items.Add(DatatableTypes.CreateWord($"song_detail_{songId}", ""));                    
                }
            }

            Logger.Info($"Setting LoadedMusicBox DataSource");

            LoadedMusicBinding = new BindingSource();
            var cleanList = MusicInfos.Items.Where(mi => mi.UniqueId != 0).OrderBy(_ => _.UniqueId).ToList();

            LoadedMusicBinding.DataSource = cleanList;
            LoadedMusicBox.DataSource = LoadedMusicBinding;
            TabControl.SelectedIndex = 1;

            /*WordList.Items.RemoveAll(w =>
            {
                var key = w.Key;

                var prefixes = new string[] { "song_sub_", "song_detail_", "song_" };

                for (int i = 0; i < 3; i++)
                    if (key.StartsWith(prefixes[i]))
                    {
                        key = key.Substring(prefixes[i].Length);
                        break;
                    }
                return !MusicInfos.Items.Any(mi => mi.Id == key);
            });*/            

            MusicOrderViewer.WordList = WordList;
            foreach (var musicOrder in MusicOrders.Items.Where(_ => MusicInfos.Items.Any(mi => mi.UniqueId == _.UniqueId)))
            {
                MusicOrderViewer.AddSong(musicOrder);
            }
        });

        #endregion                    
    }
}