// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using Microsoft.AspNetCore.DataProtection;
using Microsoft.AspNetCore.DataProtection.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;

namespace EntityFrameworkCoreSample;

class Program
{
    static void Main(string[] args)
    {
        // Configure
        var services = new ServiceCollection()
            .AddLogging(o => o.AddConsole().SetMinimumLevel(LogLevel.Debug))
            .AddDbContext<DataProtectionKeyContext>(o =>
            {
                o.UseInMemoryDatabase("DataProtection_EntityFrameworkCore");
                // Make sure to create a sql server called DataProtectionApp
                //o.UseSqlServer(@"Server=(localdb)\mssqllocaldb;Database=DataProtectionApp;Trusted_Connection=True;Connect Timeout=5;ConnectRetryCount=0");
                o.UseQueryTrackingBehavior(QueryTrackingBehavior.NoTracking);
                o.EnableSensitiveDataLogging();
            })
            .AddDataProtection()
            .PersistKeysToDbContext<DataProtectionKeyContext>()
            .SetDefaultKeyLifetime(TimeSpan.FromDays(7))
            .Services
            .BuildServiceProvider(validateScopes: true);

        using (services)
        {
            // Run a sample payload
            var protector = services.GetDataProtector("sample-purpose");
            var protectedData = protector.Protect("Hello world!");
            Console.WriteLine(protectedData);
        }
    }
}

class DataProtectionKeyContext : DbContext, IDataProtectionKeyContext
{
    public DataProtectionKeyContext(DbContextOptions<DataProtectionKeyContext> options) : base(options) { }

    public DbSet<DataProtectionKey> DataProtectionKeys { get; set; }
}
