1 files changed, 34 insertions, 0 deletions
diff --git a/src/server/Models/Database/MainDbContext.cs b/src/server/Models/Database/MainDbContext.cs
new file mode 100644
index 0000000..88b1585
--- /dev/null
+++ b/src/server/Models/Database/MainDbContext.cs
@@ -0,0 +1,34 @@
+using System.Collections.Generic;
+using Microsoft.EntityFrameworkCore;
+
+namespace Dough.Models.Database
+{
+    public class MainDbContext : DbContext
+    {
+
+        public MainDbContext(DbContextOptions<MainDbContext> options) : base(options) {
+
+        }
+
+        public DbSet<Transaction> Transactions { get; set; }
+        public DbSet<User> Users { get; set; }
+        public DbSet<Category> Categories { get; set; }
+        public DbSet<Payee> Payees { get; set; }
+
+        protected override void OnModelCreating(ModelBuilder modelBuilder)
+        {
+            modelBuilder.Entity<Transaction>().ToTable("Transactions");
+            var initUser = new User("ivar");
+            initUser.SetBaseProperties();
+            initUser.HashAndSetPassword("ivar123");
+            modelBuilder.Entity<User>().HasData(new List<User>
+            {
+                initUser
+            });
+            modelBuilder.Entity<User>().ToTable("Users");
+            modelBuilder.Entity<Category>().ToTable("Categories");
+            modelBuilder.Entity<Payee>().ToTable("Payees");
+            base.OnModelCreating(modelBuilder);
+        }
+    }
+}