sharp-chat/SharpChat.Protocol.IRC/ClientCommands/UserCommand.cs

160 lines
7.2 KiB
C#
Raw Normal View History

2022-08-30 15:05:29 +00:00
using SharpChat.Bans;
using SharpChat.Channels;
using SharpChat.Protocol.IRC.Replies;
using SharpChat.Sessions;
using SharpChat.Users;
using SharpChat.Users.Remote;
using System;
using System.Linq;
namespace SharpChat.Protocol.IRC.ClientCommands {
public class UserCommand : IClientCommand {
public const string NAME = @"USER";
private const byte MODE_W = 0x02;
private const byte MODE_I = 0x04;
public string CommandName => NAME;
public bool RequireSession => false;
private IRCServer Server { get; }
private Context Context { get; }
private UserManager Users { get; }
private ChannelManager Channels { get; }
private ChannelUserRelations ChannelUsers { get; }
private SessionManager Sessions { get; }
private IRemoteUserClient RemoteUserClient { get; }
private BanManager Bans { get; }
private WelcomeMessage WelcomeMessage { get; }
public UserCommand(
IRCServer server,
Context context,
UserManager users,
ChannelManager channels,
ChannelUserRelations channelUsers,
SessionManager sessions,
IRemoteUserClient remoteUserClient,
BanManager bans,
WelcomeMessage welcomeMessage
) {
Server = server ?? throw new ArgumentNullException(nameof(server));
Context = context ?? throw new ArgumentNullException(nameof(context));
Users = users ?? throw new ArgumentNullException(nameof(users));
Channels = channels ?? throw new ArgumentNullException(nameof(channels));
ChannelUsers = channelUsers ?? throw new ArgumentNullException(nameof(channelUsers));
Sessions = sessions ?? throw new ArgumentNullException(nameof(sessions));
RemoteUserClient = remoteUserClient ?? throw new ArgumentNullException(nameof(remoteUserClient));
Bans = bans ?? throw new ArgumentNullException(nameof(bans));
WelcomeMessage = welcomeMessage ?? throw new ArgumentNullException(nameof(welcomeMessage));
}
public void HandleCommand(ClientCommandContext ctx) {
if(ctx.Connection.HasAuthenticated) {
ctx.Connection.SendReply(new AlreadyRegisteredReply());
return;
}
// just drop subsequent calls
if(ctx.Connection.IsAuthenticating)
return;
ctx.Connection.IsAuthenticating = true;
string userName = ctx.Arguments.ElementAtOrDefault(0);
string modeStr = ctx.Arguments.ElementAtOrDefault(1);
//string param3 = ctx.Arguments.ElementAtOrDefault(2);
//string realName = ctx.Arguments.ElementAtOrDefault(3);
if(string.IsNullOrEmpty(userName) || string.IsNullOrEmpty(modeStr)) {
ctx.Connection.SendReply(new NeedMoreParamsReply(NAME));
return;
}
// TODO: should accept normal text username in the future!!!!
if(!long.TryParse(userName, out long userId)) {
ctx.Connection.SendReply(new PasswordMismatchReply());
ctx.Connection.Close();
return;
}
if(!int.TryParse(modeStr, out int mode))
mode = 0;
bool isInvisible = (mode & MODE_I) > 0;
bool receiveWallOps = (mode & MODE_W) > 0;
Action<Exception> exceptionHandler = new(ex => {
Logger.Debug($@"[{ctx.Connection}] Auth fail: {ex.Message}");
ctx.Connection.SendReply(new PasswordMismatchReply());
ctx.Connection.Close();
});
RemoteUserClient.AuthenticateUser(
new UserAuthRequest(userId, ctx.Connection.Password, ctx.Connection.RemoteAddress),
res => {
ctx.Connection.Password = null;
ctx.Connection.HasAuthenticated = true;
Bans.CheckBan(res, ctx.Connection.RemoteAddress, ban => {
if(ban.IsPermanent || ban.Expires > DateTimeOffset.Now) {
// should probably include the time
ctx.Connection.SendReply(new YouAreBannedReply(@"You have been banned."));
ctx.Connection.Close();
return;
}
Users.Connect(res, user => {
Sessions.HasAvailableSessions(user, available => {
// Enforce a maximum amount of connections per user
if(!available) {
// map somethign to this
//ctx.Connection.SendPacket(new AuthFailPacket(AuthFailReason.MaxSessions));
ctx.Connection.Close();
return;
}
Sessions.Create(ctx.Connection, user, session => {
// TODO: !!!!!!!!!!!!!!!!
ctx.Connection.Session = session;
session.Connection = ctx.Connection;
ctx.Connection.SendReply(new WelcomeReply(Server, user));
ctx.Connection.SendReply(new YourHostReply(Server));
ctx.Connection.SendReply(new CreatedReply(Context));
ctx.Connection.SendReply(new MyInfoReply(Server));
ctx.Connection.SendReply(new ISupportReply(Server));
if(WelcomeMessage.HasRandom) {
ctx.Connection.SendReply(new MotdStartReply());
string line = WelcomeMessage.GetRandomString();
if(!string.IsNullOrWhiteSpace(line))
ctx.Connection.SendReply(new MotdReply(line));
ctx.Connection.SendReply(new MotdEndReply());
} else
ctx.Connection.SendReply(new NoMotdReply());
// are these necessary?
ctx.Connection.SendReply(new ListUserClientReply());
ctx.Connection.SendReply(new ListUserOperatorsReply());
ctx.Connection.SendReply(new ListUserUnknownReply());
ctx.Connection.SendReply(new ListUserChannelsReply());
ctx.Connection.SendReply(new ListUserMeReply());
Channels.GetDefaultChannels(channels => {
foreach(IChannel channel in channels)
ChannelUsers.JoinChannel(channel, session);
});
});
});
});
}, exceptionHandler);
},
exceptionHandler
);
}
}
}