1use async_trait::async_trait;
2use sqlx::{Error, PgPool};
3use uuid::Uuid;
4
5use crate::model::{Organization, OrganizationMember, OrganizationRole};
6
7#[async_trait]
8pub trait OrganizationRepository: Send + Sync + Clone + 'static {
9    async fn create(&self, org_name: &str, owner_id: Uuid) -> Result<Organization, Error>;
10
11    async fn get(&self, org_name: &str) -> Result<Option<Organization>, Error>;
12
13    async fn is_member(&self, org_id: Uuid, user_id: Uuid) -> Result<bool, Error>;
14
15    async fn add_member(
16        &self,
17        org_name: &str,
18        user_name: &str,
19        role: OrganizationRole,
20    ) -> Result<Option<OrganizationMember>, Error>;
21
22    async fn get_member_role(
23        &self,
24        org_name: &str,
25        user_id: Uuid,
26    ) -> Result<Option<OrganizationRole>, Error>;
27}
28
29#[derive(Debug, Clone)]
30pub struct OrganizationRepositoryImpl {
31    pool: PgPool,
32}
33
34impl OrganizationRepositoryImpl {
35    pub fn new(pool: PgPool) -> OrganizationRepositoryImpl {
36        OrganizationRepositoryImpl { pool }
37    }
38}
39
40#[async_trait]
41impl OrganizationRepository for OrganizationRepositoryImpl {
42    async fn create(&self, org_name: &str, owner_id: Uuid) -> Result<Organization, Error> {
43        let mut tx = self.pool.begin().await?;
44
45        let org = sqlx::query_as::<_, Organization>(
46            "INSERT INTO organizations (name) VALUES ($1) RETURNING id, name, created_at",
47        )
48        .bind(org_name)
49        .fetch_one(&mut *tx)
50        .await?;
51
52        sqlx::query(
53            "INSERT INTO organization_members (user_id, organization_id, role) VALUES ($1, $2, 'admin')",
54        )
55        .bind(owner_id)
56        .bind(org.id)
57        .execute(&mut *tx)
58        .await?;
59
60        tx.commit().await?;
61
62        Ok(org)
63    }
64
65    async fn get(&self, org_name: &str) -> Result<Option<Organization>, Error> {
66        let org = sqlx::query_as::<_, Organization>(
67            "SELECT id, name, created_at FROM organizations WHERE name = $1",
68        )
69        .bind(org_name)
70        .fetch_optional(&self.pool)
71        .await?;
72
73        Ok(org)
74    }
75
76    async fn is_member(&self, org_id: Uuid, user_id: Uuid) -> Result<bool, Error> {
77        let result = sqlx::query_scalar::<_, bool>(
78            r#"
79            SELECT EXISTS(
80                SELECT 1 FROM organization_members
81                WHERE organization_id = $1 AND user_id = $2
82            )
83            "#,
84        )
85        .bind(org_id)
86        .bind(user_id)
87        .fetch_one(&self.pool)
88        .await?;
89
90        Ok(result)
91    }
92
93    async fn add_member(
94        &self,
95        org_name: &str,
96        user_name: &str,
97        role: OrganizationRole,
98    ) -> Result<Option<OrganizationMember>, Error> {
99        let member = sqlx::query_as::<_, OrganizationMember>(
100            r#"
101            INSERT INTO organization_members (user_id, organization_id, role)
102            SELECT u.id, o.id, $3
103            FROM users u, organizations o
104            WHERE u.name = $1 AND o.name = $2
105            ON CONFLICT (user_id, organization_id) DO NOTHING
106            RETURNING id, user_id, organization_id, role, created_at
107            "#,
108        )
109        .bind(user_name)
110        .bind(org_name)
111        .bind(role)
112        .fetch_optional(&self.pool)
113        .await?;
114
115        Ok(member)
116    }
117
118    async fn get_member_role(
119        &self,
120        org_name: &str,
121        user_id: Uuid,
122    ) -> Result<Option<OrganizationRole>, Error> {
123        let role = sqlx::query_scalar::<_, OrganizationRole>(
124            r#"
125            SELECT om.role
126            FROM organization_members om
127            JOIN organizations o ON om.organization_id = o.id
128            WHERE o.name = $1 AND om.user_id = $2
129            "#,
130        )
131        .bind(org_name)
132        .bind(user_id)
133        .fetch_optional(&self.pool)
134        .await?;
135
136        Ok(role)
137    }
138}
139