From b7e39b59fd0fc7b5610ebff29035bf622079e0d8 Mon Sep 17 00:00:00 2001
From: ivarlovlie <git@ivarlovlie.no>
Date: Wed, 5 Oct 2022 20:45:21 +0800
Subject: refactor: Change file structure

---
 code/api/src/Data/AppDbContext.cs | 51 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 code/api/src/Data/AppDbContext.cs

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

diff --git a/code/api/src/Data/AppDbContext.cs b/code/api/src/Data/AppDbContext.cs
new file mode 100644
index 0000000..c970429
--- /dev/null
+++ b/code/api/src/Data/AppDbContext.cs
@@ -0,0 +1,51 @@
+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<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<ApiAccessToken>(e => {
+			e.ToTable("api_access_tokens");
+		});
+
+		modelBuilder.Entity<Tenant>(e => {
+			e.ToTable("tenants");
+		});
+
+		base.OnModelCreating(modelBuilder);
+	}
+}
-- 
cgit v1.3