なんかもうめっちゃ変えた

1 files changed, 23 insertions, 7 deletions

diff --git a/packages/backend/src/server/api/endpoints/reset-db.ts b/packages/backend/src/server/api/endpoints/reset-db.ts
index 140f96d579..526efbc2f6 100644
--- a/packages/backend/src/server/api/endpoints/reset-db.ts
+++ b/packages/backend/src/server/api/endpoints/reset-db.ts
@@ -1,5 +1,9 @@
-import { resetDb } from '@/db/postgre.js';
-import define from '../define.js';
+import { Inject, Injectable } from '@nestjs/common';
+import { DataSource } from 'typeorm';
+import Redis from 'ioredis';
+import { Endpoint } from '@/server/api/endpoint-base.js';
+import { DI } from '@/di-symbols.js';
+import { resetDb } from '@/misc/reset-db.js';
 import { ApiError } from '../error.js';
 
 export const meta = {
@@ -21,10 +25,22 @@ export const paramDef = {
 } as const;
 
 // eslint-disable-next-line import/no-default-export
-export default define(meta, paramDef, async (ps, user) => {
-	if (process.env.NODE_ENV !== 'test') throw 'NODE_ENV is not a test';
+@Injectable()
+export default class extends Endpoint<typeof meta, typeof paramDef> {
+	constructor(
+		@Inject(DI.db)
+		private db: DataSource,
 
-	await resetDb();
+		@Inject(DI.redis)
+		private redisClient: Redis.Redis,
+	) {
+		super(meta, paramDef, async (ps, me) => {
+			if (process.env.NODE_ENV !== 'test') throw 'NODE_ENV is not a test';
 
-	await new Promise(resolve => setTimeout(resolve, 1000));
-});
+			await redisClient.flushdb();
+			await resetDb(this.db);
+
+			await new Promise(resolve => setTimeout(resolve, 1000));
+		});
+	}
+}