Merge branch 'release/0.3.0'

[rbdr/lyricli] / Sources / itunes_source.swift

import ScriptingBridge

// Protocol to obtain the track from iTunes
@objc protocol iTunesTrack {
    @objc optional var name: String {get}
    @objc optional var artist: String {get}
}

// Protocol to interact with iTunes
@objc protocol iTunesApplication {
    @objc optional var currentTrack: iTunesTrack? {get}
    @objc optional var currentStreamTitle: String? {get}
}

extension SBApplication : iTunesApplication {}

// Source that reads track artist and name from current itunes track
class ItunesSource: Source {

    // Calls the spotify API and returns the current track
    var currentTrack: Track? {

        if let iTunes: iTunesApplication = SBApplication(bundleIdentifier: "com.apple.iTunes") {

            // Attempt to fetch the title from a stream
            if let currentStreamTitle = iTunes.currentStreamTitle {
                if let track = currentStreamTitle {

                    let trackComponents = track.characters.split(separator: "-").map(String.init)

                    if trackComponents.count == 2 {
                        let artist = trackComponents[0].trimmingCharacters(in: .whitespaces)
                        let name = trackComponents[1].trimmingCharacters(in: .whitespaces)

                        return Track(withName: name, andArtist: artist)
                    }

                }
            }

            // Attempt to fetch the title from a song
            if let currentTrack = iTunes.currentTrack {
                if let track = currentTrack {
                    if let name = track.name {
                        if let artist = track.artist {

                            // track properties are empty strings if itunes is closed
                            if (!(name != "" && artist != "")) {
                                return nil
                            }
                            return Track(withName: name, andArtist: artist)
                        }
                    }
                }
            }
        }

        return nil
    }

}