Fixed Interface for Comment

2 files changed, 20 insertions, 7 deletions

diff --git a/src/DevHive.Data/DevHiveContext.cs b/src/DevHive.Data/DevHiveContext.cs
index 40b2a42..e391882 100644
--- a/src/DevHive.Data/DevHiveContext.cs
+++ b/src/DevHive.Data/DevHiveContext.cs
@@ -20,9 +20,6 @@ namespace DevHive.Data
 		{
 			/* User */
 			builder.Entity<User>()
-				.HasKey(x => x.Id);
-
-			builder.Entity<User>()
 				.HasIndex(x => x.UserName)
 				.IsUnique();
 
@@ -32,13 +29,11 @@ namespace DevHive.Data
 				.WithMany(x => x.Users);
 
 			/* Friends */
+			//TODO: Look into the User - User
 			builder.Entity<UserFriends>()
 				.HasKey(uu => new { uu.UserId, uu.FriendId });
 
 			/* Languages */
-			builder.Entity<Language>()
-				.HasKey(x => x.Id);
-
 			builder.Entity<User>()
 				.HasMany(x => x.Languages)
 				.WithMany(x => x.Users)
@@ -63,6 +58,24 @@ namespace DevHive.Data
 				.WithMany(x => x.Technologies)
 				.UsingEntity(x => x.ToTable("TechnologyUser"));
 
+			/* Post */
+			builder.Entity<Post>()
+				.HasMany(x => x.Comments)
+				.WithOne(x => x.Post);
+
+			builder.Entity<Post>()
+				.HasOne(x => x.Creator)
+				.WithMany(x => x.Posts);
+
+			/* Comment */
+			builder.Entity<Comment>()
+				.HasOne(x => x.Post)
+				.WithMany(x => x.Comments);
+
+			builder.Entity<Comment>()
+				.HasOne(x => x.Creator)
+				.WithMany(x => x.Comments);
+
 			base.OnModelCreating(builder);
 		}
 	}
diff --git a/src/DevHive.Data/Interfaces/Models/IComment.cs b/src/DevHive.Data/Interfaces/Models/IComment.cs
index e7b5ad5..97c1578 100644
--- a/src/DevHive.Data/Interfaces/Models/IComment.cs
+++ b/src/DevHive.Data/Interfaces/Models/IComment.cs
@@ -5,7 +5,7 @@ namespace DevHive.Data.Interfaces.Models
 {
 	public interface IComment : IModel
 	{
-		Post Parrent { get; set; }
+		Post Post { get; set; }
 
 		User Creator { get; set; }