Brotli-Builder/BrotliCalc/Commands/CmdGenerateMarkers.cs

using System;
using System.IO;
using System.Text;
using BrotliCalc.Commands.Base;
using BrotliCalc.Helpers;
using BrotliLib.Markers;

namespace BrotliCalc.Commands{
    class CmdGenerateMarkers : CmdAbstractFileMapper.Compressed{
        public override string FullName => "generate-markers";
        public override string ShortName => "gm";

        protected override int ExtraArgumentCount => 2;
        protected override string ExtraArgumentDesc => "<s|simple|v|verbose> <includeBitCounts>";

        protected override string WorkDesc => "Extracted markers from";
        protected override string AppendFileName => ".txt";

        private MarkerLevel markerLevel;
        private bool includeBitCounts;

        protected override void Setup(string[] args){
            markerLevel = args[0] switch{
                var x when x == "s" || x == "simple"  => MarkerLevel.Simple,
                var x when x == "v" || x == "verbose" => MarkerLevel.Verbose,
                _ => throw new ArgumentException("Invalid marker type: " + args[0])
            };

            includeBitCounts = args[1] switch{
                var x when x == "y" || x == "yes" || x == "true" => true,
                var x when x == "n" || x == "no" || x == "false" => false,
                _ => throw new ArgumentException("Invalid includeBitCounts value: " + args[1])
            };
        }

        protected override void MapFile(BrotliFileGroup group, BrotliFile.Compressed file, FileStream output){
            var markerRoot = file.GenerateMarkers(markerLevel);

            using var stream = new StreamWriter(output, Encoding.UTF8){
                NewLine = "\n"
            };

            markerRoot.WriteText(stream, includeBitCounts);
        }
    }
}