Friendship.ts

import { modelOptions, prop, getModelForClass } from '@typegoose/typegoose'
import { Field, ObjectType } from 'type-graphql'

import { BaseModel, User } from '.'
import { DateScalar } from '../types/lib'

@ObjectType()
@modelOptions({
	schemaOptions: {
		collection: 'friendships',
		timestamps: false, // turn off timestamps because we're using `acceptedAt`, `declinedAt`, `requestedAt`
	},
})
export class Friendship extends BaseModel() {
	@Field(() => User)
	@prop({ required: true, ref: User })
	addressedTo!: User // I'm not sure about the name of this field

	@Field(() => User)
	@prop({ required: true, ref: User })
	requestedBy!: User

	// Statuses

	@Field(() => DateScalar, { nullable: true })
	@prop()
	acceptedAt?: Date

	@Field(() => DateScalar, { nullable: true })
	@prop()
	blockedAt?: Date

	@Field(() => DateScalar, { nullable: true })
	@prop()
	declinedAt?: Date

	@Field(() => DateScalar)
	@prop({ required: true })
	requestedAt!: Date

	// Virtual properties

	@Field(() => Boolean, { nullable: true })
	get pending(): boolean {
		const { acceptedAt, blockedAt, declinedAt, requestedAt } =
			((this as any)._doc as Friendship) || this
		return (
			Boolean(requestedAt) || !Boolean(acceptedAt || blockedAt || declinedAt)
		)
	}
}

export const FriendshipModel = getModelForClass(Friendship)