aboutsummaryrefslogtreecommitdiff
path: root/back-end/src/Storage/MinioClientManager.cs
blob: c7867b2c47944540f13c5ab2d109d6a1197a0cb2 (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
128
129
130
131
132
133
134
135
/*
* Copyright (c) 2023 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("s3_config")]
    internal sealed class MinioClientManager : StorageBase
    {
        private readonly MinioClient Client;
        private readonly S3Config Config;

        public MinioClientManager(PluginBase pbase, IConfigScope s3Config)
        {
            //Deserialize the config
            Config = s3Config.Deserialze<S3Config>();
            Client = new();
        }

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

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

            Client.WithEndpoint(Config.ServerAddress)
                    .WithCredentials(Config.ClientId, secret.Result.ToString());

            Client.WithSSL(Config.UseSsl.HasValue && Config.UseSsl.Value);

            //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;
            }
        }
    }
}