...

Author: Decimation

SmartImage.Lib/Engines/Impl/Search/IqdbEngine.cs

@@ -12,6 +12,7 @@
 using Flurl.Http;
 using Kantan.Net.Utilities;
 using Kantan.Text;
+using Microsoft.Extensions.Logging;
 using SmartImage.Lib.Results;
 using SmartImage.Lib.Utilities;
 
@@ -34,7 +35,7 @@ protected IqdbEngine(string b, string e) : base(b, e)
 	{
 		MaxSize = MAX_FILE_SIZE; // NOTE: assuming IQDB uses kilobytes instead of kibibytes
 
-		// Timeout = TimeSpan.FromSeconds(10);
+		Timeout = TimeSpan.FromSeconds(30);
 	}
 
 	private const int MAX_FILE_SIZE = 8_388_608;
@@ -54,7 +55,7 @@ private async Task<IDocument> GetDocumentAsync(SearchQuery query, CancellationTo
 		IDocument document = null;
 
 		try {
-			var response = await Client.Request(EndpointUrl)
+			/*var response = await Client.Request(EndpointUrl)
 				               .OnError(r =>
 					               {
 						               Debug.WriteLine($"{r.Exception}", Name);
@@ -73,8 +74,22 @@ private async Task<IDocument> GetDocumentAsync(SearchQuery query, CancellationTo
 					               }
 
 					               return;
-				               }, cancellationToken: ct);
-
+				               }, cancellationToken: ct);*/
+			var response = await Client.Request(URL_QUERY)
+				               .OnError(r =>
+					               {
+						               // Debug.WriteLine($"{r.Exception}", Name);
+						               // r.ExceptionHandled = true;
+						               Logger.LogError(r.Exception, $"{Name}");
+						               Debugger.Break();
+#if !DEBUG
+						               r.ExceptionHandled = true;
+#endif
+					               }
+				               )
+				               .SetQueryParam("url", query.Upload)
+				               .WithTimeout(Timeout)
+				               .GetAsync(cancellationToken: ct);
 			if (response != null) {
 				var s = await response.GetStringAsync();
 

SmartImage.Lib/Engines/Impl/Search/SauceNaoEngine.cs

@@ -406,9 +406,13 @@ private static void Parse(INode result, SearchResult sr)
 
 		Url url = urls.FirstOrDefault();
 
-		var site = url.Host.Replace("www", "");
-		site = site.Split('.', StringSplitOptions.RemoveEmptyEntries)[0];
+		string site = null;
 
+		if (Url.IsValid(url)) {
+			site = url.Host.Replace("www", "");
+			site = site.Split('.', StringSplitOptions.RemoveEmptyEntries)[0];
+
+		}
 
 		var sndr = new SearchResultItem(sr)
 		{