feat(cli): add protocol filtering to fetch command

Author: ImuS663

src/Proxxi.Cli/Commands/Fetch/FetchCommand.cs

@@ -6,6 +6,7 @@
 using Proxxi.Core.ProxyWriters;
 using Proxxi.Plugin.Loader.Extensions;
 using Proxxi.Plugin.Loader.PluginLoaders;
+using Proxxi.Plugin.Sdk.Models;
 using Proxxi.Plugin.Sdk.ProxySources;
 
 using Spectre.Console;
@@ -29,6 +30,8 @@ public override async Task<int> ExecuteAsync(CommandContext context, FetchComman
         Stream stream;
         OutputFormat format;
 
+        var protocols = settings.Protocols ?? Protocols.None;
+
         if (settings.Output != null)
         {
             stream = File.OpenWrite(settings.Output);
@@ -54,14 +57,14 @@ public override async Task<int> ExecuteAsync(CommandContext context, FetchComman
                 if (writer is not IStreamProxyWriter streamProxyWriter)
                     throw new InvalidOperationException($"{format} output does not support stream mode.");
 
-                await FetchAndWriteProxiesAsync(streamProxySource, streamProxyWriter, ct);
+                await FetchAndWriteProxiesAsync(streamProxySource, streamProxyWriter, protocols, ct);
             }
             else
             {
                 if (writer is not IBatchProxyWriter batchProxyWriter)
                     throw new InvalidOperationException($"{format} output does not support stream mode.");
 
-                await FetchAndWriteProxiesAsync(batchProxySource, batchProxyWriter, ct);
+                await FetchAndWriteProxiesAsync(batchProxySource, batchProxyWriter, protocols, ct);
             }
 
             return 0;
@@ -105,24 +108,30 @@ public override async Task<int> ExecuteAsync(CommandContext context, FetchComman
     }
 
     private static async Task FetchAndWriteProxiesAsync(IBatchProxySource? batchProxySource,
-        IBatchProxyWriter writer, CancellationToken ct)
+        IBatchProxyWriter writer, Protocols protocols, CancellationToken ct)
     {
         if (batchProxySource == null)
             throw new InvalidOperationException("The plugin does not support batch mode.");
 
         var proxies = await batchProxySource.FetchAsync(ct);
 
+        if (protocols != Protocols.None)
+            proxies = proxies.Where(p => (p.Protocols & protocols) != Protocols.None);
+
         await writer.WriteAsync(proxies, ct);
     }
 
     private static async Task FetchAndWriteProxiesAsync(IStreamProxySource? streamProxySource,
-        IStreamProxyWriter writer, CancellationToken ct)
+        IStreamProxyWriter writer, Protocols protocols, CancellationToken ct)
     {
         if (streamProxySource == null)
             throw new InvalidOperationException("The plugin does not support stream mode.");
 
         var proxies = streamProxySource.FetchAsync(ct);
 
+        if (protocols != Protocols.None)
+            proxies = proxies.Where(p => (p.Protocols & protocols) != Protocols.None);
+
         await writer.WriteAsync(proxies, ct);
     }
 

src/Proxxi.Cli/Commands/Fetch/FetchCommandSettings.cs

@@ -1,6 +1,7 @@
 using System.ComponentModel;
 
 using Proxxi.Core.Models;
+using Proxxi.Plugin.Sdk.Models;
 
 using Spectre.Console.Cli;
 
@@ -24,6 +25,10 @@ public class FetchCommandSettings : CommandSettings
     [Description("Fetch proxies using streaming mode instead of batch mode")]
     public bool Stream { get; init; }
 
+    [CommandOption("-p|--protocols <PROTOCOLS>"), DefaultValue(Proxxi.Plugin.Sdk.Models.Protocols.None)]
+    [Description("Filter proxies by protocol: http, https, socks4, socks5")]
+    public Protocols? Protocols { get; init; }
+
     [CommandOption("--pretty")]
     [Description("Write human-readable, pretty-formatted output")]
     public bool Pretty { get; init; }