use crate::prelude::*;
use r2d2::PooledConnection;
use r2d2_sqlite::SqliteConnectionManager;
use rusqlite::{params, Row};
use crate::persist::local::models::schema_models::Project;
pub type DbConn = PooledConnection<SqliteConnectionManager>;
#[derive(Clone, Debug)]
pub struct ProjectRepo {
db: Arc<DatabaseService>,
}
impl ProjectRepo {
pub fn new(db: Arc<DatabaseService>) -> Self {
Self { db }
}
}
impl Repository<Project> for ProjectRepo {
fn get_connection(&self) -> Result<r2d2::PooledConnection<SqliteConnectionManager>> {
self.db.get_connection().map_err(Into::into)
}
fn map_row(row: &Row) -> Result<Project> {
let created_at_str: String = row
.get("created_at")
.map_err(|e| anyhow::anyhow!("{}", e))?;
let updated_at_str: String = row
.get("updated_at")
.map_err(|e| anyhow::anyhow!("{}", e))?;
let created_at = parse_timestamp(&created_at_str)?;
let updated_at = parse_timestamp(&updated_at_str)?;
Ok(Project::from_db_row(
row.get("id").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("code").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("name").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("description")
.map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("status").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("current_phase")
.map_err(|e| anyhow::anyhow!("{}", e))?,
created_at,
updated_at,
)
.map_err(|e| anyhow::anyhow!("{}", e))?)
}
fn create_impl(
&self,
entity: &Project,
conn: &PooledConnection<SqliteConnectionManager>,
) -> Result<()> {
conn.execute(
r#"
INSERT INTO projects (id, code, name, description, status, current_phase, created_at, updated_at)
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8)
"#,
params![
&entity.id,
&entity.code,
&entity.name,
&entity.description,
&entity.status,
&entity.current_phase,
&entity.created_at.to_rfc3339(),
&entity.updated_at.to_rfc3339(),
],
).map_err(|e| anyhow::anyhow!("{}", e))?;
Ok(())
}
fn find_by_id_impl(
&self,
id: &str,
conn: &PooledConnection<SqliteConnectionManager>,
) -> Result<Option<Project>> {
let mut stmt = conn.prepare(
"SELECT id, code, name, description, status, current_phase, created_at, updated_at FROM projects WHERE id = ?1",
).map_err(|e| anyhow::anyhow!("{}", e))?;
let mut rows = stmt
.query(params![id])
.map_err(|e| anyhow::anyhow!("{}", e))?;
if let Some(row) = rows.next().map_err(|e| anyhow::anyhow!("{}", e))? {
Ok(Some(Self::map_row(row)?))
} else {
Ok(None)
}
}
fn update_impl(
&self,
entity: &Project,
conn: &PooledConnection<SqliteConnectionManager>,
) -> Result<()> {
conn.execute(
r#"
UPDATE projects SET name = ?1, description = ?2, status = ?3, current_phase = ?4, updated_at = ?5
WHERE id = ?6
"#,
params![
&entity.name,
&entity.description,
&entity.status,
&entity.current_phase,
chrono::Utc::now().to_rfc3339(),
&entity.id,
],
).map_err(|e| anyhow::anyhow!("{}", e))?;
Ok(())
}
fn upsert_impl(
&self,
entity: &Project,
conn: &PooledConnection<SqliteConnectionManager>,
) -> Result<()> {
conn.execute(
r#"
INSERT OR REPLACE INTO projects (id, code, name, description, status, current_phase, created_at, updated_at)
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8)
"#,
params![
&entity.id,
&entity.code,
&entity.name,
&entity.description,
&entity.status,
&entity.current_phase,
&entity.created_at.to_rfc3339(),
&entity.updated_at.to_rfc3339(),
],
).map_err(|e| anyhow::anyhow!("{}", e))?;
Ok(())
}
fn delete_impl(
&self,
id: &str,
conn: &PooledConnection<SqliteConnectionManager>,
) -> Result<()> {
conn.execute("DELETE FROM projects WHERE id = ?", params![id])
.map_err(|e| anyhow::anyhow!("{}", e))?;
Ok(())
}
fn list_impl(
&self,
conn: &PooledConnection<SqliteConnectionManager>,
_offset: Option<i64>,
_limit: Option<i64>,
) -> Result<Vec<Project>> {
let mut stmt = conn.prepare(
"SELECT id, code, name, description, status, current_phase, created_at, updated_at FROM projects ORDER BY created_at DESC",
).map_err(|e| anyhow::anyhow!("{}", e))?;
let mut rows = stmt.query([]).map_err(|e| anyhow::anyhow!("{}", e))?;
let mut projects = Vec::new();
while let Some(row) = rows.next().map_err(|e| anyhow::anyhow!("{}", e))? {
projects.push(Self::map_row(row)?);
}
Ok(projects)
}
fn count_impl(&self, conn: &PooledConnection<SqliteConnectionManager>) -> Result<i64> {
let count: i64 = conn
.query_row("SELECT COUNT(*) FROM projects", [], |row| row.get(0))
.map_err(|e| anyhow::anyhow!("{}", e))?;
Ok(count)
}
}
fn parse_timestamp(s: &str) -> Result<chrono::DateTime<chrono::Utc>> {
if let Ok(parsed) = chrono::DateTime::parse_from_rfc3339(s) {
Ok(parsed.with_timezone(&chrono::Utc))
} else {
let parsed = chrono::NaiveDateTime::parse_from_str(s, "%Y-%m-%d %H:%M:%S")?;
Ok(parsed.and_utc())
}
}