Make unitsSummarized a Transform field within Listing (#259)

CityOfDetroit · Jul 26, 2021 · d3ae75a · d3ae75a
1 parent 2c9e847
commit d3ae75a
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/backend/core/src/listings/dto/listing.dto.ts b/backend/core/src/listings/dto/listing.dto.ts
@@ -309,12 +309,19 @@ export class ListingDto extends OmitType(Listing, [
   yearBuilt?: number | null
 
   @Expose()
-  @ApiProperty({ type: UnitsSummarized })
-  get unitsSummarized(): UnitsSummarized | undefined {
-    if (Array.isArray(this.units) && this.units.length > 0) {
-      return transformUnits(this.units as Unit[])
-    }
-  }
+  @IsDefined({ groups: [ValidationsGroupsEnum.default] })
+  @ValidateNested({ groups: [ValidationsGroupsEnum.default] })
+  @Type(() => UnitsSummarized)
+  @Transform(
+    (value, obj: Listing) => {
+      const units = obj.property.units
+      if (Array.isArray(units) && units.length > 0) {
+        return transformUnits(units)
+      }
+    },
+    { toClassOnly: true }
+  )
+  unitsSummarized: UnitsSummarized | undefined
 }
 
 export class ListingCreateDto extends OmitType(ListingDto, [