/*
This file is part of the Notesnook Sync Server project (https://notesnook.com/)

Copyright (C) 2023 Streetwriters (Private) Limited

This program is free software: you can redistribute it and/or modify
it under the terms of the Affero GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program 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
Affero GNU General Public License for more details.

You should have received a copy of the Affero GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

using Microsoft.Extensions.Configuration;
using MongoDB.Driver;
using Streetwriters.Data.Interfaces;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;

namespace Streetwriters.Data.DbContexts
{
    public class MongoDbContext(IMongoClient MongoClient) : IDbContext
    {
        public static IMongoClient CreateMongoDbClient(IDbSettings dbSettings)
        {
            var settings = MongoClientSettings.FromConnectionString(dbSettings.ConnectionString);
            settings.MaxConnectionPoolSize = 500;
            settings.MinConnectionPoolSize = 0;
            return new MongoClient(settings);
        }

        public static IMongoCollection<T> GetMongoCollection<T>(IMongoClient client, string databaseName, string collectionName)
        {
            return client.GetDatabase(databaseName).GetCollection<T>(collectionName, new MongoCollectionSettings()
            {
                AssignIdOnInsert = true,
            });
        }

        private readonly List<Func<IClientSessionHandle, CancellationToken, Task>> _commands = [];

        public async Task<int> SaveChanges()
        {
            try
            {
                var count = _commands.Count;

                using (IClientSessionHandle session = await MongoClient.StartSessionAsync())
                {
#if DEBUG
                    await Parallel.ForEachAsync(_commands, async (c, ct) => await c(session, ct));
#else
                    await session.WithTransactionAsync(async (handle, token) =>
                    {
                        await Task.WhenAll(_commands.Select(c => c(handle, token)));
                        return true;
                    });
#endif

                }
                return count;
            }
            catch (Exception ex)
            {
                // TODO use Slogger here.
                await Console.Error.WriteLineAsync(ex.ToString());
                return 0;
            }
        }

        public void AddCommand(Func<IClientSessionHandle, CancellationToken, Task> func)
        {
            _commands.Add(func);
        }

        public void Dispose()
        {
            GC.SuppressFinalize(this);
        }
    }
}