squidex/src/Squidex.Infrastructure.MongoDb/Assets/MongoGridFsAssetStore.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using System.IO;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using MongoDB.Bson;
using MongoDB.Driver;
using MongoDB.Driver.GridFS;

namespace Squidex.Infrastructure.Assets
{
    public sealed class MongoGridFsAssetStore : IAssetStore, IInitializable
    {
        private const int BufferSize = 81920;
        private readonly IGridFSBucket<string> bucket;

        public MongoGridFsAssetStore(IGridFSBucket<string> bucket)
        {
            Guard.NotNull(bucket, nameof(bucket));

            this.bucket = bucket;
        }

        public async Task InitializeAsync(CancellationToken ct = default)
        {
            try
            {
                await bucket.Database.ListCollectionsAsync(cancellationToken: ct);
            }
            catch (MongoException ex)
            {
                throw new ConfigurationException($"Cannot connect to Mongo GridFS bucket '${bucket.Options.BucketName}'.", ex);
            }
        }

        public string GeneratePublicUrl(string fileName)
        {
            return null;
        }

        public async Task CopyAsync(string sourceFileName, string targetFileName, CancellationToken ct = default)
        {
            try
            {
                var sourceName = GetFileName(sourceFileName, nameof(sourceFileName));

                using (var readStream = await bucket.OpenDownloadStreamAsync(sourceFileName, cancellationToken: ct))
                {
                    await UploadAsync(targetFileName, readStream, false, ct);
                }
            }
            catch (GridFSFileNotFoundException ex)
            {
                throw new AssetNotFoundException(sourceFileName, ex);
            }
        }

        public async Task DownloadAsync(string fileName, Stream stream, CancellationToken ct = default)
        {
            try
            {
                var name = GetFileName(fileName, nameof(fileName));

                using (var readStream = await bucket.OpenDownloadStreamAsync(name, cancellationToken: ct))
                {
                    await readStream.CopyToAsync(stream, BufferSize, ct);
                }
            }
            catch (GridFSFileNotFoundException ex)
            {
                throw new AssetNotFoundException(fileName, ex);
            }
        }

        public async Task UploadAsync(string fileName, Stream stream, bool overwrite = false, CancellationToken ct = default)
        {
            try
            {
                var name = GetFileName(fileName, nameof(fileName));

                if (overwrite)
                {
                    await DeleteAsync(fileName);
                }

                await bucket.UploadFromStreamAsync(fileName, fileName, stream, cancellationToken: ct);
            }
            catch (MongoWriteException ex) when (ex.WriteError.Category == ServerErrorCategory.DuplicateKey)
            {
                throw new AssetAlreadyExistsException(fileName);
            }
            catch (MongoBulkWriteException<BsonDocument> ex) when (ex.WriteErrors.Any(x => x.Category == ServerErrorCategory.DuplicateKey))
            {
                throw new AssetAlreadyExistsException(fileName);
            }
        }

        public async Task DeleteAsync(string fileName)
        {
            try
            {
                var name = GetFileName(fileName, nameof(fileName));

                await bucket.DeleteAsync(name);
            }
            catch (GridFSFileNotFoundException)
            {
                return;
            }
        }

        private static string GetFileName(string fileName, string parameterName)
        {
            Guard.NotNullOrEmpty(fileName, parameterName);

            return fileName;
        }
    }
}