Export information about related users

src/entities/AccountExport.ts

@@ -13,6 +13,7 @@ import { Movie } from "./Movie";
 import { ConversationMessage } from "./ConversationMessage";
 import { Conversation } from "./Conversation";
 import { Friend } from "./Friend";
+import { CommentsHelper } from "../helpers/CommentsHelper";
 
 export interface AccountExportBuilder {
 	userID: number;
@@ -47,4 +48,37 @@ export class AccountExport implements AccountExportBuilder {
 				this[key] = info[key];
 		}
 	}
+
+	/**
+	 * Get the ID of all the related users
+	 */
+	public async getRelatedUsersID() : Promise<Set<number>> {
+		const set = new Set<number>();
+
+		// Own user
+		set.add(this.userID);
+
+		// Friends
+		this.friendsList.forEach(f => set.add(f.friendID))
+
+		// Posts
+		for(const p of this.postsList) {
+			set.add(p.userID);
+			if(p.isUserPage) set.add(p.userPageID);
+
+			// Process post comments
+			(await CommentsHelper.Get(p.id)).forEach(c => set.add(c.userID))
+		}
+
+		// Comments
+		this.comments.forEach(c => set.add(c.userID))
+
+		// Conversation members
+		this.conversations.forEach(c => c.members.forEach(id => set.add(id)))
+
+		// Conversations messages
+		this.conversationsMessages.forEach(c => c.forEach(m => set.add(m.userID)))
+
+		return set;
+	}
 }