refactor: api key repository (#15491)

server/src/repositories/api-key.repository.ts

@@ -1,50 +1,36 @@
 import { Injectable } from '@nestjs/common';
-import { InjectRepository } from '@nestjs/typeorm';
 import { Insertable, Kysely, Updateable } from 'kysely';
 import { InjectKysely } from 'nestjs-kysely';
 import { ApiKeys, DB } from 'src/db';
 import { DummyValue, GenerateSql } from 'src/decorators';
-import { APIKeyEntity } from 'src/entities/api-key.entity';
-import { IKeyRepository } from 'src/interfaces/api-key.interface';
-import { AuthApiKey } from 'src/types';
 import { asUuid } from 'src/utils/database';
-import { Repository } from 'typeorm';
 
 const columns = ['id', 'name', 'userId', 'createdAt', 'updatedAt', 'permissions'] as const;
 
 @Injectable()
-export class ApiKeyRepository implements IKeyRepository {
-  constructor(
-    @InjectRepository(APIKeyEntity) private repository: Repository<APIKeyEntity>,
-    @InjectKysely() private db: Kysely<DB>,
-  ) {}
+export class ApiKeyRepository {
+  constructor(@InjectKysely() private db: Kysely<DB>) {}
 
-  async create(dto: Insertable<ApiKeys>): Promise<APIKeyEntity> {
-    const { id, name, createdAt, updatedAt, permissions } = await this.db
-      .insertInto('api_keys')
-      .values(dto)
-      .returningAll()
-      .executeTakeFirstOrThrow();
-
-    return { id, name, createdAt, updatedAt, permissions } as APIKeyEntity;
+  create(dto: Insertable<ApiKeys>) {
+    return this.db.insertInto('api_keys').values(dto).returningAll().executeTakeFirstOrThrow();
   }
 
-  async update(userId: string, id: string, dto: Updateable<ApiKeys>): Promise<APIKeyEntity> {
+  async update(userId: string, id: string, dto: Updateable<ApiKeys>) {
     return this.db
       .updateTable('api_keys')
       .set(dto)
       .where('api_keys.userId', '=', userId)
       .where('id', '=', asUuid(id))
       .returningAll()
-      .executeTakeFirstOrThrow() as unknown as Promise<APIKeyEntity>;
+      .executeTakeFirstOrThrow();
   }
 
-  async delete(userId: string, id: string): Promise<void> {
+  async delete(userId: string, id: string) {
     await this.db.deleteFrom('api_keys').where('userId', '=', userId).where('id', '=', asUuid(id)).execute();
   }
 
   @GenerateSql({ params: [DummyValue.STRING] })
-  getKey(hashedToken: string): Promise<AuthApiKey | undefined> {
+  getKey(hashedToken: string) {
     return this.db
       .selectFrom('api_keys')
       .innerJoinLateral(
@@ -72,26 +58,26 @@ export class ApiKeyRepository implements IKeyRepository {
         eb.fn.toJson('user').as('user'),
       ])
       .where('api_keys.key', '=', hashedToken)
-      .executeTakeFirst() as Promise<AuthApiKey | undefined>;
+      .executeTakeFirst();
   }
 
   @GenerateSql({ params: [DummyValue.UUID, DummyValue.UUID] })
-  getById(userId: string, id: string): Promise<APIKeyEntity | null> {
+  getById(userId: string, id: string) {
     return this.db
       .selectFrom('api_keys')
       .select(columns)
       .where('id', '=', asUuid(id))
       .where('userId', '=', userId)
-      .executeTakeFirst() as unknown as Promise<APIKeyEntity | null>;
+      .executeTakeFirst();
   }
 
   @GenerateSql({ params: [DummyValue.UUID] })
-  getByUserId(userId: string): Promise<APIKeyEntity[]> {
+  getByUserId(userId: string) {
     return this.db
       .selectFrom('api_keys')
       .select(columns)
       .where('userId', '=', userId)
       .orderBy('createdAt', 'desc')
-      .execute() as unknown as Promise<APIKeyEntity[]>;
+      .execute();
   }
 }