feat: 目前完成了订阅功能和 Xray-Core 的更新管理.

2025-01-07 09:50:02 +08:00
parent 78c7db692a
commit 99cb0a03a5
22 changed files with 1153 additions and 0 deletions
@@ -0,0 +1,116 @@
+using System.Buffers.Text;
+using System.Collections.Immutable;
+using System.Text;
+using Microsoft.Extensions.Logging;
+using Microsoft.Extensions.Options;
+using Proxd_man.Config;
+using Proxd_man.Config.Outbound;
+using Proxd_man.ShareLink;
+
+namespace Proxd_man.Subscribe;
+
+public class SubscribeManager
+{
+    
+    private readonly Dictionary<SubscribeSource, List<BaseOutboundSetting>> _subscribeOutbounds = new();
+    private CancellationTokenSource? _cts;
+    private readonly ILogger<SubscribeManager> _logger;
+    private readonly IOptionsMonitor<ProxdConfig> _config;
+
+    public SubscribeManager(ILogger<SubscribeManager> logger, IOptionsMonitor<ProxdConfig> config)
+    {
+        _logger = logger;
+        _config = config;
+
+        _config.OnChange((_, _) => Task.Run(async () => await UpdateSubscribes()));
+    }
+
+    public delegate void SubscribeUpdatedEventHandler(IDictionary<SubscribeSource, List<BaseOutboundSetting>> subscribeOutbounds);
+    public event SubscribeUpdatedEventHandler? SubscribeUpdated;
+
+    private ProxdConfig Config => _config.Get(null);
+    
+    public async Task UpdateSubscribes(CancellationToken cancellationToken = default)
+    {
+        var httpClient = new HttpClient();
+
+        _logger.LogInformation("开始更新订阅...");
+        
+        var newSubscribeOutbounds = new Dictionary<SubscribeSource, List<BaseOutboundSetting>>();
+        for (var i = 0; i < Config.Subscribe.Subscribes.Count; i++)
+        {
+            var subscribe = Config.Subscribe.Subscribes[i];
+            var outbounds = new List<BaseOutboundSetting>();
+
+            _logger.LogInformation("正在更新订阅 {subscribeName} ({current}/{total})...", subscribe.Name, i + 1, Config.Subscribe.Subscribes.Count);
+            try
+            {
+                var response = await httpClient.GetAsync(subscribe.Url, cancellationToken);
+                response.EnsureSuccessStatusCode();
+                var content = await response.Content.ReadAsStringAsync(cancellationToken);
+
+                if (Base64.IsValid(content))
+                {
+                    content = Encoding.UTF8.GetString(Convert.FromBase64String(content));
+                }
+                
+                var shareLinks = content.Split('\n').Select(l => l.Trim()).Where(l => !string.IsNullOrEmpty(l));
+                foreach (var shareLink in shareLinks)
+                {
+                    try
+                    {
+                        var outbound = ShareLinkParser.Parse(shareLink);
+                        outbounds.Add(outbound);
+                    }
+                    catch (InvalidShareLinkException e)
+                    {
+                        _logger.LogWarning(e, "解析分享链接失败: {shareLink}\n", shareLink);
+                    }
+                }
+            }
+            catch (Exception e)
+            {
+                _logger.LogWarning(e, "更新订阅失败: {subscribeName}", subscribe.Name);
+            }
+
+            newSubscribeOutbounds[subscribe] = outbounds;
+            _logger.LogInformation("订阅 {subscribeName} 更新完成. (共 {outboundCount} 个节点)", subscribe.Name, outbounds.Count);
+        }
+        
+        _subscribeOutbounds.Clear();
+        foreach (var (source, outbounds) in newSubscribeOutbounds)
+        {
+            _subscribeOutbounds[source] = outbounds;
+        }
+        
+        _logger.LogInformation("订阅更新完成.");
+        SubscribeUpdated?.Invoke(_subscribeOutbounds.ToImmutableDictionary());
+    }
+    
+    public IDictionary<SubscribeSource, List<BaseOutboundSetting>> GetSubscribeOutbounds()
+    {
+        return _subscribeOutbounds.ToImmutableDictionary();
+    }
+    
+    public void StartUpdateSubscribes()
+    {
+        var cts = new CancellationTokenSource();
+        
+        Task.Run(async () =>
+        {
+            while (!cts.IsCancellationRequested)
+            {
+                await UpdateSubscribes(cts.Token);
+                await Task.Delay(TimeSpan.FromHours(12), cts.Token);
+            }
+        }, cts.Token);
+        _cts = cts;
+        _logger.LogInformation("订阅更新任务已启动.");
+    }
+    
+    public void StopUpdateSubscribes()
+    {
+        _cts?.Cancel();
+    }
+    
+}