cf-mini-chat 0.1.31

use async_trait::async_trait;
use modkit_db::secure::DBRunner;
use modkit_macros::domain_model;
use modkit_security::AccessScope;
use uuid::Uuid;

use crate::domain::error::DomainError;
use crate::infra::db::entity::chat_vector_store::Model as VectorStoreModel;

/// Parameters for inserting a new `chat_vector_stores` row with `vector_store_id = NULL`.
#[domain_model]
pub struct InsertVectorStoreParams {
    pub id: Uuid,
    pub tenant_id: Uuid,
    pub chat_id: Uuid,
    pub provider: String,
}

/// Repository trait for chat vector store persistence operations.
///
/// Supports the insert-first CAS protocol for get-or-create:
/// 1. `insert` (may fail with unique violation on (`tenant_id`, `chat_id`))
/// 2. `cas_set_vector_store_id` (winner sets the provider ID)
/// 3. `find_by_chat` (loser polls until `vector_store_id` is non-NULL)
#[async_trait]
pub trait VectorStoreRepository: Send + Sync {
    async fn insert<C: DBRunner>(
        &self,
        runner: &C,
        scope: &AccessScope,
        params: InsertVectorStoreParams,
    ) -> Result<VectorStoreModel, DomainError>;
    async fn cas_set_vector_store_id<C: DBRunner>(
        &self,
        runner: &C,
        scope: &AccessScope,
        id: Uuid,
        vector_store_id: &str,
    ) -> Result<u64, DomainError>;
    async fn find_by_chat<C: DBRunner>(
        &self,
        runner: &C,
        scope: &AccessScope,
        chat_id: Uuid,
    ) -> Result<Option<VectorStoreModel>, DomainError>;
    /// Best-effort delete a placeholder row by ID.
    async fn delete<C: DBRunner>(
        &self,
        runner: &C,
        scope: &AccessScope,
        id: Uuid,
    ) -> Result<u64, DomainError>;

    // ── System-scoped methods (no AccessScope — background workers) ─────

    /// Find vector store row for a chat (system context, no access scope).
    async fn find_by_chat_system<C: DBRunner>(
        &self,
        runner: &C,
        chat_id: Uuid,
    ) -> Result<Option<VectorStoreModel>, DomainError>;

    /// Hard-delete vector store row (system context, no access scope).
    async fn delete_system<C: DBRunner>(&self, runner: &C, id: Uuid) -> Result<u64, DomainError>;
}