mirror of https://github.com/abpframework/abp.git
Suhaib
2 years ago
6 changed files with 222 additions and 281 deletions
@ -1,11 +1,17 @@ |
|||
using Volo.Abp.Modularity; |
|||
using System; |
|||
using Microsoft.Extensions.DependencyInjection; |
|||
using Volo.Abp.Caching; |
|||
using Volo.Abp.Modularity; |
|||
|
|||
namespace Volo.Abp.BlobStoring.Bunny; |
|||
|
|||
[DependsOn(typeof(AbpBlobStoringModule))] |
|||
[DependsOn( |
|||
typeof(AbpBlobStoringModule), |
|||
typeof(AbpCachingModule))] |
|||
public class AbpBlobStoringBunnyModule : AbpModule |
|||
{ |
|||
public override void ConfigureServices(ServiceConfigurationContext context) |
|||
{ |
|||
context.Services.AddHttpClient(); |
|||
} |
|||
} |
|||
@ -1,229 +0,0 @@ |
|||
using System; |
|||
using System.Collections.Generic; |
|||
using System.IO; |
|||
using System.Linq; |
|||
using System.Net.Http; |
|||
using System.Net.Http.Headers; |
|||
using System.Text.Json; |
|||
using System.Threading.Tasks; |
|||
using BunnyCDN.Net.Storage; |
|||
using BunnyCDN.Net.Storage.Models; |
|||
|
|||
namespace Volo.Abp.BlobStoring.Bunny; |
|||
|
|||
public class BunnyClient : IDisposable |
|||
{ |
|||
private readonly string _region; |
|||
private readonly string _storageZoneName; |
|||
private readonly HttpClient _httpClient; |
|||
|
|||
private static BunnyCDNStorage? BunnyCDNStorage = null; |
|||
|
|||
public BunnyClient(string accessKey, string storageZoneName, string region = "de", HttpMessageHandler? handler = null) |
|||
{ |
|||
if (string.IsNullOrWhiteSpace(accessKey)) |
|||
{ |
|||
throw new ArgumentException("Account access key must not be null or empty.", nameof(accessKey)); |
|||
} |
|||
_region = region; |
|||
_storageZoneName = storageZoneName; |
|||
_httpClient = handler != null ? new HttpClient(handler) : new HttpClient(); |
|||
_httpClient.Timeout = TimeSpan.FromSeconds(120); |
|||
_httpClient.DefaultRequestHeaders.Add("AccessKey", accessKey); |
|||
} |
|||
|
|||
public async Task<bool> DoesStorageZoneExistAsync(string zoneName) |
|||
{ |
|||
if (string.IsNullOrWhiteSpace(zoneName)) |
|||
{ |
|||
throw new ArgumentException("Zone name must not be null or empty.", nameof(zoneName)); |
|||
} |
|||
|
|||
try |
|||
{ |
|||
var request = new HttpRequestMessage |
|||
{ |
|||
Method = HttpMethod.Get, |
|||
RequestUri = new Uri("https://api.bunny.net/storagezone?page=0&perPage=1000&includeDeleted=false"), |
|||
Headers = { { "accept", "application/json" } } |
|||
}; |
|||
|
|||
using var response = await _httpClient.SendAsync(request); |
|||
response.EnsureSuccessStatusCode(); |
|||
|
|||
var responseBody = await response.Content.ReadAsStringAsync(); |
|||
var storageZones = JsonSerializer.Deserialize<List<StorageZone>>(responseBody); |
|||
|
|||
return storageZones?.Any(zone => zone.Name!.Equals(zoneName, StringComparison.OrdinalIgnoreCase)) == true; |
|||
} |
|||
catch (HttpRequestException ex) |
|||
{ |
|||
throw new BunnyApiException("Error occurred while checking storage zone existence.", ex); |
|||
} |
|||
catch (JsonException ex) |
|||
{ |
|||
throw new BunnyApiException("Failed to parse the response from Bunny API.", ex); |
|||
} |
|||
} |
|||
|
|||
// Ensure the storage zone exists, creating it if not found, and return the zone
|
|||
public async Task<StorageZone> EnsureStorageZoneExistsAsync(string zoneName, string? originUrl = "") |
|||
{ |
|||
try |
|||
{ |
|||
if (!await DoesStorageZoneExistAsync(zoneName)) |
|||
{ |
|||
return await CreateStorageZoneAsync(zoneName, originUrl); |
|||
} |
|||
|
|||
var request = new HttpRequestMessage |
|||
{ |
|||
Method = HttpMethod.Get, |
|||
RequestUri = new Uri("https://api.bunny.net/storagezone?page=0&perPage=1000&includeDeleted=false"), |
|||
Headers = { { "accept", "application/json" } } |
|||
}; |
|||
|
|||
using var response = await _httpClient.SendAsync(request); |
|||
response.EnsureSuccessStatusCode(); |
|||
|
|||
var responseBody = await response.Content.ReadAsStringAsync(); |
|||
var storageZones = JsonSerializer.Deserialize<List<StorageZone>>(responseBody); |
|||
|
|||
return storageZones?.FirstOrDefault(zone => zone.Name!.Equals(zoneName, StringComparison.OrdinalIgnoreCase)) |
|||
?? throw new BunnyApiException($"Storage zone '{zoneName}' not found even though it should exist."); |
|||
} |
|||
catch (BunnyApiException ex) |
|||
{ |
|||
throw new BunnyApiException($"Failed to ensure the existence of storage zone '{zoneName}'.", ex); |
|||
} |
|||
} |
|||
|
|||
public async Task UploadAsync(Stream stream, string path) => await (await GetBunnyCDNStorage()).UploadAsync(stream, path); |
|||
|
|||
public async Task<bool> DeleteObjectAsync(string path) => await (await GetBunnyCDNStorage()).DeleteObjectAsync(path); |
|||
|
|||
public async Task<List<StorageObject>> GetStorageObjectsAsync(string path) => await (await GetBunnyCDNStorage()).GetStorageObjectsAsync(path); |
|||
|
|||
public async Task<Stream> DownloadObjectAsStreamAsync(string path) => await (await GetBunnyCDNStorage()).DownloadObjectAsStreamAsync(path); |
|||
|
|||
public async Task DeleteStorageZoneAsync(string zoneName) |
|||
{ |
|||
if (string.IsNullOrWhiteSpace(zoneName)) |
|||
{ |
|||
throw new ArgumentException("Zone name must not be null or empty.", nameof(zoneName)); |
|||
} |
|||
var storageZone = (await GetStorageZoneAsync()); |
|||
|
|||
var request = new HttpRequestMessage |
|||
{ |
|||
Method = HttpMethod.Delete, |
|||
RequestUri = new Uri($"https://api.bunny.net/storagezone/{storageZone.Id}") |
|||
}; |
|||
using (var response = await _httpClient.SendAsync(request)) |
|||
{ |
|||
response.EnsureSuccessStatusCode(); |
|||
} |
|||
} |
|||
private async Task<StorageZone> GetStorageZoneAsync() |
|||
{ |
|||
var request = new HttpRequestMessage |
|||
{ |
|||
Method = HttpMethod.Get, |
|||
RequestUri = new Uri("https://api.bunny.net/storagezone") |
|||
}; |
|||
|
|||
var response = await _httpClient.SendAsync(request); |
|||
response.EnsureSuccessStatusCode(); |
|||
|
|||
var content = await response.Content.ReadAsStringAsync(); |
|||
var zones = JsonSerializer.Deserialize<List<StorageZone>>(content); |
|||
|
|||
var targetZone = zones?.FirstOrDefault(z => |
|||
z.Name.Equals(_storageZoneName, StringComparison.OrdinalIgnoreCase)); |
|||
|
|||
return targetZone |
|||
?? throw new AbpException( |
|||
$"Storage zone '{_storageZoneName}' is not found"); |
|||
} |
|||
private async Task<StorageZone> CreateStorageZoneAsync(string zoneName, string? originUrl = "", int zoneTier = 0) |
|||
{ |
|||
if (string.IsNullOrWhiteSpace(zoneName)) |
|||
{ |
|||
throw new ArgumentException("Zone name must not be null or empty.", nameof(zoneName)); |
|||
} |
|||
|
|||
try |
|||
{ |
|||
var payload = new Dictionary<string, object> |
|||
{ |
|||
{ "ZoneTier", zoneTier }, |
|||
{ "Name", zoneName }, |
|||
{ "Region", _region } |
|||
}; |
|||
|
|||
// Add OriginUrl only if it is not empty
|
|||
if (!string.IsNullOrEmpty(originUrl)) |
|||
{ |
|||
payload.Add("OriginUrl", originUrl!); |
|||
} |
|||
|
|||
var request = new HttpRequestMessage |
|||
{ |
|||
Method = HttpMethod.Post, |
|||
RequestUri = new Uri("https://api.bunny.net/storagezone"), |
|||
Headers = { { "accept", "application/json" } }, |
|||
Content = new StringContent(JsonSerializer.Serialize(payload)) |
|||
{ |
|||
Headers = { ContentType = new MediaTypeHeaderValue("application/json") } |
|||
} |
|||
}; |
|||
|
|||
using var response = await _httpClient.SendAsync(request); |
|||
if (!response.IsSuccessStatusCode) |
|||
{ |
|||
var responseBody = await response.Content.ReadAsStringAsync(); |
|||
throw new BunnyApiException($"Failed to create storage zone. Response: {responseBody}"); |
|||
} |
|||
|
|||
var createdZoneJson = await response.Content.ReadAsStringAsync(); |
|||
return JsonSerializer.Deserialize<StorageZone>(createdZoneJson) |
|||
?? throw new BunnyApiException("Failed to deserialize the created storage zone response."); |
|||
} |
|||
catch (HttpRequestException ex) |
|||
{ |
|||
throw new BunnyApiException("Error occurred while creating storage zone.", ex); |
|||
} |
|||
catch (JsonException ex) |
|||
{ |
|||
throw new BunnyApiException("Failed to parse the response for the created storage zone.", ex); |
|||
} |
|||
} |
|||
private async Task<BunnyCDNStorage> GetBunnyCDNStorage() |
|||
{ |
|||
if (BunnyCDNStorage != null) { |
|||
return BunnyCDNStorage; |
|||
} |
|||
var storageZonePassword = (await GetStorageZoneAsync()).Password; |
|||
BunnyCDNStorage = new BunnyCDNStorage(_storageZoneName, storageZonePassword, _region); |
|||
|
|||
return BunnyCDNStorage; |
|||
} |
|||
|
|||
public void Dispose() => _httpClient?.Dispose(); |
|||
|
|||
public class StorageZone |
|||
{ |
|||
public int Id { get; set; } |
|||
public string Password { get; set; } = null!; |
|||
public string Name { get; set; } = null!; |
|||
public string? Region { get; set; } = null!; |
|||
public bool Deleted { get; set; } |
|||
} |
|||
} |
|||
|
|||
public class BunnyApiException : Exception |
|||
{ |
|||
public BunnyApiException(string message) : base(message) { } |
|||
|
|||
public BunnyApiException(string message, Exception innerException) : base(message, innerException) { } |
|||
} |
|||
@ -0,0 +1,153 @@ |
|||
using System; |
|||
using System.Collections.Generic; |
|||
using System.Net.Http; |
|||
using System.Text; |
|||
using System.Text.Json; |
|||
using System.Threading.Tasks; |
|||
using BunnyCDN.Net.Storage; |
|||
using Microsoft.Extensions.Caching.Distributed; |
|||
using Volo.Abp.Caching; |
|||
using Volo.Abp.DependencyInjection; |
|||
|
|||
namespace Volo.Abp.BlobStoring.Bunny; |
|||
|
|||
public class BunnyClientFactory : IBunnyClientFactory, ITransientDependency |
|||
{ |
|||
private readonly IDistributedCache<StorageZone> _cache; |
|||
private readonly IHttpClientFactory _httpClientFactory; |
|||
private const string CacheKeyPrefix = "BunnyStorageZone:"; |
|||
private static TimeSpan CacheDuration = TimeSpan.FromHours(12); |
|||
|
|||
public BunnyClientFactory( |
|||
IHttpClientFactory httpClient, |
|||
IDistributedCache<StorageZone> cache) |
|||
{ |
|||
_cache = cache; |
|||
_httpClientFactory = httpClient; |
|||
} |
|||
|
|||
public async Task<BunnyCDNStorage> CreateAsync(string accessKey, string containerName, string region = "de") |
|||
{ |
|||
var cacheKey = $"{CacheKeyPrefix}{containerName}"; |
|||
var storageZoneInfo = await _cache.GetOrAddAsync( |
|||
cacheKey, |
|||
async () => { |
|||
var result = await GetStorageZoneAsync(accessKey, containerName); |
|||
if (result == null) |
|||
{ |
|||
throw new AbpException($"Storage zone '{containerName}' not found"); |
|||
} |
|||
return result; |
|||
}, |
|||
() => new DistributedCacheEntryOptions |
|||
{ |
|||
AbsoluteExpiration = DateTimeOffset.Now.Add(CacheDuration) |
|||
} |
|||
); |
|||
|
|||
if (storageZoneInfo == null) |
|||
{ |
|||
throw new AbpException($"Could not retrieve storage zone information for container '{containerName}'"); |
|||
} |
|||
|
|||
return new BunnyCDNStorage(containerName, storageZoneInfo.Password, region); |
|||
} |
|||
|
|||
public async Task EnsureStorageZoneExistsAsync( |
|||
string accessKey, |
|||
string containerName, |
|||
string region = "de", |
|||
bool createIfNotExists = false) |
|||
{ |
|||
var storageZone = await GetStorageZoneAsync(accessKey, containerName); |
|||
|
|||
if (storageZone == null) |
|||
{ |
|||
if (!createIfNotExists) |
|||
{ |
|||
throw new AbpException( |
|||
$"Storage zone '{containerName}' does not exist. " + |
|||
"Set createIfNotExists to true to create it automatically."); |
|||
} |
|||
|
|||
await CreateStorageZoneAsync(accessKey, containerName, region); |
|||
|
|||
// Clear the cache to force a refresh of the storage zone info
|
|||
var cacheKey = $"{CacheKeyPrefix}{containerName}"; |
|||
await _cache.RemoveAsync(cacheKey); |
|||
} |
|||
} |
|||
|
|||
private async Task<StorageZone> CreateStorageZoneAsync( |
|||
string accessKey, |
|||
string containerName, |
|||
string region) |
|||
{ |
|||
using var _client = _httpClientFactory.CreateClient("BunnyApiClient"); |
|||
_client.DefaultRequestHeaders.Add("AccessKey", accessKey); |
|||
|
|||
var payload = new Dictionary<string, object> |
|||
{ |
|||
{ "Name", containerName }, |
|||
{ "Region", region }, |
|||
{ "ZoneTier", 0 } |
|||
}; |
|||
|
|||
var content = new StringContent( |
|||
JsonSerializer.Serialize(payload), |
|||
Encoding.UTF8, |
|||
"application/json"); |
|||
|
|||
var response = await _client.PostAsync( |
|||
"https://api.bunny.net/storagezone", |
|||
content); |
|||
|
|||
if (!response.IsSuccessStatusCode) |
|||
{ |
|||
var errorContent = await response.Content.ReadAsStringAsync(); |
|||
throw new AbpException( |
|||
$"Failed to create storage zone '{containerName}'. " + |
|||
$"Status: {response.StatusCode}, Error: {errorContent}"); |
|||
} |
|||
|
|||
var responseContent = await response.Content.ReadAsStringAsync(); |
|||
var createdZone = JsonSerializer.Deserialize<StorageZone>(responseContent); |
|||
|
|||
if (createdZone == null) |
|||
{ |
|||
throw new AbpException( |
|||
$"Failed to deserialize the created storage zone response for '{containerName}'"); |
|||
} |
|||
|
|||
return createdZone; |
|||
} |
|||
|
|||
private async Task<StorageZone?> GetStorageZoneAsync(string accessKey, string containerName) |
|||
{ |
|||
using var _client = _httpClientFactory.CreateClient("BunnyApiClient"); |
|||
_client.DefaultRequestHeaders.Add("AccessKey", accessKey); |
|||
var response = await _client.GetAsync("https://api.bunny.net/storagezone"); |
|||
response.EnsureSuccessStatusCode(); |
|||
|
|||
var content = await response.Content.ReadAsStringAsync(); |
|||
var zones = JsonSerializer.Deserialize<StorageZone[]>(content); |
|||
|
|||
return Array.Find(zones!, z => |
|||
z.Name.Equals(containerName, StringComparison.OrdinalIgnoreCase) && !z.Deleted); |
|||
} |
|||
|
|||
[Serializable] |
|||
public class StorageZone |
|||
{ |
|||
public int Id { get; set; } |
|||
public string Password { get; set; } = null!; |
|||
public string Name { get; set; } = null!; |
|||
public string? Region { get; set; } |
|||
public bool Deleted { get; set; } |
|||
} |
|||
} |
|||
public class BunnyApiException : Exception |
|||
{ |
|||
public BunnyApiException(string message) : base(message) { } |
|||
public BunnyApiException(string message, Exception innerException) : base(message, innerException) { } |
|||
} |
|||
@ -0,0 +1,10 @@ |
|||
using System.Threading.Tasks; |
|||
using BunnyCDN.Net.Storage; |
|||
|
|||
namespace Volo.Abp.BlobStoring.Bunny; |
|||
|
|||
public interface IBunnyClientFactory |
|||
{ |
|||
Task<BunnyCDNStorage> CreateAsync(string accessKey, string containerName, string region = "de"); |
|||
Task EnsureStorageZoneExistsAsync(string accessKey, string containerName, string region = "de", bool createIfNotExists = false); |
|||
} |
|||
Loading…
Reference in new issue