PicoBot/src/storage/ports.rs

use super::{
    MemoryRecord, MemoryUpsert, SchedulerJobRecord, SchedulerJobState, SchedulerJobStatus,
    SchedulerJobUpsert, SkillEventRecord, StorageError,
};

pub trait MemoryRepository: Send + Sync + 'static {
    fn put_memory(&self, input: &MemoryUpsert) -> Result<MemoryRecord, StorageError>;

    fn update_memory(&self, input: &MemoryUpsert) -> Result<Option<MemoryRecord>, StorageError>;

    fn delete_memory(
        &self,
        scope_kind: &str,
        scope_key: &str,
        namespace: &str,
        memory_key: &str,
    ) -> Result<bool, StorageError>;

    fn get_memory(
        &self,
        scope_kind: &str,
        scope_key: &str,
        namespace: &str,
        memory_key: &str,
    ) -> Result<Option<MemoryRecord>, StorageError>;

    fn list_memories(
        &self,
        scope_kind: &str,
        scope_key: &str,
        namespace: Option<&str>,
        limit: usize,
    ) -> Result<Vec<MemoryRecord>, StorageError>;

    fn search_memories_any(
        &self,
        scope_kind: &str,
        scope_key: &str,
        queries: &[String],
        namespace: Option<&str>,
        limit: usize,
    ) -> Result<Vec<MemoryRecord>, StorageError>;
}

pub trait SchedulerJobRepository: Send + Sync + 'static {
    fn upsert_scheduler_job(
        &self,
        input: &SchedulerJobUpsert,
    ) -> Result<SchedulerJobRecord, StorageError>;

    fn get_scheduler_job(&self, job_id: &str) -> Result<Option<SchedulerJobRecord>, StorageError>;

    fn list_scheduler_jobs(
        &self,
        enabled_only: bool,
    ) -> Result<Vec<SchedulerJobRecord>, StorageError>;

    fn delete_scheduler_job(&self, job_id: &str) -> Result<(), StorageError>;

    fn update_scheduler_job_runtime(
        &self,
        job_id: &str,
        state: SchedulerJobState,
        last_status: Option<SchedulerJobStatus>,
        last_error: Option<&str>,
        run_count: i64,
        last_fired_at: Option<i64>,
        next_fire_at: Option<i64>,
        paused_at: Option<i64>,
        completed_at: Option<i64>,
    ) -> Result<(), StorageError>;
}

pub trait SkillEventRepository: Send + Sync + 'static {
    fn append_skill_event(
        &self,
        session_id: Option<&str>,
        event_type: &str,
        skill_name: Option<&str>,
        payload: &serde_json::Value,
    ) -> Result<(), StorageError>;

    fn list_skill_events(
        &self,
        session_id: Option<&str>,
    ) -> Result<Vec<SkillEventRecord>, StorageError>;
}

impl MemoryRepository for super::SessionStore {
    fn put_memory(&self, input: &MemoryUpsert) -> Result<MemoryRecord, StorageError> {
        super::SessionStore::put_memory(self, input)
    }

    fn update_memory(&self, input: &MemoryUpsert) -> Result<Option<MemoryRecord>, StorageError> {
        super::SessionStore::update_memory(self, input)
    }

    fn delete_memory(
        &self,
        scope_kind: &str,
        scope_key: &str,
        namespace: &str,
        memory_key: &str,
    ) -> Result<bool, StorageError> {
        super::SessionStore::delete_memory(self, scope_kind, scope_key, namespace, memory_key)
    }

    fn get_memory(
        &self,
        scope_kind: &str,
        scope_key: &str,
        namespace: &str,
        memory_key: &str,
    ) -> Result<Option<MemoryRecord>, StorageError> {
        super::SessionStore::get_memory(self, scope_kind, scope_key, namespace, memory_key)
    }

    fn list_memories(
        &self,
        scope_kind: &str,
        scope_key: &str,
        namespace: Option<&str>,
        limit: usize,
    ) -> Result<Vec<MemoryRecord>, StorageError> {
        super::SessionStore::list_memories(self, scope_kind, scope_key, namespace, limit)
    }

    fn search_memories_any(
        &self,
        scope_kind: &str,
        scope_key: &str,
        queries: &[String],
        namespace: Option<&str>,
        limit: usize,
    ) -> Result<Vec<MemoryRecord>, StorageError> {
        super::SessionStore::search_memories_any(
            self, scope_kind, scope_key, queries, namespace, limit,
        )
    }
}

impl SchedulerJobRepository for super::SessionStore {
    fn upsert_scheduler_job(
        &self,
        input: &SchedulerJobUpsert,
    ) -> Result<SchedulerJobRecord, StorageError> {
        super::SessionStore::upsert_scheduler_job(self, input)
    }

    fn get_scheduler_job(&self, job_id: &str) -> Result<Option<SchedulerJobRecord>, StorageError> {
        super::SessionStore::get_scheduler_job(self, job_id)
    }

    fn list_scheduler_jobs(
        &self,
        enabled_only: bool,
    ) -> Result<Vec<SchedulerJobRecord>, StorageError> {
        super::SessionStore::list_scheduler_jobs(self, enabled_only)
    }

    fn delete_scheduler_job(&self, job_id: &str) -> Result<(), StorageError> {
        super::SessionStore::delete_scheduler_job(self, job_id)
    }

    fn update_scheduler_job_runtime(
        &self,
        job_id: &str,
        state: SchedulerJobState,
        last_status: Option<SchedulerJobStatus>,
        last_error: Option<&str>,
        run_count: i64,
        last_fired_at: Option<i64>,
        next_fire_at: Option<i64>,
        paused_at: Option<i64>,
        completed_at: Option<i64>,
    ) -> Result<(), StorageError> {
        super::SessionStore::update_scheduler_job_runtime(
            self,
            job_id,
            state,
            last_status,
            last_error,
            run_count,
            last_fired_at,
            next_fire_at,
            paused_at,
            completed_at,
        )
    }
}

impl SkillEventRepository for super::SessionStore {
    fn append_skill_event(
        &self,
        session_id: Option<&str>,
        event_type: &str,
        skill_name: Option<&str>,
        payload: &serde_json::Value,
    ) -> Result<(), StorageError> {
        super::SessionStore::append_skill_event(self, session_id, event_type, skill_name, payload)
    }

    fn list_skill_events(
        &self,
        session_id: Option<&str>,
    ) -> Result<Vec<SkillEventRecord>, StorageError> {
        super::SessionStore::list_skill_events(self, session_id)
    }
}