diff --git a/Swiftfin/Views/ItemView/Components/AttributeHStack.swift b/Swiftfin/Views/ItemView/Components/AttributeHStack.swift index 9b308735..dd019db0 100644 --- a/Swiftfin/Views/ItemView/Components/AttributeHStack.swift +++ b/Swiftfin/Views/ItemView/Components/AttributeHStack.swift @@ -22,14 +22,13 @@ extension ItemView { .asAttributeStyle(.outline) } - // TODO: Have stream indicate this instead? - if viewModel.item.isHD ?? false { - Text("HD") - .asAttributeStyle(.fill) - } - if let mediaStreams = viewModel.selectedMediaSource?.mediaStreams { + if mediaStreams.hasHDVideo { + Text("HD") + .asAttributeStyle(.fill) + } + if mediaStreams.has4KVideo { Text("4K") .asAttributeStyle(.fill)