From: Ben Beltran Date: Fri, 19 May 2017 02:54:32 +0000 (-0500) Subject: Implement the lyrics engine X-Git-Tag: 0.1.0^2~2 X-Git-Url: https://git.r.bdr.sh/rbdr/lyricli/commitdiff_plain/85d0536f2e9a3d4596c01b263d76b2b8d9ba70ae?ds=sidebyside;hp=--cc Implement the lyrics engine --- 85d0536f2e9a3d4596c01b263d76b2b8d9ba70ae diff --git a/Package.swift b/Package.swift index 602b04b..5027708 100644 --- a/Package.swift +++ b/Package.swift @@ -4,5 +4,6 @@ let package = Package( name: "lyricli", dependencies: [ .Package(url: "https://github.com/rbdr/CommandLineKit", majorVersion: 4, minor: 0), + .Package(url: "https://github.com/IBM-Swift/swift-html-entities.git", majorVersion: 3, minor: 0) ] ) diff --git a/Sources/lyrics_engine.swift b/Sources/lyrics_engine.swift index 661ce86..d6b1985 100644 --- a/Sources/lyrics_engine.swift +++ b/Sources/lyrics_engine.swift @@ -1,15 +1,51 @@ +import Foundation +import HTMLEntities + /// Looks for lyrics on the internet class LyricsEngine { - let track: Track + private let apiURL = "https://lyrics.wikia.com/api.php?action=lyrics&func=getSong&fmt=realjson" + private let apiMethod = "GET" + private let lyricsMatcher = "class='lyricbox'>(.+) Void in + if let lyricsResult = lyricsResult { + lyrics = lyricsResult + requestFinished = true + asyncLock.signal() + } + }) + + while(!requestFinished) { + asyncLock.wait() + } + asyncLock.unlock() + } + } } - return nil + return lyrics } } @@ -17,4 +53,90 @@ class LyricsEngine { track = targetTrack } + + // Fetch the lyrics from the API and request / parse the page + + private func fetchLyricsFromAPI(withURL url: URL, completionHandler: @escaping (String?) -> Void) { + + var apiRequest = URLRequest(url: url) + apiRequest.httpMethod = "GET" + + let task = URLSession.shared.dataTask(with: apiRequest, completionHandler: {data, response, error -> Void in + + // If the response is parseable JSON, and has a url, we'll look for + // the lyrics in there + + if let data = data { + let jsonResponse = try? JSONSerialization.jsonObject(with: data) as! [String: Any] + if let jsonResponse = jsonResponse { + if let lyricsUrlString = jsonResponse["url"] as? String { + if let lyricsUrl = URL(string: lyricsUrlString) { + + // At this point we have a valid wiki url + self.fetchLyricsFromPage(withURL: lyricsUrl, completionHandler: completionHandler) + return + } + } + } + } + + completionHandler(nil) + }) + task.resume() + } + + // Fetch the lyrics from the page and parse the page + + private func fetchLyricsFromPage(withURL url: URL, completionHandler: @escaping (String?) -> Void) { + + var pageRequest = URLRequest(url: url) + pageRequest.httpMethod = "GET" + + let task = URLSession.shared.dataTask(with: pageRequest, completionHandler: {data, response, error -> Void in + + // If the response is parseable JSON, and has a url, we'll look for + // the lyrics in there + + if let data = data { + if let htmlBody = String(data: data, encoding: String.Encoding.utf8) { + self.parseHtmlBody(htmlBody, completionHandler: completionHandler) + return + } + } + + completionHandler(nil) + }) + task.resume() + } + + // Parses the wiki to obtain the lyrics + + private func parseHtmlBody(_ body: String, completionHandler: @escaping (String?) -> Void) { + + if let regex = try? NSRegularExpression(pattern: lyricsMatcher) { + let matches = regex.matches(in: body, range: NSRange(location: 0, length: body.characters.count)) + + for match in matches { + + let nsBody = body as NSString + let range = match.rangeAt(1) + let encodedLyrics = nsBody.substring(with: range) + + let decodedLyrics = decodeLyrics(encodedLyrics) + + completionHandler(decodedLyrics) + return + } + } + + completionHandler(nil) + } + + // Escapes the HTML entities + + private func decodeLyrics(_ lyrics: String) -> String { + + let unescapedLyrics = lyrics.htmlUnescape() + return unescapedLyrics.replacingOccurrences(of: "
", with: "\n") + } }