1 files changed, 10 insertions, 10 deletions

diff --git a/src/types/like.rs b/src/types/like.rs
index 1c113c1..8eec941 100644
--- a/src/types/like.rs
+++ b/src/types/like.rs
@@ -1,7 +1,7 @@
 use serde::Serialize;
 use tracing::instrument;
 
-use crate::database;
+use crate::database::Database;
 use crate::types::http::{ResponseCode, Result};
 
 #[derive(Serialize)]
@@ -11,9 +11,9 @@ pub struct Like {
 }
 
 impl Like {
-    #[instrument()]
-    pub fn add_liked(user_id: u64, post_id: u64) -> Result<()> {
-        let Ok(liked) = database::likes::add_liked(user_id, post_id) else {
+    #[instrument(skip(db))]
+    pub fn add_liked(db: &Database, user_id: u64, post_id: u64) -> Result<()> {
+        let Ok(liked) = db.add_liked(user_id, post_id) else {
             return Err(ResponseCode::BadRequest.text("Failed to add like status"))
         };
 
@@ -24,9 +24,9 @@ impl Like {
         Ok(())
     }
 
-    #[instrument()]
-    pub fn remove_liked(user_id: u64, post_id: u64) -> Result<()> {
-        let Ok(liked) = database::likes::remove_liked(user_id, post_id) else {
+    #[instrument(skip(db))]
+    pub fn remove_liked(db: &Database, user_id: u64, post_id: u64) -> Result<()> {
+        let Ok(liked) = db.remove_liked(user_id, post_id) else {
             return Err(ResponseCode::BadRequest.text("Failed to remove like status"))
         };
 
@@ -37,9 +37,9 @@ impl Like {
         Ok(())
     }
 
-    #[instrument()]
-    pub fn reterieve_all() -> Result<Vec<Self>> {
-        let Ok(likes) = database::likes::get_all_likes() else {
+    #[instrument(skip(db))]
+    pub fn reterieve_all(db: &Database) -> Result<Vec<Self>> {
+        let Ok(likes) = db.get_all_likes() else {
             return Err(ResponseCode::InternalServerError.text("Failed to fetch likes"))
         };
         Ok(likes)