submodule对gitea不管用，所以直接拉了一份拉格兰
This commit is contained in:
255
Lagrange.Core/Lagrange.OneBot/Core/Notify/NotifyService.cs
Normal file
255
Lagrange.Core/Lagrange.OneBot/Core/Notify/NotifyService.cs
Normal file
@@ -0,0 +1,255 @@
|
||||
using Lagrange.Core;
|
||||
using Lagrange.Core.Common.Entity;
|
||||
using Lagrange.Core.Common.Interface.Api;
|
||||
using Lagrange.Core.Event.EventArg;
|
||||
using Lagrange.Core.Message.Entity;
|
||||
using Lagrange.OneBot.Core.Entity.Notify;
|
||||
using Lagrange.OneBot.Core.Network;
|
||||
using Lagrange.OneBot.Database;
|
||||
using LiteDB;
|
||||
using Microsoft.Extensions.Logging;
|
||||
|
||||
namespace Lagrange.OneBot.Core.Notify;
|
||||
|
||||
public sealed class NotifyService(BotContext bot, ILogger<NotifyService> logger, LagrangeWebSvcCollection service, LiteDatabase database)
|
||||
{
|
||||
public void RegisterEvents()
|
||||
{
|
||||
bot.Invoker.OnGroupMessageReceived += async (_, @event) =>
|
||||
{
|
||||
if (@event.Chain.GetEntity<FileEntity>() is { FileId: { } id } file)
|
||||
{
|
||||
var fileInfo = new OneBotFileInfo(id, file.FileName, (ulong)file.FileSize, file.FileUrl ?? "");
|
||||
await service.SendJsonAsync(new OneBotGroupFile(bot.BotUin, @event.Chain.GroupUin ?? 0, @event.Chain.FriendUin, fileInfo));
|
||||
}
|
||||
};
|
||||
|
||||
bot.Invoker.OnFriendMessageReceived += async (_, @event) =>
|
||||
{
|
||||
if (@event.Chain.GetEntity<FileEntity>() is { FileUrl: { } url } file)
|
||||
{
|
||||
var fileInfo = new OneBotPrivateFileInfo(file.FileUuid ?? "", file.FileName, (ulong)file.FileSize, url, file.FileHash ?? "");
|
||||
await service.SendJsonAsync(new OneBotPrivateFile(bot.BotUin, @event.Chain.FriendUin, fileInfo));
|
||||
}
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupMuteEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
await service.SendJsonAsync(new OneBotGroupMute(bot.BotUin, @event.IsMuted ? "ban" : "lift_ban", @event.GroupUin, @event.OperatorUin ?? 0, 0, @event.IsMuted ? -1 : 0));
|
||||
};
|
||||
|
||||
bot.Invoker.OnFriendRequestEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
await service.SendJsonAsync(new OneBotFriendRequestNotice(bot.BotUin, @event.SourceUin));
|
||||
await service.SendJsonAsync(new OneBotFriendRequest(bot.BotUin, @event.SourceUin, @event.Message, @event.SourceUid));
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupInvitationReceived += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
|
||||
var requests = await bot.FetchGroupRequests();
|
||||
if (requests?.FirstOrDefault(x => @event.GroupUin == x.GroupUin && @event.InvitorUin == x.InvitorMemberUin) is { } request)
|
||||
{
|
||||
string flag = $"{request.Sequence}-{request.GroupUin}-{(uint)request.EventType}";
|
||||
await service.SendJsonAsync(new OneBotGroupRequest(bot.BotUin, @event.InvitorUin, @event.GroupUin, "invite", request.Comment, flag));
|
||||
}
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupJoinRequestEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
|
||||
var requests = await bot.FetchGroupRequests();
|
||||
if (requests?.FirstOrDefault(x => @event.GroupUin == x.GroupUin && @event.TargetUin == x.TargetMemberUin) is { } request)
|
||||
{
|
||||
string flag = $"{request.Sequence}-{request.GroupUin}-{(uint)request.EventType}-{Convert.ToInt32(request.IsFiltered)}";
|
||||
await service.SendJsonAsync(new OneBotGroupRequest(bot.BotUin, @event.TargetUin, @event.GroupUin, "add", request.Comment, flag));
|
||||
}
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupInvitationRequestEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
|
||||
var requests = await bot.FetchGroupRequests();
|
||||
if (requests?.FirstOrDefault(x => @event.GroupUin == x.GroupUin && @event.TargetUin == x.TargetMemberUin) is { } request)
|
||||
{
|
||||
string flag = $"{request.Sequence}-{request.GroupUin}-{(uint)request.EventType}";
|
||||
await service.SendJsonAsync(new OneBotGroupRequest(bot.BotUin, @event.TargetUin, @event.GroupUin, "add", request.Comment, flag) { InvitorId = @event.InvitorUin });
|
||||
}
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupAdminChangedEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
await service.SendJsonAsync(new OneBotGroupAdmin(bot.BotUin, @event.IsPromote ? "set" : "unset", @event.GroupUin, @event.AdminUin));
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupMemberIncreaseEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
string type = @event.Type.ToString().ToLower();
|
||||
await service.SendJsonAsync(new OneBotMemberIncrease(bot.BotUin, type, @event.GroupUin, @event.InvitorUin ?? 0, @event.MemberUin));
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupMemberDecreaseEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
|
||||
BotGroupRequest? botGroupRequest = (await bot.ContextCollection.Business.OperationLogic.FetchGroupRequests())
|
||||
?.AsParallel()
|
||||
.FirstOrDefault(r =>
|
||||
{
|
||||
return @event.Type switch
|
||||
{
|
||||
GroupMemberDecreaseEvent.EventType.Kick => r.EventType == BotGroupRequest.Type.KickMember
|
||||
&& r.TargetMemberUin == @event.MemberUin,
|
||||
GroupMemberDecreaseEvent.EventType.KickMe => r.EventType == BotGroupRequest.Type.KickSelf,
|
||||
_ => false
|
||||
} && r.GroupUin == @event.GroupUin;
|
||||
});
|
||||
|
||||
string type = @event.Type switch
|
||||
{
|
||||
GroupMemberDecreaseEvent.EventType.KickMe => "kick_me",
|
||||
GroupMemberDecreaseEvent.EventType.Disband => "disband",
|
||||
GroupMemberDecreaseEvent.EventType.Leave => "leave",
|
||||
GroupMemberDecreaseEvent.EventType.Kick => "kick",
|
||||
_ => @event.Type.ToString()
|
||||
};
|
||||
await service.SendJsonAsync(new OneBotMemberDecrease(bot.BotUin, type, @event.GroupUin, botGroupRequest?.InvitorMemberUin ?? 0, @event.MemberUin));
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupMemberMuteEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
string type = @event.Duration == 0 ? "lift_ban" : "ban";
|
||||
await service.SendJsonAsync(new OneBotGroupMute(bot.BotUin, type, @event.GroupUin, @event.OperatorUin ?? 0, @event.TargetUin, @event.Duration));
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupRecallEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
await service.SendJsonAsync(new OneBotGroupRecall(bot.BotUin)
|
||||
{
|
||||
GroupId = @event.GroupUin,
|
||||
UserId = @event.AuthorUin,
|
||||
MessageId = MessageRecord.CalcMessageHash(@event.Random, @event.Sequence),
|
||||
OperatorId = @event.OperatorUin,
|
||||
Tip = @event.Tip
|
||||
});
|
||||
};
|
||||
|
||||
bot.Invoker.OnFriendRecallEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
|
||||
var collection = database.GetCollection<MessageRecord>();
|
||||
var record = collection.FindOne(Query.And(
|
||||
Query.EQ("FriendUin", new BsonValue(@event.FriendUin)),
|
||||
Query.EQ("ClientSequence", new BsonValue(@event.ClientSequence)),
|
||||
Query.EQ("MessageId", new BsonValue(0x01000000L << 32 | @event.Random))
|
||||
));
|
||||
|
||||
await service.SendJsonAsync(new OneBotFriendRecall(bot.BotUin)
|
||||
{
|
||||
UserId = @event.FriendUin,
|
||||
MessageId = MessageRecord.CalcMessageHash(@event.Random, record.Sequence),
|
||||
Tip = @event.Tip
|
||||
});
|
||||
};
|
||||
|
||||
bot.Invoker.OnFriendPokeEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
await service.SendJsonAsync(new OneBotFriendPoke(bot.BotUin)
|
||||
{
|
||||
SenderId = @event.OperatorUin,
|
||||
UserId = @event.OperatorUin,
|
||||
TargetId = @event.TargetUin,
|
||||
Action = @event.Action,
|
||||
Suffix = @event.Suffix,
|
||||
ActionImgUrl = @event.ActionImgUrl
|
||||
});
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupPokeEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
await service.SendJsonAsync(new OneBotGroupPoke(bot.BotUin)
|
||||
{
|
||||
GroupId = @event.GroupUin,
|
||||
UserId = @event.OperatorUin,
|
||||
TargetId = @event.TargetUin,
|
||||
Action = @event.Action,
|
||||
Suffix = @event.Suffix,
|
||||
ActionImgUrl = @event.ActionImgUrl
|
||||
});
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupEssenceEvent += async (_, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
await service.SendJsonAsync(new OneBotGroupEssence(bot.BotUin)
|
||||
{
|
||||
SubType = @event.IsSet ? "add" : "delete",
|
||||
GroupId = @event.GroupUin,
|
||||
SenderId = @event.FromUin,
|
||||
OperatorId = @event.OperatorUin,
|
||||
MessageId = MessageRecord.CalcMessageHash(@event.Random, @event.Sequence),
|
||||
});
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupReactionEvent += async (bot, @event) =>
|
||||
{
|
||||
logger.LogInformation(@event.ToString());
|
||||
|
||||
var record = database.GetCollection<MessageRecord>().FindOne(Query.And(
|
||||
Query.EQ("GroupUin", new BsonValue(@event.TargetGroupUin)),
|
||||
Query.EQ("Sequence", new BsonValue(@event.TargetSequence))
|
||||
));
|
||||
|
||||
if (record == null)
|
||||
{
|
||||
logger.LogInformation(
|
||||
"Unable to find the corresponding message using GroupUin: {} and Sequence: {}",
|
||||
@event.TargetGroupUin,
|
||||
@event.TargetSequence
|
||||
);
|
||||
}
|
||||
|
||||
await service.SendJsonAsync(new OneBotGroupReaction(
|
||||
bot.BotUin,
|
||||
@event.TargetGroupUin,
|
||||
record?.MessageHash ?? 0,
|
||||
@event.OperatorUin,
|
||||
@event.IsAdd ? "add" : "remove",
|
||||
@event.Code,
|
||||
@event.Count
|
||||
));
|
||||
};
|
||||
|
||||
bot.Invoker.OnGroupNameChangeEvent += async (bot, @event) =>
|
||||
{
|
||||
logger.LogInformation("{}", @event);
|
||||
await service.SendJsonAsync(new OneBotGroupNameChange(
|
||||
bot.BotUin,
|
||||
@event.GroupUin,
|
||||
@event.Name
|
||||
));
|
||||
};
|
||||
|
||||
bot.Invoker.OnBotOnlineEvent += async (bot, @event) =>
|
||||
{
|
||||
await service.SendJsonAsync(new OneBotBotOnline(bot.BotUin, @event.Reason));
|
||||
};
|
||||
|
||||
bot.Invoker.OnBotOfflineEvent += async (bot, @event) =>
|
||||
{
|
||||
await service.SendJsonAsync(new OneBotBotOffline(bot.BotUin, @event.Tag, @event.Message));
|
||||
};
|
||||
}
|
||||
}
|
||||
90
Lagrange.Core/Lagrange.OneBot/Core/Notify/TicketService.cs
Normal file
90
Lagrange.Core/Lagrange.OneBot/Core/Notify/TicketService.cs
Normal file
@@ -0,0 +1,90 @@
|
||||
using System.Net;
|
||||
using Lagrange.Core;
|
||||
using Lagrange.Core.Common.Interface.Api;
|
||||
|
||||
namespace Lagrange.OneBot.Core.Notify;
|
||||
|
||||
public class TicketService
|
||||
{
|
||||
private readonly BotContext _context;
|
||||
|
||||
private readonly HttpClient _client;
|
||||
|
||||
private readonly CookieContainer _container;
|
||||
|
||||
private readonly Dictionary<string, (string PsSey, DateTime ExpireTime)> _psKeys;
|
||||
|
||||
private (string SKey, DateTime ExpireTime) _sKey;
|
||||
|
||||
public TicketService(BotContext context)
|
||||
{
|
||||
_container = new CookieContainer();
|
||||
_context = context;
|
||||
|
||||
var handler = new HttpClientHandler
|
||||
{
|
||||
CookieContainer = _container
|
||||
};
|
||||
_client = new HttpClient(handler);
|
||||
_psKeys = new Dictionary<string, (string, DateTime)>();
|
||||
}
|
||||
|
||||
public async Task<HttpResponseMessage> SendAsync(HttpRequestMessage message, string? domain = null)
|
||||
{
|
||||
if (!message.Headers.Contains("Cookie")) message.Headers.Add("Cookie", await GetCookies(domain ?? message.RequestUri?.Host ?? ""));
|
||||
return await _client.SendAsync(message);
|
||||
}
|
||||
|
||||
public async Task<int> GetCsrfToken()
|
||||
{
|
||||
string? sKey = await GetSKey();
|
||||
if (sKey == null) throw new InvalidDataException();
|
||||
|
||||
int hash = 5381;
|
||||
for (int i = 0, len = sKey.Length; i < len; ++i)
|
||||
{
|
||||
hash += (hash << 5) + sKey[i];
|
||||
}
|
||||
return hash & 2147483647;
|
||||
}
|
||||
|
||||
private async Task<string?> GetSKey()
|
||||
{
|
||||
if (DateTime.Now < _sKey.ExpireTime) return _sKey.SKey;
|
||||
|
||||
const string jump = "https%3A%2F%2Fh5.qzone.qq.com%2Fqqnt%2Fqzoneinpcqq%2Ffriend%3Frefresh%3D0%26clientuin%3D0%26darkMode%3D0&keyindex=19&random=2599";
|
||||
string url = $"https://ssl.ptlogin2.qq.com/jump?ptlang=1033&clientuin={_context.BotUin}&clientkey={await _context.GetClientKey()}&u1={jump}";
|
||||
await _client.GetAsync(url);
|
||||
|
||||
var cookies = _container.GetAllCookies();
|
||||
string sKey = cookies["skey"]?.Value ?? "";
|
||||
_sKey = (sKey, DateTime.Now.AddDays(1));
|
||||
|
||||
return sKey;
|
||||
}
|
||||
|
||||
public async Task<string> GetCookies(string domain)
|
||||
{
|
||||
string? skey = await GetSKey();
|
||||
string token;
|
||||
if (_psKeys.TryGetValue(domain, out var tokenTime))
|
||||
{
|
||||
if (DateTime.Now > tokenTime.ExpireTime)
|
||||
{
|
||||
token = (await _context.FetchCookies([domain]))[0];
|
||||
_psKeys[domain] = (token, DateTime.Now.AddDays(1)); // update pskey
|
||||
}
|
||||
else
|
||||
{
|
||||
token = tokenTime.PsSey;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
token = (await _context.FetchCookies([domain]))[0];
|
||||
_psKeys[domain] = (token, DateTime.Now.AddDays(1)); // get pskey
|
||||
}
|
||||
|
||||
return $"p_uin=o{_context.BotUin}; p_skey={token}; skey={skey}; uin=o{_context.BotUin}";
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user