2021-02-06 19:02:12 +01:00
package org.mosad.teapod.ui.activity.main.viewmodel
import android.app.Application
2021-07-10 23:37:16 +02:00
import android.util.Log
2021-02-06 19:02:12 +01:00
import androidx.lifecycle.AndroidViewModel
import org.mosad.teapod.parser.AoDParser
import org.mosad.teapod.util.*
import org.mosad.teapod.util.DataTypes.MediaType
2021-07-10 23:37:16 +02:00
import org.mosad.teapod.util.tmdb.TMDBApiController
import org.mosad.teapod.util.tmdb.TMDBResult
2021-07-25 19:15:31 +02:00
import org.mosad.teapod.util.tmdb.TMDBTVSeason
2021-02-06 19:02:12 +01:00
/ * *
* handle media , next ep and tmdb
2021-07-11 12:56:21 +02:00
* TODO this lives in activity , is this correct ?
2021-02-06 19:02:12 +01:00
* /
class MediaFragmentViewModel ( application : Application ) : AndroidViewModel ( application ) {
2021-08-31 19:47:18 +02:00
// var media = Media(-1, "", MediaType.OTHER)
// internal set
// var nextEpisode = Episode()
// internal set
var media2 = AoDMediaNone
2021-02-06 19:02:12 +01:00
internal set
2021-08-31 19:47:18 +02:00
var nextEpisodeId = - 1
2021-02-06 19:02:12 +01:00
internal set
2021-08-31 19:47:18 +02:00
2021-07-17 13:15:14 +02:00
var tmdbResult : TMDBResult ? = null // TODO rename
2021-02-06 19:02:12 +01:00
internal set
2021-07-25 19:15:31 +02:00
var tmdbTVSeason : TMDBTVSeason ? = null
2021-07-11 12:56:21 +02:00
internal set
var mediaMeta : Meta ? = null
internal set
2021-02-06 19:02:12 +01:00
/ * *
* set media , tmdb and nextEpisode
2021-07-11 12:56:21 +02:00
* TODO run aod and tmdb load parallel
2021-02-06 19:02:12 +01:00
* /
suspend fun load ( mediaId : Int ) {
2021-07-11 12:56:21 +02:00
val tmdbApiController = TMDBApiController ( )
2021-08-31 19:47:18 +02:00
//media = AoDParser.getMediaById(mediaId)
media2 = AoDParser . getMediaById2 ( mediaId )
2021-07-10 23:37:16 +02:00
2021-07-11 12:56:21 +02:00
// check if metaDB knows the title
2021-08-31 19:47:18 +02:00
val tmdbId : Int = if ( MetaDBController . mediaList . media . contains ( media2 . aodId ) ) {
2021-07-11 12:56:21 +02:00
// load media info from metaDB
val metaDB = MetaDBController ( )
2021-08-31 19:47:18 +02:00
mediaMeta = when ( media2 . type ) {
MediaType . MOVIE -> metaDB . getMovieMetadata ( media2 . aodId )
MediaType . TVSHOW -> metaDB . getTVShowMetadata ( media2 . aodId )
2021-07-11 12:56:21 +02:00
else -> null
}
mediaMeta ?. tmdbId ?: - 1
} else {
// use tmdb search to get media info
mediaMeta = null // set mediaMeta to null, if metaDB doesn't know the media
2021-08-31 19:47:18 +02:00
tmdbApiController . search ( stripTitleInfo ( media2 . title ) , media2 . type )
2021-07-11 12:56:21 +02:00
}
2021-07-10 23:37:16 +02:00
2021-08-31 19:47:18 +02:00
tmdbResult = when ( media2 . type ) {
2021-07-10 23:37:16 +02:00
MediaType . MOVIE -> tmdbApiController . getMovieDetails ( tmdbId )
MediaType . TVSHOW -> tmdbApiController . getTVShowDetails ( tmdbId )
2021-07-17 13:15:14 +02:00
else -> null
2021-07-10 23:37:16 +02:00
}
println ( tmdbResult ) // TODO
2021-07-11 12:56:21 +02:00
// get season info, if metaDB knows the tv show
2021-08-31 19:47:18 +02:00
tmdbTVSeason = if ( media2 . type == MediaType . TVSHOW && mediaMeta != null ) {
2021-07-11 12:56:21 +02:00
val tvShowMeta = mediaMeta as TVShowMeta
tmdbApiController . getTVSeasonDetails ( tvShowMeta . tmdbId , tvShowMeta . tmdbSeasonNumber )
} else {
null
2021-07-10 23:37:16 +02:00
}
2021-08-31 19:47:18 +02:00
if ( media2 . type == MediaType . TVSHOW ) {
//nextEpisode = media.episodes.firstOrNull{ !it.watched } ?: media.episodes.first()
nextEpisodeId = media2 . playlist . firstOrNull { ! it . watched } ?. mediaId
?: media2 . playlist . first ( ) . mediaId
2021-02-06 19:02:12 +01:00
}
}
/ * *
2021-08-31 19:47:18 +02:00
* get the next episode based on episodeId
2021-02-06 19:02:12 +01:00
* if no matching is found , use first episode
* /
2021-08-31 19:47:18 +02:00
fun updateNextEpisode ( episodeId : Int ) {
if ( media2 . type == MediaType . MOVIE ) return // return if movie
// nextEpisode = media.episodes.firstOrNull{ it.number > currentEp.number }
// ?: media.episodes.first()
2021-02-06 19:02:12 +01:00
2021-08-31 19:47:18 +02:00
nextEpisodeId = media2 . playlist . firstOrNull { it . number > media2 . getEpisodeById ( episodeId ) . number } ?. mediaId
?: media2 . playlist . first ( ) . mediaId
2021-02-06 19:02:12 +01:00
}
2021-07-10 23:37:16 +02:00
// remove unneeded info from the media title before searching
private fun stripTitleInfo ( title : String ) : String {
return title . replace ( " (Sub) " , " " )
. replace ( Regex ( " -? \\ s?[0-9]+. \\ s?(Staffel|Season) " ) , " " )
. replace ( Regex ( " (Staffel|Season) \\ s?[0-9]+ " ) , " " )
. trim ( )
}
/ * * guess Season from title
* if the title ends with a number , that could be the season
* if the title ends with Regex ( " -? \\ s?[0-9]+. \\ s?(Staffel|Season) " ) or
* Regex ( " (Staffel|Season) \\ s?[0-9]+ " ) , that is the season information
* /
private fun guessSeasonFromTitle ( title : String ) : Int {
val helpTitle = title . replace ( " (Sub) " , " " ) . trim ( )
Log . d ( " test " , " helpTitle: $helpTitle " )
return if ( helpTitle . last ( ) . isDigit ( ) ) {
helpTitle . last ( ) . digitToInt ( )
} else {
Regex ( " ([0-9]+. \\ s?(Staffel|Season))|((Staffel|Season) \\ s?[0-9]+) " )
. find ( helpTitle )
?. value ?. filter { it . isDigit ( ) } ?. toInt ( ) ?: 1
}
}
2021-02-06 19:02:12 +01:00
}