summaryrefslogtreecommitdiff
path: root/back-end/src/Storage/MinioClientManager.cs
blob: 0c161af13120c594ff0ad205e93bac2723ccb913 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/*
* Copyright (c) 2024 Vaughn Nugent
* 
* Library: CMNext
* Package: Content.Publishing.Blog.Admin
* File: MinioClientManager.cs 
*
* CMNext is free software: you can redistribute it and/or modify 
* it under the terms of the GNU Affero General Public License as 
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* CMNext is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see https://www.gnu.org/licenses/.
*/

using System.IO;
using System.Threading;
using System.Threading.Tasks;

using Minio;
using Minio.DataModel;
using Minio.DataModel.Args;

using VNLib.Utils.Memory;
using VNLib.Utils.Extensions;
using VNLib.Plugins;
using VNLib.Plugins.Extensions.Loading;

using static Content.Publishing.Blog.Admin.Model.PostManager;

namespace Content.Publishing.Blog.Admin.Storage
{

    [ConfigurationName("storage")]
    internal sealed class MinioClientManager(PluginBase pbase, IConfigScope s3Config) : StorageBase
    {
        private readonly MinioClient Client = new();
        private readonly S3Config Config = s3Config.Deserialze<S3Config>();

        ///<inheritdoc/>
        protected override string? BasePath => Config.BaseBucket;

        ///<inheritdoc/>
        public override async Task ConfigureServiceAsync(PluginBase plugin)
        {
            using ISecretResult? secret = await plugin.GetSecretAsync("storage_secret");

            Client.WithEndpoint(Config.ServerAddress)
                    .WithCredentials(Config.ClientId, secret.Result.ToString())
                    .WithSSL(Config.UseSsl == true);

            //Accept optional region
            if (!string.IsNullOrWhiteSpace(Config.Region))
            {
                Client.WithRegion(Config.Region);
            }

            //10 second timeout
            Client.WithTimeout(10 * 1000);

            //Setup debug trace
            if (plugin.IsDebug())
            {
                Client.SetTraceOn(new ReqLogger(plugin.Log));
            }

            //Build client
            Client.Build();
        }

        ///<inheritdoc/>
        public override Task DeleteFileAsync(string filePath, CancellationToken cancellation)
        {
            RemoveObjectArgs args = new();
            args.WithBucket(Config.BaseBucket)
                .WithObject(filePath);

            //Remove the object
            return Client.RemoveObjectAsync(args, cancellation);
        }

        ///<inheritdoc/>
        public override Task WriteFileAsync(string filePath, Stream data, string ct, CancellationToken cancellation)
        {
            PutObjectArgs args = new();
            args.WithBucket(Config.BaseBucket)
                .WithContentType(ct)
                .WithObject(filePath)
                .WithObjectSize(data.Length)
                .WithStreamData(data);

            //Upload the object
            return Client.PutObjectAsync(args, cancellation);
        }

        ///<inheritdoc/>
        public override async Task<long> ReadFileAsync(string filePath, Stream output, CancellationToken cancellation)
        {
            //Get the item
            GetObjectArgs args = new();
            args.WithBucket(Config.BaseBucket)
            .WithObject(filePath)
            .WithCallbackStream(async (stream, cancellation) =>
            {
                //Read the object to memory
                await stream.CopyToAsync(output, 16384, MemoryUtil.Shared, cancellation);
            });
            try
            {
                //Get the post content file 
                ObjectStat stat = await Client.GetObjectAsync(args, cancellation);
                return stat.Size;
            }
            catch (Minio.Exceptions.ObjectNotFoundException)
            {
                //File not found
                return -1L;
            }
        }
    }
}