sprinklers3/server/repositories/SprinklersDeviceRepository.ts

import { EntityRepository, Repository } from "typeorm";

import { SprinklersDevice, User } from "@server/entities";

@EntityRepository(SprinklersDevice)
export class SprinklersDeviceRepository extends Repository<SprinklersDevice> {
  findByName(name: string) {
    return this.findOne({ name });
  }

  async userHasAccess(userId: number, deviceId: number): Promise<boolean> {
    const count = await this.manager
      .createQueryBuilder(User, "user")
      .innerJoinAndSelect(
        "user.devices",
        "sprinklers_device",
        "user.id = :userId AND sprinklers_device.id = :deviceId",
        { userId, deviceId }
      )
      .getCount();
    return count > 0;
  }

  async findUserDevice(
    userId: number,
    deviceId: number
  ): Promise<SprinklersDevice | null> {
    const user = await this.manager
      .createQueryBuilder(User, "user")
      .innerJoinAndSelect(
        "user.devices",
        "sprinklers_device",
        "user.id = :userId AND sprinklers_device.id = :deviceId",
        { userId, deviceId }
      )
      .getOne();
    if (!user) {
      return null;
    }
    return user.devices![0];
  }
}