immich/server/libs/infra/src/repositories/user-token.repository.ts

import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { UserTokenEntity } from '../entities';
import { IUserTokenRepository } from '@app/domain/user-token';

@Injectable()
export class UserTokenRepository implements IUserTokenRepository {
  constructor(
    @InjectRepository(UserTokenEntity)
    private userTokenRepository: Repository<UserTokenEntity>,
  ) {}

  async get(userToken: string): Promise<UserTokenEntity | null> {
    return this.userTokenRepository.findOne({ where: { token: userToken }, relations: { user: true } });
  }

  async create(userToken: Partial<UserTokenEntity>): Promise<UserTokenEntity> {
    return this.userTokenRepository.save(userToken);
  }

  async delete(id: string): Promise<void> {
    await this.userTokenRepository.delete(id);
  }

  async deleteAll(userId: string): Promise<void> {
    await this.userTokenRepository.delete({ user: { id: userId } });
  }
}