// 好像坏掉了
/// <summary>
/// Search
/// </summary>
/// <param name="image">Base64 Encoded Image</param>
/// <param name="filter">Limit search in specific year / season</param>
/// <returns></returns>
internal static async Task <JObject> SearchAsync(string image, string filter = "")
{
JObject result = null;
try
{
if (!Cooldown.CanSearch)
{
throw new ApplicationException($"全局冷却中: 剩余 {Cooldown.RemainingSeconds} 秒");
}
result = JObject.Parse(await RequestStringAsync($"/api/search?token={ApiToken}", $"image={image}" + (filter == "" ? "" : "&filter=" + filter)));
Cooldown.SetCooldown(result["quota"].ToObject <int>(), result["expire"].ToObject <int>());
return(result);
}
catch (WebException we)
{
switch ((int)(we.Response as HttpWebResponse).StatusCode)
{
case 429:
Cooldown.TriggeredHTTP429();
break;
}
CoolQApi.AddLog(CoolQApi.LogLevel.Debug, "ErrorDebug", "Search Result: " + result?.ToString(Newtonsoft.Json.Formatting.None) ?? "[NULL]");
throw;
}
}
internal static CoolQApi.Event ProcessGroupMessageAsync(int subType, int sendTime, long fromGroup, long fromQQ, string fromAnonymous, string msg, int font)
{
try
{
var result = _ProcessGroupMessageAsync(subType, sendTime, fromGroup, fromQQ, fromAnonymous, msg, font);
result.Wait();
return(result.Result);
}
catch (Exception ex)
{
CoolQApi.SendPrivateMsg(Config.MasterQQ, ex.ToString());
return(CoolQApi.Event.Ignore);
}
}
/// <summary>
/// 下载网络图片到本地并转换成可直接发送的 CQ码
/// </summary>
/// <param name="url">图片地址</param>
/// <returns>CQ码</returns>
internal static async Task <string> WebImage2CQIMGAsync(string url)
{
var dir = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "data", "image", "whatanime");
if (!Directory.Exists(dir))
{
Directory.CreateDirectory(dir);
}
var request = (HttpWebRequest)WebRequest.Create(url);
request.Timeout = HTTP_TIMEOUT;
request.UserAgent = HTTP_USERAGENT;
var response = (HttpWebResponse)(await request.GetResponseAsync());
string filename = ((int)(DateTime.UtcNow.Subtract(new DateTime(1970, 1, 1))).TotalSeconds).ToString("X") + "-" + url.GetHashCode().ToString("X") + ".jpg";
Image.FromStream(response.GetResponseStream()).Save(Path.Combine(dir, filename), ImageFormat.Jpeg);
return(CoolQApi.CQC_Image("whatanime\\" + filename));
}
private static async Task <CoolQApi.Event> _ProcessGroupMessageAsync(int subType, int sendTime, long fromGroup, long fromQQ, string fromAnonymous, string msg, int font)
{
// 如果没有以主命令开头，则忽略
if (!msg.StartsWith(Config.MainCommand))
{
return(CoolQApi.Event.Ignore);
}
// 要回复给发送人的消息
string reply = string.Empty;
// 获取消息中的图片
var regex_result = REGEX_GETIMAGE.Match(msg);
if (regex_result.Success)
{ // 如果消息中带有图片
int cooldown = Cooldown.Check(fromQQ);
if (cooldown > 0 && fromQQ != Config.MasterQQ) // 检查发送账号是否处于冷却状态
{
reply = $"单人冷却：请等候 {cooldown} 秒后再试";
}
else
{
try
{
CoolQApi.SendGroupMsg(fromGroup, CoolQApi.CQC_At(fromQQ) + " 正在搜索中...");
ReplyContent replyContent = ReplyContent.Normal;
string image = "data:image/jpeg;base64," + await ImageProcessor.CQIMG2Base64Async(regex_result.Groups[1].Value);
var result = await WhatAnimeAPI.SearchAsync(image);
JObject info = result?["docs"]?[0] as JObject;
Cooldown.Set(fromQQ);
if (info == null)
{
reply = "未搜索到任何信息";
}
else
{
int anilist_id = info["anilist_id"].ToObject <int>();
double similarity = info["similarity"].ToObject <double>() * 100d;
var anilist_result = anilist.Query(ANILIST_QUERY, new { id = anilist_id }, Config.USER_AGENT).Get("Media");
// isAdult: boolean
if (anilist_result.isAdult.ToObject <bool>())
{
replyContent |= ReplyContent.R18;
}
// genres: string[]
var genres = new List <string>((string[])anilist_result.genres.ToObject <string[]>());
if (genres.Any(x => x == "Hentai" || x == "Ecchi"))
{
replyContent |= ReplyContent.NoPreview;
}
// 🌚
if (anilist_id == 10380 || anilist_id == 11879 || anilist_id == 20852)
{
replyContent |= ReplyContent.FuckOff;
}
if (similarity < 88)
{
replyContent |= ReplyContent.SearchTip;
}
// startDate.year: number
// season:
// WINTER
// SPRING
// SUMMER
// FALL
string anilist_season = anilist_result.startDate.year + "年";
switch (anilist_result.season.ToString())
{
case "WINTER":
anilist_season += "冬季";
break;
case "SPRING":
anilist_season += "春季";
break;
case "SUMMER":
anilist_season += "夏季";
break;
case "FALL":
anilist_season += "秋季";
break;
}
// status:
// FINISHED 已完结
// RELEASING 更新中
// NOT_YET_RALEASED 未发布
// CANCELLED 取消发布
string anilist_status;
switch (anilist_result.status.ToString())
{
case "FINISHED":
anilist_status = "已完结";
break;
case "RELEASING":
anilist_status = "更新中";
break;
case "NOT_YET_RALEASED":
anilist_status = "未发布";
break;
case "CANCELLED":
anilist_status = "取消发布";
break;
default:
anilist_status = string.Empty; // Error?
break;
}
// source:
// ORIGINAL 原创
// MANGA 漫改
// LIGHT_NOVEL 轻改
string anilist_source;
switch (anilist_result.source.ToString())
{
case "ORIGINAL":
anilist_source = "原创";
break;
case "MANGA":
anilist_source = "漫改";
break;
case "LIGHT_NOVEL":
anilist_source = "轻改";
break;
default:
anilist_source = "其他";
break;
}
if (replyContent.HasFlag(ReplyContent.FuckOff))
{
Cooldown.Set(fromQQ, 60 * 60 * 24); // 1 day
CoolQApi.SetGroupBan(fromGroup, fromQQ, Config.FuckOffDuration);
reply = Config.FuckOffMessage;
}
else if (replyContent.HasFlag(ReplyContent.R18))
{
Cooldown.Set(fromQQ, 60 * 10); // 10 minutes
reply = Config.R18Message;
}
else
{
Task <string> getimage = null;
if (!replyContent.HasFlag(ReplyContent.NoPreview))
{
getimage = WhatAnimeAPI.ThumbnailAsync(info);
}
TimeSpan time = TimeSpan.FromSeconds(info["at"].ToObject <double>());
string episode = info["episode"].Type == JTokenType.Integer
? $"第 {info["episode"].ToObject<int>()} 话"
: info["episode"].ToObject <string>();
reply = "搜索结果：" + Environment.NewLine;
reply += info["title"].ToObject <string>() + Environment.NewLine;
reply += info["title_chinese"].ToObject <string>() + Environment.NewLine;
reply += episode + " " + $"{time.Hours} 小时 {time.Minutes} 分 {time.Seconds} 秒" + Environment.NewLine;
reply += "相似度：" + similarity.ToString() + "%" + Environment.NewLine;
reply += $"{anilist_season} {anilist_status} {anilist_source}" + Environment.NewLine;
reply += Environment.NewLine;
if (replyContent.HasFlag(ReplyContent.NoPreview))
{
reply += "擦边球，没有预览";
}
else
{
reply += "对比截图预览 (Preview unavailable 是截图失败)" + Environment.NewLine;
reply += await getimage;
}
if (replyContent.HasFlag(ReplyContent.SearchTip))
{
reply += Environment.NewLine + Config.SearchTipMessage;
}
}
}
}
catch (Exception ex)
{
reply = "发生了错误：" + ex.Message;
CoolQApi.AddLog(CoolQApi.LogLevel.Debug, "番剧识别错误", ex.ToString());
}
}
}
else
{ // 如果消息中不带图片
reply = Config.HelpMessage;
}
if (reply != string.Empty)
{
CoolQApi.SendGroupMsg(fromGroup, CoolQApi.CQC_At(fromQQ) + "\n" + reply);
}
return(CoolQApi.Event.Block);
}
/// <summary> /// Load config file /// </summary> internal static void Load() => _raw = JObject.Parse(File.ReadAllText(Path.Combine(CoolQApi.GetAppDirectory(), "config.json")));