Initial commit: immich-sdk v1.137.0
This commit is contained in:
257
src/apis/assets.rs
Normal file
257
src/apis/assets.rs
Normal file
@@ -0,0 +1,257 @@
|
||||
//! Assets API - Manage photos and videos
|
||||
|
||||
use std::path::Path;
|
||||
|
||||
use crate::{
|
||||
Client,
|
||||
error::{ImmichError, Result},
|
||||
models::{AssetId, AssetResponse, AssetUploadResponse, DeleteAssetsRequest},
|
||||
};
|
||||
|
||||
/// API for managing assets (photos and videos)
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct AssetsApi {
|
||||
client: Client,
|
||||
}
|
||||
|
||||
impl AssetsApi {
|
||||
/// Create a new assets API instance
|
||||
pub const fn new(client: Client) -> Self {
|
||||
Self { client }
|
||||
}
|
||||
|
||||
/// List assets with optional filters
|
||||
pub fn list(&self) -> ListAssetsBuilder {
|
||||
ListAssetsBuilder::new(self.client.clone())
|
||||
}
|
||||
|
||||
/// Get a single asset by ID
|
||||
pub fn get(&self, id: AssetId) -> GetAssetBuilder {
|
||||
GetAssetBuilder::new(self.client.clone(), id)
|
||||
}
|
||||
|
||||
/// Upload a new asset
|
||||
pub fn upload(&self) -> UploadAssetBuilder {
|
||||
UploadAssetBuilder::new(self.client.clone())
|
||||
}
|
||||
|
||||
/// Delete assets
|
||||
pub fn delete(&self) -> DeleteAssetsBuilder {
|
||||
DeleteAssetsBuilder::new(self.client.clone())
|
||||
}
|
||||
}
|
||||
|
||||
/// Builder for listing assets
|
||||
#[derive(Debug)]
|
||||
pub struct ListAssetsBuilder {
|
||||
client: Client,
|
||||
album_id: Option<AssetId>,
|
||||
is_favorite: Option<bool>,
|
||||
is_trashed: Option<bool>,
|
||||
}
|
||||
|
||||
impl ListAssetsBuilder {
|
||||
/// Create a new list builder
|
||||
const fn new(client: Client) -> Self {
|
||||
Self {
|
||||
client,
|
||||
album_id: None,
|
||||
is_favorite: None,
|
||||
is_trashed: None,
|
||||
}
|
||||
}
|
||||
|
||||
/// Filter by album ID
|
||||
pub fn with_album_id(mut self, album_id: AssetId) -> Self {
|
||||
self.album_id = Some(album_id);
|
||||
self
|
||||
}
|
||||
|
||||
/// Filter by favorite status
|
||||
pub fn is_favorite(mut self, favorite: bool) -> Self {
|
||||
self.is_favorite = Some(favorite);
|
||||
self
|
||||
}
|
||||
|
||||
/// Filter by trash status
|
||||
pub fn is_trashed(mut self, trashed: bool) -> Self {
|
||||
self.is_trashed = Some(trashed);
|
||||
self
|
||||
}
|
||||
|
||||
/// Execute the request
|
||||
pub async fn execute(self) -> Result<Vec<AssetResponse>> {
|
||||
let mut req = self.client.get("/assets");
|
||||
|
||||
if let Some(album_id) = self.album_id {
|
||||
req = req.query(&[("albumId", album_id.to_string())]);
|
||||
}
|
||||
|
||||
if let Some(is_favorite) = self.is_favorite {
|
||||
req = req.query(&[("isFavorite", is_favorite.to_string())]);
|
||||
}
|
||||
|
||||
if let Some(is_trashed) = self.is_trashed {
|
||||
req = req.query(&[("isTrashed", is_trashed.to_string())]);
|
||||
}
|
||||
|
||||
let response = self.client.execute(req.build()?).await?;
|
||||
let assets: Vec<AssetResponse> = response.json().await?;
|
||||
|
||||
Ok(assets)
|
||||
}
|
||||
}
|
||||
|
||||
/// Builder for getting a single asset
|
||||
#[derive(Debug)]
|
||||
pub struct GetAssetBuilder {
|
||||
client: Client,
|
||||
id: AssetId,
|
||||
}
|
||||
|
||||
impl GetAssetBuilder {
|
||||
/// Create a new get builder
|
||||
const fn new(client: Client, id: AssetId) -> Self {
|
||||
Self { client, id }
|
||||
}
|
||||
|
||||
/// Execute the request
|
||||
pub async fn execute(self) -> Result<AssetResponse> {
|
||||
let path = format!("/assets/{}", self.id);
|
||||
let req = self.client.get(&path);
|
||||
let response = self.client.execute(req.build()?).await?;
|
||||
let asset: AssetResponse = response.json().await?;
|
||||
|
||||
Ok(asset)
|
||||
}
|
||||
}
|
||||
|
||||
/// Builder for uploading an asset
|
||||
#[derive(Debug)]
|
||||
pub struct UploadAssetBuilder {
|
||||
client: Client,
|
||||
file_path: Option<String>,
|
||||
device_asset_id: Option<String>,
|
||||
device_id: Option<String>,
|
||||
is_favorite: bool,
|
||||
}
|
||||
|
||||
impl UploadAssetBuilder {
|
||||
/// Create a new upload builder
|
||||
const fn new(client: Client) -> Self {
|
||||
Self {
|
||||
client,
|
||||
file_path: None,
|
||||
device_asset_id: None,
|
||||
device_id: None,
|
||||
is_favorite: false,
|
||||
}
|
||||
}
|
||||
|
||||
/// Set the file path to upload
|
||||
pub fn file(mut self, path: impl AsRef<Path>) -> Self {
|
||||
self.file_path = Some(path.as_ref().to_string_lossy().to_string());
|
||||
self
|
||||
}
|
||||
|
||||
/// Set the device asset ID (required)
|
||||
pub fn device_asset_id(mut self, id: impl Into<String>) -> Self {
|
||||
self.device_asset_id = Some(id.into());
|
||||
self
|
||||
}
|
||||
|
||||
/// Set the device ID (required)
|
||||
pub fn device_id(mut self, id: impl Into<String>) -> Self {
|
||||
self.device_id = Some(id.into());
|
||||
self
|
||||
}
|
||||
|
||||
/// Mark as favorite
|
||||
pub fn favorite(mut self) -> Self {
|
||||
self.is_favorite = true;
|
||||
self
|
||||
}
|
||||
|
||||
/// Execute the upload
|
||||
pub async fn execute(self) -> Result<AssetUploadResponse> {
|
||||
let file_path = self
|
||||
.file_path
|
||||
.ok_or_else(|| ImmichError::Validation("File path is required".to_string()))?;
|
||||
|
||||
// Read file
|
||||
let file_content = tokio::fs::read(&file_path).await?;
|
||||
let file_name = std::path::Path::new(&file_path)
|
||||
.file_name()
|
||||
.ok_or_else(|| ImmichError::Validation("Invalid file path".to_string()))?
|
||||
.to_string_lossy();
|
||||
|
||||
// Build multipart form
|
||||
let mut form = reqwest::multipart::Form::new().part(
|
||||
"assetData",
|
||||
reqwest::multipart::Part::bytes(file_content).file_name(file_name.to_string()),
|
||||
);
|
||||
|
||||
if let Some(device_asset_id) = self.device_asset_id {
|
||||
form = form.text("deviceAssetId", device_asset_id);
|
||||
}
|
||||
|
||||
if let Some(device_id) = self.device_id {
|
||||
form = form.text("deviceId", device_id);
|
||||
}
|
||||
|
||||
form = form.text("isFavorite", self.is_favorite.to_string());
|
||||
|
||||
let req = self.client.post("/assets").multipart(form);
|
||||
let response = self.client.execute(req.build()?).await?;
|
||||
let result: AssetUploadResponse = response.json().await?;
|
||||
|
||||
Ok(result)
|
||||
}
|
||||
}
|
||||
|
||||
/// Builder for deleting assets
|
||||
#[derive(Debug)]
|
||||
pub struct DeleteAssetsBuilder {
|
||||
client: Client,
|
||||
ids: Vec<AssetId>,
|
||||
}
|
||||
|
||||
impl DeleteAssetsBuilder {
|
||||
/// Create a new delete builder
|
||||
const fn new(client: Client) -> Self {
|
||||
Self {
|
||||
client,
|
||||
ids: Vec::new(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Add an asset ID to delete
|
||||
pub fn id(mut self, id: AssetId) -> Self {
|
||||
self.ids.push(id);
|
||||
self
|
||||
}
|
||||
|
||||
/// Add multiple asset IDs to delete
|
||||
pub fn ids(mut self, ids: impl IntoIterator<Item = AssetId>) -> Self {
|
||||
self.ids.extend(ids);
|
||||
self
|
||||
}
|
||||
|
||||
/// Execute the deletion
|
||||
pub async fn execute(self) -> Result<()> {
|
||||
if self.ids.is_empty() {
|
||||
return Err(ImmichError::Validation(
|
||||
"At least one asset ID is required".to_string(),
|
||||
));
|
||||
}
|
||||
|
||||
let body = DeleteAssetsRequest {
|
||||
ids: self.ids,
|
||||
force: false,
|
||||
};
|
||||
let req = self.client.delete("/assets").json(&body);
|
||||
let _response = self.client.execute(req.build()?).await?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user