mirror of
https://github.com/NNTmux/newznab-tmux.git
synced 2026-08-30 01:38:54 +00:00
DariusIII
361 lines
8.0 KiB
PHP
361 lines
8.0 KiB
PHP
<?php
|
|
/**
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program (see LICENSE.txt in the base directory. If
|
|
* not, see:
|
|
*
|
|
* @link <http://www.gnu.org/licenses/>.
|
|
* @author niel
|
|
* @copyright 2015 nZEDb
|
|
*/
|
|
namespace newznab\processing;
|
|
|
|
use newznab\db\Settings;
|
|
|
|
/**
|
|
* Parent class for TV/Film and any similar classes to inherit from.
|
|
*
|
|
* @package newznab\processing
|
|
*/
|
|
abstract class Videos
|
|
{
|
|
// Video Type Identifiers
|
|
const TYPE_TV = 0; // Type of video is a TV Programme/Show
|
|
const TYPE_FILM = 1; // Type of video is a Film/Movie
|
|
const TYPE_ANIME = 2; // Type of video is a Anime
|
|
|
|
/**
|
|
* @var \newznab\db\Settings
|
|
*/
|
|
public $pdo;
|
|
|
|
/**
|
|
* @var array sites The sites that we have an ID columns for in our video table.
|
|
*/
|
|
private $sites = ['imdb', 'tmdb', 'trakt', 'tvdb', 'tvmaze', 'tvrage'];
|
|
|
|
|
|
public function __construct(array $options = [])
|
|
{
|
|
$defaults = [
|
|
'Echo' => false,
|
|
'Settings' => null,
|
|
];
|
|
$options += $defaults;
|
|
|
|
// Sets the default timezone for this script (and its children).
|
|
date_default_timezone_set('UTC');
|
|
|
|
$this->pdo = ($options['Settings'] instanceof Settings ? $options['Settings'] : new Settings());
|
|
}
|
|
|
|
/**
|
|
* Get video info from a Site ID and column.
|
|
*
|
|
* @param string $siteColumn
|
|
* @param integer $videoID
|
|
*
|
|
* @return array|false False if invalid site, or ID not found; Site id value otherwise.
|
|
*/
|
|
protected function getSiteIDFromVideoID($siteColumn, $videoID)
|
|
{
|
|
if (in_array($siteColumn, $this->sites)) {
|
|
$result = $this->pdo->queryOneRow("SELECT $siteColumn FROM videos WHERE id = $videoID");
|
|
|
|
return isset($result[$siteColumn]) ? $result[$siteColumn] : false;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Get video info from a Site ID and column.
|
|
*
|
|
* @param string $siteColumn
|
|
* @param integer $siteID
|
|
*
|
|
* @return array|false False if invalid site, or ID not found; video.id value otherwise.
|
|
*/
|
|
protected function getVideoIDFromSiteID($siteColumn, $siteID)
|
|
{
|
|
if (in_array($siteColumn, $this->sites)) {
|
|
$result = $this->pdo->queryOneRow("SELECT id FROM videos WHERE $siteColumn = $siteID");
|
|
|
|
return isset($result['id']) ? $result['id'] : false;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Attempt a local lookup via the title first by exact match and then by like.
|
|
* Returns a false for no match or the Video ID of the match.
|
|
*
|
|
* @param $title
|
|
* @param $type
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function getByTitle($title, $type)
|
|
{
|
|
// Check if we already have an entry for this show.
|
|
$res = $this->getByTitleQuery($title, $type);
|
|
if (isset($res['id'])) {
|
|
return $res['id'];
|
|
}
|
|
|
|
$title2 = str_replace(' and ', ' & ', $title);
|
|
if ($title != $title2) {
|
|
$res = $this->getByTitleQuery($title2, $type);
|
|
if (isset($res['id'])) {
|
|
return $res['id'];
|
|
}
|
|
$pieces = explode(' ', $title2);
|
|
$title4 = '%';
|
|
foreach ($pieces as $piece) {
|
|
$title4 .= str_replace(["'", "!"], "", $piece) . '%';
|
|
}
|
|
$res = $this->getByTitleLikeQuery($title4, $type);
|
|
if (isset($res['id'])) {
|
|
return $res['id'];
|
|
}
|
|
}
|
|
|
|
// Some words are spelled correctly 2 ways
|
|
// example theatre and theater
|
|
$title3 = str_replace('er', 're', $title);
|
|
if ($title != $title3) {
|
|
$res = $this->getByTitleQuery($title3, $type);
|
|
if (isset($res['id'])) {
|
|
return $res['id'];
|
|
}
|
|
$pieces = explode(' ', $title3);
|
|
$title4 = '%';
|
|
foreach ($pieces as $piece) {
|
|
$title4 .= str_replace(["'", "!"], "", $piece) . '%';
|
|
}
|
|
$res = $this->getByTitleLikeQuery($title4, $type);
|
|
if (isset($res['id'])) {
|
|
return $res['id'];
|
|
}
|
|
}
|
|
|
|
// If there was not an exact title match, look for title with missing chars
|
|
// example release name :Zorro 1990, tvrage name Zorro (1990)
|
|
// Only search if the title contains more than one word to prevent incorrect matches
|
|
$pieces = explode(' ', $title);
|
|
if (count($pieces) > 1) {
|
|
$title4 = '%';
|
|
foreach ($pieces as $piece) {
|
|
$title4 .= str_replace(["'", "!"], "", $piece) . '%';
|
|
}
|
|
$res = $this->getByTitleLikeQuery($title4, $type);
|
|
if (isset($res['id'])) {
|
|
return $res['id'];
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Supplementary function for getByTitle that queries for exact match
|
|
*
|
|
* @param $title
|
|
* @param $type
|
|
*
|
|
* @return array|bool
|
|
*/
|
|
public function getByTitleQuery($title, $type)
|
|
{
|
|
$return = false;
|
|
if ($title) {
|
|
$return = $this->pdo->queryOneRow(
|
|
sprintf("
|
|
SELECT v.id
|
|
FROM videos v
|
|
LEFT JOIN videos_akas va ON v.id = va.videos_id
|
|
WHERE (v.title = %1\$s OR va.title = %1\$s)
|
|
AND v.type = %d",
|
|
$this->pdo->escapeString($title),
|
|
$type
|
|
)
|
|
);
|
|
}
|
|
return $return;
|
|
}
|
|
|
|
/**
|
|
* Supplementary function for getByTitle that queries for a like match
|
|
*
|
|
* @param $title
|
|
* @param $type
|
|
*
|
|
* @return array|bool
|
|
*/
|
|
public function getByTitleLikeQuery($title, $type)
|
|
{
|
|
$return = false;
|
|
$string = '"\'"';
|
|
if ($title) {
|
|
$return = $this->pdo->queryOneRow(
|
|
sprintf("
|
|
SELECT id
|
|
FROM videos
|
|
WHERE REPLACE(REPLACE(title, %s, ''), '!', '') %s
|
|
AND type = %d",
|
|
$string,
|
|
$this->pdo->likeString(rtrim($title, '%'), false, false),
|
|
$type
|
|
)
|
|
);
|
|
}
|
|
return $return;
|
|
}
|
|
|
|
/**
|
|
* Inserts aliases for videos
|
|
*
|
|
* @param $videoId
|
|
* @param array $aliasArr
|
|
*/
|
|
public function addAliases($videoId, $aliasArr = array())
|
|
{
|
|
if (!empty($aliasArr) && $videoId > 0) {
|
|
foreach ($aliasArr AS $key => $title) {
|
|
// Check if we have the AKA already
|
|
$check = $this->getAliases(0, $title);
|
|
|
|
if ($check === false) {
|
|
$this->pdo->queryInsert(
|
|
sprintf('
|
|
INSERT INTO videos_akas
|
|
(videos_id, title)
|
|
VALUES (%d, %s)',
|
|
$videoId,
|
|
$this->pdo->escapeString($title)
|
|
)
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Retrieves all aliases for given VideoID or VideoID for a given alias
|
|
*
|
|
* @param int $videoId
|
|
* @param string $alias
|
|
*
|
|
* @return bool|\PDOStatement
|
|
*/
|
|
public function getAliases($videoId = 0, $alias = '')
|
|
{
|
|
$return = false;
|
|
$sql = '';
|
|
|
|
if ($videoId > 0) {
|
|
$sql = 'videos_id = ' . $videoId;
|
|
} elseif ($alias !== '') {
|
|
$sql = 'title = ' . $this->pdo->escapeString($alias);
|
|
}
|
|
|
|
if ($sql !== '') {
|
|
$return = $this->pdo->query('
|
|
SELECT *
|
|
FROM videos_akas
|
|
WHERE ' . $sql
|
|
);
|
|
}
|
|
return $return;
|
|
}
|
|
|
|
/**
|
|
* This function turns a roman numeral into an integer
|
|
*
|
|
* @param string $string
|
|
*
|
|
* @return int $e
|
|
*/
|
|
public function convertRomanToInt($string) {
|
|
switch ($string) {
|
|
case 'i': $e = 1;
|
|
break;
|
|
case 'ii': $e = 2;
|
|
break;
|
|
case 'iii': $e = 3;
|
|
break;
|
|
case 'iv': $e = 4;
|
|
break;
|
|
case 'v': $e = 5;
|
|
break;
|
|
case 'vi': $e = 6;
|
|
break;
|
|
case 'vii': $e = 7;
|
|
break;
|
|
case 'viii': $e = 8;
|
|
break;
|
|
case 'ix': $e = 9;
|
|
break;
|
|
case 'x': $e = 10;
|
|
break;
|
|
case 'xi': $e = 11;
|
|
break;
|
|
case 'xii': $e = 12;
|
|
break;
|
|
case 'xiii': $e = 13;
|
|
break;
|
|
case 'xiv': $e = 14;
|
|
break;
|
|
case 'xv': $e = 15;
|
|
break;
|
|
case 'xvi': $e = 16;
|
|
break;
|
|
case 'xvii': $e = 17;
|
|
break;
|
|
case 'xviii': $e = 18;
|
|
break;
|
|
case 'xix': $e = 19;
|
|
break;
|
|
case 'xx': $e = 20;
|
|
break;
|
|
default:
|
|
$e = 0;
|
|
}
|
|
return $e;
|
|
}
|
|
|
|
/**
|
|
* Get a country code for a country name.
|
|
*
|
|
* @param string $country
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function countryCode($country)
|
|
{
|
|
if (!is_array($country) && strlen($country) > 2) {
|
|
$code = $this->pdo->queryOneRow(
|
|
sprintf('
|
|
SELECT id
|
|
FROM countries
|
|
WHERE country = %s',
|
|
$this->pdo->escapeString($country)
|
|
)
|
|
);
|
|
if (isset($code['id'])) {
|
|
return $code['id'];
|
|
}
|
|
}
|
|
return '';
|
|
}
|
|
}
|