xbmc/music/infoscanner/MusicInfoScraper.cpp

   1 /*
   2  *  Copyright (C) 2005-2018 Team Kodi
   3  *  This file is part of Kodi - https://kodi.tv
   4  *
   5  *  SPDX-License-Identifier: GPL-2.0-or-later
   6  *  See LICENSES/README.md for more information.
   7  */
   8
   9 #include "MusicInfoScraper.h"
  10
  11 #include "filesystem/CurlFile.h"
  12 #include "utils/log.h"
  13
  14 using namespace MUSIC_GRABBER;
  15 using namespace ADDON;
  16 using namespace std::chrono_literals;
  17
  18 CMusicInfoScraper::CMusicInfoScraper(const ADDON::ScraperPtr& scraper)
  19   : CThread("MusicInfoScraper"), m_scraper(scraper)
  20 {
  21   m_bSucceeded=false;
  22   m_bCanceled=false;
  23   m_iAlbum=-1;
  24   m_iArtist=-1;
  25   m_http = new XFILE::CCurlFile;
  26 }
  27
  28 CMusicInfoScraper::~CMusicInfoScraper(void)
  29 {
  30   StopThread();
  31   delete m_http;
  32 }
  33
  34 int CMusicInfoScraper::GetAlbumCount() const
  35 {
  36   return (int)m_vecAlbums.size();
  37 }
  38
  39 int CMusicInfoScraper::GetArtistCount() const
  40 {
  41   return (int)m_vecArtists.size();
  42 }
  43
  44 CMusicAlbumInfo& CMusicInfoScraper::GetAlbum(int iAlbum)
  45 {
  46   return m_vecAlbums[iAlbum];
  47 }
  48
  49 CMusicArtistInfo& CMusicInfoScraper::GetArtist(int iArtist)
  50 {
  51   return m_vecArtists[iArtist];
  52 }
  53
  54 void CMusicInfoScraper::FindAlbumInfo(const std::string& strAlbum, const std::string& strArtist /* = "" */)
  55 {
  56   m_strAlbum=strAlbum;
  57   m_strArtist=strArtist;
  58   m_bSucceeded=false;
  59   StopThread();
  60   Create();
  61 }
  62
  63 void CMusicInfoScraper::FindArtistInfo(const std::string& strArtist)
  64 {
  65   m_strArtist=strArtist;
  66   m_bSucceeded=false;
  67   StopThread();
  68   Create();
  69 }
  70
  71 void CMusicInfoScraper::FindAlbumInfo()
  72 {
  73   m_vecAlbums = m_scraper->FindAlbum(*m_http, m_strAlbum, m_strArtist);
  74   m_bSucceeded = !m_vecAlbums.empty();
  75 }
  76
  77 void CMusicInfoScraper::FindArtistInfo()
  78 {
  79   m_vecArtists = m_scraper->FindArtist(*m_http, m_strArtist);
  80   m_bSucceeded = !m_vecArtists.empty();
  81 }
  82
  83 void CMusicInfoScraper::LoadAlbumInfo(int iAlbum)
  84 {
  85   m_iAlbum=iAlbum;
  86   m_iArtist=-1;
  87   StopThread();
  88   Create();
  89 }
  90
  91 void CMusicInfoScraper::LoadArtistInfo(int iArtist, const std::string &strSearch)
  92 {
  93   m_iAlbum=-1;
  94   m_iArtist=iArtist;
  95   m_strSearch=strSearch;
  96   StopThread();
  97   Create();
  98 }
  99
 100 void CMusicInfoScraper::LoadAlbumInfo()
 101 {
 102   if (m_iAlbum<0 || m_iAlbum>=(int)m_vecAlbums.size())
 103     return;
 104
 105   CMusicAlbumInfo& album=m_vecAlbums[m_iAlbum];
 106   // Clear album artist credits
 107   album.GetAlbum().artistCredits.clear();
 108   if (album.Load(*m_http,m_scraper))
 109     m_bSucceeded=true;
 110 }
 111
 112 void CMusicInfoScraper::LoadArtistInfo()
 113 {
 114   if (m_iArtist<0 || m_iArtist>=(int)m_vecArtists.size())
 115     return;
 116
 117   CMusicArtistInfo& artist=m_vecArtists[m_iArtist];
 118   artist.GetArtist().strArtist.clear();
 119   if (artist.Load(*m_http,m_scraper,m_strSearch))
 120     m_bSucceeded=true;
 121 }
 122
 123 bool CMusicInfoScraper::Completed()
 124 {
 125   return Join(10ms);
 126 }
 127
 128 bool CMusicInfoScraper::Succeeded()
 129 {
 130   return !m_bCanceled && m_bSucceeded;
 131 }
 132
 133 void CMusicInfoScraper::Cancel()
 134 {
 135   m_http->Cancel();
 136   m_bCanceled=true;
 137   m_http->Reset();
 138 }
 139
 140 bool CMusicInfoScraper::IsCanceled()
 141 {
 142   return m_bCanceled;
 143 }
 144
 145 void CMusicInfoScraper::OnStartup()
 146 {
 147   m_bSucceeded=false;
 148   m_bCanceled=false;
 149 }
 150
 151 void CMusicInfoScraper::Process()
 152 {
 153   try
 154   {
 155     if (m_strAlbum.size())
 156     {
 157       FindAlbumInfo();
 158       m_strAlbum.clear();
 159       m_strArtist.clear();
 160     }
 161     else if (m_strArtist.size())
 162     {
 163       FindArtistInfo();
 164       m_strArtist.clear();
 165     }
 166     if (m_iAlbum>-1)
 167     {
 168       LoadAlbumInfo();
 169       m_iAlbum=-1;
 170     }
 171     if (m_iArtist>-1)
 172     {
 173       LoadArtistInfo();
 174       m_iArtist=-1;
 175     }
 176   }
 177   catch(...)
 178   {
 179     CLog::Log(LOGERROR, "Exception in CMusicInfoScraper::Process()");
 180   }
 181 }
 182
 183 bool CMusicInfoScraper::CheckValidOrFallback(const std::string &fallbackScraper)
 184 {
 185   return true;
 186 //! @todo Handle fallback mechanism
 187   /*
 188   if (m_scraper->Path() != fallbackScraper &&
 189       parser.Load("special://xbmc/system/scrapers/music/" + fallbackScraper))
 190   {
 191     CLog::Log(LOGWARNING, "{} - scraper {} fails to load, falling back to {}", __FUNCTION__, m_info.strPath, fallbackScraper);
 192     m_info.strPath = fallbackScraper;
 193     m_info.strContent = "albums";
 194     m_info.strTitle = parser.GetName();
 195     m_info.strDate = parser.GetDate();
 196     m_info.strFramework = parser.GetFramework();
 197     m_info.strLanguage = parser.GetLanguage();
 198     m_info.settings.LoadSettingsXML("special://xbmc/system/scrapers/music/" + m_info.strPath);
 199     return true;
 200   }
 201   return false; */
 202 }