# Copyright (C) 2015 - Philipp Temminghoff <phil65@kodi.tv>
# Modifications copyright (C) 2022 - Scott Smart <scott967@kodi.tv>
# This program is Free Software see LICENSE file for details
"""When addon is called with RunScript, executes required action
When called as plugin, sets listitems to container

Public functions:
    start_info_actions(info, params):  takes in invocation info action
        and optional param key/value pairs and exexutes the info action

"""

from __future__ import annotations

import os
import shutil
import time

import xbmc
import xbmcgui
import xbmcplugin
from resources.kutil131 import addon, busy, kodijson

from resources.kutil131 import favs, local_db, utils, youtube
from resources.lib import lastfm
from resources.lib import theaudiodb as AudioDB
from resources.lib import themoviedb as tmdb
from resources.lib import trakt
from resources.lib.windowmanager import wm


def start_info_actions(info: str, params: dict[str, str]):
    """executes an action from infos (info= list) using any params

    See README for list of possible actions

    Args:
        info (str): one of a defined infos list of possible actions
        params (dict[str,str]): Optional parameters for the action

    Returns:
        [ItemList]: a kodi utils ItemList of VideoItems/MusicItems
    """
    if "artistname" in params:
        params["artistname"] = params.get(
            "artistname", "").split(" feat. ")[0].strip()
        if not params.get("artist_mbid"):
            params["artist_mbid"] = utils.fetch_musicbrainz_id(
                params["artistname"])
    utils.log(f'process start_info_actions info: {info} params:')
    utils.pp(params)
    if "prefix" in params and not params["prefix"].endswith('.'):
        params["prefix"] = params["prefix"] + '.'

    # AudioDB / LastFM
    if info == 'discography':
        discography = AudioDB.get_artist_discography(params["artistname"])
        if not discography:
            discography = lastfm.get_artist_albums(params.get("artist_mbid"))
        return discography
    elif info == 'mostlovedtracks':
        return AudioDB.get_most_loved_tracks(params["artistname"])
    elif info == 'trackdetails':
        return AudioDB.get_track_details(params.get("id", ""))
    elif info == 'topartists':
        return lastfm.get_top_artists()
    #  The MovieDB
    elif info == 'incinemamovies':
        return tmdb.get_movies("now_playing")
    elif info == 'upcomingmovies':
        return tmdb.get_movies("upcoming")
    elif info == 'topratedmovies':
        return tmdb.get_movies("top_rated")
    elif info == 'popularmovies':
        return tmdb.get_movies("popular")
    elif info == 'ratedmovies':
        return tmdb.get_rated_media_items("movies")
    elif info == 'starredmovies':
        return tmdb.get_fav_items("movies")
    elif info == 'accountlists':
        account_lists = tmdb.handle_lists(tmdb.get_account_lists())
        for item in account_lists:
            item.set_property("directory", True)
            item.set_folder(True)
        return account_lists
    elif info == 'listmovies':
        return tmdb.get_movies_from_list(params["id"]) if params.get('id') else []
    elif info == 'airingtodaytvshows':
        return tmdb.get_tvshows("airing_today")
    elif info == 'onairtvshows':
        return tmdb.get_tvshows("on_the_air")
    elif info == 'topratedtvshows':
        return tmdb.get_tvshows("top_rated")
    elif info == 'populartvshows':
        return tmdb.get_tvshows("popular")
    elif info == 'ratedtvshows':
        return tmdb.get_rated_media_items("tv")
    elif info == 'ratedepisodes':
        return tmdb.get_rated_media_items("tv/episodes")
    elif info == 'starredtvshows':
        return tmdb.get_fav_items("tv")
    elif info == 'similarmovies':
        movie_id = params.get("id")
        if not movie_id:
            movie_id = tmdb.get_movie_tmdb_id(imdb_id=params.get("imdb_id"),
                                              dbid=params.get("dbid"))
        if movie_id:
            return tmdb.get_similar_movies(movie_id)
    elif info == 'similartvshows':
        tvshow_id = None
        dbid = params.get("dbid")
        name = params.get("name")
        tmdb_id = params.get("tmdb_id")
        tvdb_id = params.get("tvdb_id")
        imdb_id = params.get("imdb_id")
        if tmdb_id:
            tvshow_id = tmdb_id
        elif dbid and int(dbid) > 0:
            tvdb_id = local_db.get_imdb_id("tvshow", dbid)[0]
            if tvdb_id:
                tvshow_id = tmdb.get_show_tmdb_id(tvdb_id)
        elif tvdb_id:
            tvshow_id = tmdb.get_show_tmdb_id(tvdb_id)
        elif imdb_id:
            tvshow_id = tmdb.get_show_tmdb_id(imdb_id, "imdb_id")
        elif name:
            tvshow_id = tmdb.search_media(media_name=name,
                                          year="",
                                          media_type="tv")
        if tvshow_id:
            return tmdb.get_similar_tvshows(tvshow_id)
    elif info == 'studio':
        if params.get("id"):
            return tmdb.get_company_data(params["id"])
        elif params.get("studio"):
            company_data = tmdb.search_companies(params["studio"])
            if company_data:
                return tmdb.get_company_data(company_data[0]["id"])
    elif info == 'set':
        if params.get("dbid"):
            name = local_db.get_set_name(params["dbid"])
            if name:
                params["setid"] = tmdb.get_set_id(name)
        if params.get("setid"):
            set_data, _ = tmdb.get_set_movies(params["setid"])
            return set_data
    elif info == 'movielists':
        movie_id = params.get("id")
        if not movie_id:
            movie_id = tmdb.get_movie_tmdb_id(imdb_id=params.get("imdb_id"),
                                              dbid=params.get("dbid"))
        if movie_id:
            return tmdb.get_movie_lists(movie_id)
    elif info == 'keywords':
        movie_id = params.get("id")
        if not movie_id:
            movie_id = tmdb.get_movie_tmdb_id(imdb_id=params.get("imdb_id"),
                                              dbid=params.get("dbid"))
        if movie_id:
            return tmdb.get_keywords(movie_id)
    elif info == 'trailers':
        movie_id = params.get("id")
        if not movie_id:
            movie_id = tmdb.get_movie_tmdb_id(imdb_id=params.get("imdb_id"),
                                              dbid=params.get("dbid"))
        if movie_id:
            return tmdb.handle_videos(tmdb.get_movie_videos(movie_id))
    elif info == 'popularpeople':
        return tmdb.get_popular_actors()
    elif info == 'personmovies':
        person = tmdb.get_person_info(person_label=params.get("person"),
                                      skip_dialog=True)
        if person and person.get("id"):
            movies = tmdb.get_person_movies(person["id"])
            if not movies:
                return None
            for item in movies:
                del item["credit_id"]
            return movies.reduce(key="department")
    # Trakt info
    elif info == 'traktsimilarmovies':
        if params.get("id") or params.get("dbid"):
            if params.get("dbid"):
                movie_id = local_db.get_imdb_id("movie", params["dbid"])[0]
            else:
                movie_id = params["id"]
            return trakt.get_similar("movie", movie_id)
    elif info == 'traktsimilartvshows':
        if params.get("id") or params.get("dbid"):
            if params.get("dbid"):
                if params.get("type") == "episode":
                    tvshow_id = local_db.get_tvshow_id_by_episode(
                        params["dbid"])
                else:
                    tvshow_id = local_db.get_imdb_id(media_type="tvshow",
                                                     dbid=params["dbid"])[0]
            else:
                tvshow_id = params["id"]
            return trakt.get_similar("show", tvshow_id)
    elif info == 'airingepisodes':
        return trakt.get_episodes("shows")
    elif info == 'premiereepisodes':
        return trakt.get_episodes("premieres")
    elif info == 'trendingshows':
        return trakt.get_shows("trending")
    elif info == 'popularshows':
        return trakt.get_shows("popular")
    elif info == 'anticipatedshows':
        return trakt.get_shows("anticipated")
    elif info == 'mostcollectedshows':
        return trakt.get_shows_from_time("collected")
    elif info == 'mostplayedshows':
        return trakt.get_shows_from_time("played")
    elif info == 'mostwatchedshows':
        return trakt.get_shows_from_time("watched")
    elif info == 'trendingmovies':
        return trakt.get_movies("trending")
    elif info == 'traktpopularmovies':
        return trakt.get_movies("popular")
    elif info == 'mostplayedmovies':
        return trakt.get_movies_from_time("played")
    elif info == 'mostwatchedmovies':
        return trakt.get_movies_from_time("watched")
    elif info == 'mostcollectedmovies':
        return trakt.get_movies_from_time("collected")
    elif info == 'mostanticipatedmovies':
        return trakt.get_movies("anticipated")
    elif info == 'traktboxofficemovies':
        return trakt.get_movies("boxoffice")
    elif info == 'similarartistsinlibrary':
        return local_db.get_similar_artists(params.get("artist_mbid"))
    # LastFM
    elif info == 'trackinfo':
        addon.clear_global(f'{params.get("prefix", "")}Summary')
        if params["artistname"] and params["trackname"]:
            track_info = lastfm.get_track_info(artist_name=params["artistname"],
                                               track=params["trackname"])
            addon.set_global(f'{params.get("prefix", "")}Summary',
                            track_info["summary"])
    # Bands in town  API no longer provides event access
    #  elif info == 'topartistsnearevents':
    #    artists = local_db.get_artists()
    #    from . import  bandsintown
    #    return  bandsintown.get_near_events(artists[0:49])
    # Youtube
    elif info == 'youtubesearchvideos':
        addon.set_global(f'{params.get("prefix", "")}SearchValue',
                        params.get("id", ""))
        user_key = addon.setting("Youtube API Key")
        if params.get("id"):
            return youtube.search(search_str=params.get("id", ""),
                                  hd=params.get("hd"),
                                  orderby=params.get("orderby", "relevance"),
                                  api_key=user_key)
    elif info == 'youtubeplaylistvideos':
        return youtube.get_playlist_videos(params.get("id", ""))
    elif info == 'youtubeusersearchvideos':
        user_name = params.get("id")
        if user_name:
            playlists = youtube.get_user_playlists(user_name)
            return youtube.get_playlist_videos(playlists["uploads"])
    # Kodi JSON API
    elif info == 'favourites':
        if params.get("id"):
            items = favs.get_favs_by_type(params["id"])
        else:
            items = favs.get_favs()
            addon.set_global('favourite.count', str(len(items)))
            if items:
                addon.set_global('favourite.1.name', items[-1]["label"])
        return items
    elif info == "addonsbyauthor":
        items = favs.get_addons_by_author(params.get("id"))
    elif info == 'similarlocalmovies' and "dbid" in params:
        return local_db.get_similar_movies(params["dbid"])
    elif info == 'iconpanel':
        return favs.get_icon_panel(int(params["id"])), "IconPanel" + str(params["id"])
    # ACTIONS
    if params.get("handle"):
        xbmcplugin.setResolvedUrl(handle=int(params.get("handle")),
                                  succeeded=False,
                                  listitem=xbmcgui.ListItem())
    if info in ['playmovie', 'playepisode', 'playmusicvideo', 'playalbum', 'playsong']:
        kodijson.play_media(media_type=info.replace("play", ""),
                            dbid=params.get("dbid"),
                            resume=params.get("resume", "true"))
    elif info == "openinfodialog":
        # ListItem.DBType is used to determine what dialog to open
        # For Kodi video library items dbid and title are used
        if xbmc.getCondVisibility("System.HasActiveModalDialog"):
            container_id = ""
        else:
            container_id = f'Container({utils.get_infolabel("System.CurrentControlId")})'
        dbid:str = utils.get_infolabel(f'{container_id}ListItem.DBID')
        db_type:str = utils.get_infolabel(f'{container_id}ListItem.DBType')
        if db_type == "movie":
            params = {"dbid": dbid,
                      "id": utils.get_infolabel(f'{container_id}ListItem.Property(id)'),
                      "name": utils.get_infolabel(f'{container_id}ListItem.Title')}
            #utils.log(f'process.start_info_actions for movie call exendedinfo with {params}')
            start_info_actions("extendedinfo", params)
        elif db_type == "tvshow":
            params = {"dbid": dbid,
                      "tvdb_id": utils.get_infolabel(f'{container_id}ListItem.Property(tvdb_id)'),
                      "id": utils.get_infolabel(f'{container_id}ListItem.Property(id)'),
                      "name": utils.get_infolabel(f'{container_id}ListItem.Title')}
            start_info_actions("extendedtvinfo", params)
        elif db_type == "season":
            params = {"tvshow": utils.get_infolabel(f'{container_id}ListItem.TVShowTitle'),
                      "dbid": utils.get_infolabel(f'{container_id}ListItem.DBID'),
                      "season": utils.get_infolabel(f'{container_id}ListItem.Season')}
            start_info_actions("seasoninfo", params)
        elif db_type == "episode":
            params = {"tvshow": utils.get_infolabel(f'{container_id}ListItem.TVShowTitle'),
                      "season": utils.get_infolabel(f'{container_id}ListItem.Season'),
                      "episode": utils.get_infolabel(f'{container_id}ListItem.Episode')}
            start_info_actions("extendedepisodeinfo", params)
        elif db_type in ["actor", "director"]:
            params = {"name": utils.get_infolabel(f'{container_id}ListItem.Label')}
            start_info_actions("extendedactorinfo", params)
        else:
            utils.notify("Error", "Could not find valid content type")
    elif info == "ratedialog":
        if xbmc.getCondVisibility("System.HasModalDialog"):
            container_id = ""
        else:
            container_id = f'Container({utils.get_infolabel("System.CurrentControlId")})'
        dbid = utils.get_infolabel(f'{container_id}ListItem.DBID')
        db_type = utils.get_infolabel(f'{container_id}ListItem.DBType')
        if db_type == "movie":
            params = {"dbid": dbid,
                      "id": utils.get_infolabel(f'{container_id}ListItem.Property(id)'),
                      "type": "movie"}
            start_info_actions("ratemedia", params)
        elif db_type == "tvshow":
            params = {"dbid": dbid,
                      "id": utils.get_infolabel(f'{container_id}ListItem.Property(id)'),
                      "type": "tv"}
            start_info_actions("ratemedia", params)
        if db_type == "episode":
            params = {"tvshow": utils.get_infolabel(f'{container_id}ListItem.TVShowTitle'),
                      "season": utils.get_infolabel(f'{container_id}ListItem.Season'),
                      "type": "episode"}
            start_info_actions("ratemedia", params)
    elif info == 'youtubebrowser':
        wm.open_youtube_list(search_str=params.get("id", ""))
    elif info == 'moviedbbrowser':
        if addon.get_global('infodialogs.active'):
            return None
        addon.set_global('infodialogs.active', "true")
        try:
            search_str = params.get("id", "")
            if not search_str and params.get("search"):
                result = xbmcgui.Dialog().input(heading=addon.LANG(16017),
                                                type=xbmcgui.INPUT_ALPHANUM)
                if result:
                    search_str = result
                else:
                    addon.clear_global('infodialogs.active')
                    return None
            wm.open_video_list(search_str=search_str,
                            mode="search")
        finally:
            addon.clear_global('infodialogs.active')
    elif info == 'extendedinfo': # called with movie id
        if addon.get_global('infodialogs.active'):
            return None
        addon.set_global('infodialogs.active', "true")
        try:
            wm.open_movie_info(movie_id=params.get("id"),
                            dbid=params.get("dbid"),
                            imdb_id=params.get("imdb_id"),
                            name=params.get("name"))
        finally:
            addon.clear_global('infodialogs.active')
    elif info == 'extendedactorinfo':
        if addon.get_global('infodialogs.active'):
            return None
        addon.set_global('infodialogs.active', "true")
        try:
            wm.open_actor_info(actor_id=params.get("id"),
                            name=params.get("name"))
        finally:
            addon.clear_global('infodialogs.active')
    elif info == 'extendedtvinfo':
        if addon.get_global('infodialogs.active'):
            return None
        addon.set_global('infodialogs.active', "true")
        try:
            wm.open_tvshow_info(tmdb_id=params.get("id"),
                                tvdb_id=params.get("tvdb_id"),
                                dbid=params.get("dbid"),
                                imdb_id=params.get("imdb_id"),
                                name=params.get("name"))
        finally:
            addon.clear_global('infodialogs.active')
    elif info == 'seasoninfo':
        if addon.get_global('infodialogs.active'):
            return None
        addon.set_global('infodialogs.active', "true")
        try:
            wm.open_season_info(tvshow=params.get("tvshow"),
                                dbid=params.get("dbid"),
                                season=int(params.get("season")))
        finally:
            addon.clear_global('infodialogs.active')
    elif info == 'extendedepisodeinfo':
        if addon.get_global('infodialogs.active'):
            return None
        addon.set_global('infodialogs.active', "true")
        try:
            wm.open_episode_info(tvshow=params.get("tvshow"),
                                tvshow_id=params.get("tvshow_id"),
                                dbid=params.get("dbid"),
                                episode=int(params.get("episode")),
                                season=int(params.get("season")))
        finally:
            addon.clear_global('infodialogs.active')
    elif info == 'albuminfo':
        if params.get("id"):
            album_details = AudioDB.get_album_details(params.get("id"))
            utils.dict_to_windowprops(album_details, params.get("prefix", ""))
    elif info == 'artistdetails':
        artist_details = AudioDB.get_artist_details(params["artistname"])
        utils.dict_to_windowprops(artist_details, params.get("prefix", ""))
    elif info == 'ratemedia':
        media_type = params.get("type")
        if not media_type:
            return None
        if params.get("id"):
            tmdb_id = params["id"]
        elif media_type == "movie":
            tmdb_id = tmdb.get_movie_tmdb_id(imdb_id=params.get("imdb_id"),
                                             dbid=params.get("dbid"),
                                             name=params.get("name"))
        elif media_type == "tv" and params.get("dbid"):
            tvdb_id = local_db.get_imdb_id(media_type="tvshow",
                                           dbid=params["dbid"])[0]
            tmdb_id = tmdb.get_show_tmdb_id(tvdb_id)
        else:
            return False
        rating = utils.input_userrating()
        if rating == -1:
            return None
        tmdb.set_rating(media_type=media_type,
                        media_id=tmdb_id,
                        rating=rating,
                        dbid=params.get("dbid"))
    elif info == 'action':
        for builtin in params.get("id", "").split("$$"):
            xbmc.executebuiltin(builtin)
    elif info == "youtubevideo":
        xbmc.executebuiltin("Dialog.Close(all,true)")
        wm.play_youtube_video(params.get("id", ""))
    elif info == 'playtrailer':
        busy.show_busy()
        if params.get("id"):
            movie_id = params["id"]
        elif int(params.get("dbid", -1)) > 0:
            movie_id = local_db.get_imdb_id(media_type="movie",
                                            dbid=params["dbid"])[0]
        elif params.get("imdb_id"):
            movie_id = tmdb.get_movie_tmdb_id(params["imdb_id"])
        else:
            movie_id = ""
        if movie_id:
            trailers = tmdb.get_movie_videos(movie_id)
            busy.hide_busy()
            time.sleep(0.1)
            if trailers:
                wm.play_youtube_video(trailers[0]["key"])
            elif params.get("title"):
                wm.open_youtube_list(search_str=params["title"])
            else:
                busy.hide_busy()
    elif info == 'deletecache':
        addon.clear_globals()
        for rel_path in os.listdir(addon.DATA_PATH):
            path = os.path.join(addon.DATA_PATH, rel_path)
            try:
                if os.path.isdir(path):
                    shutil.rmtree(path)
            except Exception as err:
                utils.log(f'Failed to remove cache due to {err}')
        utils.notify("Cache deleted")
    elif info == 'tmdbpassword':
        addon.set_password_prompt("tmdb_password")
    elif info == 'syncwatchlist':
        pass
