From bcb59896dbb7449cb89190c829322914959271e6 Mon Sep 17 00:00:00 2001 From: Terry Yiu Date: Tue, 6 May 2025 10:58:10 -0400 Subject: [PATCH] Optimize classify_url function Changelog-Fixed: Optimized classify_url function Signed-off-by: Terry Yiu --- damus/Models/NoteContent.swift | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/damus/Models/NoteContent.swift b/damus/Models/NoteContent.swift index 8a4e4971..ea3ed120 100644 --- a/damus/Models/NoteContent.swift +++ b/damus/Models/NoteContent.swift @@ -206,17 +206,16 @@ func url_str(_ url: URL) -> CompatibleText { } func classify_url(_ url: URL) -> UrlType { - let str = url.lastPathComponent.lowercased() - - if str.hasSuffix(".png") || str.hasSuffix(".jpg") || str.hasSuffix(".jpeg") || str.hasSuffix(".gif") || str.hasSuffix(".webp") { + let fileExtension = url.lastPathComponent.lowercased().components(separatedBy: ".").last + + switch fileExtension { + case "png", "jpg", "jpeg", "gif", "webp": return .media(.image(url)) - } - - if str.hasSuffix(".mp4") || str.hasSuffix(".mov") || str.hasSuffix(".m3u8") { + case "mp4", "mov", "m3u8": return .media(.video(url)) + default: + return .link(url) } - - return .link(url) } func attributed_string_attach_icon(_ astr: inout AttributedString, img: UIImage) {