use crate::prelude::*;
use r2d2::PooledConnection;
use r2d2_sqlite::SqliteConnectionManager;
use rusqlite::{params, Row};
use crate::persist::local::models::schema_models::Directive;
pub type DbConn = PooledConnection<SqliteConnectionManager>;
#[derive(Clone, Debug)]
pub struct DirectiveRepo {
db: Arc<DatabaseService>,
}
impl DirectiveRepo {
pub fn new(db: Arc<DatabaseService>) -> Self {
Self { db }
}
}
impl Repository<Directive> for DirectiveRepo {
fn get_connection(&self) -> Result<r2d2::PooledConnection<SqliteConnectionManager>> {
self.db.get_connection().map_err(Into::into)
}
fn map_row(row: &Row) -> Result<Directive> {
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)?;
Directive::from_db_row(
row.get("id").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("project_id")
.map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("code").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("title").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("rule").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("priority").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("status").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("context").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("rationale").map_err(|e| anyhow::anyhow!("{}", e))?,
row.get("category").map_err(|e| anyhow::anyhow!("{}", e))?,
created_at,
updated_at,
)
.map_err(|e| anyhow::anyhow!("{}", e))
}
fn create_impl(
&self,
entity: &Directive,
conn: &PooledConnection<SqliteConnectionManager>,
) -> Result<()> {
conn.execute(
r#"
INSERT INTO directives (id, project_id, code, title, rule, category, priority, status, context, rationale, created_at, updated_at)
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11, ?12)
"#,
params![
&entity.id,
&entity.project_id,
&entity.code,
&entity.title,
&entity.rule,
&entity.category,
&entity.priority,
"active",
&entity.context,
&entity.rationale,
&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<Directive>> {
let mut stmt = conn.prepare(
"SELECT id, project_id, code, title, rule, category, priority, status, context, rationale, created_at, updated_at FROM directives 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: &Directive,
conn: &PooledConnection<SqliteConnectionManager>,
) -> Result<()> {
conn.execute(
r#"
UPDATE directives SET title = ?1, rule = ?2, context = ?3, rationale = ?4, updated_at = ?5
WHERE id = ?6
"#,
params![
&entity.title,
&entity.rule,
&entity.context,
&entity.rationale,
chrono::Utc::now().to_rfc3339(),
&entity.id,
],
).map_err(|e| anyhow::anyhow!("{}", e))?;
Ok(())
}
fn upsert_impl(
&self,
entity: &Directive,
conn: &PooledConnection<SqliteConnectionManager>,
) -> Result<()> {
conn.execute(
r#"
INSERT OR REPLACE INTO directives (id, project_id, code, title, rule, category, priority, status, context, rationale, created_at, updated_at)
VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, ?11, ?12)
"#,
params![
&entity.id,
&entity.project_id,
&entity.code,
&entity.title,
&entity.rule,
&entity.category,
&entity.priority,
"active",
&entity.context,
&entity.rationale,
&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 directives 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<Directive>> {
let mut stmt = conn.prepare(
"SELECT id, project_id, code, title, rule, category, priority, status, context, rationale, created_at, updated_at FROM directives ORDER BY created_at DESC",
).map_err(|e| anyhow::anyhow!("{}", e))?;
let mut rows = stmt.query([]).map_err(|e| anyhow::anyhow!("{}", e))?;
let mut directives = Vec::new();
while let Some(row) = rows.next().map_err(|e| anyhow::anyhow!("{}", e))? {
directives.push(Self::map_row(row)?);
}
Ok(directives)
}
fn count_impl(&self, conn: &PooledConnection<SqliteConnectionManager>) -> Result<i64> {
let count: i64 = conn
.query_row("SELECT COUNT(*) FROM directives", [], |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())
}
}