using Microsoft.Data.Sqlite;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging.Abstractions;
using System.Threading.Tasks;
using Timeline.Entities;
using Timeline.Migrations;
using Timeline.Models;
using Timeline.Services;
using Xunit;

namespace Timeline.Tests.Helpers
{
    public class TestDatabase : IAsyncLifetime
    {
        private readonly bool _createUser;

        public TestDatabase(bool createUser = true)
        {
            _createUser = createUser;
            Connection = new SqliteConnection("Data Source=:memory:;");
        }

        public async Task InitializeAsync()
        {
            await Connection.OpenAsync();

            using (var context = CreateContext())
            {
                await context.Database.EnsureCreatedAsync();
                context.JwtToken.Add(new JwtTokenEntity
                {
                    Key = JwtTokenGenerateHelper.GenerateKey()
                });
                await context.SaveChangesAsync();

                if (_createUser)
                {
                    var passwordService = new PasswordService();
                    var userService = new UserService(NullLogger<UserService>.Instance, context, passwordService, new Clock());

                    await userService.CreateUser(new User
                    {
                        Username = "admin",
                        Password = "adminpw",
                        Administrator = true,
                        Nickname = "administrator"
                    });

                    await userService.CreateUser(new User
                    {
                        Username = "user",
                        Password = "userpw",
                        Administrator = false,
                        Nickname = "imuser"
                    });
                }
            }
        }

        public async Task DisposeAsync()
        {
            await Connection.CloseAsync();
            await Connection.DisposeAsync();
        }

        public SqliteConnection Connection { get; }

        public DatabaseContext CreateContext()
        {
            var options = new DbContextOptionsBuilder<DatabaseContext>()
                .UseSqlite(Connection).Options;

            return new DatabaseContext(options);
        }
    }
}