fix: fix url encode/decode bug (#628)

Signed-off-by: Lzzzt <[email protected]>
dragonflyoss · Jul 26, 2024 · b32b685 · b32b685
1 parent 97791be
commit b32b685
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/dragonfly-client-backend/Cargo.toml b/dragonfly-client-backend/Cargo.toml
@@ -22,6 +22,7 @@ tonic.workspace = true
 url.workspace = true
 tracing.workspace = true
 opendal.workspace = true
+percent-encoding.workspace = true
 futures = "0.3.28"
 libloading = "0.8.4"
 

diff --git a/dragonfly-client-backend/src/object_storage.rs b/dragonfly-client-backend/src/object_storage.rs
@@ -18,6 +18,7 @@ use dragonfly_api::common;
 use dragonfly_client_core::error::BackendError;
 use dragonfly_client_core::{Error as ClientError, Result as ClientResult};
 use opendal::{raw::HttpClient, Metakey, Operator};
+use percent_encoding::percent_decode_str;
 use std::fmt;
 use std::result::Result;
 use std::str::FromStr;
@@ -136,14 +137,15 @@ impl TryFrom<Url> for ParsedURL {
         let key = url
             .path()
             .strip_prefix('/')
-            .ok_or_else(|| ClientError::InvalidURI(url.to_string()))?
-            .to_string();
+            .ok_or_else(|| ClientError::InvalidURI(url.to_string()))?;
+        // Decode the key.
+        let decoded_key = percent_decode_str(key).decode_utf8_lossy().to_string();
 
         Ok(Self {
             url,
             scheme,
             bucket,
-            key,
+            key: decoded_key,
         })
     }
 }

diff --git a/dragonfly-client/src/bin/dfget/main.rs b/dragonfly-client/src/bin/dfget/main.rs
@@ -762,7 +762,7 @@ async fn download(
         }
     }
 
-    progress_bar.finish_with_message(format!("{} downloaded", download_path));
+    progress_bar.finish();
     Ok(())
 }
 
@@ -813,7 +813,7 @@ fn make_output_by_entry(url: Url, output: &Path, entry: DirEntry) -> Result<Path
     let entry_url: Url = entry.url.parse().or_err(ErrorType::ParseError)?;
     let decoded_entry_url = percent_decode_str(entry_url.path()).decode_utf8_lossy();
     Ok(decoded_entry_url
-        .replace(root_dir.as_str(), output_root_dir.as_str())
+        .replacen(root_dir.as_str(), output_root_dir.as_str(), 1)
         .into())
 }