using Microsoft.AspNetCore.DataProtection.EntityFrameworkCore;

namespace IOL.GreatOffice.Api.Data;

public class AppDbContext : DbContext, IDataProtectionKeyContext
{
	public AppDbContext(DbContextOptions<AppDbContext> options) : base(options) { }
	public DbSet<User> Users { get; set; }
	public DbSet<ForgotPasswordRequest> ForgotPasswordRequests { get; set; }
	public DbSet<TimeLabel> TimeLabels { get; set; }
	public DbSet<TimeEntry> TimeEntries { get; set; }
	public DbSet<TimeCategory> TimeCategories { get; set; }
	public DbSet<GithubUserMapping> GithubUserMappings { get; set; }
	public DbSet<ApiAccessToken> AccessTokens { get; set; }
	public DbSet<Tenant> Tenants { get; set; }
	public DbSet<DataProtectionKey> DataProtectionKeys { get; set; }

	protected override void OnModelCreating(ModelBuilder modelBuilder) {
		modelBuilder.Entity<User>(e => {
			e.ToTable("users");
		});

		modelBuilder.Entity<ForgotPasswordRequest>(e => {
			e.HasOne(c => c.User);
			e.ToTable("forgot_password_requests");
		});

		modelBuilder.Entity<TimeCategory>(e => {
			e.ToTable("time_categories");
		});

		modelBuilder.Entity<TimeLabel>(e => {
			e.ToTable("time_labels");
		});

		modelBuilder.Entity<TimeEntry>(e => {
			e.HasOne(c => c.Category);
			e.HasMany(c => c.Labels);
			e.ToTable("time_entries");
		});

		modelBuilder.Entity<GithubUserMapping>(e => {
			e.HasOne(c => c.User);
			e.HasKey(c => c.GithubId);
			e.ToTable("github_user_mappings");
		});

		modelBuilder.Entity<ApiAccessToken>(e => {
			e.ToTable("api_access_tokens");
		});

		modelBuilder.Entity<Tenant>(e => {
			e.ToTable("tenants");
		});

		base.OnModelCreating(modelBuilder);
	}
}