diff --git a/Shared/PostList/PostListModel.swift b/Shared/PostList/PostListModel.swift index 8b221bc..686f3bf 100644 --- a/Shared/PostList/PostListModel.swift +++ b/Shared/PostList/PostListModel.swift @@ -1,86 +1,120 @@ import SwiftUI import CoreData class PostListModel: ObservableObject { func remove(_ post: WFAPost) { LocalStorageManager.persistentContainer.viewContext.delete(post) LocalStorageManager().saveContext() } func purgePublishedPosts() { let fetchRequest: NSFetchRequest = NSFetchRequest(entityName: "WFAPost") fetchRequest.predicate = NSPredicate(format: "status != %i", 0) let deleteRequest = NSBatchDeleteRequest(fetchRequest: fetchRequest) do { try LocalStorageManager.persistentContainer.viewContext.executeAndMergeChanges(using: deleteRequest) } catch { print("Error: Failed to purge cached posts.") } } func getBodyPreview(of post: WFAPost) -> String { var elidedPostBody: String = "" // Strip any markdown from the post body. let strippedPostBody = stripMarkdown(from: post.body) // Get the first 80 characters. let firstEightyChars = String(strippedPostBody.prefix(80)) // Extract lede from post. elidedPostBody = extractLede(from: firstEightyChars) return elidedPostBody } } private extension PostListModel { func stripMarkdown(from string: String) -> String { var strippedString = string strippedString = stripHeadingOctothorpes(from: strippedString) + strippedString = stripImages(from: strippedString, keepAltText: true) return strippedString } func stripHeadingOctothorpes(from string: String) -> String { let newLines = CharacterSet.newlines var processedComponents: [String] = [] let components = string.components(separatedBy: newLines) for component in components { if component.isEmpty { continue } var newString = component while newString.first == "#" { newString.removeFirst() } if newString.hasPrefix(" ") { newString.removeFirst() } processedComponents.append(newString) } let headinglessString = processedComponents.joined(separator: "\n\n") return headinglessString } + func stripImages(from string: String, keepAltText: Bool = false) -> String { + let pattern = #"!\[[\"]?(.*?)[\"|]?\]\(.*?\)"# + var processedComponents: [String] = [] + let components = string.components(separatedBy: .newlines) + for component in components { + if component.isEmpty { continue } + var processedString: String = component + if keepAltText { + let regex = try? NSRegularExpression(pattern: pattern, options: []) + if let matches = regex?.matches( + in: component, options: [], range: NSRange(location: 0, length: component.utf16.count) + ) { + for match in matches { + if let range = Range(match.range(at: 1), in: component) { + processedString = "\(component[range])" + } + } + } + } else { + let range = component.startIndex.. String { let terminatingCharacters = CharacterSet(charactersIn: ".。?").union(.newlines) var lede: String let sentences = string.components(separatedBy: terminatingCharacters) let firstSentence = sentences.filter { !$0.isEmpty }[0] if firstSentence == string { let endOfStringIndex = string.lastIndex(of: " ") lede = String(string[..<(endOfStringIndex ?? string.index(string.endIndex, offsetBy: -2))]) + "…" } else { lede = firstSentence } return lede } }