dev: lmgtfy should now be properly shortened, small refactor of .shorten command

This commit is contained in:
Kwoth
2024-08-04 17:58:04 +00:00
parent f4ed907134
commit 690c03b396
3 changed files with 62 additions and 41 deletions

View File

@@ -14,7 +14,6 @@ namespace NadekoBot.Modules.Searches;
public partial class Searches : NadekoModule<SearchesService> public partial class Searches : NadekoModule<SearchesService>
{ {
private static readonly ConcurrentDictionary<string, string> _cachedShortenedLinks = new();
private readonly IBotCredentials _creds; private readonly IBotCredentials _creds;
private readonly IGoogleApiService _google; private readonly IGoogleApiService _google;
private readonly IHttpClientFactory _httpFactory; private readonly IHttpClientFactory _httpFactory;
@@ -172,7 +171,8 @@ public partial class Searches : NadekoModule<SearchesService>
if (!await ValidateQuery(smh)) if (!await ValidateQuery(smh))
return; return;
var shortenedUrl = await _google.ShortenUrl($"https://letmegooglethat.com/?q={Uri.EscapeDataString(smh)}"); var link = $"https://letmegooglethat.com/?q={Uri.EscapeDataString(smh)}";
var shortenedUrl = await _service.ShortenLink(link) ?? link;
await Response().Confirm($"<{shortenedUrl}>").SendAsync(); await Response().Confirm($"<{shortenedUrl}>").SendAsync();
} }
@@ -182,35 +182,12 @@ public partial class Searches : NadekoModule<SearchesService>
if (!await ValidateQuery(query)) if (!await ValidateQuery(query))
return; return;
query = query.Trim(); var shortLink = await _service.ShortenLink(query);
if (!_cachedShortenedLinks.TryGetValue(query, out var shortLink))
if (shortLink is null)
{ {
try await Response().Error(strs.error_occured).SendAsync();
{ return;
using var http = _httpFactory.CreateClient();
using var req = new HttpRequestMessage(HttpMethod.Post, "https://goolnk.com/api/v1/shorten");
var formData = new MultipartFormDataContent
{
{ new StringContent(query), "url" }
};
req.Content = formData;
using var res = await http.SendAsync(req);
var content = await res.Content.ReadAsStringAsync();
var data = JsonConvert.DeserializeObject<ShortenData>(content);
if (!string.IsNullOrWhiteSpace(data?.ResultUrl))
_cachedShortenedLinks.TryAdd(query, data.ResultUrl);
else
return;
shortLink = data.ResultUrl;
}
catch (Exception ex)
{
Log.Error(ex, "Error shortening a link: {Message}", ex.Message);
return;
}
} }
await Response() await Response()
@@ -221,6 +198,7 @@ public partial class Searches : NadekoModule<SearchesService>
.SendAsync(); .SendAsync();
} }
[Cmd] [Cmd]
public async Task MagicTheGathering([Leftover] string search) public async Task MagicTheGathering([Leftover] string search)
{ {
@@ -328,7 +306,7 @@ public partial class Searches : NadekoModule<SearchesService>
var maybeItems = await _service.GetDefinitionsAsync(word); var maybeItems = await _service.GetDefinitionsAsync(word);
if(!maybeItems.TryPickT0(out var defs, out var error)) if (!maybeItems.TryPickT0(out var defs, out var error))
{ {
await HandleErrorAsync(error); await HandleErrorAsync(error);
return; return;
@@ -487,10 +465,4 @@ public partial class Searches : NadekoModule<SearchesService>
await Response().Error(strs.specify_search_params).SendAsync(); await Response().Error(strs.specify_search_params).SendAsync();
return false; return false;
} }
public class ShortenData
{
[JsonProperty("result_url")]
public string ResultUrl { get; set; }
}
} }

View File

@@ -2,8 +2,8 @@
using NadekoBot.Modules.Searches.Common; using NadekoBot.Modules.Searches.Common;
using Newtonsoft.Json; using Newtonsoft.Json;
using Newtonsoft.Json.Linq; using Newtonsoft.Json.Linq;
using System.Text.Json;
using OneOf; using OneOf;
using System.Text.Json;
namespace NadekoBot.Modules.Searches.Services; namespace NadekoBot.Modules.Searches.Services;
@@ -30,6 +30,7 @@ public class SearchesService : INService
private readonly object _yomamaLock = new(); private readonly object _yomamaLock = new();
private int yomamaJokeIndex; private int yomamaJokeIndex;
private readonly ConcurrentDictionary<string, string> _cachedShortenedLinks = new();
public SearchesService( public SearchesService(
IGoogleApiService google, IGoogleApiService google,
@@ -579,4 +580,42 @@ public class SearchesService : INService
return ErrorType.Unknown; return ErrorType.Unknown;
} }
} }
public async Task<string> ShortenLink(string query)
{
query = query.Trim();
if (_cachedShortenedLinks.TryGetValue(query, out var shortLink))
return shortLink;
try
{
using var http = _httpFactory.CreateClient();
using var req = new HttpRequestMessage(HttpMethod.Post, "https://goolnk.com/api/v1/shorten");
var formData = new MultipartFormDataContent
{
{ new StringContent(query), "url" }
};
req.Content = formData;
using var res = await http.SendAsync(req);
var content = await res.Content.ReadAsStringAsync();
var data = JsonConvert.DeserializeObject<ShortenData>(content);
if (!string.IsNullOrWhiteSpace(data?.ResultUrl))
_cachedShortenedLinks.TryAdd(query, data.ResultUrl);
else
return query;
shortLink = data.ResultUrl;
}
catch (Exception ex)
{
Log.Error(ex, "Error shortening a link: {Message}", ex.Message);
return null;
}
return shortLink;
throw new NotImplementedException();
}
} }

View File

@@ -0,0 +1,10 @@
#nullable disable
using Newtonsoft.Json;
namespace NadekoBot.Modules.Searches.Services;
public class ShortenData
{
[JsonProperty("result_url")]
public string ResultUrl { get; set; }
}