summary refs log tree commit diff
path: root/src/util/entities/Relationship.ts
blob: 740095c26ffe1375131ad8d479cb8abdbe598ed1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
/*
	Spacebar: A FOSS re-implementation and extension of the Discord.com backend.
	Copyright (C) 2023 Spacebar and Spacebar Contributors
	
	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU Affero General Public License as published
	by the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.
	
	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU Affero General Public License for more details.
	
	You should have received a copy of the GNU Affero General Public License
	along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

import {
	Column,
	Entity,
	Index,
	JoinColumn,
	ManyToOne,
	RelationId,
} from "typeorm";
import { BaseClass } from "./BaseClass";
import { User } from "./User";

export enum RelationshipType {
	outgoing = 4,
	incoming = 3,
	blocked = 2,
	friends = 1,
}

@Entity("relationships")
@Index(["from_id", "to_id"], { unique: true })
export class Relationship extends BaseClass {
	@Column({})
	@RelationId((relationship: Relationship) => relationship.from)
	from_id: string;

	@JoinColumn({ name: "from_id" })
	@ManyToOne(() => User, {
		onDelete: "CASCADE",
	})
	from: User;

	@Column({})
	@RelationId((relationship: Relationship) => relationship.to)
	to_id: string;

	@JoinColumn({ name: "to_id" })
	@ManyToOne(() => User, {
		onDelete: "CASCADE",
	})
	to: User;

	@Column({ nullable: true })
	nickname?: string;

	@Column({ type: "int" })
	type: RelationshipType;

	toPublicRelationship() {
		return {
			id: this.to?.id || this.to_id,
			type: this.type,
			nickname: this.nickname,
			user: this.to?.toPublicUser(),
		};
	}
}