vlc-ios/Sources/LibrarySearchDataSource.swift

/*****************************************************************************
 * LibrarySearchDataSource.swift
 * VLC for iOS
 *****************************************************************************
 * Copyright (c) 2019 VideoLAN. All rights reserved.
 * $Id$
 *
 * Authors: Carola Nitz <caro # videolan.org>
 *
 * Refer to the COPYING file of the official project for license.
 *****************************************************************************/

class LibrarySearchDataSource: NSObject {

    var searchData = [VLCMLObject]()
    var model: MediaLibraryBaseModel

    init(model: MediaLibraryBaseModel) {
        self.model = model
        super.init()
        shouldReloadFor(searchString: "")
    }

    func objectAtIndex(index: Int) -> VLCMLObject? {
         return index < searchData.count ? searchData[index] : nil
    }

    func shouldReloadFor(searchString: String) {
        guard searchString != "" else {
            searchData = model.anyfiles
            return
        }
        searchData.removeAll()
        let lowercaseSearchString = searchString.lowercased()
        model.anyfiles.forEach {
            if let media = $0 as? VLCMLMedia {
                if media.contains(lowercaseSearchString) { searchData.append($0) }
            } else if let album = $0 as? VLCMLAlbum {
                if album.contains(lowercaseSearchString) { searchData.append($0) }
            } else if let genre = $0 as? VLCMLGenre {
                if genre.contains(lowercaseSearchString) { searchData.append($0) }
            } else if let playlist = $0 as? VLCMLPlaylist {
                if playlist.contains(lowercaseSearchString) { searchData.append($0) }
            } else if let artist = $0 as? VLCMLArtist {
                if artist.contains(lowercaseSearchString) { searchData.append($0) }
            } else if let albumtrack = $0 as? VLCMLAlbumTrack {
                if albumtrack.contains(lowercaseSearchString) { searchData.append($0) }
            } else {
                assertionFailure("unhandled type")
            }
        }
    }
}

extension VLCMLObject {
    func contains(_ searchString: String) -> Bool {
        return false
    }
}

extension VLCMLMedia {
    func contains(_ searchString: String) -> Bool {
        return title.lowercased().contains(searchString)
    }
}

extension VLCMLAlbum {
    func contains(_ searchString: String) -> Bool {
        var matches = false
        matches = matches || title.lowercased().contains(searchString)
        matches = matches || String(releaseYear()).lowercased().contains(searchString)
        matches = matches || shortSummary.lowercased().contains(searchString)
        matches = matches || albumArtist?.contains(searchString) ?? false
        matches = matches || tracks?.filter({ $0.contains(searchString)}).isEmpty == false
        return matches
    }
}

extension VLCMLGenre {
    func contains(_ searchString: String) -> Bool {
        return name.lowercased().contains(searchString)
    }
}

extension VLCMLPlaylist {
    func contains(_ searchString: String) -> Bool {
        return name.lowercased().contains(searchString)
    }
}

extension VLCMLArtist {
    func contains(_ searchString: String) -> Bool {
        return name.lowercased().contains(searchString)
    }
}

extension VLCMLAlbumTrack {
    func contains(_ searchString: String) -> Bool {
        var matches = false
        matches = matches || artist?.contains(searchString) ?? false
        matches = matches || genre?.contains(searchString) ?? false
        matches = matches || album?.contains(searchString) ?? false
        return matches
    }
}