add: following timeline

This commit is contained in:
trisua 2025-04-09 22:00:45 -04:00
parent 6141910059
commit 1ee1f29cdf
15 changed files with 176 additions and 9 deletions

View file

@ -106,6 +106,30 @@ impl DataManager {
Ok(res.unwrap())
}
/// Get users the given user is following.
///
/// # Arguments
/// * `id` - the ID of the user
pub async fn get_userfollows_by_initiator_all(&self, id: usize) -> Result<Vec<UserFollow>> {
let conn = match self.connect().await {
Ok(c) => c,
Err(e) => return Err(Error::DatabaseConnection(e.to_string())),
};
let res = query_rows!(
&conn,
"SELECT * FROM userfollows WHERE initiator = $1",
&[&(id as i64)],
|x| { Self::get_userfollow_from_row(x) }
);
if res.is_err() {
return Err(Error::GeneralNotFound("user follow".to_string()));
}
Ok(res.unwrap())
}
/// Get users following the given user.
///
/// # Arguments
@ -137,6 +161,30 @@ impl DataManager {
Ok(res.unwrap())
}
/// Get users following the given user.
///
/// # Arguments
/// * `id` - the ID of the user
pub async fn get_userfollows_by_receiver_all(&self, id: usize) -> Result<Vec<UserFollow>> {
let conn = match self.connect().await {
Ok(c) => c,
Err(e) => return Err(Error::DatabaseConnection(e.to_string())),
};
let res = query_rows!(
&conn,
"SELECT * FROM userfollows WHERE receiver = $1",
&[&(id as i64)],
|x| { Self::get_userfollow_from_row(x) }
);
if res.is_err() {
return Err(Error::GeneralNotFound("user follow".to_string()));
}
Ok(res.unwrap())
}
/// Complete a vector of just userfollows with their receiver as well.
pub async fn fill_userfollows_with_receiver(
&self,