From d63436b1d91f32df3114faceac496882df825be1 Mon Sep 17 00:00:00 2001
From: Duong Thai <dvthai.dev@gmail.com>
Date: Mon, 11 Mar 2024 17:45:45 +0700
Subject: [PATCH] fix `getFirstImageOf(..)`

---
 .../Sources/RSS/Models/RSSItem+SendableData.swift |  2 +-
 Packages/RSS/Sources/RSS/RSSTools.swift           | 15 ++++++++++++---
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/Packages/RSS/Sources/RSS/Models/RSSItem+SendableData.swift b/Packages/RSS/Sources/RSS/Models/RSSItem+SendableData.swift
index f47aa4a7..554f2f54 100644
--- a/Packages/RSS/Sources/RSS/Models/RSSItem+SendableData.swift
+++ b/Packages/RSS/Sources/RSS/Models/RSSItem+SendableData.swift
@@ -63,7 +63,7 @@ extension ParsedItem {
       (url: imageURL, size: image.size)
     } else if
       let contentHTML = self.contentHTML,
-      let imageURL = RSSTools.getFirstImageOf(html: contentHTML),
+      let imageURL = RSSTools.getFirstImageOf(html: contentHTML, baseURL: self.getRSSURL()),
       let imageData = try? Data(contentsOf: imageURL),
       let image = UIImage(data: imageData)
     {
diff --git a/Packages/RSS/Sources/RSS/RSSTools.swift b/Packages/RSS/Sources/RSS/RSSTools.swift
index 3d23d02e..a12488fa 100644
--- a/Packages/RSS/Sources/RSS/RSSTools.swift
+++ b/Packages/RSS/Sources/RSS/RSSTools.swift
@@ -103,9 +103,18 @@ public enum RSSTools {
     return NonEmptyString(String(match.1))
   }
 
-  public static func getFirstImageOf(html: String) -> URL? {
-    guard let match = html.firstMatch(of: /<img[\s\S]+?src="(.+?)"/) else { return nil }
-    return URL(string: String(match.1))
+  public static func getFirstImageOf(html: String, baseURL: URL?) -> URL? {
+    guard let match = html.firstMatch(of: /<img[\s\S]+?src="(.+?)"/),
+          let url =  URL(string: String(match.1))
+    else { return nil }
+
+    return if url.host != nil {
+      url
+    } else if let host = baseURL?.host() {
+      URL(string: "https://\(host)\(url.absoluteString)")
+    } else {
+      nil
+    }
   }
 
   public static func getFeedData(from feedURL: URL) async -> RSSFeed.SendableData? {