ultrasonic-app-subsonic-android/ultrasonic/src/main/kotlin/org/moire/ultrasonic/model/SearchListModel.kt

package org.moire.ultrasonic.model

import android.app.Application
import android.os.Bundle
import androidx.lifecycle.MutableLiveData
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.withContext
import org.moire.ultrasonic.domain.SearchCriteria
import org.moire.ultrasonic.domain.SearchResult
import org.moire.ultrasonic.fragment.SearchFragment
import org.moire.ultrasonic.service.MusicService
import org.moire.ultrasonic.service.MusicServiceFactory
import org.moire.ultrasonic.util.Settings

class SearchListModel(application: Application) : GenericListModel(application) {

    var searchResult: MutableLiveData<SearchResult?> = MutableLiveData()

    override fun load(
        isOffline: Boolean,
        useId3Tags: Boolean,
        musicService: MusicService,
        refresh: Boolean,
        args: Bundle
    ) {
        super.load(isOffline, useId3Tags, musicService, refresh, args)
    }

    suspend fun search(query: String) {
        val maxArtists = Settings.maxArtists
        val maxAlbums = Settings.maxAlbums
        val maxSongs = Settings.maxSongs

        withContext(Dispatchers.IO) {
            val criteria = SearchCriteria(query, maxArtists, maxAlbums, maxSongs)
            val service = MusicServiceFactory.getMusicService()
            val result = service.search(criteria)

            if (result != null) searchResult.postValue(result)
        }
    }

    fun trimResultLength(
        result: SearchResult,
        maxArtists: Int = SearchFragment.DEFAULT_ARTISTS,
        maxAlbums: Int = SearchFragment.DEFAULT_ALBUMS,
        maxSongs: Int = SearchFragment.DEFAULT_SONGS
    ): SearchResult {
        return SearchResult(
            artists = result.artists.take(maxArtists),
            albums = result.albums.take(maxAlbums),
            songs = result.songs.take(maxSongs)
        )
    }
}