funkwhale-app-android/app/src/main/java/audio/funkwhale/ffa/activities/SearchActivity.kt

package audio.funkwhale.ffa.activities

import android.os.Bundle
import android.view.View
import androidx.appcompat.app.AppCompatActivity
import androidx.appcompat.widget.SearchView
import androidx.lifecycle.Lifecycle
import androidx.lifecycle.lifecycleScope
import androidx.recyclerview.widget.LinearLayoutManager
import audio.funkwhale.ffa.adapters.FavoriteListener
import audio.funkwhale.ffa.adapters.SearchAdapter
import audio.funkwhale.ffa.databinding.ActivitySearchBinding
import audio.funkwhale.ffa.fragments.AddToPlaylistDialog
import audio.funkwhale.ffa.fragments.AlbumsFragment
import audio.funkwhale.ffa.fragments.ArtistsFragment
import audio.funkwhale.ffa.model.Album
import audio.funkwhale.ffa.model.Artist
import audio.funkwhale.ffa.repositories.AlbumsSearchRepository
import audio.funkwhale.ffa.repositories.ArtistsSearchRepository
import audio.funkwhale.ffa.repositories.FavoritesRepository
import audio.funkwhale.ffa.repositories.Repository
import audio.funkwhale.ffa.repositories.TracksSearchRepository
import audio.funkwhale.ffa.utils.Command
import audio.funkwhale.ffa.utils.CommandBus
import audio.funkwhale.ffa.utils.Event
import audio.funkwhale.ffa.utils.EventBus
import audio.funkwhale.ffa.utils.getMetadata
import audio.funkwhale.ffa.utils.untilNetwork
import com.google.android.exoplayer2.offline.Download
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.launch
import kotlinx.coroutines.withContext
import java.net.URLEncoder
import java.util.Locale

class SearchActivity : AppCompatActivity() {
  private lateinit var adapter: SearchAdapter

  private lateinit var artistsRepository: ArtistsSearchRepository
  private lateinit var albumsRepository: AlbumsSearchRepository
  private lateinit var tracksRepository: TracksSearchRepository
  private lateinit var favoritesRepository: FavoritesRepository
  private lateinit var binding: ActivitySearchBinding

  var done = 0

  override fun onCreate(savedInstanceState: Bundle?) {
    super.onCreate(savedInstanceState)

    artistsRepository = ArtistsSearchRepository(this@SearchActivity, "")
    albumsRepository = AlbumsSearchRepository(this@SearchActivity, "")
    tracksRepository = TracksSearchRepository(this@SearchActivity, "")
    favoritesRepository = FavoritesRepository(this@SearchActivity)

    binding = ActivitySearchBinding.inflate(layoutInflater)

    setContentView(binding.root)

    binding.search.requestFocus()
  }

  override fun onResume() {
    super.onResume()

    lifecycleScope.launch(Dispatchers.Main) {
      CommandBus.get().collect { command ->
        if (command is Command.AddToPlaylist) {

          if (lifecycle.currentState.isAtLeast(Lifecycle.State.RESUMED)) {
            AddToPlaylistDialog.show(
              layoutInflater,
              this@SearchActivity,
              lifecycleScope,
              command.tracks
            )
          }
        }
      }
    }

    lifecycleScope.launch(Dispatchers.IO) {
      EventBus.get().collect { event ->
        if (event is Event.DownloadChanged) refreshDownloadedTrack(event.download)
      }
    }

    adapter =
      SearchAdapter(
        layoutInflater,
        this,
        SearchResultClickListener(),
        FavoriteListener(favoritesRepository)
      ).also {
        binding.results.layoutManager = LinearLayoutManager(this)
        binding.results.adapter = it
      }

    binding.search.setOnQueryTextListener(object : SearchView.OnQueryTextListener {

      override fun onQueryTextSubmit(rawQuery: String?): Boolean {
        binding.search.clearFocus()

        rawQuery?.let {
          done = 0

          val query = URLEncoder.encode(it, "UTF-8")

          artistsRepository.query = query.lowercase(Locale.ROOT)
          albumsRepository.query = query.lowercase(Locale.ROOT)
          tracksRepository.query = query.lowercase(Locale.ROOT)

          binding.searchSpinner.visibility = View.VISIBLE
          binding.searchEmpty.visibility = View.GONE
          binding.searchNoResults.visibility = View.GONE

          adapter.artists.clear()
          adapter.albums.clear()
          adapter.tracks.clear()
          adapter.notifyDataSetChanged()

          artistsRepository.fetch(Repository.Origin.Network.origin)
            .untilNetwork(lifecycleScope) { artists, _, _, _ ->
              done++

              adapter.artists.addAll(artists)
              refresh()
            }

          albumsRepository.fetch(Repository.Origin.Network.origin)
            .untilNetwork(lifecycleScope) { albums, _, _, _ ->
              done++

              adapter.albums.addAll(albums)
              refresh()
            }

          tracksRepository.fetch(Repository.Origin.Network.origin)
            .untilNetwork(lifecycleScope) { tracks, _, _, _ ->
              done++

              adapter.tracks.addAll(tracks)
              refresh()
            }
        }

        return true
      }

      override fun onQueryTextChange(newText: String?) = true
    })
  }

  private fun refresh() {
    adapter.notifyDataSetChanged()

    if (adapter.artists.size + adapter.albums.size + adapter.tracks.size == 0) {
      binding.searchNoResults.visibility = View.VISIBLE
    } else {
      binding.searchNoResults.visibility = View.GONE
    }

    if (done == 3) {
      binding.searchSpinner.visibility = View.INVISIBLE
    }
  }

  private suspend fun refreshDownloadedTrack(download: Download) {
    if (download.state == Download.STATE_COMPLETED) {
      download.getMetadata()?.let { info ->
        adapter.tracks.withIndex().associate { it.value to it.index }
          .filter { it.key.id == info.id }.toList().getOrNull(0)?.let { match ->
            withContext(Dispatchers.Main) {
              adapter.tracks[match.second].downloaded = true
              adapter.notifyItemChanged(
                adapter.getPositionOf(
                  SearchAdapter.ResultType.Track,
                  match.second
                )
              )
            }
          }
      }
    }
  }

  inner class SearchResultClickListener : SearchAdapter.OnSearchResultClickListener {
    override fun onArtistClick(holder: View?, artist: Artist) {
      ArtistsFragment.openAlbums(this@SearchActivity, artist)
    }

    override fun onAlbumClick(holder: View?, album: Album) {
      AlbumsFragment.openTracks(this@SearchActivity, album)
    }
  }
}