mirror of
https://github.com/BililiveRecorder/BililiveRecorder.git
synced 2024-11-16 03:32:20 +08:00
463 lines
19 KiB
C#
463 lines
19 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.CommandLine;
|
|
using System.CommandLine.Invocation;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using System.Security.Authentication;
|
|
using System.Security.Cryptography;
|
|
using System.Security.Cryptography.X509Certificates;
|
|
using System.Text.RegularExpressions;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using BililiveRecorder.Cli.Configure;
|
|
using BililiveRecorder.Core;
|
|
using BililiveRecorder.Core.Config;
|
|
using BililiveRecorder.Core.Config.V3;
|
|
using BililiveRecorder.DependencyInjection;
|
|
using BililiveRecorder.ToolBox;
|
|
using BililiveRecorder.Web;
|
|
using Microsoft.AspNetCore.Hosting;
|
|
using Microsoft.AspNetCore.Server.Kestrel.Core;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Hosting;
|
|
using Serilog;
|
|
using Serilog.Core;
|
|
using Serilog.Events;
|
|
using Serilog.Exceptions;
|
|
using Serilog.Formatting.Compact;
|
|
|
|
namespace BililiveRecorder.Cli
|
|
{
|
|
internal class Program
|
|
{
|
|
private static int Main(string[] args)
|
|
{
|
|
ServicePointManager.Expect100Continue = false;
|
|
|
|
var cmd_run = new Command("run", "Run BililiveRecorder in standard mode")
|
|
{
|
|
new Option<string?>(new []{ "--http-bind", "--bind", "-b" }, () => null, "Bind address for http service"),
|
|
// new Option<int?>(new []{ "--http-port", "--port", "-p" }, () => null, "Port number for http service"),
|
|
new Option<LogEventLevel>(new []{ "--loglevel", "--log", "-l" }, () => LogEventLevel.Information, "Minimal log level output to console"),
|
|
new Option<LogEventLevel>(new []{ "--logfilelevel", "--flog" }, () => LogEventLevel.Debug, "Minimal log level output to file"),
|
|
new Option<string?>(new []{ "--cert-pem-path", "--pem" }, "Path of the certificate pem file"),
|
|
new Option<string?>(new []{ "--cert-key-path", "--key" }, "Path of the certificate key file"),
|
|
new Option<string?>(new []{ "--cert-pfx-path", "--pfx" }, "Path of the certificate pfx file"),
|
|
new Option<string?>(new []{ "--cert-password"}, "Password of the certificate"),
|
|
|
|
new Argument<string>("path"),
|
|
};
|
|
cmd_run.AddAlias("r");
|
|
cmd_run.Handler = CommandHandler.Create<RunModeArguments>(RunConfigModeAsync);
|
|
|
|
var cmd_portable = new Command("portable", "Run BililiveRecorder in config-less mode")
|
|
{
|
|
new Option<string?>(new []{ "--http-bind", "--bind", "-b" }, () => null, "Bind address for http service"),
|
|
// new Option<int?>(new []{ "--http-port", "--port", "-p" }, () => null, "Port number for http service"),
|
|
new Option<LogEventLevel>(new []{ "--loglevel", "--log", "-l" }, () => LogEventLevel.Information, "Minimal log level output to console"),
|
|
new Option<LogEventLevel>(new []{ "--logfilelevel", "--flog" }, () => LogEventLevel.Debug, "Minimal log level output to file"),
|
|
new Option<string?>(new []{ "--cert-pem-path", "--pem" }, "Path of the certificate pem file"),
|
|
new Option<string?>(new []{ "--cert-key-path", "--key" }, "Path of the certificate key file"),
|
|
new Option<string?>(new []{ "--cert-pfx-path", "--pfx" }, "Path of the certificate pfx file"),
|
|
new Option<string?>(new []{ "--cert-password"}, "Password of the certificate"),
|
|
|
|
new Option<RecordMode>(new []{ "--record-mode", "--mode" }, () => RecordMode.Standard, "Recording mode"),
|
|
new Option<string>(new []{ "--cookie", "-c" }, "Cookie string for api requests"),
|
|
new Option<string>(new []{ "--filename", "-f" }, "File name format"),
|
|
new Option<PortableModeArguments.PortableDanmakuMode>(new []{ "--danmaku", "-d" }, "Flags for danmaku recording"),
|
|
new Option<string>("--webhook-url", "URL of webhoook"),
|
|
new Option<string>("--live-api-host"),
|
|
new Argument<string>("output-path"),
|
|
new Argument<int[]>("room-ids", () => Array.Empty<int>())
|
|
};
|
|
cmd_portable.AddAlias("p");
|
|
cmd_portable.Handler = CommandHandler.Create<PortableModeArguments>(RunPortableModeAsync);
|
|
|
|
var root = new RootCommand("A Stream Recorder For Bilibili Live")
|
|
{
|
|
cmd_run,
|
|
cmd_portable,
|
|
new ConfigureCommand(),
|
|
new ToolCommand()
|
|
};
|
|
|
|
return root.Invoke(args);
|
|
}
|
|
|
|
private static async Task<int> RunConfigModeAsync(RunModeArguments args)
|
|
{
|
|
var path = Path.GetFullPath(args.Path);
|
|
|
|
using var logger = BuildLogger(args.LogLevel, args.LogFileLevel);
|
|
Log.Logger = logger;
|
|
|
|
path = Path.GetFullPath(path);
|
|
var config = ConfigParser.LoadFrom(path);
|
|
if (config is null)
|
|
{
|
|
logger.Error("Initialize Error");
|
|
return -1;
|
|
}
|
|
|
|
config.Global.WorkDirectory = path;
|
|
|
|
var serviceProvider = BuildServiceProvider(config, logger);
|
|
|
|
return await RunRecorderAsync(serviceProvider, args);
|
|
}
|
|
|
|
private static async Task<int> RunPortableModeAsync(PortableModeArguments args)
|
|
{
|
|
using var logger = BuildLogger(args.LogLevel, args.LogFileLevel);
|
|
Log.Logger = logger;
|
|
|
|
var config = new ConfigV3()
|
|
{
|
|
DisableConfigSave = true,
|
|
};
|
|
|
|
{
|
|
var global = config.Global;
|
|
|
|
if (!string.IsNullOrWhiteSpace(args.Cookie))
|
|
global.Cookie = args.Cookie;
|
|
|
|
if (!string.IsNullOrWhiteSpace(args.LiveApiHost))
|
|
global.LiveApiHost = args.LiveApiHost;
|
|
|
|
if (!string.IsNullOrWhiteSpace(args.Filename))
|
|
global.FileNameRecordTemplate = args.Filename;
|
|
|
|
if (!string.IsNullOrWhiteSpace(args.WebhookUrl))
|
|
global.WebHookUrlsV2 = args.WebhookUrl;
|
|
|
|
global.RecordMode = args.RecordMode;
|
|
|
|
var danmaku = args.Danmaku;
|
|
global.RecordDanmaku = danmaku != PortableModeArguments.PortableDanmakuMode.None;
|
|
global.RecordDanmakuSuperChat = danmaku.HasFlag(PortableModeArguments.PortableDanmakuMode.SuperChat);
|
|
global.RecordDanmakuGuard = danmaku.HasFlag(PortableModeArguments.PortableDanmakuMode.Guard);
|
|
global.RecordDanmakuGift = danmaku.HasFlag(PortableModeArguments.PortableDanmakuMode.Gift);
|
|
global.RecordDanmakuRaw = danmaku.HasFlag(PortableModeArguments.PortableDanmakuMode.RawData);
|
|
|
|
global.WorkDirectory = Path.GetFullPath(args.OutputPath);
|
|
config.Rooms = args.RoomIds.Select(x => new RoomConfig { RoomId = x, AutoRecord = true }).ToList();
|
|
}
|
|
|
|
var serviceProvider = BuildServiceProvider(config, logger);
|
|
|
|
return await RunRecorderAsync(serviceProvider, args);
|
|
}
|
|
|
|
private static async Task<int> RunRecorderAsync(IServiceProvider serviceProvider, SharedArguments sharedArguments)
|
|
{
|
|
var logger = serviceProvider.GetRequiredService<ILogger>();
|
|
IRecorder recorderAccessProxy(IServiceProvider x) => serviceProvider.GetRequiredService<IRecorder>();
|
|
|
|
// recorder setup done
|
|
// check if web service required
|
|
IHost? host = null;
|
|
if (sharedArguments.HttpBind is null)
|
|
{
|
|
logger.Information("Web API not enabled");
|
|
}
|
|
else
|
|
{
|
|
host = new HostBuilder()
|
|
.UseSerilog(logger: logger)
|
|
.ConfigureServices(services =>
|
|
{
|
|
services.AddSingleton(recorderAccessProxy);
|
|
})
|
|
.ConfigureWebHost(webBuilder =>
|
|
{
|
|
webBuilder
|
|
.UseKestrel(option =>
|
|
{
|
|
(var scheme, var host, var port) = ParseBindArgument(sharedArguments.HttpBind, logger);
|
|
|
|
if (host.Equals("localhost", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
option.ListenLocalhost(port, ListenConfigure);
|
|
}
|
|
else if (IPAddress.TryParse(host, out var ip))
|
|
{
|
|
option.Listen(ip, port, ListenConfigure);
|
|
}
|
|
else
|
|
{
|
|
option.ListenAnyIP(port, ListenConfigure);
|
|
}
|
|
|
|
void ListenConfigure(ListenOptions listenOptions)
|
|
{
|
|
if (scheme == "https")
|
|
{
|
|
listenOptions.UseHttps(LoadCertificate(sharedArguments, logger) ?? GenerateSelfSignedCertificate(logger), https =>
|
|
{
|
|
https.SslProtocols = SslProtocols.Tls11 | SslProtocols.Tls12 | SslProtocols.Tls13;
|
|
});
|
|
}
|
|
}
|
|
})
|
|
.UseStartup<Startup>();
|
|
})
|
|
.Build();
|
|
}
|
|
|
|
ConsoleCancelEventHandler p = null!;
|
|
var cts = new CancellationTokenSource();
|
|
p = (sender, e) =>
|
|
{
|
|
logger.Information("Ctrl+C pressed. Exiting");
|
|
Console.CancelKeyPress -= p;
|
|
e.Cancel = true;
|
|
cts.Cancel();
|
|
};
|
|
Console.CancelKeyPress += p;
|
|
|
|
IRecorder? recorder = null;
|
|
|
|
try
|
|
{
|
|
var token = cts.Token;
|
|
if (host is not null)
|
|
{
|
|
try
|
|
{
|
|
await host.StartAsync(token);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
logger.Fatal(ex, "Failed to start web server.");
|
|
return -1;
|
|
}
|
|
logger.Information("Web host started.");
|
|
|
|
recorder = serviceProvider.GetRequiredService<IRecorder>();
|
|
|
|
await Task.WhenAny(Task.Delay(-1, token), host.WaitForShutdownAsync()).ConfigureAwait(false);
|
|
|
|
logger.Information("Shutdown in progress.");
|
|
|
|
await host.StopAsync().ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
recorder = serviceProvider.GetRequiredService<IRecorder>();
|
|
await Task.Delay(-1, token).ConfigureAwait(false);
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
recorder?.Dispose();
|
|
// TODO 修复这里 Dispose 之后不会停止房间继续初始化
|
|
}
|
|
await Task.Delay(1000 * 3).ConfigureAwait(false);
|
|
return 0;
|
|
}
|
|
|
|
private static X509Certificate2? LoadCertificate(SharedArguments arguments, ILogger logger)
|
|
{
|
|
if (arguments.CertPfxPath is not null)
|
|
{
|
|
if (arguments.CertPemPath is not null || arguments.CertKeyPath is not null)
|
|
{
|
|
logger.Warning("Both cert-pfx and cert-pem/cert-key are specified. Using cert-pfx.");
|
|
}
|
|
|
|
if (!File.Exists(arguments.CertPfxPath))
|
|
{
|
|
logger.Error("Certificate file {Path} not found.", arguments.CertPfxPath);
|
|
return null;
|
|
}
|
|
|
|
return new X509Certificate2(arguments.CertPfxPath, arguments.CertPassword);
|
|
}
|
|
else if (arguments.CertPemPath is not null || arguments.CertKeyPath is not null)
|
|
{
|
|
if (arguments.CertPemPath is null)
|
|
{
|
|
logger.Error("Certificate PEM file not specified.");
|
|
return null;
|
|
}
|
|
|
|
if (arguments.CertKeyPath is null)
|
|
{
|
|
logger.Error("Certificate key file not specified.");
|
|
return null;
|
|
}
|
|
|
|
if (!File.Exists(arguments.CertPemPath))
|
|
{
|
|
logger.Error("Certificate PEM file {Path} not found.", arguments.CertPemPath);
|
|
return null;
|
|
}
|
|
|
|
if (!File.Exists(arguments.CertKeyPath))
|
|
{
|
|
logger.Error("Certificate key file {Path} not found.", arguments.CertKeyPath);
|
|
return null;
|
|
}
|
|
|
|
var cert = arguments.CertPassword is null
|
|
? X509Certificate2.CreateFromPemFile(arguments.CertPemPath, arguments.CertKeyPath)
|
|
: X509Certificate2.CreateFromEncryptedPemFile(arguments.CertPemPath, arguments.CertPassword, arguments.CertKeyPath);
|
|
|
|
return new X509Certificate2(cert.Export(X509ContentType.Pfx));
|
|
}
|
|
else
|
|
{
|
|
logger.Debug("No certificate specified.");
|
|
return null;
|
|
}
|
|
}
|
|
|
|
private static X509Certificate2 GenerateSelfSignedCertificate(ILogger logger)
|
|
{
|
|
logger.Warning("使用录播姬生成的自签名证书");
|
|
|
|
var firstDayofCurrentYear = new DateTimeOffset(DateTime.Now.Year, 1, 1, 0, 0, 0, TimeSpan.Zero);
|
|
X509Certificate2? CA = null;
|
|
try
|
|
{
|
|
{
|
|
using var key = RSA.Create();
|
|
var req = new CertificateRequest("CN=自签名证书，每次启动都会重新生成", key, HashAlgorithmName.SHA256, RSASignaturePadding.Pkcs1);
|
|
req.CertificateExtensions.Add(new X509BasicConstraintsExtension(true, false, 0, false));
|
|
CA = new X509Certificate2(req.CreateSelfSigned(firstDayofCurrentYear, firstDayofCurrentYear.AddYears(10)).Export(X509ContentType.Pfx));
|
|
}
|
|
|
|
{
|
|
using var key = RSA.Create();
|
|
var req = new CertificateRequest("CN=B站录播姬", key, HashAlgorithmName.SHA256, RSASignaturePadding.Pkcs1);
|
|
|
|
var subjectAltName = new SubjectAlternativeNameBuilder();
|
|
subjectAltName.AddDnsName("BililiveRecorder");
|
|
subjectAltName.AddDnsName("localhost");
|
|
subjectAltName.AddIpAddress(IPAddress.Loopback);
|
|
subjectAltName.AddIpAddress(IPAddress.IPv6Loopback);
|
|
subjectAltName.AddDnsName("*.nip.io");
|
|
subjectAltName.AddDnsName("*.sslip.io");
|
|
req.CertificateExtensions.Add(subjectAltName.Build());
|
|
|
|
using var cert = req.Create(CA, firstDayofCurrentYear, firstDayofCurrentYear.AddYears(10), "BililiveRecorder".Select(x => (byte)x).ToArray());
|
|
using var withPrivateKey = cert.CopyWithPrivateKey(key);
|
|
return new X509Certificate2(withPrivateKey.Export(X509ContentType.Pfx));
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
CA?.Dispose();
|
|
}
|
|
}
|
|
|
|
private static (string schema, string host, int port) ParseBindArgument(string bind, ILogger logger)
|
|
{
|
|
if (int.TryParse(bind, out var value))
|
|
{
|
|
// 只传入了一个端口号
|
|
return ("http", "localhost", value);
|
|
}
|
|
|
|
var match = Regex.Match(bind, @"^(?<schema>https?):\/\/(?<host>[^\:\/\?\#]+)(?:\:(?<port>\d+))?(?:\/.*)?$", RegexOptions.Singleline | RegexOptions.CultureInvariant, TimeSpan.FromSeconds(5));
|
|
if (match.Success)
|
|
{
|
|
var schema = match.Groups["schema"].Value.ToLower();
|
|
var host = match.Groups["host"].Value;
|
|
var port = match.Groups["port"].Success ? int.Parse(match.Groups["port"].Value) : 2356;
|
|
return (schema, host, port);
|
|
}
|
|
else
|
|
{
|
|
logger.Warning("侦听参数解析失败，使用默认值 {DefaultBindLocation}", "http://localhost:2356");
|
|
return ("http", "localhost", 2356);
|
|
}
|
|
}
|
|
|
|
private static IServiceProvider BuildServiceProvider(ConfigV3 config, ILogger logger) => new ServiceCollection()
|
|
.AddSingleton(logger)
|
|
.AddFlv()
|
|
.AddRecorderConfig(config)
|
|
.AddRecorder()
|
|
.BuildServiceProvider();
|
|
|
|
private static Logger BuildLogger(LogEventLevel logLevel, LogEventLevel logFileLevel) => new LoggerConfiguration()
|
|
.MinimumLevel.Verbose()
|
|
.Enrich.WithProcessId()
|
|
.Enrich.WithThreadId()
|
|
.Enrich.WithThreadName()
|
|
.Enrich.FromLogContext()
|
|
.Enrich.WithExceptionDetails()
|
|
.Destructure.AsScalar<IPAddress>()
|
|
.Destructure.ByTransforming<Flv.Xml.XmlFlvFile.XmlFlvFileMeta>(x => new
|
|
{
|
|
x.Version,
|
|
x.ExportTime,
|
|
x.FileSize,
|
|
x.FileCreationTime,
|
|
x.FileModificationTime,
|
|
})
|
|
.WriteTo.Console(restrictedToMinimumLevel: logLevel, outputTemplate: "[{Timestamp:HH:mm:ss} {Level:u3}] [{RoomId}] {Message:lj}{NewLine}{Exception}")
|
|
.WriteTo.File(new CompactJsonFormatter(), "./logs/bilirec.txt", restrictedToMinimumLevel: logFileLevel, shared: true, rollingInterval: RollingInterval.Day, rollOnFileSizeLimit: true)
|
|
.CreateLogger();
|
|
|
|
public abstract class SharedArguments
|
|
{
|
|
public LogEventLevel LogLevel { get; set; } = LogEventLevel.Information;
|
|
|
|
public LogEventLevel LogFileLevel { get; set; } = LogEventLevel.Information;
|
|
|
|
public string? HttpBind { get; set; } = null;
|
|
|
|
// public int? HttpPort { get; set; } = null;
|
|
|
|
public string? CertPemPath { get; set; } = null;
|
|
|
|
public string? CertKeyPath { get; set; } = null;
|
|
|
|
public string? CertPfxPath { get; set; } = null;
|
|
|
|
public string? CertPassword { get; set; } = null;
|
|
}
|
|
|
|
public sealed class RunModeArguments : SharedArguments
|
|
{
|
|
public string Path { get; set; } = string.Empty;
|
|
}
|
|
|
|
public sealed class PortableModeArguments : SharedArguments
|
|
{
|
|
public RecordMode RecordMode { get; set; } = RecordMode.Standard;
|
|
|
|
public string OutputPath { get; set; } = string.Empty;
|
|
|
|
public string? Cookie { get; set; }
|
|
|
|
public string? LiveApiHost { get; set; }
|
|
|
|
public string? Filename { get; set; }
|
|
|
|
public string? WebhookUrl { get; set; }
|
|
|
|
public PortableDanmakuMode Danmaku { get; set; }
|
|
|
|
public IEnumerable<int> RoomIds { get; set; } = Enumerable.Empty<int>();
|
|
|
|
[Flags]
|
|
public enum PortableDanmakuMode
|
|
{
|
|
None = 0,
|
|
Danmaku = 1 << 0,
|
|
SuperChat = 1 << 1,
|
|
Guard = 1 << 2,
|
|
Gift = 1 << 3,
|
|
RawData = 1 << 4,
|
|
All = Danmaku | SuperChat | Guard | Gift | RawData
|
|
}
|
|
}
|
|
}
|
|
}
|