1use async_trait::async_trait;
2use sqlx::{Error, PgPool};
3use uuid::Uuid;
4
5use crate::model::{Repository, RepositoryOwnerType, RepositoryVisibility};
6
7#[async_trait]
8pub trait RepositoryRepository: Send + Sync + Clone + 'static {
9    async fn create(
10        &self,
11        name: &str,
12        owner_id: Uuid,
13        owner_name: &str,
14        owner_type: &RepositoryOwnerType,
15        visibility: &RepositoryVisibility,
16    ) -> Result<Repository, Error>;
17
18    async fn get(&self, owner: &str, repo: &str) -> Result<Option<Repository>, Error>;
19
20    async fn get_by_id(&self, id: Uuid) -> Result<Option<Repository>, Error>;
21
22    async fn list_by_owner(&self, owner_name: &str) -> Result<Vec<Repository>, Error>;
23}
24
25#[derive(Debug, Clone)]
26pub struct RepositoryRepositoryImpl {
27    pool: PgPool,
28}
29
30impl RepositoryRepositoryImpl {
31    pub fn new(pool: PgPool) -> RepositoryRepositoryImpl {
32        RepositoryRepositoryImpl { pool }
33    }
34}
35
36#[async_trait]
37impl RepositoryRepository for RepositoryRepositoryImpl {
38    async fn create(
39        &self,
40        name: &str,
41        owner_id: Uuid,
42        owner_name: &str,
43        owner_type: &RepositoryOwnerType,
44        visibility: &RepositoryVisibility,
45    ) -> Result<Repository, Error> {
46        let repository = sqlx::query_as::<_, Repository>(
47            r#"
48            INSERT INTO repositories (name, owner_id, owner_name, owner_type, visibility)
49            VALUES ($1, $2, $3, $4, $5)
50            RETURNING id, name, owner_id, owner_name, owner_type, visibility, created_at
51            "#,
52        )
53        .bind(name)
54        .bind(owner_id)
55        .bind(owner_name)
56        .bind(owner_type)
57        .bind(visibility)
58        .fetch_one(&self.pool)
59        .await?;
60
61        Ok(repository)
62    }
63
64    async fn get(&self, owner: &str, repo: &str) -> Result<Option<Repository>, Error> {
65        let repository = sqlx::query_as::<_, Repository>(
66            r#"
67            SELECT id, name, owner_id, owner_name, owner_type, visibility, created_at
68            FROM repositories
69            WHERE owner_name = $1 AND name = $2
70            "#,
71        )
72        .bind(owner)
73        .bind(repo)
74        .fetch_optional(&self.pool)
75        .await?;
76
77        Ok(repository)
78    }
79
80    async fn get_by_id(&self, id: Uuid) -> Result<Option<Repository>, Error> {
81        let repository = sqlx::query_as::<_, Repository>(
82            r#"
83            SELECT id, name, owner_id, owner_name, owner_type, visibility, created_at
84            FROM repositories
85            WHERE id = $1
86            "#,
87        )
88        .bind(id)
89        .fetch_optional(&self.pool)
90        .await?;
91
92        Ok(repository)
93    }
94
95    async fn list_by_owner(&self, owner_name: &str) -> Result<Vec<Repository>, Error> {
96        let repositories = sqlx::query_as::<_, Repository>(
97            r#"
98            SELECT id, name, owner_id, owner_name, owner_type, visibility, created_at
99            FROM repositories
100            WHERE owner_name = $1
101            ORDER BY created_at DESC
102            "#,
103        )
104        .bind(owner_name)
105        .fetch_all(&self.pool)
106        .await?;
107
108        Ok(repositories)
109    }
110}
111