1 files changed, 36 insertions, 0 deletions
diff --git a/code/api/Database/AppDatabase.cs b/code/api/Database/AppDatabase.cs
new file mode 100644
index 0000000..722a3e0
--- /dev/null
+++ b/code/api/Database/AppDatabase.cs
@@ -0,0 +1,36 @@
+using File = I2R.Storage.Api.Database.Models.File;
+
+namespace I2R.Storage.Api.Database;
+
+public class AppDatabase : DbContext
+{
+    public AppDatabase(DbContextOptions<AppDatabase> options) : base(options) { }
+    public DbSet<User> Users { get; set; }
+    public DbSet<File> Files { get; set; }
+    public DbSet<Folder> Folders { get; set; }
+    public DbSet<Permission> Permissions { get; set; }
+    public DbSet<PermissionGroup> PermissionGroups { get; set; }
+
+    protected override void OnModelCreating(ModelBuilder modelBuilder) {
+        modelBuilder.Entity<User>(e => { e.ToTable("users"); });
+        modelBuilder.Entity<File>(e => {
+            e.HasMany(c => c.Permissions);
+            e.HasOne(c => c.Folder);
+            e.ToTable("files");
+        });
+        modelBuilder.Entity<Folder>(e => {
+            e.HasMany(c => c.Files);
+            e.HasMany(c => c.Permissions);
+            e.ToTable("folders");
+        });
+        modelBuilder.Entity<PermissionGroup>(e => {
+            e.HasMany(c => c.Users);
+            e.ToTable("permission_groups");
+        });
+        modelBuilder.Entity<Permission>(e => {
+            e.HasOne(c => c.Group);
+            e.ToTable("permissions");
+        });
+        base.OnModelCreating(modelBuilder);
+    }
+}
+\ No newline at end of file