Files
simapi-net/Helpers/SimApiStorage.cs
T

101 lines
3.0 KiB
C#
Raw Normal View History

using System;
using System.IO;
using Microsoft.AspNetCore.Http;
using Minio;
using SimApi.Configs;
2024-03-23 07:29:43 +08:00
namespace SimApi.Helpers;
public class SimApiStorage
{
2024-03-23 07:29:43 +08:00
private MinioClient Mc { get; }
public MinioClient Client => Mc;
private string ServeUrl { get; }
public string Bucket { get; }
private IHttpContextAccessor HttpContextAccessor { get; }
public SimApiStorage(SimApiOptions apiOptions, IHttpContextAccessor httpContextAccessor)
{
2024-03-23 07:29:43 +08:00
var options = apiOptions.SimApiStorageOptions;
HttpContextAccessor = httpContextAccessor;
var useSsl = false;
string endpoint;
if (options.Endpoint.StartsWith("http://"))
{
2024-03-23 07:29:43 +08:00
endpoint = options.Endpoint.Replace("http://", string.Empty);
}
else if (options.Endpoint.StartsWith("https://"))
{
endpoint = options.Endpoint.Replace("https://", string.Empty);
useSsl = true;
}
else
{
throw new Exception("SimApiStorage: Error Endpoint");
}
2024-03-23 07:29:43 +08:00
ServeUrl = options.ServeUrl;
Bucket = options.Bucket;
var mcb = new MinioClient().WithEndpoint(endpoint)
.WithCredentials(options.AccessKey, options.SecretKey);
if (useSsl)
2021-09-29 02:37:14 +08:00
{
2024-03-23 07:29:43 +08:00
mcb = mcb.WithSSL();
2021-09-29 02:37:14 +08:00
}
2024-03-23 07:29:43 +08:00
Mc = mcb.Build();
2021-09-29 02:37:14 +08:00
2024-03-23 07:29:43 +08:00
var found = Mc.BucketExistsAsync(new BucketExistsArgs().WithBucket(Bucket)).Result;
if (!found)
{
2024-03-23 07:29:43 +08:00
Mc.MakeBucketAsync(new MakeBucketArgs().WithBucket(Bucket)).Wait();
}
}
2024-03-23 07:29:43 +08:00
/// <summary>
/// 获取上传的URL
/// </summary>
/// <param name="path"></param>
/// <param name="expire"></param>
/// <returns></returns>
public string GetUploadUrl(string path, int expire = 7200)
{
return Mc.PresignedPutObjectAsync(new PresignedPutObjectArgs().WithBucket(Bucket)
.WithObject(path).WithExpiry(expire)).Result;
}
/// <summary>
/// 获取下载的URL
/// </summary>
/// <param name="path"></param>
/// <param name="expire"></param>
/// <returns></returns>
public string GetDownloadUrl(string path, int expire = 600)
{
return Mc.PresignedGetObjectAsync(new PresignedGetObjectArgs().WithBucket(Bucket).WithObject(path)
.WithExpiry(expire)).Result;
}
public string UploadFile(string path, Stream stream, string contentType = "image/png")
{
Mc.PutObjectAsync(new PutObjectArgs().WithBucket(Bucket).WithObject(path).WithObjectSize(stream.Length)
.WithStreamData(stream).WithContentType(contentType)).Wait();
return null;
}
public string FullUrl(string path)
{
var httpRequest = HttpContextAccessor.HttpContext?.Request;
var url = $"{httpRequest?.Scheme}://{httpRequest?.Host}";
if (string.IsNullOrEmpty(path))
{
return path;
}
return path.StartsWith('~') ? string.Concat(url, path.AsSpan(1, path.Length - 1)) : $"{ServeUrl}{path}";
}
}