using DebugMenuEditorUI.Model; using GameFormatReader.Common; using Microsoft.Win32; using System; using System.Collections.Generic; using System.Collections.ObjectModel; using System.ComponentModel; using System.IO; using System.Windows; using System.Windows.Data; using System.Windows.Input; namespace DebugMenuEditorUI.ViewModel { public class MainWindowViewModel : INotifyPropertyChanged { public event PropertyChangedEventHandler PropertyChanged; public const string ApplicationName = "Debug Menu Editor"; #region Command Callbacks ///

User has requested that we create a new file from scratch. Save current, then create new.

public ICommand OnRequestNewFile { get { return new RelayCommand(x => CreateNewFile()); } } ///

User has requested that we open a file from disk. Ask user for file, save current, then load.

public ICommand OnRequestFileOpen { get { return new RelayCommand(x => OpenFile()); } } ///

User has requested that we close the current file. Save if applicable, then close.

public ICommand OnRequestFileClose { get { return new RelayCommand(x => CloseFile(), x => LoadedFile != null); } } ///

User has requested that we save the current file.

public ICommand OnRequestFileSave { get { return new RelayCommand(x => SaveFile(), x => LoadedFile != null); } } ///

User has requested that we save the current file under a different name.

public ICommand OnRequestFileSaveAs { get { return new RelayCommand(x => SaveFileAs(), x => LoadedFile != null); } } ///

The user has requested that we close the application. Save file (if required).

public ICommand OnRequestApplicationClose { get { return new RelayCommand(x => QuitApplication()); } } ///

Add a new category to the loaded .

public ICommand OnRequestNewCategory { get { return new RelayCommand(x => AddCategory(), x => LoadedFile != null); } } ///

Add a new entry to the currently selected category.

public ICommand OnRequestNewEntry { get { return new RelayCommand(x => CreateNewEntry(), x => LoadedFile != null && SelectedCategory != null); } } /////

Take the currently selected entries from the List and add them to the copy buffer.

//public ICommand OnRequestCopyEntries; /////

Take the current entries in the copy buffer and paste them into the currently selected .

//public ICommand OnRequestPasteEntries; ///

Delete the currently selected category.

public ICommand OnRequestDeleteCategory { get { return new RelayCommand(x => RemoveCategory(), x => LoadedFile != null && SelectedCategory != null); } } ///

Delete the currently selected entries from the currently selected category.

public ICommand OnRequestDeleteEntries { get { return new RelayCommand(x => RemoveEntry(), x=> LoadedFile != null && SelectedEntries.Count > 0); } } #endregion #region Properties ///

/// What is the title of the ? Adjusted based on the name of the currently loaded file. ///

public string WindowTitle { get { return m_windowTitle; } set { m_windowTitle = value; OnPropertyChanged("WindowTitle"); } } ///

/// Indicates the last action performed by the application and displayed in the lower corner of the UI. ///

public string ApplicationStatus { get { return m_applicationStatus; } set { m_applicationStatus = value; OnPropertyChanged("ApplicationStatus"); } } ///

/// The text to search the category entries by. ///

public string SearchFilter { get { return m_searchFilter; } set { m_searchFilter = value; if (!string.IsNullOrEmpty(m_searchFilter)) { AddFilter(); // If they're searching for something, we want to set the selected category to null, // as there is no specific selected category anymore and instead it is showing all // entries. SelectedCategory = null; } CollViewSource.View.Refresh(); OnPropertyChanged("SearchFilter"); } } public CollectionViewSource CollViewSource { get; set; } ///

/// The currently loaded Debug Menu file, null if not loaded. ///

public Menu LoadedFile { get { return m_loadedFile; } set { m_loadedFile = value; UpdateWindowTitle(); if(m_loadedFile != null) { if (LoadedFile.Categories.Count > 0) SelectedCategory = LoadedFile.Categories[0]; } else { SelectedCategory = null; } OnPropertyChanged("LoadedFile"); } } public Category SelectedCategory { get { return m_selectedCategory; } set { m_selectedCategory = value; // If they're assigning a valid Category to this list, we clear the search filter // so that it doesn't fight with us. if (m_selectedCategory != null) { // Copy to a List as a BindingList doesn't support filtering (doh...) List entries = new List(m_selectedCategory.Entries); // Then filter our collection by only showing items from this one. CollViewSource.Source = entries; // Wipe out the search filter text, they can't search and select categories at the same time. SearchFilter = string.Empty; } else { // If there is no longer a selected category, set the items source to be the entire list from the file. List allEntries = new List(); foreach (var cat in LoadedFile.Categories) { foreach (var entry in cat.Entries) allEntries.Add(entry); } CollViewSource.Source = allEntries; } OnPropertyChanged("SelectedCategory"); } } public ObservableCollection SelectedEntries { get { return m_selectedEntries; } set { m_selectedEntries = value; OnPropertyChanged("SelectedEntries"); // Only set the selected entry if we have one item selected. Otherwise it sets it to null // as we don't support multi-entry editing. if (SelectedEntries.Count == 1) SelectedEntry = SelectedEntries[0]; else SelectedEntry = null; } } public CategoryEntry SelectedEntry { get { return m_selectedEntry; } set { m_selectedEntry = value; OnPropertyChanged("SelectedEntry"); } } #endregion private string m_windowTitle; private string m_applicationStatus; private string m_searchFilter; private Menu m_loadedFile; private Category m_selectedCategory; private ObservableCollection m_selectedEntries; private CategoryEntry m_selectedEntry; public MainWindowViewModel() { CollViewSource = new CollectionViewSource(); SelectedEntries = new ObservableCollection(); SelectedEntries.CollectionChanged += SelectedEntries_CollectionChanged; UpdateWindowTitle(); } void SelectedEntries_CollectionChanged(object sender, System.Collections.Specialized.NotifyCollectionChangedEventArgs e) { // Only set the selected entry if we have one item selected. Otherwise it sets it to null // as we don't support multi-entry editing. if (SelectedEntries.Count == 1) SelectedEntry = SelectedEntries[0]; else SelectedEntry = null; } #region New, Open, Save, Save As, Load ///

User has requested that we create a new file from scratch. Save current, then create new.

private void CreateNewFile() { ConfirmSaveDesireAndSave(); LoadedFile = new Menu(); ApplicationStatus = string.Format("Created {0}", LoadedFile.FileName); } ///

User has requested that we open a file from disk. Ask user for file, save current, then load.

private void OpenFile() { OpenFileDialog ofd = new OpenFileDialog(); ofd.ValidateNames = true; ofd.Filter = "Debug Menu Files (*.dat)|*.dat|All Files (*.*)|*.*"; if(ofd.ShowDialog() == true) { ConfirmSaveDesireAndSave(); ApplicationStatus = string.Format("Loading {0}...", ofd.SafeFileName); string folderName = Path.GetDirectoryName(ofd.FileName); string fileName = Path.GetFileName(ofd.FileName); Menu newMenu = new Menu(); newMenu.FileName = fileName; newMenu.FolderPath = folderName; try { using(EndianBinaryReader reader = new EndianBinaryReader(File.Open(ofd.FileName, FileMode.Open, FileAccess.Read, FileShare.ReadWrite), Endian.Big)) { newMenu.Load(reader); LoadedFile = newMenu; } } catch (Exception ex) { Console.WriteLine("Exception while loading: {0}", ex.ToString()); ApplicationStatus = string.Format("Exception while loading: {0}", ex.ToString()); } ApplicationStatus = string.Format("Loaded {0}", ofd.SafeFileName); } } ///

User has requested that we close the current file. Save if applicable, then close.

private void CloseFile() { if (LoadedFile == null) return; ConfirmSaveDesireAndSave(); ApplicationStatus = string.Format("Unloaded {0}", LoadedFile.FileName); LoadedFile = null; } ///

User has requested that we save the current file.

private void SaveFile() { if (LoadedFile == null) throw new InvalidOperationException("No file loaded to save!"); if (string.IsNullOrEmpty(LoadedFile.FolderPath) || string.IsNullOrEmpty(LoadedFile.FileName)) { // Invoke a save-file dialog and make them choose where to save it. SaveFileAs(); } else { SaveFileAtPath(Path.Combine(LoadedFile.FolderPath, LoadedFile.FileName)); } } ///

User has requested that we save the current file under a different name.

private void SaveFileAs() { SaveFileDialog sfd = new SaveFileDialog(); sfd.RestoreDirectory = true; sfd.ValidateNames = true; sfd.Filter = "Debug Menu Files (*.dat)|*.dat|All Files (*.*)|*.*"; if(sfd.ShowDialog() == true) { SaveFileAtPath(sfd.FileName); } } private void SaveFileAtPath(string filePath) { if (LoadedFile == null) throw new InvalidOperationException("No file loaded to save as!"); string folderName = Path.GetDirectoryName(filePath); string fileName = Path.GetFileName(filePath); LoadedFile.FileName = fileName; LoadedFile.FolderPath = folderName; UpdateWindowTitle(); ApplicationStatus = string.Format("Saving file {0}...", fileName); try { using(EndianBinaryWriter writer = new EndianBinaryWriter(File.Open(filePath, FileMode.OpenOrCreate, FileAccess.Write, FileShare.ReadWrite), Endian.Big)) { LoadedFile.Save(writer); } } catch (Exception ex) { ApplicationStatus = string.Format("Exception while saving: {0}", ex.ToString()); } LoadedFile.FileName = fileName; ApplicationStatus = string.Format("Saved file {0}", LoadedFile.FileName); } private void ConfirmSaveDesireAndSave() { if (LoadedFile != null) { // Ask the user if they'd like to save before making a new file. if (System.Windows.MessageBox.Show("Save changes to the file?", ApplicationName, System.Windows.MessageBoxButton.YesNo, System.Windows.MessageBoxImage.Question) == System.Windows.MessageBoxResult.Yes) { SaveFile(); } } } #endregion #region Categories private void AddCategory() { if (LoadedFile == null) throw new InvalidOperationException("Cannot add category when no file is loaded!"); Category newCategory = new Category(); LoadedFile.Categories.Add(newCategory); SelectedCategory = newCategory; } private void RemoveCategory() { if(LoadedFile == null) throw new InvalidOperationException("Cannot remove category when no file is loaded!"); // Get the index of the currently selected one so that when we remove it, we can select the category one before. int index = LoadedFile.Categories.IndexOf(SelectedCategory); LoadedFile.Categories.RemoveAt(index); // Removing the category shifts everyone up by one, so if its still in a valid range, use it, otherwise subtract one (ie: was last thing) // and ensure that is in range. if(index >= LoadedFile.Categories.Count) index--; if(index >= 0) SelectedCategory = LoadedFile.Categories[index]; } #endregion #region Entries private void CreateNewEntry() { if (LoadedFile == null || SelectedCategory == null) throw new InvalidOperationException("Cannot add entry to unloaded file or null category!"); CategoryEntry entry = new CategoryEntry(); SelectedCategory.Entries.Add(entry); // Add it to our collection view source so it shows up in the filtered results... ((List)CollViewSource.Source).Add(entry); // Clear the list of currently selected entries and then add the newly created one as the only selected one. SelectedEntries.Clear(); SelectedEntries.Add(entry); } private void RemoveEntry() { if (LoadedFile == null || SelectedCategory == null) throw new InvalidOperationException("Cannot remove entry from unloaded file or null category!"); if(SelectedEntries.Count > 0) { // Get the index of the currently selected ones - presumably the first. int index = SelectedCategory.Entries.IndexOf(SelectedEntries[0]); // Remove all selected foreach(var entry in SelectedEntries) SelectedCategory.Entries.Remove(entry); SelectedEntries.Clear(); // It shifted us up by one or more, so if still in valid range, use it, otherwise we subtract one (ie: it was the last one) if (index >= SelectedCategory.Entries.Count) index--; if (index >= 0) SelectedEntries.Add(SelectedCategory.Entries[index]); } } #endregion private void AddFilter() { CollViewSource.Filter -= Filter; CollViewSource.Filter += Filter; } private void Filter(object sender, FilterEventArgs e) { var src = e.Item as CategoryEntry; e.Accepted = false; if (src == null) return; if (src.DisplayName.Contains(SearchFilter)) e.Accepted = true; else if (src.MapName.Contains(SearchFilter)) e.Accepted = true; } ///

The user has requested that we close the application. Save file (if required).

private void QuitApplication() { // Check if they want to save changes to the file before exiting. ConfirmSaveDesireAndSave(); Application.Current.MainWindow.Close(); } private void UpdateWindowTitle() { if (LoadedFile == null) { WindowTitle = ApplicationName; } else { WindowTitle = string.Format("{0} - {1}", LoadedFile.FileName, ApplicationName); } } internal void OnWindowClosing(object sender, CancelEventArgs e) { // Check to see if they want to save changes to the file before exiting. ConfirmSaveDesireAndSave(); } protected void OnPropertyChanged(string propertyName) { if (PropertyChanged != null) PropertyChanged(this, new PropertyChangedEventArgs(propertyName)); } } }