mirror of
https://gitlab.com/Kwoth/nadekobot.git
synced 2025-09-10 09:18:27 -04:00
441 lines
13 KiB
C#
441 lines
13 KiB
C#
#nullable disable
|
|
using NadekoBot.Modules.Searches.Common;
|
|
using Newtonsoft.Json.Linq;
|
|
|
|
namespace NadekoBot.Modules.Nsfw;
|
|
|
|
#if !GLOBAL_NADEKO
|
|
[NoPublicBot]
|
|
public partial class NSFW : NadekoModule<ISearchImagesService>
|
|
{
|
|
private static readonly ConcurrentHashSet<ulong> _hentaiBombBlacklist = new();
|
|
private readonly IHttpClientFactory _httpFactory;
|
|
private readonly NadekoRandom _rng;
|
|
|
|
public NSFW(IHttpClientFactory factory)
|
|
{
|
|
_httpFactory = factory;
|
|
_rng = new();
|
|
}
|
|
|
|
private async Task InternalBoobs()
|
|
{
|
|
try
|
|
{
|
|
JToken obj;
|
|
using (var http = _httpFactory.CreateClient())
|
|
{
|
|
obj = JArray.Parse(
|
|
await http.GetStringAsync($"http://api.oboobs.ru/boobs/{new NadekoRandom().Next(0, 10330)}"))[0];
|
|
}
|
|
|
|
await ctx.Channel.SendMessageAsync($"http://media.oboobs.ru/{obj["preview"]}");
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await SendErrorAsync(ex.Message);
|
|
}
|
|
}
|
|
|
|
private async Task InternalButts(IMessageChannel channel)
|
|
{
|
|
try
|
|
{
|
|
JToken obj;
|
|
using (var http = _httpFactory.CreateClient())
|
|
{
|
|
obj = JArray.Parse(
|
|
await http.GetStringAsync($"http://api.obutts.ru/butts/{new NadekoRandom().Next(0, 4335)}"))[0];
|
|
}
|
|
|
|
await channel.SendMessageAsync($"http://media.obutts.ru/{obj["preview"]}");
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await SendErrorAsync(ex.Message);
|
|
}
|
|
}
|
|
|
|
[Cmd]
|
|
[RequireNsfw]
|
|
[RequireContext(ContextType.Guild)]
|
|
[UserPerm(ChannelPerm.ManageMessages)]
|
|
public async partial Task AutoHentai(int interval = 0, [Leftover] string tags = null)
|
|
{
|
|
Timer t;
|
|
|
|
if (interval == 0)
|
|
{
|
|
if (!_service.AutoHentaiTimers.TryRemove(ctx.Channel.Id, out t))
|
|
return;
|
|
|
|
t.Change(Timeout.Infinite, Timeout.Infinite); //proper way to disable the timer
|
|
await ReplyConfirmLocalizedAsync(strs.stopped);
|
|
return;
|
|
}
|
|
|
|
if (interval < 20)
|
|
return;
|
|
|
|
t = new(async _ =>
|
|
{
|
|
try
|
|
{
|
|
if (tags is null || tags.Length == 0)
|
|
await InternalDapiCommand(null, true, _service.Hentai);
|
|
else
|
|
{
|
|
var groups = tags.Split('|');
|
|
var group = groups[_rng.Next(0, groups.Length)];
|
|
await InternalDapiCommand(group.Split(' '), true, _service.Hentai);
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
// ignored
|
|
}
|
|
},
|
|
null,
|
|
interval * 1000,
|
|
interval * 1000);
|
|
|
|
_service.AutoHentaiTimers.AddOrUpdate(ctx.Channel.Id,
|
|
t,
|
|
(_, old) =>
|
|
{
|
|
old.Change(Timeout.Infinite, Timeout.Infinite);
|
|
return t;
|
|
});
|
|
|
|
await ReplyConfirmLocalizedAsync(strs.autohentai_started(interval, string.Join(", ", tags)));
|
|
}
|
|
|
|
[Cmd]
|
|
[RequireNsfw]
|
|
[RequireContext(ContextType.Guild)]
|
|
[UserPerm(ChannelPerm.ManageMessages)]
|
|
public async partial Task AutoBoobs(int interval = 0)
|
|
{
|
|
Timer t;
|
|
|
|
if (interval == 0)
|
|
{
|
|
if (!_service.AutoBoobTimers.TryRemove(ctx.Channel.Id, out t))
|
|
return;
|
|
|
|
t.Change(Timeout.Infinite, Timeout.Infinite);
|
|
await ReplyConfirmLocalizedAsync(strs.stopped);
|
|
return;
|
|
}
|
|
|
|
if (interval < 20)
|
|
return;
|
|
|
|
t = new(async _ =>
|
|
{
|
|
try
|
|
{
|
|
await InternalBoobs();
|
|
}
|
|
catch
|
|
{
|
|
// ignored
|
|
}
|
|
},
|
|
null,
|
|
interval * 1000,
|
|
interval * 1000);
|
|
|
|
_service.AutoBoobTimers.AddOrUpdate(ctx.Channel.Id,
|
|
t,
|
|
(_, old) =>
|
|
{
|
|
old.Change(Timeout.Infinite, Timeout.Infinite);
|
|
return t;
|
|
});
|
|
|
|
await ReplyConfirmLocalizedAsync(strs.started(interval));
|
|
}
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
[UserPerm(ChannelPerm.ManageMessages)]
|
|
public async partial Task AutoButts(int interval = 0)
|
|
{
|
|
Timer t;
|
|
|
|
if (interval == 0)
|
|
{
|
|
if (!_service.AutoButtTimers.TryRemove(ctx.Channel.Id, out t))
|
|
return;
|
|
|
|
t.Change(Timeout.Infinite, Timeout.Infinite); //proper way to disable the timer
|
|
await ReplyConfirmLocalizedAsync(strs.stopped);
|
|
return;
|
|
}
|
|
|
|
if (interval < 20)
|
|
return;
|
|
|
|
t = new(async _ =>
|
|
{
|
|
try
|
|
{
|
|
await InternalButts(ctx.Channel);
|
|
}
|
|
catch
|
|
{
|
|
// ignored
|
|
}
|
|
},
|
|
null,
|
|
interval * 1000,
|
|
interval * 1000);
|
|
|
|
_service.AutoButtTimers.AddOrUpdate(ctx.Channel.Id,
|
|
t,
|
|
(_, old) =>
|
|
{
|
|
old.Change(Timeout.Infinite, Timeout.Infinite);
|
|
return t;
|
|
});
|
|
|
|
await ReplyConfirmLocalizedAsync(strs.started(interval));
|
|
}
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public partial Task Hentai(params string[] tags)
|
|
=> InternalDapiCommand(tags, true, _service.Hentai);
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public async partial Task HentaiBomb(params string[] tags)
|
|
{
|
|
if (!_hentaiBombBlacklist.Add(ctx.Guild?.Id ?? ctx.User.Id))
|
|
return;
|
|
try
|
|
{
|
|
var images = await Task.WhenAll(_service.Yandere(ctx.Guild?.Id, true, tags),
|
|
_service.Danbooru(ctx.Guild?.Id, true, tags),
|
|
_service.Konachan(ctx.Guild?.Id, true, tags),
|
|
_service.Gelbooru(ctx.Guild?.Id, true, tags));
|
|
|
|
var linksEnum = images.Where(l => l is not null).ToArray();
|
|
if (!linksEnum.Any())
|
|
{
|
|
await ReplyErrorLocalizedAsync(strs.no_results);
|
|
return;
|
|
}
|
|
|
|
await ctx.Channel.SendMessageAsync(string.Join("\n\n", linksEnum.Select(x => x.Url)));
|
|
}
|
|
finally
|
|
{
|
|
_hentaiBombBlacklist.TryRemove(ctx.Guild?.Id ?? ctx.User.Id);
|
|
}
|
|
}
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public partial Task Yandere(params string[] tags)
|
|
=> InternalDapiCommand(tags, false, _service.Yandere);
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public partial Task Konachan(params string[] tags)
|
|
=> InternalDapiCommand(tags, false, _service.Konachan);
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public partial Task Sankaku(params string[] tags)
|
|
=> InternalDapiCommand(tags, false, _service.Sankaku);
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public partial Task E621(params string[] tags)
|
|
=> InternalDapiCommand(tags, false, _service.E621);
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public partial Task Rule34(params string[] tags)
|
|
=> InternalDapiCommand(tags, false, _service.Rule34);
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public partial Task Danbooru(params string[] tags)
|
|
=> InternalDapiCommand(tags, false, _service.Danbooru);
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public partial Task Gelbooru(params string[] tags)
|
|
=> InternalDapiCommand(tags, false, _service.Gelbooru);
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public partial Task Derpibooru(params string[] tags)
|
|
=> InternalDapiCommand(tags, false, _service.DerpiBooru);
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public partial Task Safebooru(params string[] tags)
|
|
=> InternalDapiCommand(tags, false, _service.SafeBooru);
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public async partial Task Boobs()
|
|
{
|
|
try
|
|
{
|
|
JToken obj;
|
|
using (var http = _httpFactory.CreateClient())
|
|
{
|
|
obj = JArray.Parse(
|
|
await http.GetStringAsync($"http://api.oboobs.ru/boobs/{new NadekoRandom().Next(0, 12000)}"))[0];
|
|
}
|
|
|
|
await ctx.Channel.SendMessageAsync($"http://media.oboobs.ru/{obj["preview"]}");
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await SendErrorAsync(ex.Message);
|
|
}
|
|
}
|
|
|
|
[Cmd]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
public async partial Task Butts()
|
|
{
|
|
try
|
|
{
|
|
JToken obj;
|
|
using (var http = _httpFactory.CreateClient())
|
|
{
|
|
obj = JArray.Parse(
|
|
await http.GetStringAsync($"http://api.obutts.ru/butts/{new NadekoRandom().Next(0, 6100)}"))[0];
|
|
}
|
|
|
|
await ctx.Channel.SendMessageAsync($"http://media.obutts.ru/{obj["preview"]}");
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await SendErrorAsync(ex.Message);
|
|
}
|
|
}
|
|
|
|
[Cmd]
|
|
[RequireContext(ContextType.Guild)]
|
|
[UserPerm(GuildPerm.ManageMessages)]
|
|
public async partial Task NsfwTagBlacklist([Leftover] string tag = null)
|
|
{
|
|
if (string.IsNullOrWhiteSpace(tag))
|
|
{
|
|
var blTags = await _service.GetBlacklistedTags(ctx.Guild.Id);
|
|
await SendConfirmAsync(GetText(strs.blacklisted_tag_list), blTags.Any() ? string.Join(", ", blTags) : "-");
|
|
}
|
|
else
|
|
{
|
|
tag = tag.Trim().ToLowerInvariant();
|
|
var added = await _service.ToggleBlacklistTag(ctx.Guild.Id, tag);
|
|
|
|
if (added)
|
|
await ReplyPendingLocalizedAsync(strs.blacklisted_tag_add(tag));
|
|
else
|
|
await ReplyPendingLocalizedAsync(strs.blacklisted_tag_remove(tag));
|
|
}
|
|
}
|
|
|
|
[Cmd]
|
|
[RequireContext(ContextType.Guild)]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
[Priority(1)]
|
|
public async partial Task Nhentai(uint id)
|
|
{
|
|
var g = await _service.GetNhentaiByIdAsync(id);
|
|
|
|
if (g is null)
|
|
{
|
|
await ReplyErrorLocalizedAsync(strs.not_found);
|
|
return;
|
|
}
|
|
|
|
await SendNhentaiGalleryInternalAsync(g);
|
|
}
|
|
|
|
[Cmd]
|
|
[RequireContext(ContextType.Guild)]
|
|
[RequireNsfw(Group = "nsfw_or_dm")]
|
|
[RequireContext(ContextType.DM, Group = "nsfw_or_dm")]
|
|
[Priority(0)]
|
|
public async partial Task Nhentai([Leftover] string query)
|
|
{
|
|
var g = await _service.GetNhentaiBySearchAsync(query);
|
|
|
|
if (g is null)
|
|
{
|
|
await ReplyErrorLocalizedAsync(strs.not_found);
|
|
return;
|
|
}
|
|
|
|
await SendNhentaiGalleryInternalAsync(g);
|
|
}
|
|
|
|
private async Task SendNhentaiGalleryInternalAsync(Gallery g)
|
|
{
|
|
var count = 0;
|
|
var tagString = g.Tags.Shuffle()
|
|
.Select(tag => $"[{tag.Name}]({tag.Url})")
|
|
.TakeWhile(tag => (count += tag.Length) < 1000)
|
|
.Join(" ");
|
|
|
|
var embed = _eb.Create()
|
|
.WithTitle(g.Title)
|
|
.WithDescription(g.FullTitle)
|
|
.WithImageUrl(g.Thumbnail)
|
|
.WithUrl(g.Url)
|
|
.AddField(GetText(strs.favorites), g.Likes, true)
|
|
.AddField(GetText(strs.pages), g.PageCount, true)
|
|
.AddField(GetText(strs.tags), tagString, true)
|
|
.WithFooter(g.UploadedAt.ToString("f"))
|
|
.WithOkColor();
|
|
|
|
await ctx.Channel.EmbedAsync(embed);
|
|
}
|
|
|
|
private async Task InternalDapiCommand(
|
|
string[] tags,
|
|
bool forceExplicit,
|
|
Func<ulong?, bool, string[], Task<UrlReply>> func)
|
|
{
|
|
var data = await func(ctx.Guild?.Id, forceExplicit, tags);
|
|
|
|
if (data is null || !string.IsNullOrWhiteSpace(data.Error))
|
|
{
|
|
await ReplyErrorLocalizedAsync(strs.no_results);
|
|
return;
|
|
}
|
|
|
|
await ctx.Channel.EmbedAsync(_eb.Create(ctx)
|
|
.WithOkColor()
|
|
.WithImageUrl(data.Url)
|
|
.WithDescription($"[link]({data.Url})")
|
|
.WithFooter(
|
|
$"{data.Rating} ({data.Provider}) | {string.Join(" | ", data.Tags.Where(x => !string.IsNullOrWhiteSpace(x)).Take(5))}"));
|
|
}
|
|
}
|
|
#endif |