import { Attr } from 'ts-framework'
import { MobilityResponseDTO } from './mobility'

export class CreateMobilityCategoryDTO {

  @Attr({ type: String })
  name: string

}

export class MobilityCategoryResponseDTO {

  id: string
  name: string
  mobilities?: MobilityResponseDTO[]

  constructor(model: MobilityCategoryResponseDTO) {
    this.id = model.id
    this.name = model.name
    if (model.mobilities) {
      this.mobilities = model.mobilities.map((it) => MobilityResponseDTO.of(it))
    }
  }

  static of(model: MobilityCategoryResponseDTO): MobilityCategoryResponseDTO {
    return new MobilityCategoryResponseDTO(model)
  }

}