1
0
Fork 0
mirror of https://gitlab.com/famedly/conduit.git synced 2025-08-06 17:40:59 +00:00
conduit/src/service/rooms/alias/mod.rs

67 lines
2.4 KiB
Rust
Raw Normal View History

#[tracing::instrument(skip(self, globals))]
2020-05-25 23:24:13 +02:00
pub fn set_alias(
&self,
alias: &RoomAliasId,
room_id: Option<&RoomId>,
globals: &super::globals::Globals,
) -> Result<()> {
if let Some(room_id) = room_id {
2020-06-03 20:55:11 +02:00
// New alias
2020-05-25 23:24:13 +02:00
self.alias_roomid
.insert(alias.alias().as_bytes(), room_id.as_bytes())?;
let mut aliasid = room_id.as_bytes().to_vec();
2021-02-11 13:16:14 +01:00
aliasid.push(0xff);
2020-05-25 23:24:13 +02:00
aliasid.extend_from_slice(&globals.next_count()?.to_be_bytes());
2021-06-08 18:10:00 +02:00
self.aliasid_alias.insert(&aliasid, &*alias.as_bytes())?;
2020-05-25 23:24:13 +02:00
} else {
2020-06-03 20:55:11 +02:00
// room_id=None means remove alias
if let Some(room_id) = self.alias_roomid.get(alias.alias().as_bytes())? {
2021-06-08 18:10:00 +02:00
let mut prefix = room_id.to_vec();
prefix.push(0xff);
for (key, _) in self.aliasid_alias.scan_prefix(prefix) {
self.aliasid_alias.remove(&key)?;
}
self.alias_roomid.remove(alias.alias().as_bytes())?;
2021-06-08 18:10:00 +02:00
} else {
return Err(Error::BadRequest(
2020-06-09 15:13:17 +02:00
ErrorKind::NotFound,
"Alias does not exist.",
2021-06-08 18:10:00 +02:00
));
2020-05-25 23:24:13 +02:00
}
}
Ok(())
}
#[tracing::instrument(skip(self))]
2021-11-26 20:36:40 +01:00
pub fn id_from_alias(&self, alias: &RoomAliasId) -> Result<Option<Box<RoomId>>> {
2020-05-25 23:24:13 +02:00
self.alias_roomid
2021-06-08 18:10:00 +02:00
.get(alias.alias().as_bytes())?
.map(|bytes| {
2021-11-27 00:30:28 +01:00
RoomId::parse(utils::string_from_bytes(&bytes).map_err(|_| {
Error::bad_database("Room ID in alias_roomid is invalid unicode.")
})?)
.map_err(|_| Error::bad_database("Room ID in alias_roomid is invalid."))
2020-05-25 23:24:13 +02:00
})
.transpose()
2020-05-25 23:24:13 +02:00
}
#[tracing::instrument(skip(self))]
2021-06-08 18:10:00 +02:00
pub fn room_aliases<'a>(
&'a self,
room_id: &RoomId,
2021-11-26 20:36:40 +01:00
) -> impl Iterator<Item = Result<Box<RoomAliasId>>> + 'a {
let mut prefix = room_id.as_bytes().to_vec();
2020-05-25 23:24:13 +02:00
prefix.push(0xff);
2021-06-08 18:10:00 +02:00
self.aliasid_alias.scan_prefix(prefix).map(|(_, bytes)| {
2021-06-17 20:34:14 +02:00
utils::string_from_bytes(&bytes)
2021-06-08 18:10:00 +02:00
.map_err(|_| Error::bad_database("Invalid alias bytes in aliasid_alias."))?
.try_into()
2021-06-17 20:34:14 +02:00
.map_err(|_| Error::bad_database("Invalid alias in aliasid_alias."))
2021-06-08 18:10:00 +02:00
})
2020-05-25 23:24:13 +02:00
}