296 lines
11 KiB
C#
296 lines
11 KiB
C#
using Microsoft.AspNetCore.Authorization;
|
|
using Microsoft.AspNetCore.Identity.Data;
|
|
using Microsoft.AspNetCore.SignalR;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using StackExchange.Redis;
|
|
using System.Diagnostics;
|
|
using System.Security.Cryptography.Xml;
|
|
using WebAppServer1.ApplicationDbContext;
|
|
using WebAppServer1.Authentication;
|
|
using WebAppServer1.Models;
|
|
using WebAppServer1.Tool;
|
|
|
|
namespace WebAppServer1.Chat
|
|
{
|
|
public class Chat : Hub
|
|
{
|
|
|
|
private readonly ILogger<Chat> logger;
|
|
private readonly IConnectionMultiplexer _redis;
|
|
private readonly AppDbContext pgSql;
|
|
private readonly TokenService jwtService;
|
|
|
|
|
|
public Chat(ILogger<Chat> _logger, IConnectionMultiplexer redis, AppDbContext _pgsql,TokenService _jwtService)
|
|
{
|
|
logger = _logger;
|
|
_redis = redis;
|
|
pgSql = _pgsql;
|
|
jwtService = _jwtService;
|
|
}
|
|
|
|
//wss子端点API
|
|
|
|
public override async Task OnConnectedAsync()
|
|
{
|
|
var db = _redis.GetDatabase();
|
|
await db.SetAddAsync("AllConnections", Context.ConnectionId);
|
|
await BroadcastOnlineCount();
|
|
await base.OnConnectedAsync();
|
|
}
|
|
|
|
public override async Task OnDisconnectedAsync(Exception? exception)
|
|
{
|
|
var db = _redis.GetDatabase();
|
|
await db.SetRemoveAsync("AllConnections", Context.ConnectionId);
|
|
// 遍历所有群组，移除该连接
|
|
var server = _redis.GetServer(_redis.GetEndPoints().First());
|
|
foreach (var key in server.Keys(pattern: "Group:*"))
|
|
{
|
|
await db.SetRemoveAsync(key, Context.ConnectionId);
|
|
var groupName = key.ToString().Replace("Group:", "");
|
|
await BroadcastGroupCount(groupName);
|
|
}
|
|
await BroadcastOnlineCount();
|
|
|
|
await base.OnDisconnectedAsync(exception);
|
|
}
|
|
|
|
//简单广播消息
|
|
[Authorize]
|
|
public async Task SendMessage(string user, string message)
|
|
{
|
|
logger.LogWarning("广播消息");
|
|
//测试发送
|
|
var exuser = Context.User?.Identity?.Name ?? "匿名";
|
|
await Clients.All.SendAsync("ReceiveMessage", exuser, message);
|
|
}
|
|
|
|
//群聊分组消息
|
|
public async Task JoinGroup(string groupName)
|
|
{
|
|
var db = _redis.GetDatabase();
|
|
await Groups.AddToGroupAsync(Context.ConnectionId, groupName);
|
|
await db.SetAddAsync($"Group:{groupName}", Context.ConnectionId);
|
|
await BroadcastGroupCount(groupName);
|
|
|
|
logger.LogWarning($"用户: {Context.ConnectionId}加入了{groupName}群组!");
|
|
await Clients.Group(groupName).SendAsync("ReceiveMessage", groupName, "系统", $"{Context.ConnectionId} 加入了群组");
|
|
}
|
|
|
|
public async Task LeaveGroup(string groupName)
|
|
{
|
|
var db = _redis.GetDatabase();
|
|
await Groups.RemoveFromGroupAsync(Context.ConnectionId, groupName);
|
|
await db.SetRemoveAsync($"Group:{groupName}", Context.ConnectionId);
|
|
await BroadcastGroupCount(groupName);
|
|
logger.LogInformation($"用户{Context.ConnectionId}---离开了---{groupName}群组");
|
|
await Clients.Group(groupName).SendAsync("ReceiveMessage",groupName, "系统", $"{Context.ConnectionId} 离开了群聊 {groupName}");
|
|
}
|
|
[Authorize]
|
|
public async Task SendMessageToGroup(string groupName, string user, string message)
|
|
{
|
|
logger.LogInformation($"用户{user}发来的的消息:{message}");
|
|
await Clients.Group(groupName).SendAsync("ReceiveMessage", groupName, user, message);
|
|
}
|
|
|
|
//发送已读回执
|
|
public async Task SendReadReceipt(int messageId)
|
|
{
|
|
var update = await pgSql.Messages.FindAsync(messageId);
|
|
if (update == null) { return; };
|
|
update.IsRead = true;
|
|
pgSql.Messages.Update(update);
|
|
await pgSql.SaveChangesAsync();
|
|
await Clients.User(update.SenderId.ToString()).SendAsync("SendReadReceipt",messageId);
|
|
}
|
|
|
|
//发送私聊消息
|
|
[Authorize]
|
|
public async Task SendPrivateMessage(string userid, string message)
|
|
{
|
|
var receiver = await pgSql.Users.FindAsync(int.Parse(userid));
|
|
if (receiver == null)
|
|
{
|
|
await Clients.Caller.SendAsync("SendPrivateMessage", false, "用户不存在", Context.UserIdentifier);
|
|
return;
|
|
}
|
|
var newMesssage = new Message
|
|
{
|
|
SenderId = int.Parse(Context.UserIdentifier!),
|
|
ReceiverId = receiver.Id,
|
|
Content = message,
|
|
CreatedAt = DateTime.UtcNow,
|
|
MessageType = MessageType.Text,
|
|
IsDeleted = false,
|
|
IsRead = false
|
|
};
|
|
pgSql.Messages.Add(newMesssage);
|
|
await pgSql.SaveChangesAsync();
|
|
await Clients.User(userid).SendAsync("SendPrivateMessage", true, message,Context.UserIdentifier,newMesssage.Id);
|
|
await Clients.Caller.SendAsync("SendPrivateMessage", false, message, Context.UserIdentifier,newMesssage.Id);
|
|
}
|
|
|
|
//添加好友申请
|
|
public async Task<bool> AddFriend(int userId)
|
|
{
|
|
//var user = await pgSql.Users.FindAsync(userId);
|
|
pgSql.Friends.Add(new Friend
|
|
{
|
|
UserId = int.Parse(Context.UserIdentifier!),
|
|
FriendId = userId,
|
|
CreatedAt = DateTime.UtcNow,
|
|
Status = FriendStatus.Pending
|
|
});
|
|
await pgSql.SaveChangesAsync();
|
|
return true;
|
|
}
|
|
|
|
//搜索好友
|
|
public async Task<object> SearchFriends(string username)
|
|
{
|
|
User? user = null;
|
|
|
|
if (int.TryParse(username, out int userId))
|
|
{
|
|
user = await pgSql.Users.FindAsync(userId);
|
|
Console.WriteLine("按 ID 查询: " + userId);
|
|
}
|
|
if (user == null)
|
|
{
|
|
user = await pgSql.Users.FirstOrDefaultAsync(a => a.Username == username);
|
|
}
|
|
return user == null
|
|
? new { success = false }
|
|
: new
|
|
{
|
|
success = true,
|
|
userid = user.Id,
|
|
username = user.Username,
|
|
nickname = user.Nickname,
|
|
avatarurl = user.AvatarUrl,
|
|
signature = user.Signature,
|
|
isonline = user.IsOnline
|
|
};
|
|
}
|
|
|
|
//获取用户表单测试
|
|
public async Task<List<object>> GetFormData()
|
|
{
|
|
var users = await pgSql.Users
|
|
.Select(u => new {
|
|
Id = u.Id,
|
|
Name = u.Username,
|
|
Password = u.PasswordHash,
|
|
CreatedDate = u.CreatedAt,
|
|
LastModifiedDate = u.LastActive
|
|
})
|
|
.ToListAsync();
|
|
|
|
return users.Cast<object>().ToList();
|
|
}
|
|
|
|
|
|
//刷新Token
|
|
// 刷新 Token
|
|
public async Task RefreshAccessToken(string refreshToken)
|
|
{
|
|
var token = await pgSql.Tokens.FirstOrDefaultAsync(t => t.RefreshToken == refreshToken);
|
|
|
|
if (token == null || token.IsRevoked || token.ExpiresAt < DateTime.UtcNow)
|
|
{
|
|
await Clients.Caller.SendAsync("RefreshFailed", "Refresh token invalid or expired");
|
|
return;
|
|
}
|
|
|
|
var newAccessToken = jwtService.GenerateAccessToken(token.UserName, token.UserId);
|
|
var newRefreshToken = await jwtService.GenerateRefreshToken(token.UserId, token.UserName);
|
|
|
|
token.IsRevoked = true;
|
|
await pgSql.SaveChangesAsync();
|
|
|
|
await Clients.Caller.SendAsync("ReceiveNewAccessToken", newAccessToken, newRefreshToken);
|
|
}
|
|
|
|
//登录认证
|
|
public async Task<LoginResultResponse> LoginAuthentication(string username, string password, string ua)
|
|
{
|
|
var user = await pgSql.Users.FirstOrDefaultAsync(a => a.Username == username && a.PasswordHash == password);
|
|
if (user == null)
|
|
{
|
|
//await Clients.Client(Context.ConnectionId).SendAsync("LoginResult",false,"用户名或密码错误!");
|
|
return new LoginResultResponse { Success = false };
|
|
}
|
|
if (user == null) { return new LoginResultResponse { Success = false }; }
|
|
var accessToken = jwtService.GenerateAccessToken(user.Username,user.Id);
|
|
var refreshToken = await jwtService.GenerateRefreshToken(user.Id, username);
|
|
//await Clients.Client(Context.ConnectionId).SendAsync("LoginResult", true, accessToken, refreshToken);
|
|
logger.LogInformation($"用户:{username}正在进行登录认证!密码:{password}===accessToken:{accessToken}");
|
|
var ip = Context.GetHttpContext()?.Connection?.RemoteIpAddress?.ToString();
|
|
if (ip == null ) { return new LoginResultResponse { Success = false }; }
|
|
pgSql.LoginRecords.Add(new LoginRecord
|
|
{
|
|
UserId = user.Id,
|
|
LoginTime = DateTime.UtcNow,
|
|
IpAddress = ip,
|
|
DeviceInfo = ua,
|
|
IsSuccess = true
|
|
});
|
|
await pgSql.SaveChangesAsync();
|
|
return new LoginResultResponse
|
|
{
|
|
Success = true,
|
|
AccessToken = accessToken,
|
|
RefreshToken = refreshToken,
|
|
Nickname = user.Nickname,
|
|
Id = user.Id,
|
|
IsOnline = true,
|
|
AvatarUrl = user.AvatarUrl,
|
|
Signature = user.Signature
|
|
};
|
|
}
|
|
//新用户注册
|
|
public async Task<object> Register()
|
|
{
|
|
var newUser = new User
|
|
{
|
|
Username = "user_" + Guid.NewGuid().ToString("N").Substring(0, 8),
|
|
Nickname = GenerateTool.GenerateRandomNickname(),
|
|
PasswordHash = GenerateTool.GeneratePassword(),
|
|
Signature = GenerateTool.GenerateRandomSignature(),
|
|
AvatarUrl = GenerateTool.GenerateRandomAvatarUrl(),
|
|
CreatedAt = DateTime.UtcNow,
|
|
LastActive = DateTime.UtcNow
|
|
};
|
|
pgSql.Add(newUser);
|
|
await pgSql.SaveChangesAsync();
|
|
logger.LogInformation($"新用户:{newUser.Username}正在进行登录注册操作!密码:{newUser.Nickname}");
|
|
return new
|
|
{
|
|
Username = newUser.Username,
|
|
Password = newUser.PasswordHash,
|
|
Success = true
|
|
};
|
|
|
|
}
|
|
|
|
//统计在线人数
|
|
private async Task BroadcastOnlineCount()
|
|
{
|
|
var db = _redis.GetDatabase();
|
|
var count = await db.SetLengthAsync("AllConnections");
|
|
await Clients.All.SendAsync("OnlineCount", count);
|
|
}
|
|
|
|
//统计群组在线人数
|
|
private async Task BroadcastGroupCount(string groupName)
|
|
{
|
|
var db = _redis.GetDatabase();
|
|
var count = await db.SetLengthAsync($"Group:{groupName}");
|
|
await Clients.Group(groupName).SendAsync("GroupCount", groupName, count);
|
|
}
|
|
|
|
|
|
}
|
|
}
|