[Storage] ContainerClient (create_container, delete_container) and BlobClient (download_blob, upload_blob) (#2246)

Author: vincenttran-msft

eng/emitter-package-lock.json

@@ -1,7 +1,7 @@
 {
   "main": "dist/src/index.js",
   "dependencies": {
-    "@azure-tools/typespec-rust": "0.9.0"
+    "@azure-tools/typespec-rust": "0.9.1"
   },
   "devDependencies": {
     "@azure-tools/typespec-azure-core": "0.50.0",
@@ -14,4 +14,4 @@
     "@typespec/versioning": "0.64.0",
     "@typespec/xml": "0.64.0"
   }
-}
+}

eng/emitter-package.json

@@ -0,0 +1,6 @@
+{
+  "AssetsRepo": "Azure/azure-sdk-assets",
+  "AssetsRepoPrefixPath": "rust",
+  "Tag": "rust/azure_storage_blob_d76e2bb82c",
+  "TagPrefix": "rust/azure_storage_blob"
+}

sdk/storage/assets.json

@@ -2,10 +2,18 @@
 // Licensed under the MIT License.
 
 use crate::{
-    models::BlobProperties, pipeline::StorageHeadersPolicy, BlobBlobClientGetPropertiesOptions,
-    BlobClientOptions, GeneratedBlobClient,
+    clients::GeneratedBlobClient,
+    models::{
+        BlobBlobClientDownloadOptions, BlobBlobClientGetPropertiesOptions,
+        BlobBlockBlobClientUploadOptions, BlobProperties,
+    },
+    pipeline::StorageHeadersPolicy,
+    BlobClientOptions,
+};
+use azure_core::{
+    credentials::TokenCredential, BearerTokenCredentialPolicy, Bytes, Policy, RequestContent,
+    Response, Result, Url,
 };
-use azure_core::{credentials::TokenCredential, BearerTokenCredentialPolicy, Policy, Result, Url};
 use std::sync::Arc;
 
 pub struct BlobClient {
@@ -53,6 +61,14 @@ impl BlobClient {
         &self.endpoint
     }
 
+    pub fn container_name(&self) -> &str {
+        &self.container_name
+    }
+
+    pub fn blob_name(&self) -> &str {
+        &self.blob_name
+    }
+
     pub async fn get_blob_properties(
         &self,
         options: Option<BlobBlobClientGetPropertiesOptions<'_>>,
@@ -66,4 +82,39 @@ impl BlobClient {
         let blob_properties: BlobProperties = response.headers().get()?;
         Ok(blob_properties)
     }
+
+    pub async fn download_blob(
+        &self,
+        options: Option<BlobBlobClientDownloadOptions<'_>>,
+    ) -> Result<Response> {
+        let response = self
+            .client
+            .get_blob_blob_client(self.container_name.clone(), self.blob_name.clone())
+            .download(options)
+            .await?;
+        Ok(response)
+    }
+
+    // For now, this is single-shot, block blob hot path only.
+    pub async fn upload_blob(
+        &self,
+        data: RequestContent<Bytes>,
+        overwrite: bool,
+        content_length: i64,
+        options: Option<BlobBlockBlobClientUploadOptions<'_>>,
+    ) -> Result<Response<()>> {
+        let mut options = options.unwrap_or_default();
+
+        // Check if they want overwrite, by default overwrite=False
+        if !overwrite {
+            options.if_none_match = Some(String::from("*"));
+        }
+
+        let response = self
+            .client
+            .get_blob_block_blob_client(self.container_name.clone(), self.blob_name.clone())
+            .upload(data, content_length, Some(options))
+            .await?;
+        Ok(response)
+    }
 }

sdk/storage/azure_storage_blob/src/clients/blob_client.rs

@@ -1,2 +1,83 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
+
+use crate::clients::GeneratedBlobClient;
+use crate::models::{BlobContainerClientCreateOptions, BlobContainerClientDeleteOptions};
+use crate::pipeline::StorageHeadersPolicy;
+use crate::BlobClientOptions;
+use azure_core::{
+    credentials::TokenCredential, BearerTokenCredentialPolicy, Policy, Response, Result, Url,
+};
+use std::sync::Arc;
+
+pub struct BlobContainerClient {
+    endpoint: Url,
+    container_name: String,
+    client: GeneratedBlobClient,
+}
+
+impl BlobContainerClient {
+    pub fn new(
+        endpoint: &str,
+        container_name: String,
+        credential: Arc<dyn TokenCredential>,
+        options: Option<BlobClientOptions>,
+    ) -> Result<Self> {
+        let mut options = options.unwrap_or_default();
+
+        let storage_headers_policy = Arc::new(StorageHeadersPolicy);
+        options
+            .client_options
+            .per_call_policies
+            .push(storage_headers_policy);
+
+        let oauth_token_policy = BearerTokenCredentialPolicy::new(
+            credential.clone(),
+            ["https://storage.azure.com/.default"],
+        );
+        options
+            .client_options
+            .per_try_policies
+            .push(Arc::new(oauth_token_policy) as Arc<dyn Policy>);
+
+        let client = GeneratedBlobClient::new(endpoint, credential, Some(options))?;
+
+        Ok(Self {
+            endpoint: endpoint.parse()?,
+            container_name,
+            client,
+        })
+    }
+
+    pub fn endpoint(&self) -> &Url {
+        &self.endpoint
+    }
+
+    pub fn container_name(&self) -> &str {
+        &self.container_name
+    }
+
+    pub async fn create_container(
+        &self,
+        options: Option<BlobContainerClientCreateOptions<'_>>,
+    ) -> Result<Response<()>> {
+        let response = self
+            .client
+            .get_blob_container_client(self.container_name.clone())
+            .create(options)
+            .await?;
+        Ok(response)
+    }
+
+    pub async fn delete_container(
+        &self,
+        options: Option<BlobContainerClientDeleteOptions<'_>>,
+    ) -> Result<Response<()>> {
+        let response = self
+            .client
+            .get_blob_container_client(self.container_name.clone())
+            .delete(options)
+            .await?;
+        Ok(response)
+    }
+}