1 files changed, 43 insertions, 28 deletions
diff --git a/packages/backend/src/server/api/endpoints/pages/like.ts b/packages/backend/src/server/api/endpoints/pages/like.ts
index 269b539f74..f3c55fed8b 100644
--- a/packages/backend/src/server/api/endpoints/pages/like.ts
+++ b/packages/backend/src/server/api/endpoints/pages/like.ts
@@ -1,6 +1,8 @@
-import { Pages, PageLikes } from '@/models/index.js';
-import { genId } from '@/misc/gen-id.js';
-import define from '../../define.js';
+import { Inject, Injectable } from '@nestjs/common';
+import { PagesRepository, PageLikesRepository } from '@/models/index.js';
+import { IdService } from '@/core/IdService.js';
+import { Endpoint } from '@/server/api/endpoint-base.js';
+import { DI } from '@/di-symbols.js';
 import { ApiError } from '../../error.js';
 
 export const meta = {
@@ -40,33 +42,46 @@ export const paramDef = {
 } as const;
 
 // eslint-disable-next-line import/no-default-export
-export default define(meta, paramDef, async (ps, user) => {
-	const page = await Pages.findOneBy({ id: ps.pageId });
-	if (page == null) {
-		throw new ApiError(meta.errors.noSuchPage);
-	}
+@Injectable()
+export default class extends Endpoint<typeof meta, typeof paramDef> {
+	constructor(
+		@Inject(DI.pagesRepository)
+		private pagesRepository: PagesRepository,
 
-	if (page.userId === user.id) {
-		throw new ApiError(meta.errors.yourPage);
-	}
+		@Inject(DI.pageLikesRepository)
+		private pageLikesRepository: PageLikesRepository,
 
-	// if already liked
-	const exist = await PageLikes.findOneBy({
-		pageId: page.id,
-		userId: user.id,
-	});
+		private idService: IdService,
+	) {
+		super(meta, paramDef, async (ps, me) => {
+			const page = await this.pagesRepository.findOneBy({ id: ps.pageId });
+			if (page == null) {
+				throw new ApiError(meta.errors.noSuchPage);
+			}
 
-	if (exist != null) {
-		throw new ApiError(meta.errors.alreadyLiked);
-	}
+			if (page.userId === me.id) {
+				throw new ApiError(meta.errors.yourPage);
+			}
 
-	// Create like
-	await PageLikes.insert({
-		id: genId(),
-		createdAt: new Date(),
-		pageId: page.id,
-		userId: user.id,
-	});
+			// if already liked
+			const exist = await this.pageLikesRepository.findOneBy({
+				pageId: page.id,
+				userId: me.id,
+			});
 
-	Pages.increment({ id: page.id }, 'likedCount', 1);
-});
+			if (exist != null) {
+				throw new ApiError(meta.errors.alreadyLiked);
+			}
+
+			// Create like
+			await this.pageLikesRepository.insert({
+				id: this.idService.genId(),
+				createdAt: new Date(),
+				pageId: page.id,
+				userId: me.id,
+			});
+
+			this.pagesRepository.increment({ id: page.id }, 'likedCount', 1);
+		});
+	}
+}