MbrCommandFactory.cs

using Hst.Imager.Core.Models;

namespace Hst.Imager.ConsoleApp
{
    using System.CommandLine;

    public static class MbrCommandFactory
    {
        public static Command CreateMbrCommand()
        {
            var command = new Command("mbr", "Master Boot Record.");

            command.AddCommand(CreateMbrInfo());
            command.AddCommand(CreateMbrInit());
            command.AddCommand(CreateMbrPart());

            return command;
        }

        private static Command CreateMbrInfo()
        {
            var pathArgument = new Argument<string>(
                name: "Path",
                description: "Path to physical drive or image file.");

            var showUnallocatedOption = new Option<bool>(
                new[] { "--unallocated", "-u" },
                description: "Show unallocated.",
                getDefaultValue: () => true);
            
            var command = new Command("info", "Display info about Master Boot Record.");
            command.SetHandler(CommandHandler.MbrInfo, pathArgument, showUnallocatedOption);
            command.AddArgument(pathArgument);
            command.AddOption(showUnallocatedOption);

            return command;
        }

        private static Command CreateMbrInit()
        {
            var pathArgument = new Argument<string>(
                name: "Path",
                description: "Path to physical drive or image file.");

            var mbrInitCommand = new Command("initialize", "Initialize disk with empty Master Boot Record.");
            mbrInitCommand.AddAlias("init");
            mbrInitCommand.SetHandler(CommandHandler.MbrInit, pathArgument);
            mbrInitCommand.AddArgument(pathArgument);

            return mbrInitCommand;
        }

        private static Command CreateMbrPart()
        {
            var partCommand = new Command("part", "Partition.");

            partCommand.AddCommand(CreateMbrPartAdd());
            partCommand.AddCommand(CreateMbrPartDel());
            partCommand.AddCommand(CreateMbrPartFormat());
            partCommand.AddCommand(CreateMbrPartExport());
            partCommand.AddCommand(CreateMbrPartImport());
            partCommand.AddCommand(CreateMbrPartClone());

            return partCommand;
        }

        private static Command CreateMbrPartAdd()
        {
            var pathArgument = new Argument<string>(
                name: "Path",
                description: "Path to physical drive or image file.");

            var typeArgument = new Argument<string>(
                name: "Type",
                description: "Type of the partition as name or number (e.g. name FAT32 or value 0xb for FAT32).");

            var sizeArgument = new Argument<string>(
                name: "Size",
                description: "Size of the partition.");

            var startSectorOption = new Option<long?>(
                new[] { "--start-sector", "-s" },
                description: "Start sector.");

            var activeOption = new Option<bool>(
                new[] { "--active", "-a" },
                description: "Set partition active (bootable).",
                getDefaultValue: () => false);

            var mbrPartAddCommand = new Command("add", "Add partition.");
            mbrPartAddCommand.SetHandler(CommandHandler.MbrPartAdd, pathArgument, typeArgument, sizeArgument,
                startSectorOption, activeOption);
            mbrPartAddCommand.AddArgument(pathArgument);
            mbrPartAddCommand.AddArgument(typeArgument);
            mbrPartAddCommand.AddArgument(sizeArgument);
            mbrPartAddCommand.AddOption(startSectorOption);
            mbrPartAddCommand.AddOption(activeOption);

            return mbrPartAddCommand;
        }

        private static Command CreateMbrPartDel()
        {
            var path = new Argument<string>(
                name: "Path",
                description: "Path to physical drive or image file.");

            var partitionNumber = new Argument<int>(
                name: "PartitionNumber",
                description: "Partition number to delete.");

            var command = new Command("delete", "Delete partition.");
            command.AddAlias("del");
            command.SetHandler(CommandHandler.MbrPartDel, path, partitionNumber);
            command.AddArgument(path);
            command.AddArgument(partitionNumber);

            return command;
        }

        private static Command CreateMbrPartFormat()
        {
            var pathArgument = new Argument<string>(
                name: "Path",
                description: "Path to physical drive or image file.");

            var partitionNumberArgument = new Argument<int>(
                name: "PartitionNumber",
                description: "Partition number to delete.");

            var nameArgument = new Argument<string>(
                name: "Name",
                description: "Name of the partition.");

            var fileSystemOption = new Option<string>(
                new[] { "--file-system", "-fs" },
                description: "File system format partition with.");

            var formatCommand = new Command("format", "Format partition.");
            formatCommand.SetHandler(CommandHandler.MbrPartFormat, pathArgument, partitionNumberArgument, nameArgument, fileSystemOption);
            formatCommand.AddArgument(pathArgument);
            formatCommand.AddArgument(partitionNumberArgument);
            formatCommand.AddArgument(nameArgument);

            return formatCommand;
        }

        private static Command CreateMbrPartExport()
        {
            var sourcePathArgument = new Argument<string>(
                name: "SourcePath",
                description: "Path to source physical drive or image file.");

            var partition = new Argument<string>(
                name: "Partition",
                description: "Partition to export from (\"2\" for partition number 2 or \"fat32\" for first fat32 partition).");

            var destinationPathArgument = new Argument<string>(
                name: "DestinationPath",
                description: "Path to destination physical drive or image file.");

            var command = new Command("export", "Export partition to a file.");
            command.SetHandler(CommandHandler.MbrPartExport, sourcePathArgument, partition,
                destinationPathArgument);
            command.AddArgument(sourcePathArgument);
            command.AddArgument(partition);
            command.AddArgument(destinationPathArgument);

            return command;
        }

        private static Command CreateMbrPartImport()
        {
            var sourcePathArgument = new Argument<string>(
                name: "SourcePath",
                description: "Path to source file.");

            var destinationPathArgument = new Argument<string>(
                name: "DestinationPath",
                description: "Path to destination physical drive or image file.");

            var partition = new Argument<string>(
                name: "Partition",
                description: "Partition to import to (\"2\" for partition number 2 or \"fat32\" for first fat32 partition).");

            var command = new Command("import", "Import partition from a file.");
            command.SetHandler(CommandHandler.MbrPartImport, sourcePathArgument, destinationPathArgument, partition);
            command.AddArgument(sourcePathArgument);
            command.AddArgument(destinationPathArgument);
            command.AddArgument(partition);

            return command;
        }
        
        private static Command CreateMbrPartClone()
        {
            var srcPathArgument = new Argument<string>(
                name: "SourcePath",
                description: "Path to source physical drive or image file.");

            var srcPartitionNumber = new Argument<int>(
                name: "Partition",
                description: "Source partition to clone from.");

            var destPathArgument = new Argument<string>(
                name: "DestinationPath",
                description: "Path to destination physical drive or image file.");

            var destPartitionNumber = new Argument<int>(
                name: "DestinationPartitionNumber",
                description: "Destination partition number to clone to.");

            var command = new Command("clone", "Clone partition from a physical drive or image file.");
            command.SetHandler(CommandHandler.MbrPartClone, srcPathArgument, srcPartitionNumber, destPathArgument, destPartitionNumber);
            command.AddArgument(srcPathArgument);
            command.AddArgument(srcPartitionNumber);
            command.AddArgument(destPathArgument);
            command.AddArgument(destPartitionNumber);

            return command;
        }
    }
}