Image scanning; cli

Author: Decimation

SmartImage.Lib/Searching/ImageResult.cs

@@ -232,7 +232,7 @@ public void FindDirectImages()
 			}
 		}
 
-		private void UpdateImageData()
+		public void UpdateImageData()
 		{
 			if (Image is { }) {
 

SmartImage.Lib/Utilities/ImageHelper.cs

@@ -11,6 +11,7 @@
 using AngleSharp.Html.Parser;
 using SimpleCore.Net;
 using SimpleCore.Utilities;
+
 // ReSharper disable UnusedParameter.Local
 
 // ReSharper disable PossibleMultipleEnumeration
@@ -91,7 +92,6 @@ public static bool IsDirect(string url, DirectType directType = DirectType.Regex
 			{
 				DirectType.Binary => MediaTypes.IsDirect(url, MimeType.Image),
 				DirectType.Regex =>
-					// todo
 					/*
 					 * https://github.com/PactInteractive/image-downloader
 					 */
@@ -136,13 +136,6 @@ public static List<string> FindDirectImages(string url, out List<Image> images,
 			//                 $"{fragmentSize} | {readImage} | "      +
 			//                 $"{imageFilter} | {count}");
 
-
-			//<img.*?src="(.*?)"
-			//href\s*=\s*"(.+?)"
-			//var src  = "<img.*?src=\"(.*?)\"";
-			//var href = "href\\s*=\\s*\"(.+?)\"";
-
-
 			IHtmlDocument document;
 
 			try {
@@ -165,6 +158,8 @@ public static List<string> FindDirectImages(string url, out List<Image> images,
 			flat.AddRange(document.QuerySelectorAttributes("a", "href"));
 			flat.AddRange(document.QuerySelectorAttributes("img", "src"));
 
+			flat = flat.Distinct().ToList();
+
 			var fragments = flat.Chunk(fragmentSize).ToArray();
 
 			var tasks = new List<Task>();

SmartImage/Core/MainDialog.cs

@@ -27,15 +27,18 @@ internal static class MainDialog
 	{
 		#region Colors
 
-		private static readonly Color ColorMain  = Color.Yellow;
-		private static readonly Color ColorOther = Color.Aquamarine;
-		private static readonly Color ColorYes   = Color.GreenYellow;
-		private static readonly Color ColorNo    = Color.Red;
+		internal static readonly Color ColorMain  = Color.Yellow;
+		internal static readonly Color ColorOther = Color.Aquamarine;
+		internal static readonly Color ColorYes   = Color.GreenYellow;
+		internal static readonly Color ColorNo    = Color.Red;
 
 		#endregion
 
 		#region Elements
 
+		public const string Description = "Press the result number to open in browser\n" +
+		                                     "Ctrl: Load direct | Alt: Show other | Shift: Open raw | Alt+Ctrl: Download";
+
 		private static readonly string Enabled = StringConstants.CHECK_MARK.ToString().AddColor(ColorYes);
 
 		private static readonly string Disabled = StringConstants.MUL_SIGN.ToString().AddColor(ColorNo);

SmartImage/Core/NConsoleFactory.cs

@@ -89,7 +89,7 @@ public static NConsoleOption Create(SearchResult result)
 			return option;
 		}
 
-		private static NConsoleOption Create(ImageResult result, int i, Color c)
+		public static NConsoleOption Create(ImageResult result, int i, Color c)
 		{
 
 			const float correctionFactor = -.3f;

SmartImage/Program.cs

@@ -30,6 +30,8 @@
 using SmartImage.Lib.Utilities;
 using SmartImage.Utilities;
 
+// ReSharper disable AssignNullToNotNullAttribute
+
 // ReSharper disable ConvertSwitchStatementToSwitchExpression
 
 // ReSharper disable UnusedParameter.Local
@@ -53,7 +55,7 @@ private static async Task Main(string[] args)
 		{
 #if DEBUG
 			if (!args.Any()) {
-				//args = new[] {""};
+				args = new[] {"find-direct", "https://danbooru.donmai.us/posts/3987008"};
 			}
 
 
@@ -105,7 +107,42 @@ private static async Task Main(string[] args)
 					object? arg = enumerator.Current;
 
 					switch (arg) {
+						case "find-direct":
+							enumerator.MoveNext();
+							var argValue = (string) enumerator.Current;
+
+							var directImages   = ImageHelper.FindDirectImages(argValue, out var im);
+							var imageResults = new List<ImageResult>();
+
+							for (int i = 0; i < directImages.Count; i++) {
+								string directUrl = directImages[i];
+
+								var ir = new ImageResult
+								{
+									Image  = im[i],
+									Url    = new Uri(directUrl),
+									Direct = new Uri(directUrl)
+								};
+
+								ir.UpdateImageData();
+
+								imageResults.Add(ir);
+							}
+
+							int i2 = 0;
+
+							var options = imageResults
+							              .Select(r => NConsoleFactory.Create(r, i2++, MainDialog.ColorOther))
+							              .ToArray();
+
+
+							NConsole.ReadOptions(new NConsoleDialog
+							{
+								Options = options,
+								Description  = MainDialog.Description
+							});
 
+							return;
 						default:
 							Config.Query = args[0];
 							break;
@@ -179,9 +216,8 @@ private static void OnResultCompleted(object? sender, SearchResultEventArgs even
 
 		public static readonly NConsoleDialog ResultDialog = new()
 		{
-			Options = new List<NConsoleOption>(),
-			Description = "Press the result number to open in browser\n" +
-			              "Ctrl: Load direct | Alt: Show other | Shift: Open raw | Alt+Ctrl: Download"
+			Options     = new List<NConsoleOption>(),
+			Description = MainDialog.Description
 		};
 
 		public static readonly SearchConfig Config = new();