babelforce-manager-sdk 0.43.0

use crate::error::{map_manager_err, ManagerError};
use crate::gen::manager::apis::configuration::Configuration;
use crate::gen::manager::apis::file_api;
use crate::gen::manager::models;
use crate::http::{collect_all, fetch_page, Page};
use crate::retry::{with_retry, RetryPolicy};
use crate::token::SharedCfg;

/// Stored files — `/api/v2/files` (recordings, prompts and backups).
pub struct FilesResource {
    pub(crate) cfg: SharedCfg<Configuration>,
    pub(crate) retry: RetryPolicy,
}

impl FilesResource {
    /// List all files (auto-paginated).
    pub async fn list_all(&self) -> Result<Vec<models::StoredFile>, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        collect_all(&self.retry, map_manager_err, |page| async move {
            let r = file_api::list_files(cfg, Some(page), None, None, None, None, None, None, None)
                .await?;
            Ok((r.items, r.pagination.pages, r.pagination.current))
        })
        .await
    }

    /// List one page of files, optionally narrowed to one storage type.
    pub async fn list_page(
        &self,
        page: i32,
        per_page: Option<i32>,
        file_type: Option<models::StorageType>,
    ) -> Result<Page<models::StoredFile>, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        fetch_page(&self.retry, map_manager_err, || async move {
            let r = file_api::list_files(
                cfg,
                Some(page),
                per_page,
                None,
                None,
                file_type,
                None,
                None,
                None,
            )
            .await?;
            Ok((
                r.items,
                r.pagination.pages,
                r.pagination.current,
                Some(r.pagination.total),
            ))
        })
        .await
    }

    /// List files of a given storage type (single, non-paginated response).
    pub async fn list_by_type(
        &self,
        file_type: models::StorageType,
    ) -> Result<Vec<models::StoredFile>, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        let r = with_retry(&self.retry, true, || {
            file_api::list_files_by_type(cfg, file_type)
        })
        .await
        .map_err(map_manager_err)?;
        Ok(r.items)
    }

    /// List all backup files.
    pub async fn backups(&self) -> Result<Vec<models::StoredFile>, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        let r = with_retry(&self.retry, true, || file_api::list_backup_files(cfg))
            .await
            .map_err(map_manager_err)?;
        Ok(r.items)
    }

    /// List all recording files.
    pub async fn recordings(&self) -> Result<Vec<models::StoredFile>, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        let r = with_retry(&self.retry, true, || file_api::list_recording_files(cfg))
            .await
            .map_err(map_manager_err)?;
        Ok(r.items)
    }

    /// List all prompt files.
    pub async fn prompts(&self) -> Result<Vec<models::StoredFile>, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        let r = with_retry(&self.retry, true, || file_api::list_prompt_files(cfg))
            .await
            .map_err(map_manager_err)?;
        Ok(r.items)
    }

    /// Get a file by id.
    pub async fn get(&self, id: &str) -> Result<models::StoredFileItemResponse, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        with_retry(&self.retry, true, || file_api::get_file(cfg, id))
            .await
            .map_err(map_manager_err)
    }

    /// Delete a file by id.
    pub async fn delete(&self, id: &str) -> Result<models::DefaultV2MessageResponse, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        with_retry(&self.retry, false, || file_api::delete_file(cfg, id))
            .await
            .map_err(map_manager_err)
    }

    /// Trigger a download for a file by id.
    ///
    /// The underlying endpoint redirects to a download link; the generated client discards the
    /// response body, so this returns `()` once the server-side download has been triggered.
    /// Use [`bulk_download`](Self::bulk_download) / [`bulk_download_post`](Self::bulk_download_post)
    /// when you need the actual bytes.
    pub async fn download(&self, id: &str) -> Result<(), ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        with_retry(&self.retry, true, || file_api::download_file(cfg, id))
            .await
            .map_err(map_manager_err)?;
        Ok(())
    }

    /// Bulk-delete files by id. Each `id` must be a valid UUID.
    pub async fn bulk_delete(
        &self,
        ids: Vec<String>,
    ) -> Result<models::DefaultV2MessageResponse, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        let ids = ids
            .iter()
            .map(|id| {
                uuid::Uuid::parse_str(id)
                    .map_err(|e| ManagerError::InvalidArgument(format!("id {id:?}: {e}")))
            })
            .collect::<Result<Vec<_>, _>>()?;
        let body = models::FileBulkDeleteRequest { ids };
        with_retry(&self.retry, false, || {
            file_api::bulk_delete_files(cfg, body.clone())
        })
        .await
        .map_err(map_manager_err)
    }

    /// Download multiple files as a ZIP archive (GET), returning the raw bytes.
    pub async fn bulk_download(&self, ids: Vec<String>) -> Result<Vec<u8>, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        let ids_joined = ids.join(",");
        let resp = file_api::get_bulk_file_download(cfg, &ids_joined)
            .await
            .map_err(map_manager_err)?;
        let bytes = resp
            .bytes()
            .await
            .map_err(|e| ManagerError::Network(e.to_string()))?
            .to_vec();
        Ok(bytes)
    }

    /// Download multiple files as a ZIP archive (POST), returning the raw bytes.
    /// Each `id` must be a valid UUID.
    pub async fn bulk_download_post(&self, ids: Vec<String>) -> Result<Vec<u8>, ManagerError> {
        let cfg = self.cfg.get().await?;
        let cfg = cfg.as_ref();
        let ids = ids
            .iter()
            .map(|id| {
                uuid::Uuid::parse_str(id)
                    .map_err(|e| ManagerError::InvalidArgument(format!("id {id:?}: {e}")))
            })
            .collect::<Result<Vec<_>, _>>()?;
        let body = models::FileBulkDownloadRequest { ids };
        let resp = file_api::post_bulk_file_download(cfg, body)
            .await
            .map_err(map_manager_err)?;
        let bytes = resp
            .bytes()
            .await
            .map_err(|e| ManagerError::Network(e.to_string()))?
            .to_vec();
        Ok(bytes)
    }
}