feat: basic api & model for post

src/models.rs

@@ -0,0 +1,82 @@
+#![allow(clippy::all)]
+
+use chrono::NaiveDateTime;
+use diesel::{insert_into, Connection, ExpressionMethods, QueryDsl, RunQueryDsl, SqliteConnection};
+use std::env;
+
+use crate::schema::posts;
+
+type MR<T> = Result<T, diesel::result::Error>;
+
+no_arg_sql_function!(RANDOM, (), "Represents the sql RANDOM() function");
+
+pub fn establish_connection() -> SqliteConnection {
+    let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set");
+    SqliteConnection::establish(&database_url)
+        .expect(&format!("Error connecting to {}", database_url))
+}
+
+#[derive(Queryable, Debug)]
+pub struct Post {
+    pub id: i32,
+    pub author_hash: String,
+    pub content: String,
+    pub cw: String,
+    pub author_title: String,
+    pub n_likes: i32,
+    pub n_comments: i32,
+    pub create_time: NaiveDateTime,
+    pub last_comment_time: NaiveDateTime,
+    pub is_deleted: bool,
+    pub is_reported: bool,
+    pub hot_score: i32,
+    pub allow_search: bool,
+}
+
+#[derive(Insertable)]
+#[table_name = "posts"]
+pub struct NewPost<'a> {
+    pub content: &'a str,
+    pub cw: &'a str,
+    pub author_hash: &'a str,
+    pub author_title: &'a str,
+    // TODO: tags
+}
+
+impl Post {
+    pub fn get(conn: &SqliteConnection, id: i32) -> MR<Self> {
+        posts::table.find(id).first(conn)
+    }
+
+    pub fn gets_by_page(
+        conn: &SqliteConnection,
+        order_mode: u8,
+        page: u32,
+        page_size: u32,
+        is_admin: bool,
+    ) -> MR<Vec<Self>> {
+        let mut query = posts::table.into_boxed();
+        if !is_admin {
+            query = query.filter(posts::is_deleted.eq(false));
+            if order_mode > 0 {
+                query = query.filter(posts::is_reported.eq(false))
+            }
+        }
+
+        match order_mode {
+            1 => query = query.order(posts::last_comment_time.desc()),
+            2 => query = query.order(posts::hot_score.desc()),
+            3 => query = query.order(RANDOM),
+            _ => query = query.order(posts::id.desc()),
+        }
+        query
+            .offset(((page - 1) * page_size).into())
+            .limit(page_size.into())
+            .load(conn)
+    }
+
+    pub fn create(conn: &SqliteConnection, new_post: NewPost) -> MR<usize> {
+        // TODO: tags
+        insert_into(posts::table).values(&new_post).execute(conn)
+    }
+}