87 lines
3.1 KiB
TypeScript
87 lines
3.1 KiB
TypeScript
import { Injectable } from '@nestjs/common';
|
|
import { ExpressionBuilder, Insertable, Kysely } from 'kysely';
|
|
import { jsonObjectFrom } from 'kysely/helpers/postgres';
|
|
import { InjectKysely } from 'nestjs-kysely';
|
|
import { columns } from 'src/database';
|
|
import { Activity, DB } from 'src/db';
|
|
import { DummyValue, GenerateSql } from 'src/decorators';
|
|
import { asUuid } from 'src/utils/database';
|
|
|
|
export interface ActivitySearch {
|
|
albumId?: string;
|
|
assetId?: string | null;
|
|
userId?: string;
|
|
isLiked?: boolean;
|
|
}
|
|
|
|
const withUser = (eb: ExpressionBuilder<DB, 'activity'>) => {
|
|
return jsonObjectFrom(
|
|
eb
|
|
.selectFrom('users')
|
|
.select(columns.userDto)
|
|
.whereRef('users.id', '=', 'activity.userId')
|
|
.where('users.deletedAt', 'is', null),
|
|
).as('user');
|
|
};
|
|
|
|
@Injectable()
|
|
export class ActivityRepository {
|
|
constructor(@InjectKysely() private db: Kysely<DB>) {}
|
|
|
|
@GenerateSql({ params: [{ albumId: DummyValue.UUID }] })
|
|
search(options: ActivitySearch) {
|
|
const { userId, assetId, albumId, isLiked } = options;
|
|
|
|
return this.db
|
|
.selectFrom('activity')
|
|
.selectAll('activity')
|
|
.select(withUser)
|
|
.leftJoin('assets', (join) => join.onRef('assets.id', '=', 'activity.assetId').on('assets.deletedAt', 'is', null))
|
|
.$if(!!userId, (qb) => qb.where('activity.userId', '=', userId!))
|
|
.$if(assetId === null, (qb) => qb.where('assetId', 'is', null))
|
|
.$if(!!assetId, (qb) => qb.where('activity.assetId', '=', assetId!))
|
|
.$if(!!albumId, (qb) => qb.where('activity.albumId', '=', albumId!))
|
|
.$if(isLiked !== undefined, (qb) => qb.where('activity.isLiked', '=', isLiked!))
|
|
.orderBy('activity.createdAt', 'asc')
|
|
.execute();
|
|
}
|
|
|
|
async create(activity: Insertable<Activity>) {
|
|
return this.save(activity);
|
|
}
|
|
|
|
async delete(id: string) {
|
|
await this.db.deleteFrom('activity').where('id', '=', asUuid(id)).execute();
|
|
}
|
|
|
|
@GenerateSql({ params: [{ albumId: DummyValue.UUID, assetId: DummyValue.UUID }] })
|
|
async getStatistics({ albumId, assetId }: { albumId: string; assetId?: string }): Promise<number> {
|
|
const { count } = await this.db
|
|
.selectFrom('activity')
|
|
.select((eb) => eb.fn.countAll().as('count'))
|
|
.innerJoin('users', (join) => join.onRef('users.id', '=', 'activity.userId').on('users.deletedAt', 'is', null))
|
|
.leftJoin('assets', 'assets.id', 'activity.assetId')
|
|
.$if(!!assetId, (qb) => qb.where('activity.assetId', '=', assetId!))
|
|
.where('activity.albumId', '=', albumId)
|
|
.where('activity.isLiked', '=', false)
|
|
.where('assets.deletedAt', 'is', null)
|
|
.where('assets.fileCreatedAt', 'is not', null)
|
|
.where('assets.fileModifiedAt', 'is not', null)
|
|
.where('assets.localDateTime', 'is not', null)
|
|
.executeTakeFirstOrThrow();
|
|
|
|
return count as number;
|
|
}
|
|
|
|
private async save(entity: Insertable<Activity>) {
|
|
const { id } = await this.db.insertInto('activity').values(entity).returning('id').executeTakeFirstOrThrow();
|
|
|
|
return this.db
|
|
.selectFrom('activity')
|
|
.selectAll('activity')
|
|
.select(withUser)
|
|
.where('activity.id', '=', asUuid(id))
|
|
.executeTakeFirstOrThrow();
|
|
}
|
|
}
|