[ 'url_pattern' => '/(?:https?:\/\/)?(?:www\.)?imdb\.com\/title\/(tt\d+)\/?/', 'home_page' => 'https://www.imdb.com/', 'rating_average_img' => 'pic/imdb2.png', 'rating_pattern_in_desc' => "/IMDb评分.*([\d\.]+)\//isU", ], self::SITE_DOUBAN => [ 'url_pattern' => '/(?:https?:\/\/)?(?:(?:movie|www)\.)?douban\.com\/(?:subject|movie)\/(\d+)\/?/', 'home_page' => 'https://www.douban.com/', 'rating_average_img' => 'pic/douban2.png', 'rating_pattern_in_desc' => "/豆瓣评分.*([\d\.]+)\//isU", ], self::SITE_BANGUMI => [ 'url_pattern' => '/(?:https?:\/\/)?(?:bgm\.tv|bangumi\.tv|chii\.in)\/subject\/(\d+)\/?/', 'home_page' => 'https://bangumi.tv/', 'rating_average_img' => 'pic/bangumi.jpg', ], ]; public function __construct() { $setting = get_setting('main'); $this->setApiPoint($setting['pt_gen_api_point'] ?? ''); } public function getApiPoint(): string { return $this->apiPoint; } public function setApiPoint(string $apiPoint) { $this->apiPoint = trim($apiPoint); } public function generate(string $url, bool $withoutCache = false): array { $parsed = $this->parse($url); $targetUrl = trim($this->apiPoint, '/'); if (Str::contains($targetUrl, '?')) { $targetUrl .= "&"; } else { $targetUrl .= "?"; } $targetUrl .= sprintf('site=%s&sid=%s&url=%s', $parsed['site'] , $parsed['id'], urlencode($parsed['url'])); return $this->request($targetUrl, $withoutCache); } public function parse(string $url): array { foreach (self::$validSites as $site => $info) { if (preg_match($info['url_pattern'], $url, $matches)) { return [ 'site' => $site, 'url' => $matches[0], 'id' => $matches[1] ]; } } throw new PTGenException("invalid url: $url"); } private function buildDetailsPageTableRow($torrentId, $ptGenArr, $site): string { global $lang_details; if ($this->isRawPTGen($ptGenArr)) { $ptGenFormatted = $ptGenArr['format']; } elseif ($this->isIyuu($ptGenArr)) { $ptGenFormatted = $ptGenArr['data']['format']; } else { do_log("Invalid pt gen data", 'error'); return ''; } $poster = ''; if (!empty($ptGenArr['poster'])) { $poster = $ptGenArr['poster']; } elseif (preg_match('/\[img\](.*)\[\/img\]/iU', $ptGenFormatted, $matches)) { $poster = $matches[1]; } if ($poster) { $prefix = sprintf("[img]%s[/img]\n", $poster); $ptGenFormatted = mb_substr($ptGenFormatted, mb_strlen($prefix, 'utf-8') + 1); } $ptGenFormatted = format_comment($ptGenFormatted); $ptGenFormatted .= sprintf( '%s %s%s%s', $lang_details['text_information_updated_at'], $ptGenArr['__updated_at'], $lang_details['text_might_be_outdated'], $torrentId, $site, $lang_details['text_here_to_update'] ); $titleShowOrHide = $lang_details['title_show_or_hide'] ?? ''; $id = 'pt-gen-' . $site; $posterHtml = ""; if ($poster) { $posterHtml = sprintf('
Show/Hide PT-Gen-{$site} $posterHtml
{$ptGenFormatted}
HTML; return $html; } private function request(string $url, bool $withoutCache = false): array { global $Cache; $begin = microtime(true); $logPrefix = "url: $url"; $cacheKey = $this->getApiPointResultCacheKey($url); if (!$withoutCache) { $cache = $Cache->get_value($cacheKey); if ($cache) { do_log("$logPrefix, from cache"); return $cache; } } do_log("$logPrefix, going to send request..."); $http = new Client(); $response = $http->get($url, ['timeout' => 5]); $statusCode = $response->getStatusCode(); if ($statusCode != 200) { $msg = "api point response http status code: $statusCode"; do_log("$logPrefix, $msg"); throw new PTGenException($msg); } $bodyString = (string)$response->getBody(); if (empty($bodyString)) { $msg = "response body empty"; do_log("$logPrefix, $msg"); throw new PTGenException($msg); } $bodyArr = json_decode($bodyString, true); if (empty($bodyArr) || !is_array($bodyArr)) { $msg = "response body error: $bodyString"; do_log("$logPrefix, $msg"); throw new PTGenException($msg); } if ($this->isRawPTGen($bodyArr) || $this->isIyuu($bodyArr)) { $Cache->cache_value($cacheKey, $bodyArr, 24 * 3600); do_log("$logPrefix, success get from api point, use time: " . (microtime(true) - $begin)); $bodyArr['__updated_at'] = now()->toDateTimeString(); return $bodyArr; } else { $msg = "error: " . $bodyArr['error'] ?? ''; do_log("$logPrefix, response: $bodyString"); throw new PTGenException($msg); } } public function deleteApiPointResultCache($url) { global $Cache; $Cache->delete_value($this->getApiPointResultCacheKey($url)); } private function getApiPointResultCacheKey($url) { return __METHOD__ . "_$url"; } public function renderUploadPageFormInput($ptGen = '') { global $lang_functions; $html = ''; $ptGen = (array)json_decode($ptGen, true); foreach (self::$validSites as $site => $info) { $value = $ptGen[$site]['link'] ?? ''; $x = $lang_functions["row_pt_gen_{$site}_url"]; $y = $this->buildInput("pt_gen[{$site}][link]", $value, $lang_functions["text_pt_gen_{$site}_url_note"], $lang_functions['pt_gen_get_description']); $html .= tr($x, $y, 1); } return $html; } public function buildInput($name, $value, $note, $btnText): string { $btn = ''; if ($this->apiPoint != '') { $btn = '
'; } $input = <<
{$note}
$btn
HTML; return $input; } public function renderDetailsPageDescription($torrentId, $torrentPtGenArr): array { $html = ''; $jsonArr = []; $update = false; $torrentPtGenArr = (array)$torrentPtGenArr; foreach (self::$validSites as $site => $info) { if (empty($torrentPtGenArr[$site]['link'])) { continue; } $link = $torrentPtGenArr[$site]['link']; $data = $torrentPtGenArr[$site]['data'] ?? []; if (!empty($data)) { $jsonArr[$site] = [ 'link' => $link, 'data' => $data, ]; $html .= $this->buildDetailsPageTableRow($torrentId, $data, $site); } else { try { $ptGenArr = $this->generate($torrentPtGenArr[$site]['link']); } catch (\Exception $e) { $log = $e->getMessage() . ", trace: " . $e->getTraceAsString(); do_log($log,'error'); $ptGenArr = [ 'format' => $e->getMessage() ]; } $jsonArr[$site] = [ 'link' => $link, 'data' => $ptGenArr, ]; $html .= $this->buildDetailsPageTableRow($torrentId, $ptGenArr, $site); if (!$update) { $update = true; } } } return ['json_arr' => $jsonArr, 'html' => $html, 'update' => $update]; } public function buildRatingSpan(array $siteIdAndRating): string { $result = '
'; $count = 1; $ratingIcons = []; foreach (self::$validSites as $site => $info) { if (!isset($siteIdAndRating[$site])) { continue; } $rating = $siteIdAndRating[$site]; if (empty($rating)) { continue; } if ($count > 2) { //only show the first two break; } $ratingIcons[] = $this->getRatingIcon($site, $rating); $count++; } if (empty($ratingIcons)) { $ratingIcons[] = $this->getRatingIcon(self::SITE_IMDB, 'N/A'); $ratingIcons[] = $this->getRatingIcon(self::SITE_DOUBAN, 'N/A'); } $result .= implode("", $ratingIcons) . '
'; return $result; } public function getRatingIcon($siteId, $rating): string { if (is_numeric($rating)) { $rating = number_format($rating, 1); } $result = sprintf( '
%s%s
', self::$validSites[$siteId]['rating_average_img'], $siteId, $siteId, $rating ); return $result; } public function isRawPTGen(array $bodyArr): bool { return isset($bodyArr['success']) && $bodyArr['success']; } public function isIyuu(array $bodyArr): bool { return isset($bodyArr['ret']) && $bodyArr['ret'] == 200; } public function listRatings(array $ptGenData, string $imdbLink, string $desc = ''): array { $results = []; $log = ""; //First, get from PTGen foreach (self::$validSites as $site => $info) { if (!isset($ptGenData[$site]['data'])) { continue; } $data = $ptGenData[$site]['data']; $log .= ", handling site: $site"; $rating = ''; if (isset($data['__rating'])) { //__rating is new add $rating = $data['__rating']; $log .= ", from __rating"; } else { // from original structure fetch if ($this->isRawPTGen($data)) { $log .= ", isRawPTGen"; $rating = $data["{$site}_rating_average"] ?? ''; } elseif ($this->isIyuu($data)) { $log .= ", isIyuu"; $pattern = $info['rating_pattern_in_desc'] ?? null; if ($pattern && preg_match($pattern,$data['data']['format'], $matches)) { $rating = $matches[1]; } } } if (!empty($rating)) { $results[$site] = $rating; $log .= ", get rating: $rating"; } else { $log .= ", can't get rating"; } } //Second, imdb can get from imdb api if (!isset($results[self::SITE_IMDB]) && !empty($imdbLink)) { $imdb = new Imdb(); $imdbRating = $imdb->getRating($imdbLink); $results[self::SITE_IMDB] = $imdbRating; $log .= ", again 'imdb' from: $imdbLink} -> $imdbRating"; } //Otherwise, get from desc if (!empty($desc)) { foreach (self::$validSites as $site => $info) { if (isset($results[$site])) { continue; } if (empty($info['rating_pattern_in_desc'])) { continue; } $pattern = $info['rating_pattern_in_desc']; $log .= ", at last, trying to get '$site' from desc with pattern: $pattern"; if (preg_match($pattern, $desc, $matches)) { $log .= ", get " . $matches[1]; $results[$site] = $matches[1]; } else { $log .= ", not match"; } } } do_log($log); return $results; } public function updateTorrentPtGen(array $torrentInfo, $siteId = null) { $ptGenInfo = json_decode($torrentInfo['pt_gen'], true); foreach (self::$validSites as $site => $siteConfig) { if ($siteId !== null && $siteId != $site) { //If specific, only update it continue; } if (empty($ptGenInfo[$site]['link'])) { do_log("site: $site no link..."); continue; } try { $response = $this->generate($ptGenInfo[$site]['link'], true); $ptGenInfo[$site]['data'] = $response; } catch (\Exception $exception) { do_log("site: $site can not be updated: " . $exception->getMessage(), 'error'); } } $siteIdAndRating = $this->listRatings($ptGenInfo, $torrentInfo['url'], $torrentInfo['descr']); foreach ($siteIdAndRating as $key => $value) { $ptGenInfo[$key]['data']["__rating"] = $value; } Torrent::query()->where('id', $torrentInfo['id'])->update(['pt_gen' => $ptGenInfo]); return $ptGenInfo; } }