NetNewsWire/Shared/Favicons/FaviconURLFinder.swift

67 lines
1.7 KiB
Swift
Raw Normal View History

//
// FaviconURLFinder.swift
2018-08-29 07:18:24 +02:00
// NetNewsWire
//
// Created by Brent Simmons on 11/20/17.
// Copyright © 2017 Ranchero Software. All rights reserved.
//
import Foundation
import CoreServices
import RSParser
2024-02-25 05:39:44 +01:00
import UniformTypeIdentifiers
// The favicon URLs may be specified in the head section of the home page.
struct FaviconURLFinder {
2020-01-31 23:33:35 +01:00
/// Uniform types to ignore when finding favicon URLs.
2024-02-25 05:39:44 +01:00
static var ignoredTypes = [UTType.svg]
/// Finds favicon URLs in a web page.
/// - Parameters:
/// - homePageURL: The page to search.
/// - completion: A closure called when the links have been found.
/// - urls: An array of favicon URLs as strings.
2020-02-02 02:19:33 +01:00
static func findFaviconURLs(with homePageURL: String, _ completion: @escaping (_ urls: [String]?) -> Void) {
guard let _ = URL(unicodeString: homePageURL) else {
completion(nil)
return
}
// If the favicon has an explicit type, check that for an ignored type; otherwise, check the file extension.
2024-02-25 05:39:44 +01:00
HTMLMetadataDownloader.downloadMetadata(for: homePageURL) { htmlMetadata in
let faviconURLs = htmlMetadata?.favicons.compactMap { favicon -> String? in
2024-02-25 05:39:44 +01:00
guard shouldAllowFavicon(favicon) else {
return nil
}
return favicon.urlString
2024-02-25 05:39:44 +01:00
}
completion(faviconURLs)
}
}
2024-02-25 05:39:44 +01:00
static func shouldAllowFavicon(_ favicon: RSHTMLMetadataFavicon) -> Bool {
// Check mime type.
if let mimeType = favicon.type, let utType = UTType(mimeType: mimeType) {
if ignoredTypes.contains(utType) {
return false
}
}
// Check file extension.
if let urlString = favicon.urlString, let url = URL(string: urlString), let utType = UTType(filenameExtension: url.pathExtension) {
if ignoredTypes.contains(utType) {
return false
}
}
return true
}
}