import { Transform } from 'class-transformer'

import { parseDate } from '../utils'

import { BaseModel } from './base.model'
import { SmartfitUnitType } from './types'

export class SmartfitUnit extends BaseModel<SmartfitUnit> {
  id: number
  name: string
  acronym: string
  type: SmartfitUnitType
  country_code: string
  city: string
  address: string
  state: string
  nickname: string
  allow_offline: boolean
  allow_teacher_indication: boolean
  @Transform(value => parseDate(value), { toClassOnly: true })
  created_at: Date
  @Transform(value => parseDate(value), { toClassOnly: true })
  updated_at: Date
}