AppFlowy/frontend/rust-lib/flowy-document/src/sql_tables/doc/doc_sql.rs

use crate::{
    errors::{internal_error, DocError},
    sql_tables::doc::{DocTable, DocTableChangeset},
};
use flowy_database::{
    prelude::*,
    schema::{doc_table, doc_table::dsl},
    ConnectionPool,
    SqliteConnection,
};
use std::sync::Arc;

pub struct DocTableSql {}

impl DocTableSql {
    pub(crate) fn create_doc_table(&self, doc_table: DocTable, conn: &SqliteConnection) -> Result<(), DocError> {
        let _ = diesel::insert_into(doc_table::table).values(doc_table).execute(conn)?;
        Ok(())
    }

    pub(crate) fn update_doc_table(
        &self,
        changeset: DocTableChangeset,
        conn: &SqliteConnection,
    ) -> Result<(), DocError> {
        diesel_update_table!(doc_table, changeset, conn);
        Ok(())
    }

    pub(crate) fn read_doc_table(&self, doc_id: &str, pool: Arc<ConnectionPool>) -> Result<DocTable, DocError> {
        let conn = &*pool.get().map_err(internal_error)?;
        let doc_table = dsl::doc_table
            .filter(doc_table::id.eq(doc_id))
            .first::<DocTable>(conn)?;

        Ok(doc_table)
    }

    #[allow(dead_code)]
    pub(crate) fn delete_doc(&self, doc_id: &str, conn: &SqliteConnection) -> Result<DocTable, DocError> {
        let doc_table = dsl::doc_table
            .filter(doc_table::id.eq(doc_id))
            .first::<DocTable>(conn)?;
        diesel_delete_table!(doc_table, doc_id, conn);
        Ok(doc_table)
    }
}