1 files changed, 51 insertions, 33 deletions
diff --git a/packages/backend/src/server/api/endpoints/stats.ts b/packages/backend/src/server/api/endpoints/stats.ts
index cc94f8bf26..17af75578b 100644
--- a/packages/backend/src/server/api/endpoints/stats.ts
+++ b/packages/backend/src/server/api/endpoints/stats.ts
@@ -1,7 +1,8 @@
-import { Instances, NoteReactions, Notes, Users } from '@/models/index.js';
-import define from '../define.js';
-import { } from '@/services/chart/index.js';
+import { Inject, Injectable } from '@nestjs/common';
 import { IsNull } from 'typeorm';
+import { InstancesRepository, NotesRepository, UsersRepository } from '@/models/index.js';
+import { Endpoint } from '@/server/api/endpoint-base.js';
+import { DI } from '@/di-symbols.js';
 
 export const meta = {
 	requireCredential: false,
@@ -51,34 +52,51 @@ export const paramDef = {
 } as const;
 
 // eslint-disable-next-line import/no-default-export
-export default define(meta, paramDef, async () => {
-	const [
-		notesCount,
-		originalNotesCount,
-		usersCount,
-		originalUsersCount,
-		reactionsCount,
-		//originalReactionsCount,
-		instances,
-	] = await Promise.all([
-		Notes.count({ cache: 3600000 }), // 1 hour
-		Notes.count({ where: { userHost: IsNull() }, cache: 3600000 }),
-		Users.count({ cache: 3600000 }),
-		Users.count({ where: { host: IsNull() }, cache: 3600000 }),
-		NoteReactions.count({ cache: 3600000 }), // 1 hour
-		//NoteReactions.count({ where: { userHost: IsNull() }, cache: 3600000 }),
-		Instances.count({ cache: 3600000 }),
-	]);
+@Injectable()
+export default class extends Endpoint<typeof meta, typeof paramDef> {
+	constructor(
+		@Inject(DI.usersRepository)
+		private usersRepository: UsersRepository,
 
-	return {
-		notesCount,
-		originalNotesCount,
-		usersCount,
-		originalUsersCount,
-		reactionsCount,
-		//originalReactionsCount,
-		instances,
-		driveUsageLocal: 0,
-		driveUsageRemote: 0,
-	};
-});
+		@Inject(DI.notesRepository)
+		private notesRepository: NotesRepository,
+
+		@Inject(DI.instancesRepository)
+		private instancesRepository: InstancesRepository,
+
+		@Inject(DI.noteReactionsRepository)
+		private noteReactionsRepository: NoteReactionsRepository,
+	) {
+		super(meta, paramDef, async () => {
+			const [
+				notesCount,
+				originalNotesCount,
+				usersCount,
+				originalUsersCount,
+				reactionsCount,
+				//originalReactionsCount,
+				instances,
+			] = await Promise.all([
+				this.notesRepository.count({ cache: 3600000 }), // 1 hour
+				this.notesRepository.count({ where: { userHost: IsNull() }, cache: 3600000 }),
+				this.usersRepository.count({ cache: 3600000 }),
+				this.usersRepository.count({ where: { host: IsNull() }, cache: 3600000 }),
+				this.noteReactionsRepository.count({ cache: 3600000 }), // 1 hour
+				//this.noteReactionsRepository.count({ where: { userHost: IsNull() }, cache: 3600000 }),
+				this.instancesRepository.count({ cache: 3600000 }),
+			]);
+
+			return {
+				notesCount,
+				originalNotesCount,
+				usersCount,
+				originalUsersCount,
+				reactionsCount,
+				//originalReactionsCount,
+				instances,
+				driveUsageLocal: 0,
+				driveUsageRemote: 0,
+			};
+		});
+	}
+}