Vizzuality/landgriffon

import { Injectable, Logger, NotFoundException } from '@nestjs/common';
import {
  AppBaseService,
  JSONAPISerializerConfig,
} from 'utils/app-base.service';
import { Unit, unitResource } from 'modules/units/unit.entity';
import { AppInfoDTO } from 'dto/info.dto';
import { UnitRepository } from 'modules/units/unit.repository';
import { CreateUnitDto } from 'modules/units/dto/create.unit.dto';
import { UpdateUnitDto } from 'modules/units/dto/update.unit.dto';

@Injectable()
export class UnitsService extends AppBaseService<
  Unit,
  CreateUnitDto,
  UpdateUnitDto,
  AppInfoDTO
> {
  constructor(protected readonly unitRepository: UnitRepository) {
    super(unitRepository, unitResource.name.singular, unitResource.name.plural);
  }

  get serializerConfig(): JSONAPISerializerConfig<Unit> {
    return {
      attributes: ['name', 'symbol', 'description'],
      keyForAttribute: 'camelCase',
    };
  }

  async getUnitById(id: string): Promise<Unit> {
    const found: Unit | null = await this.unitRepository.findOne({
      where: { id },
    });

    if (!found) {
      throw new NotFoundException(` Unit with ID "${id}" not found`);
    }

    return found;
  }
}