SabreTools/RombaSharp/Features/Dir2Dat.cs

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

using SabreTools.Core;
using SabreTools.Core.Tools;
using SabreTools.DatFiles;
using SabreTools.DatTools;
using SabreTools.FileTypes;
using SabreTools.Filtering;
using SabreTools.Help;
using SabreTools.IO;

namespace RombaSharp.Features
{
    internal class Dir2Dat : BaseFeature
    {
        public const string Value = "Dir2Dat";

        public Dir2Dat()
        {
            Name = Value;
            Flags = new List<string>() { "dir2dat" };
            Description = "Creates a DAT file for the specified input directory and saves it to the -out filename.";
            _featureType = ParameterType.Flag;
            LongDescription = "Creates a DAT file for the specified input directory and saves it to the -out filename.";
            Features = new Dictionary<string, Feature>();

            AddFeature(OutStringInput);
            AddFeature(SourceStringInput);
            AddFeature(NameStringInput); // Defaults to "untitled"
            AddFeature(DescriptionStringInput);
        }

        public override void ProcessFeatures(Dictionary<string, Feature> features)
        {
            base.ProcessFeatures(features);

            // Get feature flags
            string name = GetString(features, NameStringValue);
            string description = GetString(features, DescriptionStringValue);
            string source = GetString(features, SourceStringValue);
            string outdat = GetString(features, OutStringValue);

            // Ensure the output directory
            outdat.Ensure(create: true);

            // Check that all required directories exist
            if (!Directory.Exists(source))
            {
                logger.Error($"File '{source}' does not exist!");
                return;
            }

            // Create and write the encapsulating datfile
            DatFile datfile = DatFile.Create();
            datfile.Header.Name = string.IsNullOrWhiteSpace(name) ? "untitled" : name;
            datfile.Header.Description = description;
            DatFromDir.PopulateFromDir(datfile, source, asFiles: TreatAsFile.NonArchive);
            Remover remover = new Remover();
            remover.PopulateExclusionsFromList(new List<string> 
            {
                "DatItem.SHA256",
                "DatItem.SHA384",
                "DatItem.SHA512",
                "DatItem.SpamSum",
            });
            remover.ApplyRemovals(datfile);
            Writer.Write(datfile, outdat);
        }
    }
}