1 files changed, 85 insertions, 0 deletions
diff --git a/code/api/src/Data/Database/MainAppDatabase.cs b/code/api/src/Data/Database/MainAppDatabase.cs
new file mode 100644
index 0000000..b529791
--- /dev/null
+++ b/code/api/src/Data/Database/MainAppDatabase.cs
@@ -0,0 +1,85 @@
+using IOL.GreatOffice.Api.Endpoints.V1.Projects;
+using Microsoft.AspNetCore.DataProtection.EntityFrameworkCore;
+
+namespace IOL.GreatOffice.Api.Data.Database;
+
+public class MainAppDatabase : DbContext, IDataProtectionKeyContext
+{
+    public MainAppDatabase(DbContextOptions<MainAppDatabase> 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<ApiAccessToken> AccessTokens { get; set; }
+    public DbSet<Tenant> Tenants { get; set; }
+    public DbSet<DataProtectionKey> DataProtectionKeys { get; set; }
+    public DbSet<Project> Projects { get; set; }
+    public DbSet<ProjectLabel> ProjectLabels { get; set; }
+    public DbSet<Customer> Customers { get; set; }
+    public DbSet<CustomerContact> CustomersContacts { get; set; }
+    public DbSet<CustomerEvent> CustomerEvents { get; set; }
+    public DbSet<CustomerGroup> CustomerGroups { get; set; }
+
+    protected override void OnModelCreating(ModelBuilder modelBuilder) {
+        modelBuilder.Entity<User>(e => {
+            e.HasMany(n => n.Tenants);
+            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<ApiAccessToken>(e => {
+            e.HasOne(n => n.User);
+            e.ToTable("api_access_tokens");
+        });
+        modelBuilder.Entity<Tenant>(e => {
+            e.HasMany(n => n.Users);
+            e.ToTable("tenants");
+        });
+        modelBuilder.Entity<Project>(e => {
+            e.HasMany(n => n.Members);
+            e.HasMany(n => n.Customers);
+            e.ToTable("projects");
+        });
+        modelBuilder.Entity<ProjectMember>(e => {
+            e.HasOne(n => n.Project);
+            e.HasOne(n => n.User);
+            e.ToTable("project_members");
+        });
+        modelBuilder.Entity<ProjectLabel>(e => {
+            e.HasOne(n => n.Project);
+            e.ToTable("project_labels");
+        });
+        modelBuilder.Entity<Customer>(e => {
+            e.HasOne(n => n.Owner);
+            e.HasMany(n => n.Events);
+            e.HasMany(n => n.Contacts);
+            e.HasMany(n => n.Groups);
+            e.HasMany(n => n.Projects);
+            e.ToTable("customers");
+        });
+        modelBuilder.Entity<CustomerContact>(e => {
+            e.HasOne(n => n.Customer);
+            e.ToTable("customer_contacts");
+        });
+        modelBuilder.Entity<CustomerEvent>(e => {
+            e.HasOne(n => n.Customer);
+            e.ToTable("customer_events");
+        });
+        modelBuilder.Entity<CustomerGroup>(e => {
+            e.HasMany(n => n.Customers);
+            e.ToTable("customer_groups");
+        });
+
+        base.OnModelCreating(modelBuilder);
+    }
+}
+\ No newline at end of file