1 files changed, 58 insertions, 0 deletions
diff --git a/server/src/Data/AppDbContext.cs b/server/src/Data/AppDbContext.cs
new file mode 100644
index 0000000..3f949dd
--- /dev/null
+++ b/server/src/Data/AppDbContext.cs
@@ -0,0 +1,58 @@
+namespace IOL.GreatOffice.Api.Data;
+
+public class AppDbContext : DbContext
+{
+	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; }
+
+	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.HasOne(c => c.User);
+			e.ToTable("time_categories");
+		});
+
+		modelBuilder.Entity<TimeLabel>(e => {
+			e.HasOne(c => c.User);
+			e.ToTable("time_labels");
+		});
+
+		modelBuilder.Entity<TimeEntry>(e => {
+			e.HasOne(c => c.User);
+			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);
+	}
+}