immich/server/src/repositories/move.repository.ts

import { Injectable } from '@nestjs/common';
import { Insertable, Kysely, sql, Updateable } from 'kysely';
import { InjectKysely } from 'nestjs-kysely';
import { DB, MoveHistory } from 'src/db';
import { DummyValue, GenerateSql } from 'src/decorators';
import { MoveEntity } from 'src/entities/move.entity';
import { AssetPathType, PathType } from 'src/enum';

export type MoveCreate = Pick<MoveEntity, 'oldPath' | 'newPath' | 'entityId' | 'pathType'> & Partial<MoveEntity>;

@Injectable()
export class MoveRepository {
  constructor(@InjectKysely() private db: Kysely<DB>) {}

  create(entity: Insertable<MoveHistory>): Promise<MoveEntity> {
    return this.db
      .insertInto('move_history')
      .values(entity)
      .returningAll()
      .executeTakeFirstOrThrow() as Promise<MoveEntity>;
  }

  @GenerateSql({ params: [DummyValue.UUID, DummyValue.STRING] })
  getByEntity(entityId: string, pathType: PathType): Promise<MoveEntity | undefined> {
    return this.db
      .selectFrom('move_history')
      .selectAll()
      .where('entityId', '=', entityId)
      .where('pathType', '=', pathType)
      .executeTakeFirst() as Promise<MoveEntity | undefined>;
  }

  update(id: string, entity: Updateable<MoveHistory>): Promise<MoveEntity> {
    return this.db
      .updateTable('move_history')
      .set(entity)
      .where('id', '=', id)
      .returningAll()
      .executeTakeFirstOrThrow() as unknown as Promise<MoveEntity>;
  }

  @GenerateSql({ params: [DummyValue.UUID] })
  delete(id: string): Promise<MoveEntity> {
    return this.db
      .deleteFrom('move_history')
      .where('id', '=', id)
      .returningAll()
      .executeTakeFirstOrThrow() as unknown as Promise<MoveEntity>;
  }

  @GenerateSql()
  async cleanMoveHistory(): Promise<void> {
    await this.db
      .deleteFrom('move_history')
      .where((eb) =>
        eb(
          'move_history.entityId',
          'not in',
          eb.selectFrom('assets').select('id').whereRef('assets.id', '=', 'move_history.entityId'),
        ),
      )
      .where('move_history.pathType', '=', sql.lit(AssetPathType.ORIGINAL))
      .execute();
  }

  @GenerateSql()
  async cleanMoveHistorySingle(assetId: string): Promise<void> {
    await this.db
      .deleteFrom('move_history')
      .where('move_history.pathType', '=', sql.lit(AssetPathType.ORIGINAL))
      .where('entityId', '=', assetId)
      .execute();
  }
}