[irc] used nonempty prefixed nicks in 353 reply

This commit is contained in:
JustTestingV 2023-09-07 22:35:19 +03:00
parent 53f218c58f
commit 9c020e970e
5 changed files with 43 additions and 13 deletions

7
Cargo.lock generated
View File

@ -761,6 +761,7 @@ dependencies = [
"hyper 1.0.0-rc.3", "hyper 1.0.0-rc.3",
"lazy_static", "lazy_static",
"nom", "nom",
"nonempty",
"prometheus", "prometheus",
"quick-xml", "quick-xml",
"regex", "regex",
@ -887,6 +888,12 @@ dependencies = [
"minimal-lexical", "minimal-lexical",
] ]
[[package]]
name = "nonempty"
version = "0.8.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "aeaf4ad7403de93e699c191202f017118df734d3850b01e13a3a8b2e6953d3c9"
[[package]] [[package]]
name = "nu-ansi-term" name = "nu-ansi-term"
version = "0.46.0" version = "0.46.0"

View File

@ -19,6 +19,7 @@ prometheus = { version = "0.13.3", default-features = false }
regex = "1.7.1" regex = "1.7.1"
lazy_static = "1.4.0" lazy_static = "1.4.0"
nom = "7.1.3" nom = "7.1.3"
nonempty = "0.8.1"
tokio-rustls = "0.24.1" tokio-rustls = "0.24.1"
rustls-pemfile = "1.0.2" rustls-pemfile = "1.0.2"
quick-xml = { version = "0.30.0", features = ["async-tokio"] } quick-xml = { version = "0.30.0", features = ["async-tokio"] }

View File

@ -27,7 +27,7 @@ use crate::{
/// Opaque player identifier. Cannot contain spaces, must be shorter than 32. /// Opaque player identifier. Cannot contain spaces, must be shorter than 32.
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize)] #[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize)]
pub struct PlayerId(Str); pub struct PlayerId(pub Str);
impl PlayerId { impl PlayerId {
pub fn from(str: impl Into<Str>) -> Result<PlayerId> { pub fn from(str: impl Into<Str>) -> Result<PlayerId> {
let bytes = str.into(); let bytes = str.into();
@ -46,6 +46,22 @@ impl PlayerId {
self.0 self.0
} }
} }
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct Prefix(pub Str);
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct PrefixedNick {
pub prefix: Prefix,
pub nick: PlayerId,
}
impl PrefixedNick {
pub fn fromPlayerId(id: PlayerId) -> Result<PrefixedNick> {
Ok(PrefixedNick { prefix: Prefix(Str::from("")), nick: id } )
}
pub fn fromStr(nick: Str) -> Result<PrefixedNick> {
Ok(PrefixedNick { prefix: Prefix(Str::from("")), nick: PlayerId(nick) } )
}
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)] #[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct ConnectionId(pub AnonKey); pub struct ConnectionId(pub AnonKey);

View File

@ -2,6 +2,8 @@ use std::collections::HashMap;
use std::net::SocketAddr; use std::net::SocketAddr;
use futures_util::future::join_all; use futures_util::future::join_all;
use nonempty::nonempty;
use nonempty::NonEmpty;
use prometheus::{IntCounter, IntGauge, Registry as MetricsRegistry}; use prometheus::{IntCounter, IntGauge, Registry as MetricsRegistry};
use serde::Deserialize; use serde::Deserialize;
use tokio::io::{AsyncBufReadExt, AsyncWrite, AsyncWriteExt, BufReader, BufWriter}; use tokio::io::{AsyncBufReadExt, AsyncWrite, AsyncWriteExt, BufReader, BufWriter};
@ -666,22 +668,16 @@ async fn produce_on_join_cmd_messages(
} }
.write_async(writer) .write_async(writer)
.await?; .await?;
let mut members: String = if let Some(head) = room_info.members.first() { let prefixedMembers: Vec<PrefixedNick> = room_info.members.iter().map(|member| PrefixedNick::fromPlayerId(member.clone()).unwrap()).collect();
head.as_inner().clone() let nonEmptyMembers: NonEmpty<PrefixedNick> = NonEmpty::from_vec(prefixedMembers).unwrap_or(nonempty![PrefixedNick::fromStr(user.nickname.clone()).unwrap()]);
} else {
user.nickname.clone()
}.as_ref().into();
for i in &room_info.members[1..] {
members.push(' ');
members.push_str(i.as_inner());
}
ServerMessage { ServerMessage {
tags: vec![], tags: vec![],
sender: Some(config.server_name.clone()), sender: Some(config.server_name.clone()),
body: ServerMessageBody::N353NamesReply { body: ServerMessageBody::N353NamesReply {
client: user.nickname.clone(), client: user.nickname.clone(),
chan: chan.clone(), chan: chan.clone(),
members: members.into(), members: nonEmptyMembers.into(),
}, },
} }
.write_async(writer) .write_async(writer)

View File

@ -1,7 +1,9 @@
use nonempty::NonEmpty;
use tokio::io::AsyncWrite; use tokio::io::AsyncWrite;
use tokio::io::AsyncWriteExt; use tokio::io::AsyncWriteExt;
use super::*; use super::*;
use crate::core::player::PrefixedNick;
/// Server-to-client message. /// Server-to-client message.
#[derive(Clone, Debug, PartialEq, Eq)] #[derive(Clone, Debug, PartialEq, Eq)]
@ -121,7 +123,7 @@ pub enum ServerMessageBody {
N353NamesReply { N353NamesReply {
client: Str, client: Str,
chan: Chan, chan: Chan,
members: Str, // TODO make this a non-empty list with prefixes members: NonEmpty<PrefixedNick>,
}, },
N366NamesReplyEnd { N366NamesReplyEnd {
client: Str, client: Str,
@ -279,12 +281,20 @@ impl ServerMessageBody {
chan, chan,
members, members,
} => { } => {
let mut membersTail: Vec<PrefixedNick> = members.tail.clone();
membersTail.insert(0, members.clone().head);
let membersStr: Vec<String> = membersTail.iter().map(| prefixedNick| {
let pref: String = prefixedNick.prefix.0.clone().to_string();
let nick: String = prefixedNick.nick.0.clone().to_string();
pref + &nick
}).collect();
writer.write_all(b"353 ").await?; writer.write_all(b"353 ").await?;
writer.write_all(client.as_bytes()).await?; writer.write_all(client.as_bytes()).await?;
writer.write_all(b" = ").await?; writer.write_all(b" = ").await?;
chan.write_async(writer).await?; chan.write_async(writer).await?;
writer.write_all(b" :").await?; writer.write_all(b" :").await?;
writer.write_all(members.as_bytes()).await?; writer.write_all(membersStr.join(", ").as_bytes()).await?;
} }
ServerMessageBody::N366NamesReplyEnd { client, chan } => { ServerMessageBody::N366NamesReplyEnd { client, chan } => {
writer.write_all(b"366 ").await?; writer.write_all(b"366 ").await?;