Merge branch 'delete-offline-assets' into 'main'

Delete offline assets

See merge request kernald/immich-tools!15

Cargo.lock

@@ -304,6 +304,19 @@ version = "1.0.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "d3fd119d74b830634cea2a0f58bbd0d54540518a14397557951e79340abc28c0"
 
+[[package]]
+name = "console"
+version = "0.15.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0e1f83fc076bd6dd27517eacdf25fef6c4dfe5f1d7448bafaaf3a26f13b5e4eb"
+dependencies = [
+ "encode_unicode",
+ "lazy_static",
+ "libc",
+ "unicode-width",
+ "windows-sys 0.52.0",
+]
+
 [[package]]
 name = "core-foundation"
 version = "0.9.4"
@@ -329,6 +342,19 @@ dependencies = [
  "powerfmt",
 ]
 
+[[package]]
+name = "dialoguer"
+version = "0.11.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "658bce805d770f407bc62102fca7c2c64ceef2fbcb2b8bd19d2765ce093980de"
+dependencies = [
+ "console",
+ "shell-words",
+ "tempfile",
+ "thiserror",
+ "zeroize",
+]
+
 [[package]]
 name = "directories"
 version = "5.0.1"
@@ -356,6 +382,12 @@ version = "1.0.17"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "0d6ef0072f8a535281e4876be788938b528e9a1d43900b82c2569af7da799125"
 
+[[package]]
+name = "encode_unicode"
+version = "0.3.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a357d28ed41a50f9c765dbfe56cbc04a64e53e5fc58ba79fbc34c10ef3df831f"
+
 [[package]]
 name = "encoding_rs"
 version = "0.8.34"
@@ -758,6 +790,7 @@ dependencies = [
  "chrono",
  "clap",
  "color-eyre",
+ "dialoguer",
  "directories",
  "figment",
  "figment_file_provider_adapter",
@@ -1645,6 +1678,12 @@ dependencies = [
  "lazy_static",
 ]
 
+[[package]]
+name = "shell-words"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "24188a676b6ae68c3b2cb3a01be17fbf7240ce009799bb56d5b1409051e78fde"
+
 [[package]]
 name = "shlex"
 version = "1.3.0"

Cargo.toml

@@ -15,6 +15,7 @@ ignored = ["progenitor-client", "regress"]
 chrono = { version = "0.4.38", features = ["serde"] }
 clap = { version = "4.5.20", features = ["derive"] }
 color-eyre = "0.6.3"
+dialoguer = "0.11.0"
 directories = "5.0.1"
 figment = { version = "0.10.19", features = ["env", "toml"] }
 figment_file_provider_adapter = "0.1.1"

README.md

@@ -4,6 +4,8 @@
 
  - List named faces that do not have an associated date of birth
  - Synchronise date of births from a vcard file
+ - List and delete assets, including filtering offline assets only
+ - Show details about the server
 
 Some other features are planned, mainly around using external libraries. Feature ideas and contributions welcome.
 

src/args.rs

@@ -20,6 +20,10 @@ pub(crate) struct Opts {
     #[arg(short, long)]
     pub dry_run: bool,
 
+    /// If enabled, don't ask for confirmation of destructive actions
+    #[arg(short, long)]
+    pub no_confirm: bool,
+
     #[command(subcommand)]
     #[serde(flatten)]
     pub command: Commands,
@@ -53,6 +57,13 @@ pub(crate) enum Commands {
 
 #[derive(Serialize, Subcommand)]
 pub(crate) enum AssetsCommands {
+    /// Delete all assets
+    #[serde(rename = "delete")]
+    Delete {
+        /// Delete only offline assets
+        #[arg(short, long, action)]
+        offline: bool,
+    },
     /// List all assets
     #[serde(rename = "list")]
     List {

src/commands/delete_assets.rs

@@ -0,0 +1,58 @@
+use crate::{
+    utils::assets::{fetch_all_assets, AssetQuery},
+    Client,
+};
+use color_eyre::eyre::Result;
+use dialoguer::{theme::ColorfulTheme, Confirm};
+use log::info;
+use uuid::Uuid;
+
+pub async fn delete_assets(
+    client: &Client,
+    offline: bool,
+    dry_run: bool,
+    no_confirm: bool,
+) -> Result<()> {
+    let query = AssetQuery {
+        is_offline: if offline { Some(true) } else { None },
+        ..Default::default()
+    };
+
+    let assets = fetch_all_assets(query, client).await?;
+
+    for asset in &assets {
+        info!(
+            "Deleting asset {} located at {}",
+            asset.id, asset.original_path
+        );
+    }
+
+    if !dry_run {
+        if !(no_confirm
+            || Confirm::with_theme(&ColorfulTheme::default())
+                .with_prompt("Do you want to continue?")
+                .interact()?)
+        {
+            return Ok(());
+        }
+
+        let asset_ids: Result<Vec<_>, uuid::Error> = assets
+            .iter()
+            .map(|asset| Uuid::parse_str(&asset.id))
+            .collect();
+        client
+            .delete_assets(&crate::types::AssetBulkDeleteDto {
+                force: Some(true),
+                ids: asset_ids?,
+            })
+            .await?;
+    }
+
+    info!(
+        "Deleted {} assets{}",
+        assets.len(),
+        if dry_run { " (dry run)" } else { "" }
+    );
+
+    Ok(())
+}

src/commands/mod.rs

@@ -1,3 +1,4 @@
+pub mod delete_assets;
 pub mod list_assets;
 pub mod missing_date_of_birth;
 pub mod server_features;

src/main.rs

@@ -5,6 +5,7 @@ use args::{AssetsCommands, PeopleCommands, ServerCommands};
 use clap::Parser;
 use color_eyre::eyre::{Result, WrapErr};
 use color_eyre::Section;
+use commands::delete_assets::delete_assets;
 use commands::list_assets::list_assets;
 use commands::missing_date_of_birth::missing_date_of_birth;
 use commands::server_features::server_features;
@@ -54,6 +55,9 @@ async fn main() -> Result<()> {
 
     match &args.command {
         Commands::Assets { assets_command } => match assets_command {
+            AssetsCommands::Delete { offline } => {
+                delete_assets(&client, *offline, args.dry_run, args.no_confirm).await
+            }
             AssetsCommands::List { offline } => list_assets(*offline, &client).await,
         },
         Commands::People { people_command } => match people_command {