BililiveRecorder/BililiveRecorder.ToolBox/Commands/Analyze.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.IO.Pipelines;
using System.Linq;
using System.Threading.Tasks;
using BililiveRecorder.Flv;
using BililiveRecorder.Flv.Amf;
using BililiveRecorder.Flv.Grouping;
using BililiveRecorder.Flv.Parser;
using BililiveRecorder.Flv.Pipeline;
using BililiveRecorder.Flv.Writer;
using Microsoft.Extensions.DependencyInjection;
using Serilog;

namespace BililiveRecorder.ToolBox.Commands
{
    public class AnalyzeRequest : ICommandRequest<AnalyzeResponse>
    {
        public string Input { get; set; } = string.Empty;
    }

    public class AnalyzeResponse
    {
        public string InputPath { get; set; } = string.Empty;

        public bool NeedFix { get; set; }
        public bool Unrepairable { get; set; }

        public int OutputFileCount { get; set; }

        public int IssueTypeOther { get; set; }
        public int IssueTypeUnrepairable { get; set; }
        public int IssueTypeTimestampJump { get; set; }
        public int IssueTypeDecodingHeader { get; set; }
        public int IssueTypeRepeatingData { get; set; }
    }

    public class AnalyzeHandler : ICommandHandler<AnalyzeRequest, AnalyzeResponse>
    {
        private static readonly ILogger logger = Log.ForContext<AnalyzeHandler>();

        public Task<AnalyzeResponse> Handle(AnalyzeRequest request) => this.Handle(request, null);

        public async Task<AnalyzeResponse> Handle(AnalyzeRequest request, Func<double, Task>? progress)
        {
            var inputPath = Path.GetFullPath(request.Input);

            var memoryStreamProvider = new DefaultMemoryStreamProvider();
            var tagWriter = new AnalyzeMockFlvTagWriter();
            var comments = new List<ProcessingComment>();
            var context = new FlvProcessingContext();
            var session = new Dictionary<object, object?>();

            {
                using var inputStream = File.OpenRead(inputPath);

                using var grouping = new TagGroupReader(new FlvTagPipeReader(PipeReader.Create(inputStream), memoryStreamProvider, skipData: false, logger: logger));
                using var writer = new FlvProcessingContextWriter(tagWriter: tagWriter, allowMissingHeader: true);
                var pipeline = new ProcessingPipelineBuilder(new ServiceCollection().BuildServiceProvider()).AddDefault().AddRemoveFillerData().Build();

                var count = 0;
                while (true)
                {
                    var group = await grouping.ReadGroupAsync(default).ConfigureAwait(false);
                    if (group is null)
                        break;

                    context.Reset(group, session);
                    pipeline(context);

                    if (context.Comments.Count > 0)
                    {
                        comments.AddRange(context.Comments);
                        logger.Debug("分析逻辑输出 {@Comments}", context.Comments);
                    }

                    await writer.WriteAsync(context).ConfigureAwait(false);

                    foreach (var action in context.Actions)
                        if (action is PipelineDataAction dataAction)
                            foreach (var tag in dataAction.Tags)
                                tag.BinaryData?.Dispose();

                    if (count++ % 10 == 0)
                    {
                        progress?.Invoke((double)inputStream.Position / inputStream.Length);
                    }
                }
            }

            var countableComments = comments.Where(x => x.T != CommentType.Logging);

            var response = new AnalyzeResponse
            {
                InputPath = inputPath,

                NeedFix = tagWriter.OutputFileCount != 1 || countableComments.Any(),
                Unrepairable = countableComments.Any(x => x.T == CommentType.Unrepairable),

                OutputFileCount = tagWriter.OutputFileCount,

                IssueTypeOther = countableComments.Count(x => x.T == CommentType.Other),
                IssueTypeUnrepairable = countableComments.Count(x => x.T == CommentType.Unrepairable),
                IssueTypeTimestampJump = countableComments.Count(x => x.T == CommentType.TimestampJump),
                IssueTypeDecodingHeader = countableComments.Count(x => x.T == CommentType.DecodingHeader),
                IssueTypeRepeatingData = countableComments.Count(x => x.T == CommentType.RepeatingData)
            };

            return response;
        }

        public void PrintResponse(AnalyzeResponse response)
        {
            Console.Write("Input: ");
            Console.WriteLine(response.InputPath);

            Console.WriteLine(response.NeedFix ? "File needs repair" : "File doesn't need repair");

            if (response.Unrepairable)
                Console.WriteLine("File contains error(s) that are unrepairable (yet), please send sample to the author of this program.");

            Console.WriteLine("Will output {0} file(s) if repaired", response.OutputFileCount);

            Console.WriteLine("Types of error:");
            Console.Write("Other: ");
            Console.WriteLine(response.IssueTypeOther);
            Console.Write("Unrepairable: ");
            Console.WriteLine(response.IssueTypeUnrepairable);
            Console.Write("TimestampJump: ");
            Console.WriteLine(response.IssueTypeTimestampJump);
            Console.Write("DecodingHeader: ");
            Console.WriteLine(response.IssueTypeDecodingHeader);
            Console.Write("RepeatingData: ");
            Console.WriteLine(response.IssueTypeRepeatingData);
        }

        private class AnalyzeMockFlvTagWriter : IFlvTagWriter
        {
            public long FileSize => 0;
            public object? State => null;

            public int OutputFileCount { get; private set; }

            public bool CloseCurrentFile() => true;
            public Task CreateNewFile()
            {
                this.OutputFileCount++;
                return Task.CompletedTask;
            }

            public void Dispose() { }
            public Task OverwriteMetadata(ScriptTagBody metadata) => Task.CompletedTask;
            public Task WriteAlternativeHeaders(IEnumerable<Tag> tags) => Task.CompletedTask;
            public Task WriteTag(Tag tag) => Task.CompletedTask;
        }
    }
}