From f0cea15a4f85b47bf13b6db392f4ff97449d53c2 Mon Sep 17 00:00:00 2001
From: ivarlovlie <git@ivarlovlie.no>
Date: Tue, 1 Nov 2022 00:16:38 +0700
Subject: feat: More models and schema changes

---
 code/api/src/Data/Database/MainAppDatabase.cs | 31 +++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

(limited to 'code/api/src/Data/Database/MainAppDatabase.cs')

diff --git a/code/api/src/Data/Database/MainAppDatabase.cs b/code/api/src/Data/Database/MainAppDatabase.cs
index b529791..d6409e6 100644
--- a/code/api/src/Data/Database/MainAppDatabase.cs
+++ b/code/api/src/Data/Database/MainAppDatabase.cs
@@ -20,6 +20,11 @@ public class MainAppDatabase : DbContext, IDataProtectionKeyContext
     public DbSet<CustomerContact> CustomersContacts { get; set; }
     public DbSet<CustomerEvent> CustomerEvents { get; set; }
     public DbSet<CustomerGroup> CustomerGroups { get; set; }
+    public DbSet<TodoLabel> TodoLabels { get; set; }
+    public DbSet<TodoCollectionAccessControl> TodoProjectAccessControls { get; set; }
+    public DbSet<TodoCollection> TodoProjects { get; set; }
+    public DbSet<TodoComment> TodoComments { get; set; }
+    public DbSet<Todo> Todos { get; set; }
 
     protected override void OnModelCreating(ModelBuilder modelBuilder) {
         modelBuilder.Entity<User>(e => {
@@ -79,6 +84,32 @@ public class MainAppDatabase : DbContext, IDataProtectionKeyContext
             e.HasMany(n => n.Customers);
             e.ToTable("customer_groups");
         });
+        modelBuilder.Entity<Todo>(e => {
+            e.HasOne(n => n.Collection);
+            e.HasOne(n => n.AssignedTo);
+            e.HasOne(n => n.ClosedBy);
+            e.HasMany(n => n.Labels);
+            e.HasMany(n => n.Comments);
+            e.ToTable("todos");
+        });
+        modelBuilder.Entity<TodoCollection>(e => {
+            e.HasOne(n => n.Project);
+            e.HasMany(n => n.AccessControls);
+            e.ToTable("todo_collections");
+        });
+        modelBuilder.Entity<TodoComment>(e => {
+            e.HasOne(n => n.Todo);
+            e.ToTable("todo_comments");
+        });
+        modelBuilder.Entity<TodoLabel>(e => {
+            e.HasOne(n => n.Todo);
+            e.ToTable("todo_labels");
+        });
+        modelBuilder.Entity<TodoCollectionAccessControl>(e => {
+            e.HasOne(n => n.User);
+            e.HasOne(n => n.Collection);
+            e.ToTable("todo_collection_access_controls");
+        });
 
         base.OnModelCreating(modelBuilder);
     }
-- 
cgit v1.3