morrisjdev/FileContextCore

// Copyright (c) morrisjdev. All rights reserved.
// Original copyright (c) .NET Foundation. All rights reserved.
// Modified version by morrisjdev
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System;
using System.Collections.Generic;
using System.Globalization;
using System.Text;
using FileContextCore.FileManager;
using FileContextCore.Serializer;
using FileContextCore.Storage;
using FileContextCore.StoreManager;
using JetBrains.Annotations;
using Microsoft.EntityFrameworkCore.Infrastructure;
using Microsoft.Extensions.DependencyInjection;

namespace FileContextCore.Infrastructure.Internal
{

    public class FileContextOptionsExtension : IDbContextOptionsExtension
    {
        private FileContextDatabaseRoot _databaseRoot;
        private DbContextOptionsExtensionInfo _info;
        private FileContextScopedOptions _options;

    
        public FileContextOptionsExtension()
        {
            _options = new FileContextScopedOptions(null, null, null,
                typeof(DefaultStoreManager), typeof(JSONSerializer), typeof(DefaultFileManager));
        }

    
        protected FileContextOptionsExtension([NotNull] FileContextOptionsExtension copyFrom)
        {
            _options = copyFrom._options;
            _databaseRoot = copyFrom._databaseRoot;
        }

    
        public virtual DbContextOptionsExtensionInfo Info
            => _info ??= new ExtensionInfo(this);

    
        protected virtual FileContextOptionsExtension Clone() => new FileContextOptionsExtension(this);

    
        public virtual FileContextScopedOptions Options => _options;

    
        public virtual FileContextOptionsExtension WithCustomOptions(string databaseName, string location, string password, Type storeManagerType, Type serializerType, Type fileManagerType)
        {
            var clone = Clone();
            clone._options = new FileContextScopedOptions(databaseName, location, password, storeManagerType, serializerType, fileManagerType);
            return clone;
        }

    
        public virtual FileContextDatabaseRoot DatabaseRoot => _databaseRoot;

    
        public virtual FileContextOptionsExtension WithDatabaseRoot([NotNull] FileContextDatabaseRoot databaseRoot)
        {
            var clone = Clone();

            clone._databaseRoot = databaseRoot;

            return clone;
        }

    
        public virtual void ApplyServices(IServiceCollection services)
        {
            services.AddEntityFrameworkFileContextDatabase();
        }

    
        public virtual void Validate(IDbContextOptions options)
        {
        }

        private sealed class ExtensionInfo : DbContextOptionsExtensionInfo
        {
            private string _logFragment;

            public ExtensionInfo(IDbContextOptionsExtension extension)
                : base(extension)
            {
            }

            private new FileContextOptionsExtension Extension
                => (FileContextOptionsExtension)base.Extension;

            public override bool IsDatabaseProvider => true;

            public override string LogFragment
            {
                get
                {
                    if (_logFragment == null)
                    {
                        var builder = new StringBuilder();

                        builder.Append("Location=").Append(Extension.Options.Location).Append(' ');
                        builder.Append("DatabaseName=").Append(Extension.Options.DatabaseName).Append(' ');
                        builder.Append("StoreManager=").Append(Extension.Options.StoreManagerType).Append(' ');
                        builder.Append("Serializer=").Append(Extension.Options.SerializerType).Append(' ');
                        builder.Append("FileManager=").Append(Extension.Options.FileManagerType).Append(' ');
                        builder.Append("StoreManager=").Append(Extension.Options.StoreManagerType).Append(' ');
                        builder.Append("Password=").Append("<Password>").Append(' ');

                        _logFragment = builder.ToString();
                    }

                    return _logFragment;
                }
            }

            public override long GetServiceProviderHashCode() => Extension._databaseRoot?.GetHashCode() ?? 0L;

            public override void PopulateDebugInfo(IDictionary<string, string> debugInfo)
                => debugInfo["FileContextDatabase:DatabaseRoot"]
                    = (Extension._databaseRoot?.GetHashCode() ?? 0L).ToString(CultureInfo.InvariantCulture);
        }
    }
}