mirror of
https://gitee.com/ledc/IYUUAutoReseed
synced 2025-05-20 08:25:22 +00:00
401 lines
12 KiB
PHP
401 lines
12 KiB
PHP
<?php
|
||
/**
|
||
* IYUU自动辅种脚本
|
||
*/
|
||
use Curl\Curl;
|
||
require_once __DIR__ . '/app/init.php';
|
||
iyuuAutoReseed::init();
|
||
$hashArray = iyuuAutoReseed::get();
|
||
if ( iyuuAutoReseed::$move != null ) {
|
||
echo "种子移动完毕，请重新编辑配置，再尝试辅种！ \n\n";
|
||
exit;
|
||
}
|
||
iyuuAutoReseed::call($hashArray);
|
||
/**
|
||
* iyuu自动辅种类
|
||
*/
|
||
class iyuuAutoReseed
|
||
{
|
||
/**
|
||
* 版本号
|
||
* @var string
|
||
*/
|
||
const VER = '2019年12月21日20:26:48';
|
||
// RPC连接池
|
||
public static $links = array();
|
||
/**
|
||
* 客户端配置
|
||
*/
|
||
public static $clients = array();
|
||
/**
|
||
* 不辅种的站点
|
||
*/
|
||
public static $noReseed = ['ourbits','hdchina'];
|
||
/**
|
||
* 缓存路径
|
||
*/
|
||
public static $cacheDir = TORRENT_PATH.'cache'.DS;
|
||
/**
|
||
* API接口配置
|
||
*/
|
||
public static $apiUrl = 'http://iyuu.cn:2122';
|
||
public static $endpoints = array(
|
||
'add' => '/api/add',
|
||
'update' => '/api/update',
|
||
'reseed' => '/api/reseed',
|
||
);
|
||
/**
|
||
* 退出状态码
|
||
*/
|
||
public static $ExitCode = 0;
|
||
/**
|
||
* 客户端转移做种 状态码[请把transmission配置为第一个客户端]
|
||
*/
|
||
public static $move = null;
|
||
/**
|
||
* 初始化
|
||
*/
|
||
public static function init(){
|
||
global $configALL;
|
||
|
||
self::$clients = isset($configALL['default']['clients']) && $configALL['default']['clients'] ? $configALL['default']['clients'] : array();
|
||
// 递归删除上次历史记录
|
||
IFile::rmdir(self::$cacheDir, true);
|
||
// 建立目录
|
||
IFile::mkdir(self::$cacheDir);
|
||
self::links();
|
||
}
|
||
/**
|
||
* 连接远端RPC服务器
|
||
*
|
||
* @param string
|
||
* @return array
|
||
*/
|
||
public static function links()
|
||
{
|
||
if(empty(self::$links)){
|
||
foreach ( self::$clients as $k => $v ){
|
||
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;
|
||
default:
|
||
echo '[ERROR] '.$v['type'];
|
||
exit(1);
|
||
break;
|
||
}
|
||
self::$links[$k]['type'] = $v['type'];
|
||
#self::$links[$k]['downloadDir'] = $v['downloadDir'];
|
||
// 检查是否转移种子的做种客户端？
|
||
if ( isset($v['move']) && $v['move'] ) {
|
||
self::$move = array($k,$v['type']);
|
||
}
|
||
} catch (Exception $e) {
|
||
echo '[ERROR] ' . $e->getMessage() . PHP_EOL;
|
||
exit(1);
|
||
}
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
/**
|
||
* 从客户端获取种子的哈希列表
|
||
*/
|
||
public static function get(){
|
||
$hashArray = array();
|
||
foreach ( self::$clients as $k => $v ){
|
||
$result = array();
|
||
$res = $info_hash = array();
|
||
$json = $sha1 = '';
|
||
try
|
||
{
|
||
switch($v['type']){
|
||
case 'transmission':
|
||
$ids = $fields = array();
|
||
#$fields = array( "id", "status", "name", "hashString", "downloadDir", "torrentFile" );
|
||
$fields = array( "id", "status", "hashString", "downloadDir");
|
||
$result = self::$links[$k]['rpc']->get($ids, $fields);
|
||
if ( empty($result->result) || $result->result != 'success' ){
|
||
// 获取种子列表 失败
|
||
echo "获取种子列表失败，原因可能是transmission暂时无响应，请稍后重试！ \n";
|
||
break;
|
||
}
|
||
if( empty($result->arguments) ){
|
||
echo "未获取到需要辅种的数据，请多多保种，然后重试！ \n";
|
||
break;
|
||
}
|
||
// 对象转数组
|
||
$res = object_array($result->arguments->torrents);
|
||
// 过滤，只保留正常做种
|
||
$res = array_filter($res, "filterStatus");
|
||
// 提取数组：hashString
|
||
$info_hash = array_column($res, 'hashString');
|
||
// 升序排序
|
||
sort($info_hash);
|
||
$json = json_encode($info_hash, JSON_UNESCAPED_UNICODE);
|
||
// 去重 应该从文件读入，防止重复提交
|
||
$sha1 = sha1( $json );
|
||
if ( isset($hashArray['sha1']) && (in_array($sha1, $hashArray['sha1']) != false) ) {
|
||
break;
|
||
}
|
||
// 组装返回数据
|
||
$hashArray['hash']['clients_'.$k] = $json;
|
||
$hashArray['sha1'][] = $sha1;
|
||
// 变换数组：hashString为键
|
||
self::$links[$k]['hash'] = array_column($res, "downloadDir", 'hashString');
|
||
#p(self::$links[$k]['hash']);exit;
|
||
break;
|
||
case 'qBittorrent':
|
||
$result = self::$links[$k]['rpc']->torrentList();
|
||
$res = json_decode($result,true);
|
||
if ( empty($res) ) {
|
||
echo "未获取到需要辅种的数据，请多多保种，然后重试！ \n";
|
||
break;
|
||
}
|
||
#p($res);exit;
|
||
// 过滤，只保留正常做种
|
||
$res = array_filter($res, "qbfilterStatus");
|
||
// 提取数组：hashString
|
||
$info_hash = array_column($res, 'hash');
|
||
// 升序排序
|
||
sort($info_hash);
|
||
$json = json_encode($info_hash, JSON_UNESCAPED_UNICODE);
|
||
// 去重 应该从文件读入，防止重复提交
|
||
$sha1 = sha1( $json );
|
||
if ( isset($hashArray['sha1']) && (in_array($sha1, $hashArray['sha1']) != false) ) {
|
||
break;
|
||
}
|
||
// 组装返回数据
|
||
$hashArray['hash']['clients_'.$k] = $json;
|
||
$hashArray['sha1'][] = $sha1;
|
||
// 变换数组：hash为键
|
||
self::$links[$k]['hash'] = array_column($res, "save_path", 'hash');
|
||
#p(self::$links[$k]['hash']);exit;
|
||
break;
|
||
default:
|
||
echo '[ERROR] '.$v['type'];
|
||
exit(1);
|
||
break;
|
||
}
|
||
// 是否执行转移种子做种客户端？
|
||
if ( self::$move != null && (empty($v['move'])) ) {
|
||
self::move($res, $v['type']);
|
||
}
|
||
} catch (Exception $e) {
|
||
echo '[ERROR] ' . $e->getMessage() . PHP_EOL;
|
||
exit(1);
|
||
}
|
||
}
|
||
// 写日志：文件句柄
|
||
$resource = fopen(self::$cacheDir.'hashString.txt', "wb");
|
||
// 成功：返回写入字节数，失败返回false
|
||
$worldsnum = fwrite($resource, p($hashArray, false));
|
||
fclose($resource);
|
||
return $hashArray;
|
||
}
|
||
/**
|
||
* @brief 添加下载任务
|
||
* @param string $torrent 种子元数据
|
||
* @param string $save_path 保存路径
|
||
* @return bool
|
||
*/
|
||
public static function add($rpcKey, $torrent, $save_path = '', $extra_options = array())
|
||
{
|
||
try
|
||
{
|
||
$result = self::$links[$rpcKey]['rpc']->add( $torrent, $save_path, $extra_options ); // 种子URL添加下载任务
|
||
// 调试
|
||
#p($result);
|
||
// 下载服务器类型 判断
|
||
switch(self::$links[$rpcKey]['type']){
|
||
case 'transmission':
|
||
if(isset($result->result) && $result->result == 'success'){
|
||
$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;
|
||
}
|
||
print "********RPC添加下载任务成功 [{$result->result}] (id=$id) \n";
|
||
print "种子：".$torrent. "\n";
|
||
print "名字：".$name."\n\n";
|
||
return true;
|
||
}else{
|
||
$errmsg = isset($result->result) ? $result->result : '未知错误，请稍后重试！';
|
||
print "-----RPC添加种子任务，失败 [{$errmsg}] \n";
|
||
print "种子：".$torrent. "\n";
|
||
}
|
||
break;
|
||
case 'qBittorrent':
|
||
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) {
|
||
echo '[ERROR] ' . $e->getMessage() . PHP_EOL;
|
||
}
|
||
return false;
|
||
}
|
||
/**
|
||
* @brief 提交种子hash给远端API，用来获取辅种数据
|
||
* @param array $hashArray 种子hash数组
|
||
* @return
|
||
*/
|
||
public static function call($hashArray = array())
|
||
{
|
||
global $configALL;
|
||
|
||
$resArray = $sites = array();
|
||
$curl = new Curl();
|
||
$curl->setOpt(CURLOPT_SSL_VERIFYPEER, false);
|
||
// 签名
|
||
$hashArray['timestamp'] = time();
|
||
$hashArray['sign'] = sign($hashArray['timestamp']);
|
||
// 发起请求
|
||
echo "正在提交辅种信息…… \n";
|
||
$res = $curl->post(self::$apiUrl . self::$endpoints['reseed'], $hashArray);
|
||
|
||
$resArray = json_decode($res->response, true);
|
||
// 写日志
|
||
if(true){
|
||
// 文件句柄
|
||
$resource = fopen(self::$cacheDir.'reseed.txt', "wb");
|
||
// 成功：返回写入字节数，失败返回false
|
||
$worldsnum = fwrite($resource, p($resArray, false));
|
||
fclose($resource);
|
||
}
|
||
// 判断返回值
|
||
if ( isset($resArray['errmsg']) && ($resArray['errmsg'] == 'ok') ) {
|
||
echo "辅种信息提交成功！！！ \n\n";
|
||
}else{
|
||
$errmsg = isset($resArray['errmsg']) ? $resArray['errmsg'] : '远端服务器无响应，请稍后重试！';
|
||
echo '-----辅种失败，原因：' .$errmsg. " \n\n";
|
||
exit(1);
|
||
}
|
||
// 可辅种站点信息列表
|
||
$sites = $resArray['sites'];
|
||
#p($sites);
|
||
foreach (self::$links as $k => $v) {
|
||
$reseed = $infohash_Dir = array();
|
||
if (empty($resArray['clients_'.$k])) {
|
||
echo "clients_".$k."没有查询到可辅种数据 \n\n";
|
||
continue;
|
||
}
|
||
$reseed = $resArray['clients_'.$k];
|
||
// info_hash 对应的下载目录
|
||
$infohash_Dir = self::$links[$k]['hash'];
|
||
foreach ($reseed as $info_hash => $vv) {
|
||
$downloadDir = $infohash_Dir[$info_hash];
|
||
foreach ($vv['torrent'] as $id => $value) {
|
||
$sitesID = $value['sid'];
|
||
$url = $_url = '';
|
||
$download_page = '';
|
||
$download_page = str_replace('{}', $value['torrent_id'], $sites[$sitesID]['download_page']);
|
||
$_url = 'https://' .$sites[$sitesID]['base_url']. '/' .$download_page;
|
||
if (empty($configALL[$sites[$sitesID]['site']]['passkey'])) {
|
||
echo '-------因当前' .$sites[$sitesID]['site']. '站点未设置passkey，已跳过！！' . "\n\n";
|
||
continue;
|
||
}
|
||
// 种子URL组合方式区分
|
||
switch ($sites[$sitesID]['site']) {
|
||
case 'ttg':
|
||
$url = $_url."/". $configALL[$sites[$sitesID]['site']]['passkey'];
|
||
break;
|
||
case 'm-team':
|
||
$url = $_url."&passkey=". $configALL[$sites[$sitesID]['site']]['passkey'] ."&https=1";
|
||
break;
|
||
// case 'hdchina':
|
||
// break;
|
||
default:
|
||
$url = $_url."&passkey=". $configALL[$sites[$sitesID]['site']]['passkey'];
|
||
break;
|
||
}
|
||
// 检查不辅种的站点
|
||
// 判断是否VIP/特殊权限？
|
||
$is_vip = isset($configALL[$sites[$sitesID]['site']]['is_vip']) && $configALL[$sites[$sitesID]['site']]['is_vip'] ? 1 : 0;
|
||
if ( (in_array($sites[$sitesID]['site'], self::$noReseed)==false) || $is_vip ) {
|
||
// 可以辅种
|
||
if ( isset($infohash_Dir[$value['info_hash']]) ) {
|
||
// 与客户端现有种子重复
|
||
echo '-------与客户端现有种子重复：'.$_url."\n\n";
|
||
}else{
|
||
// 把拼接的种子URL，推送给下载器
|
||
self::add($k, $url, $downloadDir);
|
||
}
|
||
}else{
|
||
// 不辅种
|
||
echo '-------已跳过不辅种的站点：'.$_url."\n\n";
|
||
// 写入日志文件，供用户手动辅种
|
||
if ( !isset($infohash_Dir[$value['info_hash']]) ) {
|
||
// 文件句柄
|
||
$resource = fopen(self::$cacheDir . $sites[$sitesID]['site'].'.txt', 'a');
|
||
// 成功：返回写入字节数，失败返回false
|
||
$worldsnum = fwrite($resource, $downloadDir."\n".$url."\n\n");
|
||
fclose($resource);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
public static function move($torrent=array(), $type = 'qBittorrent'){
|
||
switch($type){
|
||
case 'transmission':
|
||
break;
|
||
case 'qBittorrent':
|
||
foreach ($torrent as $k => $v) {
|
||
#$v['save_path'] = '/volume3' . $v['save_path']; // docker路径转换
|
||
self::add(self::$move[0], $v['magnet_uri'], $v['save_path'] );
|
||
}
|
||
break;
|
||
default:
|
||
echo '[ERROR] '.$type;
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
// transmission过滤函数，只保留正常做种
|
||
function filterStatus( $v ){
|
||
return isset($v['status']) && $v['status']===6;
|
||
}
|
||
// qBittorrent过滤函数，只保留正常做种
|
||
function qbfilterStatus( $v ){
|
||
if( ($v['state']=='uploading') || ($v['state'] == 'stalledUP') ){
|
||
return true;
|
||
}
|
||
return false;
|
||
}
|
||
//PHP stdClass Object转array
|
||
function object_array($array) {
|
||
if(is_object($array)) {
|
||
$array = (array)$array;
|
||
}
|
||
if(is_array($array)) {
|
||
foreach($array as $key=>$value) {
|
||
$array[$key] = object_array($value);
|
||
}
|
||
}
|
||
return $array;
|
||
}
|
||
// 对象转数组
|
||
function object2array(&$object) {
|
||
return json_decode( json_encode( $object ), true );
|
||
} |