2022-08-14 13:38:21 +02:00
|
|
|
mod data;
|
|
|
|
pub use data::Data;
|
2022-09-06 23:15:09 +02:00
|
|
|
use ruma::{RoomId, UserId};
|
2021-04-12 12:40:16 +02:00
|
|
|
|
2022-08-14 13:38:21 +02:00
|
|
|
use crate::service::*;
|
2021-04-12 12:40:16 +02:00
|
|
|
|
2022-08-14 13:38:21 +02:00
|
|
|
pub struct Service<D: Data> {
|
|
|
|
db: D,
|
|
|
|
}
|
2021-04-13 15:00:45 +02:00
|
|
|
|
2022-08-14 13:38:21 +02:00
|
|
|
impl Service<_> {
|
|
|
|
pub fn reset_notification_counts(&self, user_id: &UserId, room_id: &RoomId) -> Result<()> {
|
|
|
|
self.db.reset_notification_counts(user_id, room_id)
|
2021-04-13 15:00:45 +02:00
|
|
|
}
|
|
|
|
|
2022-06-19 22:56:14 +02:00
|
|
|
pub fn notification_count(&self, user_id: &UserId, room_id: &RoomId) -> Result<u64> {
|
2022-08-14 13:38:21 +02:00
|
|
|
self.db.notification_count(user_id, room_id)
|
2020-05-25 23:24:13 +02:00
|
|
|
}
|
|
|
|
|
2022-06-19 22:56:14 +02:00
|
|
|
pub fn highlight_count(&self, user_id: &UserId, room_id: &RoomId) -> Result<u64> {
|
2022-08-14 13:38:21 +02:00
|
|
|
self.db.highlight_count(user_id, room_id)
|
2020-05-24 08:30:57 +02:00
|
|
|
}
|
|
|
|
|
2022-06-19 22:56:14 +02:00
|
|
|
pub fn associate_token_shortstatehash(
|
|
|
|
&self,
|
2020-08-18 12:15:27 +02:00
|
|
|
room_id: &RoomId,
|
2022-06-19 22:56:14 +02:00
|
|
|
token: u64,
|
|
|
|
shortstatehash: u64,
|
|
|
|
) -> Result<()> {
|
2022-08-14 13:38:21 +02:00
|
|
|
self.db.associate_token_shortstatehash(user_id, room_id)
|
2022-06-19 22:56:14 +02:00
|
|
|
}
|
2021-06-17 20:44:29 +02:00
|
|
|
|
2022-06-19 22:56:14 +02:00
|
|
|
pub fn get_token_shortstatehash(&self, room_id: &RoomId, token: u64) -> Result<Option<u64>> {
|
2022-08-14 13:38:21 +02:00
|
|
|
self.db.get_token_shortstatehash(room_id, token)
|
2020-08-21 21:22:59 +02:00
|
|
|
}
|
2020-08-18 12:15:27 +02:00
|
|
|
|
2020-08-21 21:22:59 +02:00
|
|
|
pub fn get_shared_rooms<'a>(
|
|
|
|
&'a self,
|
2021-11-26 20:36:40 +01:00
|
|
|
users: Vec<Box<UserId>>,
|
|
|
|
) -> Result<impl Iterator<Item = Result<Box<RoomId>>> + 'a> {
|
2022-08-14 13:38:21 +02:00
|
|
|
self.db.get_shared_rooms(users)
|
2020-08-18 12:15:27 +02:00
|
|
|
}
|
2022-08-14 13:38:21 +02:00
|
|
|
}
|