Content rating test

Replace episode art with content rating icon

src/EPG.cpp

@@ -220,10 +220,67 @@ PVR_ERROR EPG::GetEPGForChannel(int channelUid, time_t start, time_t end, kodi::
           }
         }
       }
+      rating.clear();
+      XMLUtils::GetString(pListingNode, "rating", rating);
+      SetContentRating(broadcast, rating);
       results.Add(broadcast);
     }
     return PVR_ERROR_NO_ERROR;
   }
 
   return PVR_ERROR_NO_ERROR;
 }
+
+void EPG::SetContentRating(kodi::addon::PVREPGTag& broadcast, std::string contentRating)
+{
+  const std::string contentDirectory = "special://home/addons/pvr.nextpvr/resources/content_ratings";
+  // std::string checkname = kodi::vfs::TranslateSpecialProtocol(content);
+  if (!contentRating.empty() && kodi::vfs::DirectoryExists(contentDirectory))
+  {
+    broadcast.SetParentalRatingCode(contentRating);
+    std::regex base_regex("([1-9]\\d?)\\+?$");
+    std::smatch base_match;
+    if (std::regex_match(contentRating, base_match, base_regex))
+    {
+      int code = std::stoi(base_match[1].str());
+      if (code < 20)
+        broadcast.SetParentalRating(code);
+    }
+    std::map<std::string, std::string>::iterator lookupRating = m_contentRatings.find(contentRating);
+    if ( lookupRating == m_contentRatings.end())
+    {
+      std::list <std::string> regions{ "US", "MPAA", "AU", "UK", "NL", "NO","CA", "DE", "NZ", "FR"};
+      std::string parsedRating = contentRating;
+      kodi::tools::StringUtils::Trim(parsedRating);
+      kodi::tools::StringUtils::ToUpper(parsedRating);
+      if (parsedRating[0] != '-')
+        kodi::tools::StringUtils::Replace(parsedRating, "-", "");
+      kodi::tools::StringUtils::Replace(parsedRating, " ", "");
+
+      // get from a setting
+      std::string priorityRegion = "CA";
+      regions.emplace_front(priorityRegion);
+
+      int counter = 0;
+      for (std::string region : regions)
+      {
+        if (counter++ > 0 && region == priorityRegion)
+          continue;
+        std::string ratingName = kodi::tools::StringUtils::Format("%s/%s/%s.png", contentDirectory.c_str(), region.c_str(), parsedRating.c_str());
+        if (kodi::vfs::FileExists(ratingName))
+        {
+          m_contentRatings.emplace(contentRating, ratingName);
+          broadcast.SetIconPath(ratingName);
+          return;
+        }
+      }
+      kodi::Log(ADDON_LOG_DEBUG, "Content rating not found %s|%s",parsedRating.c_str(), contentRating.c_str());
+      // Optionally set a default
+    }
+    else
+    {
+      broadcast.SetIconPath(lookupRating->second);
+      return;
+    }
+  }
+}

src/EPG.h

@@ -30,5 +30,7 @@ namespace NextPVR
     Request& m_request;
     Recordings& m_recordings;
     Channels& m_channels;
+    void SetContentRating(kodi::addon::PVREPGTag& broadcast, std::string rating);
+    std::map<std::string, std::string> m_contentRatings;
   };
 } // namespace NextPVR