BililiveRecorder/BililiveRecorder.Cli/Program.cs

using System;
using System.ComponentModel.DataAnnotations;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using Autofac;
using BililiveRecorder.Core;
using BililiveRecorder.Core.Config;
using BililiveRecorder.FlvProcessor;
using CommandLine;
using NLog;

namespace BililiveRecorder.Cli
{
    class Program
    {
        private static IContainer Container { get; set; }
        private static ILifetimeScope RootScope { get; set; }
        private static IRecorder Recorder { get; set; }
        private static readonly Logger logger = LogManager.GetCurrentClassLogger();

        static void Main(string[] _)
        {
            var builder = new ContainerBuilder();
            builder.RegisterModule<FlvProcessorModule>();
            builder.RegisterModule<CoreModule>();
            builder.RegisterType<CommandConfigV1>().As<ConfigV1>().InstancePerMatchingLifetimeScope("recorder_root");
            Container = builder.Build();

            RootScope = Container.BeginLifetimeScope("recorder_root");
            Recorder = RootScope.Resolve<IRecorder>();
            if (!Recorder.Initialize(System.IO.Directory.GetCurrentDirectory()))
            {
                Console.WriteLine("Initialize Error");
                return;
            }

            Parser.Default
                .ParseArguments<CommandConfigV1>(() => (CommandConfigV1)Recorder.Config, Environment.GetCommandLineArgs())
                .WithParsed(Run);
        }

        private static void Run(ConfigV1 option)
        {
            foreach (var room in option.RoomList)
            {
                if (Recorder.Where(r => r.RoomId == room.Roomid).Count() == 0)
                {
                    Recorder.AddRoom(room.Roomid);
                }
            }
            
            logger.Info("开始录播");
            Task.WhenAll(Recorder.Select(x => Task.Run(() => x.Start()))).Wait();
            Console.CancelKeyPress += (sender, e) =>
            {
                Task.WhenAll(Recorder.Select(x => Task.Run(() => x.StopRecord()))).Wait();
                logger.Info("停止录播");
            };
            while (true)
            {
                Thread.Sleep(TimeSpan.FromSeconds(10));
            }
        }
    }

    class ConfigV1Metadata
    {
        [Option('o', "dir", Default = ".", HelpText = "Output directory", Required = false)]
        [Utils.DoNotCopyProperty]
        public object WorkDirectory { get; set; }

        [Option("cookie", HelpText = "Provide custom cookies", Required = false)]
        public object Cookie { get; set; }

        [Option("avoidtxy", HelpText = "Avoid Tencent Cloud server", Required = false)]
        public object AvoidTxy { get; set; }

        [Option("live_api_host", HelpText = "Use custom api host", Required = false)]
        public object LiveApiHost { get; set; }

        [Option("record_filename_format", HelpText = "Recording name format", Required = false)]
        public object RecordFilenameFormat { get; set; }



    }

    [MetadataType(typeof(ConfigV1Metadata))]
    class CommandConfigV1 : ConfigV1
    {
        [Option('i', "id", HelpText = "room id", Required = true)]
        [Utils.DoNotCopyProperty]
        public string _RoomList
        {
            set
            {
                var roomids = value.Split(',');
                RoomList.Clear();

                foreach (var roomid in roomids)
                {
                    var room = new RoomV1();
                    room.Roomid = Int32.Parse(roomid);
                    room.Enabled = false;
                    RoomList.Add(room);
                }
            }

        }
    }
}