Gabriel Kaszewski
8b3dfffd3b
Some checks failed
lint / lint (push) Has been cancelled
test / unit (push) Has been cancelled
test / integration (push) Has been cancelled
lint / lint (pull_request) Failing after 9m39s
test / unit (pull_request) Failing after 10m48s
test / integration (pull_request) Failing after 16m54s
51 lines
2.3 KiB
Rust
51 lines
2.3 KiB
Rust
use async_trait::async_trait;
|
|
use chrono::{DateTime, Utc};
|
|
use domain::{
|
|
errors::DomainError, models::remote_actor::RemoteActor, ports::RemoteActorRepository,
|
|
};
|
|
use sqlx::PgPool;
|
|
|
|
pub struct PgRemoteActorRepository {
|
|
pool: PgPool,
|
|
}
|
|
impl PgRemoteActorRepository {
|
|
pub fn new(pool: PgPool) -> Self {
|
|
Self { pool }
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl RemoteActorRepository for PgRemoteActorRepository {
|
|
async fn upsert(&self, a: &RemoteActor) -> Result<(), DomainError> {
|
|
sqlx::query(
|
|
"INSERT INTO remote_actors(url,handle,display_name,inbox_url,shared_inbox_url,public_key,avatar_url,last_fetched_at)
|
|
VALUES($1,$2,$3,$4,$5,$6,$7,$8)
|
|
ON CONFLICT(url) DO UPDATE SET handle=EXCLUDED.handle,display_name=EXCLUDED.display_name,
|
|
inbox_url=EXCLUDED.inbox_url,shared_inbox_url=EXCLUDED.shared_inbox_url,
|
|
public_key=EXCLUDED.public_key,avatar_url=EXCLUDED.avatar_url,last_fetched_at=EXCLUDED.last_fetched_at"
|
|
)
|
|
.bind(&a.url).bind(&a.handle).bind(&a.display_name).bind(&a.inbox_url)
|
|
.bind(&a.shared_inbox_url).bind(&a.public_key).bind(&a.avatar_url).bind(a.last_fetched_at)
|
|
.execute(&self.pool).await.map_err(|e| DomainError::Internal(e.to_string())).map(|_| ())
|
|
}
|
|
|
|
async fn find_by_url(&self, url: &str) -> Result<Option<RemoteActor>, DomainError> {
|
|
#[derive(sqlx::FromRow)]
|
|
struct Row {
|
|
url: String,
|
|
handle: String,
|
|
display_name: Option<String>,
|
|
inbox_url: String,
|
|
shared_inbox_url: Option<String>,
|
|
public_key: String,
|
|
avatar_url: Option<String>,
|
|
last_fetched_at: DateTime<Utc>,
|
|
}
|
|
sqlx::query_as::<_, Row>(
|
|
"SELECT url,handle,display_name,inbox_url,shared_inbox_url,public_key,avatar_url,last_fetched_at FROM remote_actors WHERE url=$1"
|
|
).bind(url).fetch_optional(&self.pool).await
|
|
.map_err(|e| DomainError::Internal(e.to_string()))
|
|
.map(|o| o.map(|r| RemoteActor { url: r.url, handle: r.handle, display_name: r.display_name, inbox_url: r.inbox_url, shared_inbox_url: r.shared_inbox_url, public_key: r.public_key, avatar_url: r.avatar_url, last_fetched_at: r.last_fetched_at, bio: None, banner_url: None, also_known_as: None, outbox_url: None, followers_url: None, following_url: None, attachment: vec![] }))
|
|
}
|
|
}
|