AppFlowy/frontend/rust-lib/flowy-sqlite/src/lib.rs

pub use diesel::*;
pub use diesel_derives::*;
use diesel_migrations::*;
use std::{fmt::Debug, io, path::Path};
pub mod kv;
mod sqlite;

use crate::sqlite::PoolConfig;
pub use crate::sqlite::{ConnectionPool, DBConnection, Database};

pub mod schema;

#[macro_use]
pub mod macros;

#[macro_use]
extern crate diesel;
#[macro_use]
extern crate diesel_derives;
#[macro_use]
extern crate diesel_migrations;

pub type Error = diesel::result::Error;
pub mod prelude {
  pub use super::UserDatabaseConnection;
  pub use crate::*;
  pub use diesel::SqliteConnection;
  pub use diesel::{query_dsl::*, BelongingToDsl, ExpressionMethods, RunQueryDsl};
}

embed_migrations!("../flowy-sqlite/migrations/");
pub const DB_NAME: &str = "flowy-database.db";

pub fn init(storage_path: &str) -> Result<Database, io::Error> {
  if !Path::new(storage_path).exists() {
    std::fs::create_dir_all(storage_path)?;
  }
  let pool_config = PoolConfig::default();
  let database = Database::new(storage_path, DB_NAME, pool_config).map_err(as_io_error)?;
  let conn = database.get_connection().map_err(as_io_error)?;
  embedded_migrations::run(&*conn).map_err(as_io_error)?;
  Ok(database)
}

fn as_io_error<E>(e: E) -> io::Error
where
  E: Into<crate::sqlite::Error> + Debug,
{
  let msg = format!("{:?}", e);
  io::Error::new(io::ErrorKind::NotConnected, msg)
}

pub trait UserDatabaseConnection: Send + Sync {
  fn get_connection(&self) -> Result<DBConnection, String>;
}