SabreTools/DATabase/TrimMerge.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;

using SabreTools.Helper;

namespace SabreTools
{
	public class TrimMerge
	{
		// Instance variables
		private string _filename;
		private string _path;
		private bool _rename;
		private bool _forceunpack;
		private Logger _logger;

		/// <summary>
		/// Create a new TrimMerge object
		/// </summary>
		/// <param name="filename">Name of the file or folder to be processed</param>
		/// <param name="path">Root path to use for trimming</param>
		/// <param name="rename">True if games should be renamed into a uniform string, false otherwise</param>
		/// <param name="forceunpack">True if forcepacking="unzip" should be set on the output, false otherwise</param>
		/// <param name="logger">Logger object for console and file output</param>
		public TrimMerge(string filename, string path, bool rename, bool forceunpack, Logger logger)
		{
			_filename = filename.Replace("\"", "");
			_path = path;
			_rename = rename;
			_forceunpack = forceunpack;
			_logger = logger;
		}

		/// <summary>
		/// Trim and process the given DAT or folder of DATs
		/// </summary>
		/// <returns>True if the DAT could be updated, false otherwise</returns>
		public bool Process()
		{
			// If file doesn't exist, error and return
			if (!File.Exists(_filename) && !Directory.Exists(_filename))
			{
				_logger.Error("File or folder '" + _filename + "' doesn't exist");
				return false;
			}

			// We want the full path of the file, just in case
			_filename = Path.GetFullPath(_filename);

			// If it's a single file, handle it as such
			if (!Directory.Exists(_filename) && File.Exists(_filename))
			{
				_logger.Log("File found: " + _filename);
				ProcessDAT(_filename, _path, _rename);
			}
			// If it's a directory, loop through the files and see if any are DATs
			else if (Directory.Exists(_filename))
			{
				// Make sure the path ends with the proper character
				if (!_filename.EndsWith(Path.DirectorySeparatorChar.ToString()))
				{
					_filename += Path.DirectorySeparatorChar;
				}

				_logger.Log("Directory found: " + _filename);
				foreach (string file in Directory.EnumerateFiles(_filename, "*", SearchOption.AllDirectories))
				{
					_logger.Log("File found: " + file);
					ProcessDAT(file, _path, _rename);
				}
			}

			return true;
		}

		/// <summary>
		/// Import the existing DAT(s)
		/// </summary>
		/// <param name="filename">Name of the file to be processed</param>
		/// <param name="path">The base path to be used for comparison</param>
		/// <param name="rename">True if roms are to be renamed</param>
		private void ProcessDAT(string filename, string path, bool rename)
		{
			DatData datdata = new DatData
			{
				ForcePacking = (_forceunpack ? ForcePacking.Unzip : ForcePacking.None),
				OutputFormat = RomManipulation.GetOutputFormat(filename),
			};
			datdata = RomManipulation.Parse(filename, 0, 0, datdata, _logger);

			// Trim all file names according to the path that's set
			List<string> keys = datdata.Roms.Keys.ToList();
			foreach (string key in keys)
			{
				List<RomData> newroms = new List<RomData>();
				foreach (RomData rom in datdata.Roms[key])
				{
					RomData newrom = rom;

					// If we are in single game mode, rename all games
					if (rename)
					{
						newrom.Game = "!";
					}

					// Windows max name length is 260
					int usableLength = 260 - newrom.Game.Length - _path.Length;
					if (newrom.Name.Length > usableLength)
					{
						string ext = Path.GetExtension(newrom.Name);
						newrom.Name = newrom.Name.Substring(0, usableLength - ext.Length);
						newrom.Name += ext;
					}

					newroms.Add(newrom);
				}
				datdata.Roms[key] = newroms;
			}

			// Now write the file out accordingly
			Output.WriteDatfile(datdata, Path.GetDirectoryName(filename), _logger);

			// Remove the original file if different and inform the user
			if (filename != Style.CreateOutfileName(" ", datdata).Remove(0, 1))
			{
				try
				{
					File.Delete(filename);
					_logger.Log("Original file \"" + filename + "\" deleted");
				}
				catch (Exception ex)
				{
					_logger.Error(ex.ToString());
				}
			}
		}
	}
}