mirror of
https://gitee.com/ledc/IYUUAutoReseed
synced 2025-04-26 04:13:25 +00:00
368 lines
14 KiB
PHP
368 lines
14 KiB
PHP
<?php
|
||
/**
|
||
* Rpc操作类
|
||
*/
|
||
class Rpc
|
||
{
|
||
/**
|
||
* 版本号
|
||
* @var string
|
||
*/
|
||
const VER = '0.0.1';
|
||
|
||
// 下载种子的请求类型 GET POST
|
||
public static $method = 'GET';
|
||
// RPC连接池
|
||
public static $links = array();
|
||
/**
|
||
* cookie
|
||
*/
|
||
public static $cookies = '';
|
||
/**
|
||
* 浏览器 User-Agent
|
||
*/
|
||
public static $userAgent = '';
|
||
/**
|
||
* passkey
|
||
*/
|
||
public static $passkey = '';
|
||
/**
|
||
* 客户端配置
|
||
*/
|
||
public static $clients = '';
|
||
/**
|
||
* 监控目录
|
||
*/
|
||
public static $watch = '';
|
||
/**
|
||
* 种子存放路径
|
||
*/
|
||
public static $torrentDir = '';
|
||
/**
|
||
* 工作模式
|
||
*/
|
||
public static $workingMode = '';
|
||
// 站点标识
|
||
public static $site = '';
|
||
/**
|
||
* 负载均衡 控制变量
|
||
*/
|
||
public static $RPC_Key = 0;
|
||
/**
|
||
* 退出状态码
|
||
*/
|
||
public static $ExitCode = 0;
|
||
/**
|
||
* 初始化
|
||
*/
|
||
public static function init($site = '', $method = 'GET')
|
||
{
|
||
global $configALL;
|
||
|
||
self::$site = $site;
|
||
self::$method = strtoupper($method);
|
||
|
||
$config = $configALL[$site];
|
||
self::$cookies = $config['cookie'];
|
||
self::$userAgent = isset($config['userAgent']) && $config['userAgent'] ? $config['userAgent'] : $configALL['default']['userAgent'];
|
||
self::$clients = isset($config['clients']) && $config['clients'] ? $config['clients'] : $configALL['default']['clients'];
|
||
self::$workingMode = isset($config['workingMode']) && $config['workingMode'] ? $config['workingMode'] : 0;
|
||
$watch = isset($config['watch']) && $config['watch'] ? $config['watch'] : $configALL['default']['watch'];
|
||
self::$watch = rtrim($watch, '/') . DS;
|
||
self::$torrentDir = TORRENT_PATH . $site . DS;
|
||
// 建立目录
|
||
IFile::mkdir(self::$torrentDir);
|
||
|
||
self::links();
|
||
}
|
||
|
||
/**
|
||
* 连接远端RPC服务器
|
||
*
|
||
* @param string
|
||
* @return array
|
||
*/
|
||
public static function links()
|
||
{
|
||
if (self::$workingMode === 1 && empty(self::$links)) {
|
||
foreach (self::$clients as $k => $v) {
|
||
// 跳过未配置的客户端
|
||
if (empty($v['username']) || empty($v['password'])) {
|
||
unset(self::$clients[$k]);
|
||
echo "clients_".$k." 用户名或密码未配置，已跳过 \n\n";
|
||
continue;
|
||
}
|
||
try {
|
||
switch ($v['type']) {
|
||
case 'transmission':
|
||
self::$links[$k]['rpc'] = new TransmissionRPC($v['host'], $v['username'], $v['password']);
|
||
$result = self::$links[$k]['rpc']->sstats();
|
||
print $v['type'].'：'.$v['host']." Rpc连接成功 [{$result->result}] \n";
|
||
break;
|
||
case 'qBittorrent':
|
||
self::$links[$k]['rpc'] = new qBittorrent($v['host'], $v['username'], $v['password']);
|
||
$result = self::$links[$k]['rpc']->appVersion();
|
||
print $v['type'].'：'.$v['host']." Rpc连接成功 [{$result}] \n";
|
||
break;
|
||
case 'uTorrent':
|
||
self::$links[$k]['rpc'] = new uTorrent($v['host'], $v['username'], $v['password']);
|
||
$result = self::$links[$k]['rpc']->getBuild();
|
||
print $v['type'].'：'.$v['host']." Rpc连接 [{$result}] \n";
|
||
break;
|
||
default:
|
||
echo '[ERROR] '.$v['type'];
|
||
exit(1);
|
||
break;
|
||
}
|
||
self::$links[$k]['type'] = $v['type'];
|
||
self::$links[$k]['downloadDir'] = $v['downloadDir'];
|
||
} catch (Exception $e) {
|
||
echo '[ERROR] ' . $e->getMessage() . PHP_EOL;
|
||
exit(1);
|
||
}
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* @brief 添加下载任务
|
||
* @param string $torrent 种子元数据
|
||
* @param string $save_path 保存路径
|
||
* @return bool
|
||
*/
|
||
public static function add($torrent, $save_path = '', $extra_options = array())
|
||
{
|
||
switch ((int)self::$workingMode) {
|
||
case 0: // watch默认工作模式
|
||
// 复制到watch目录
|
||
copy($torrent, $save_path);
|
||
if (is_file($save_path)) {
|
||
print "********watch模式，下载任务添加成功 \n\n";
|
||
return true;
|
||
} else {
|
||
print "-----watch模式，下载任务添加失败!!! \n\n";
|
||
}
|
||
break;
|
||
case 1: //负载均衡模式
|
||
try {
|
||
$is_url = false;
|
||
if ((strpos($torrent, 'http://')===0) || (strpos($torrent, 'https://')===0) || (strpos($torrent, 'magnet:?xt=urn:btih:')===0)) {
|
||
$is_url = true;
|
||
}
|
||
// 负载均衡
|
||
$rpcKey = self::$RPC_Key;
|
||
echo '选中：负载均衡'.$rpcKey."\n";
|
||
self::rpcSelect();
|
||
// 调试
|
||
#p($result);
|
||
// 下载服务器类型 判断
|
||
$type = self::$links[$rpcKey]['type'];
|
||
switch ($type) {
|
||
case 'transmission':
|
||
if ($is_url) {
|
||
echo 'add';
|
||
$result = self::$links[$rpcKey]['rpc']->add($torrent, self::$links[$rpcKey]['downloadDir'], $extra_options); // 种子URL添加下载任务
|
||
} else {
|
||
echo 'add_metainfo';
|
||
$result = self::$links[$rpcKey]['rpc']->add_metainfo($torrent, self::$links[$rpcKey]['downloadDir'], $extra_options); // 种子文件添加下载任务
|
||
}
|
||
$id = $name = '';
|
||
if (isset($result->arguments->torrent_duplicate)) {
|
||
$id = $result->arguments->torrent_duplicate->id;
|
||
$name = $result->arguments->torrent_duplicate->name;
|
||
} elseif (isset($result->arguments->torrent_added)) {
|
||
$id = $result->arguments->torrent_added->id;
|
||
$name = $result->arguments->torrent_added->name;
|
||
}
|
||
if (!$id) {
|
||
print "-----RPC添加种子任务，失败 [{$result->result}] \n\n";
|
||
} else {
|
||
print "********RPC添加下载任务成功 [{$result->result}] (id=$id) \n\n";
|
||
// 新添加的任务，开始
|
||
self::$links[$rpcKey]['rpc']->start($id);
|
||
return true;
|
||
}
|
||
break;
|
||
case 'qBittorrent':
|
||
if ($is_url) {
|
||
echo 'add';
|
||
$result = self::$links[$rpcKey]['rpc']->add($torrent, self::$links[$rpcKey]['downloadDir'], $extra_options); // 种子URL添加下载任务
|
||
} else {
|
||
echo 'add_metainfo';
|
||
$result = self::$links[$rpcKey]['rpc']->add_metainfo($torrent, self::$links[$rpcKey]['downloadDir'], $extra_options); // 种子文件添加下载任务
|
||
}
|
||
if ($result === 'Ok.') {
|
||
print "********RPC添加下载任务成功 [{$result}] \n\n";
|
||
return true;
|
||
} else {
|
||
print "-----RPC添加种子任务，失败 [{$result}] \n\n";
|
||
}
|
||
break;
|
||
default:
|
||
echo '[ERROR] '.$type;
|
||
break;
|
||
}
|
||
} catch (Exception $e) {
|
||
die('[ERROR] ' . $e->getMessage() . PHP_EOL);
|
||
}
|
||
break;
|
||
case 2:
|
||
echo "\n\n";
|
||
# 暂未开放
|
||
break;
|
||
default:
|
||
echo "\n\n";
|
||
break;
|
||
}
|
||
return false;
|
||
}
|
||
|
||
/**
|
||
* 负载均衡 选择算法
|
||
*
|
||
* @param
|
||
* @return
|
||
*/
|
||
public static function rpcSelect()
|
||
{
|
||
$clientsConut = count(self::$clients);
|
||
if ($clientsConut > 1) {
|
||
if ($clientsConut > (self::$RPC_Key+1)) {
|
||
self::$RPC_Key++;
|
||
} else {
|
||
self::$RPC_Key = 0;
|
||
}
|
||
}
|
||
}
|
||
/**
|
||
* @brief 种子处理函数
|
||
* @param array $data 种子数组
|
||
* Array
|
||
(
|
||
[id] => 118632
|
||
[h1] => CCTV5+ 2019 ATP Men's Tennis Final 20191115B HDTV 1080i H264-HDxxx
|
||
[title] => 央视体育赛事频道 2019年ATP男子网球年终总决赛 单打小组赛 纳达尔VS西西帕斯 20191115[优惠剩余时间：4时13分]
|
||
[details] => https://XXX.me/details.php?id=118632
|
||
[download] => https://XXX.me/download.php?id=118632
|
||
[filename] => 118632.torrent
|
||
[type] => 0
|
||
[sticky] => 1
|
||
[time] => Array
|
||
(
|
||
[0] => "2019-11-16 20:41:53">4时13分
|
||
[1] => "2019-11-16 14:41:53">1时<br />46分
|
||
)
|
||
[comments] => 0
|
||
[size] => 5232.64MB
|
||
[seeders] => 69
|
||
[leechers] => 10
|
||
[completed] => 93
|
||
[percentage] => 100%
|
||
[owner] => 匿名
|
||
)
|
||
* @return
|
||
*/
|
||
public static function call($data = array())
|
||
{
|
||
foreach ($data as $key => $value) {
|
||
// 控制台打印
|
||
echo '主标题：'.$value['h1']."\n";
|
||
echo '副标题：'.$value['title']."\n";
|
||
echo '详情页：'.$value['details']."\n";
|
||
if ($value['type'] != 0) {
|
||
echo "-----非免费，已忽略！ \n\n";
|
||
continue;
|
||
}
|
||
if (isset($value['hr']) && ($value['hr'] == 1)) {
|
||
echo "-----HR种子，已忽略！ \n\n";
|
||
continue;
|
||
}
|
||
// 下载任务的可选参数
|
||
$extra_options = array();
|
||
// 保存的文件名
|
||
$filename = $value['id'] . '.torrent';
|
||
// 默认watch工作模式，复制到此目录
|
||
$to = self::$watch . $filename;
|
||
// 种子完整存放路径
|
||
$torrentFile = self::$torrentDir . $filename;
|
||
if (is_file($torrentFile)) {
|
||
$fileSize = filesize($torrentFile); //失败会返回false 或 0（0代表上次下载失败）
|
||
if (!empty($fileSize)) {
|
||
//种子已经存在
|
||
echo '-----存在旧种子：'.$filename."\n\n";
|
||
continue;
|
||
}
|
||
// 删除下载错误的文件
|
||
IFile::unlink($torrentFile);
|
||
}
|
||
|
||
// 调用过滤函数
|
||
$isFilter = filter(self::$site, $value);
|
||
if (is_string($isFilter)) {
|
||
echo "-----" .$isFilter. "\n\n";
|
||
continue;
|
||
}
|
||
//种子不存在
|
||
echo '正在下载新种子... '.$value['download']." \n";
|
||
// 创建文件、下载种子以二进制写入
|
||
$content = '';
|
||
$content = download($value['download'], self::$cookies, self::$userAgent, self::$method);
|
||
#p($content);
|
||
// 文件句柄
|
||
$resource = fopen($torrentFile, "wb");
|
||
// 成功：返回写入字节数，失败返回false
|
||
$worldsnum = fwrite($resource, $content);
|
||
// 关闭
|
||
fclose($resource);
|
||
// 判断
|
||
if (is_bool($worldsnum)) {
|
||
print "种子下载失败！！！ \n\n";
|
||
IFile::unlink($torrentFile);
|
||
continue;
|
||
} else {
|
||
print "成功下载种子" . $filename . '，共计：' . $worldsnum . "字节 \n";
|
||
sleep(mt_rand(2, 10));
|
||
$ret = false;
|
||
$rpcKey = self::$RPC_Key;
|
||
switch ((int)self::$workingMode) {
|
||
case 0: //默认工作模式
|
||
$ret = self::add($torrentFile, $to);
|
||
break;
|
||
case 1: //负载均衡模式
|
||
$type = self::$links[$rpcKey]['type'];
|
||
// 下载服务器类型
|
||
switch ($type) {
|
||
case 'transmission':
|
||
# code...
|
||
break;
|
||
case 'qBittorrent':
|
||
$extra_options['name'] = 'torrents';
|
||
$extra_options['filename'] = $filename;
|
||
$extra_options['autoTMM'] = 'false'; //关闭自动种子管理
|
||
break;
|
||
default:
|
||
# code...
|
||
break;
|
||
}
|
||
// 种子文件添加下载任务
|
||
$ret = self::add($content, $to, $extra_options);
|
||
break;
|
||
case 2:
|
||
echo "\n\n";
|
||
# 暂未开放
|
||
break;
|
||
default:
|
||
echo "\n\n";
|
||
break;
|
||
}
|
||
global $configALL;
|
||
if (isset($configALL['iyuu.cn']) && ($ret === true)) {
|
||
send(self::$site, $value);
|
||
}
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
}
|