xssbook/src/types/post.rs

use core::fmt;
use serde::Serialize;
use tracing::instrument;

use crate::database;
use crate::types::http::{ResponseCode, Result};

use super::comment::Comment;

#[derive(Serialize)]
pub struct Post {
    pub post_id: u64,
    pub user_id: u64,
    pub content: String,
    pub date: u64,
    pub likes: u64,
    pub liked: bool,
    pub comments: Vec<Comment>,
}

impl fmt::Debug for Post {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("Post")
            .field("post_id", &self.post_id)
            .finish()
    }
}

impl Post {
    #[instrument()]
    pub fn from_post_id(post_id: u64) -> Result<Self> {
        let Ok(Some(post)) = database::posts::get_post(post_id) else {
            return Err(ResponseCode::BadRequest.text("Post does not exist"))
        };

        Ok(post)
    }

    #[instrument()]
    pub fn from_post_page(page: u64) -> Result<Vec<Self>> {
        let Ok(posts) = database::posts::get_post_page(page) else {
            return Err(ResponseCode::BadRequest.text("Failed to fetch posts"))
        };
        Ok(posts)
    }

    #[instrument()]
    pub fn from_user_post_page(user_id: u64, page: u64) -> Result<Vec<Self>> {
        let Ok(posts) = database::posts::get_users_post_page(user_id, page) else {
            return Err(ResponseCode::BadRequest.text("Failed to fetch posts"))
        };
        Ok(posts)
    }

    #[instrument()]
    pub fn reterieve_all() -> Result<Vec<Self>> {
        let Ok(posts) = database::posts::get_all_posts() else {
            return Err(ResponseCode::InternalServerError.text("Failed to fetch posts"))
        };
        Ok(posts)
    }

    #[instrument()]
    pub fn new(user_id: u64, content: String) -> Result<Self> {
        let Ok(post) = database::posts::add_post(user_id, &content) else {
            tracing::error!("Failed to create post");
            return Err(ResponseCode::InternalServerError.text("Failed to create post"))
        };

        Ok(post)
    }
}