import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository, FindOptionsWhere } from 'typeorm';
import { ClsService } from 'nestjs-cls';
import { TenantAwareRepository } from '../../../common/repositories/tenant-aware.repository';
import { ServiceTypeEntity } from '../../../entities/service-type.entity';

@Injectable()
export class ServiceTypeRepository extends TenantAwareRepository<ServiceTypeEntity> {
  constructor(
    @InjectRepository(ServiceTypeEntity) repo: Repository<ServiceTypeEntity>,
    cls: ClsService,
  ) {
    super(repo, cls);
  }

  async findAllActive(): Promise<ServiceTypeEntity[]> {
    return this.find({
      where: { is_active: true } as FindOptionsWhere<ServiceTypeEntity>,
      order: { sort_order: 'ASC', name: 'ASC' },
    });
  }

  async findAllSorted(): Promise<ServiceTypeEntity[]> {
    return this.find({ order: { name: 'ASC' } });
  }

  async findByCode(code: string): Promise<ServiceTypeEntity | null> {
    return this.findOne({
      where: { code },
    });
  }
}