OutFleet/src/database/repository/user.rs

use anyhow::Result;
use sea_orm::{DatabaseConnection, EntityTrait, QueryFilter, ColumnTrait, QueryOrder, PaginatorTrait};
use uuid::Uuid;

use crate::database::entities::user::{Entity as User, Column, Model, ActiveModel, CreateUserDto, UpdateUserDto};

pub struct UserRepository {
    db: DatabaseConnection,
}

impl UserRepository {
    pub fn new(db: DatabaseConnection) -> Self {
        Self { db }
    }

    /// Get all users with pagination
    pub async fn get_all(&self, page: u64, per_page: u64) -> Result<Vec<Model>> {
        let users = User::find()
            .order_by_desc(Column::CreatedAt)
            .paginate(&self.db, per_page)
            .fetch_page(page.saturating_sub(1))
            .await?;

        Ok(users)
    }

    /// Get user by ID
    pub async fn get_by_id(&self, id: Uuid) -> Result<Option<Model>> {
        let user = User::find_by_id(id).one(&self.db).await?;
        Ok(user)
    }

    /// Find user by ID (alias for get_by_id)
    pub async fn find_by_id(&self, id: Uuid) -> Result<Option<Model>> {
        self.get_by_id(id).await
    }

    /// Get user by telegram ID
    pub async fn get_by_telegram_id(&self, telegram_id: i64) -> Result<Option<Model>> {
        let user = User::find()
            .filter(Column::TelegramId.eq(telegram_id))
            .one(&self.db)
            .await?;
        Ok(user)
    }

    /// Search users by name
    pub async fn search_by_name(&self, query: &str, page: u64, per_page: u64) -> Result<Vec<Model>> {
        let users = User::find()
            .filter(Column::Name.contains(query))
            .order_by_desc(Column::CreatedAt)
            .paginate(&self.db, per_page)
            .fetch_page(page.saturating_sub(1))
            .await?;

        Ok(users)
    }

    /// Create a new user
    pub async fn create(&self, dto: CreateUserDto) -> Result<Model> {
        let active_model: ActiveModel = dto.into();
        let user = User::insert(active_model).exec_with_returning(&self.db).await?;
        Ok(user)
    }

    /// Update user by ID
    pub async fn update(&self, id: Uuid, dto: UpdateUserDto) -> Result<Option<Model>> {
        if let Some(user) = self.get_by_id(id).await? {
            let active_model = user.apply_update(dto);
            User::update(active_model).exec(&self.db).await?;
            // Fetch the updated user
            self.get_by_id(id).await
        } else {
            Ok(None)
        }
    }

    /// Delete user by ID
    pub async fn delete(&self, id: Uuid) -> Result<bool> {
        let result = User::delete_by_id(id).exec(&self.db).await?;
        Ok(result.rows_affected > 0)
    }

    /// Get total count of users
    pub async fn count(&self) -> Result<u64> {
        let count = User::find().count(&self.db).await?;
        Ok(count)
    }

    /// Check if telegram ID is already used
    pub async fn telegram_id_exists(&self, telegram_id: i64) -> Result<bool> {
        let count = User::find()
            .filter(Column::TelegramId.eq(telegram_id))
            .count(&self.db)
            .await?;
        Ok(count > 0)
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::database::DatabaseManager;
    use crate::config::DatabaseConfig;

    async fn setup_test_db() -> Result<UserRepository> {
        let config = DatabaseConfig {
            url: std::env::var("DATABASE_URL").unwrap_or_else(|_| 
                "sqlite::memory:".to_string()
            ),
            max_connections: 5,
            connection_timeout: 30,
            auto_migrate: true,
        };

        let db_manager = DatabaseManager::new(&config).await?;
        Ok(UserRepository::new(db_manager.connection().clone()))
    }

    #[tokio::test]
    async fn test_user_crud() {
        let repo = match setup_test_db().await {
            Ok(repo) => repo,
            Err(_) => return, // Skip test if no database available
        };

        // Create user
        let create_dto = CreateUserDto {
            name: "Test User".to_string(),
            comment: Some("Test comment".to_string()),
            telegram_id: Some(123456789),
        };

        let created_user = repo.create(create_dto).await.unwrap();
        assert_eq!(created_user.name, "Test User");
        assert_eq!(created_user.telegram_id, Some(123456789));

        // Get by ID
        let fetched_user = repo.get_by_id(created_user.id).await.unwrap();
        assert!(fetched_user.is_some());
        assert_eq!(fetched_user.unwrap().name, "Test User");

        // Update user
        let update_dto = UpdateUserDto {
            name: Some("Updated User".to_string()),
            comment: None,
            telegram_id: None,
        };

        let updated_user = repo.update(created_user.id, update_dto).await.unwrap();
        assert!(updated_user.is_some());
        assert_eq!(updated_user.unwrap().name, "Updated User");

        // Delete user
        let deleted = repo.delete(created_user.id).await.unwrap();
        assert!(deleted);

        // Verify deletion
        let deleted_user = repo.get_by_id(created_user.id).await.unwrap();
        assert!(deleted_user.is_none());
    }
}