zod-backend/src/user/repositories/device.repository.ts

import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { IsNull, Not, Repository } from 'typeorm';
import { Device } from '../../user/entities';

@Injectable()
export class DeviceRepository {
  constructor(@InjectRepository(Device) private readonly deviceRepository: Repository<Device>) {}

  findUserDeviceById(deviceId: string, userId: string) {
    return this.deviceRepository.findOne({ where: { deviceId, userId } });
  }

  findByDeviceId(deviceId: string) {
    return this.deviceRepository.findOne({ where: { deviceId } });
  }

  createDevice(data: Partial<Device>) {
    return this.deviceRepository.save(data);
  }

  updateDevice(deviceId: string, data: Partial<Device>) {
    return this.deviceRepository.save({ deviceId, ...data });
  }

  getTokens(userId: string) {
    return this.deviceRepository.find({ where: { userId, fcmToken: Not(IsNull()) }, select: ['fcmToken'] });
  }
}