pisto/ncdownloader
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
huangjx
2021-09-09 22:03:08 +08:00
commit f2a2365102
74 changed files with 33916 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

220
lib/Controller/Aria2Controller.php Normal file
View File

@@ -0,0 +1,220 @@
<?php
namespace OCA\NcDownloader\Controller;
use OCA\NcDownloader\Tools\Aria2;
use OCA\NcDownloader\Tools\DBConn;
use OCA\NcDownloader\Tools\File;
use OCA\NcDownloader\Tools\Helper;
use OCA\NcDownloader\Tools\Settings;
use OCP\AppFramework\Controller;
use OCP\AppFramework\Http\JSONResponse;
use OCP\Files\IRootFolder;
use OCP\IL10N;
use OCP\IRequest;
use OC_Util;
use \OC\Files\Filesystem;
class Aria2Controller extends Controller
{
private $userId;
private $settings = null;
//@config OC\AppConfig
private $config;
private $aria2Opts;
private $l10n;
public function __construct($appName, IRequest $request, $UserId, IL10N $IL10N, IRootFolder $rootFolder, Aria2 $aria2)
{
parent::__construct($appName, $request);
$this->appName = $appName;
$this->uid = $UserId;
$this->l10n = $IL10N;
$this->rootFolder = $rootFolder;
$this->urlGenerator = \OC::$server->getURLGenerator();
$this->settings = new Settings($UserId);
$this->downloadDir = $this->settings->get('ncd_downloader_dir') ?? "/Downloads";
OC_Util::setupFS();
//$this->config = \OC::$server->getAppConfig();
$this->aria2 = $aria2;
$this->aria2->init();
$this->dbconn = new DBConn();
}
public function Action($path)
{
$path = strtolower(trim($path));
if (!in_array($path, ['start', 'check']) && !($gid = $this->request->getParam('gid'))) {
return new JSONResponse(['error' => "no gid value is received!"]);
}
switch (strtolower($path)) {
case "check":
$resp = $this->aria2->isRunning();
break;
case "start":
$resp = $this->Start();
break;
case "pause":
$resp = $this->aria2->pause($gid);
break;
case "remove":
$resp = $this->aria2->remove($gid);
break;
case "unpause":
$resp = $this->aria2->unpause($gid);
break;
case "get":
$resp = $this->aria2->tellStatus($gid);
break;
case 'purge':
$resp = $this->aria2->removeDownloadResult($gid);
}
return new JSONResponse($resp);
}
private function Start()
{
if ($this->aria2->isRunning()) {
$data = ['status' => (bool) $this->aria2->stop()];
return $data;
}
$data = $this->aria2->start();
return $data;
}
public function Update()
{
File::syncFolder($this->downloadDir);
return new JSONResponse([]);
}
private function createActionItem($name, $path)
{
return array(
'name' => $name,
'path' => $this->urlGenerator->linkToRoute('ncdownloader.Aria2.Action', ['path' => $path]),
);
}
public function getStatus($path)
{
//$path = $this->request->getRequestUri();
$counter = $this->aria2->getCounters();
switch (strtolower($path)) {
case "active":
$resp = $this->aria2->tellActive();
break;
case "waiting":
$resp = $this->aria2->tellWaiting([0, 999]);
break;
case "complete":
$resp = $this->aria2->tellStopped([0, 999]);
break;
case "fail":
$resp = $this->aria2->tellFail([0, 999]);
break;
default:
$resp = $this->aria2->tellActive();
}
if (isset($resp['error'])) {
return new JSONResponse($resp);
}
$data = $this->prepareResp($resp);
$data['counter'] = $counter;
return new JSONResponse($data);
}
private function prepareResp($resp)
{
$data = [];
if (empty($resp)) {
return $data;
}
$data['row'] = [];
foreach ($resp as $value) {
$gid = $value['following'] ?? $value['gid'];
if ($row = $this->dbconn->getByGid($gid)) {
$filename = $row['filename'];
$timestamp = $row['timestamp'];
} else if (isset($value['files'][0]['path'])) {
$parts = explode("/", ($path = $value['files'][0]['path']));
if (count($parts) > 1) {
$filename = basename(dirname($path));
} else {
$filename = basename($path);
}
} else {
$filename = "Unknown";
}
if (!isset($value['completedLength'])) {
continue;
}
//internal nextcloud absolute path for nodeExists
//$file = $this->userFolder . $this->downloadDir . "/" . $filename;
// $dir = $this->rootFolder->nodeExists($file) ? $this->downloadDir . "/" . $filename : $this->downloadDir;
$file = $this->downloadDir . "/" . $filename;
$params = ['dir' => $this->downloadDir];
$fileInfo = Filesystem::getFileInfo($file);
if ($fileInfo) {
$fileType = $fileInfo->getType();
if ($fileType === "dir") {
$params = ['dir' => $file];
}
}
$folderLink = $this->urlGenerator->linkToRoute('files.view.index', $params);
//$peers = ($this->getPeers($info['gid']));
$completed = Helper::formatBytes($value['completedLength']);
$percentage = $value['completedLength'] ? 100 * ($value['completedLength'] / $value['totalLength']) : 0;
$completed = Helper::formatBytes($value['completedLength']);
$total = Helper::formatBytes($value['totalLength']);
$remaining = (int) $value['totalLength'] - (int) $value['completedLength'];
$remaining = ($value['downloadSpeed'] > 0) ? ($remaining / $value['downloadSpeed']) : 0;
$left = Helper::formatInterval($remaining);
$numSeeders = $value['numSeeders'] ?? 0;
$extraInfo = "Seeders: $numSeeders";
// $numPeers = isset($peers['result']) ? count($peers['result']) : 0;
$value['progress'] = array(sprintf("%s(%.2f%%)", $completed, $percentage), $extraInfo);
$timestamp = $timestamp ?? 0;
//$prefix = $value['files'][0]['path'];
$filename = sprintf('<a class="download-file-folder" href="%s">%s</a>', $folderLink, $filename);
$fileInfo = sprintf("%s | %s", $total, date("Y-m-d H:i:s", $timestamp));
$tmp = [];
$actions = [];
if ($this->aria2->methodName === "tellStopped") {
$actions[] = $this->createActionItem('purge', 'purge');
} else {
$actions[] = $this->createActionItem('delete', 'remove');
}
if ($this->aria2->methodName === "tellWaiting") {
$actions[] = $this->createActionItem('unpause', 'unpause');
}
$tmp['filename'] = array($filename, $fileInfo);
if ($this->aria2->methodName === "tellActive") {
$speed = [Helper::formatBytes($value['downloadSpeed']), $left . " left"];
$tmp['speed'] = $speed;
$tmp['progress'] = $value['progress'];
$actions[] = $this->createActionItem('pause', 'pause');
}
if (strtolower($value['status']) === 'error') {
$tmp['status'] = $value['errorMessage'];
} else if ($this->aria2->methodName !== "tellActive") {
$tmp['status'] = $value['status'];
}
$tmp['data_gid'] = $value['gid'] ?? 0;
$tmp['actions'] = $actions;
//$tmp['actions'] = '';
array_push($data['row'], $tmp);
}
if ($this->aria2->methodName === "tellActive") {
$data['title'] = Helper::getTableTitles('active');
} else {
$data['title'] = Helper::getTableTitles();
}
return $data;
}
}

135
lib/Controller/MainController.php Normal file
View File

@@ -0,0 +1,135 @@
<?php
namespace OCA\NcDownloader\Controller;
use OCA\NcDownloader\Search\torrentSearch;
use OCA\NcDownloader\Tools\Aria2;
use OCA\NcDownloader\Tools\DBConn;
use OCA\NcDownloader\Tools\Helper;
use OCA\NcDownloader\Tools\YouTube;
use OCP\AppFramework\Controller;
use OCP\AppFramework\Http\JSONResponse;
use OCP\AppFramework\Http\TemplateResponse;
use OCP\Files\IRootFolder;
use OCP\IL10N;
use OCP\IRequest;
use OC_Util;
use \OCP\AppFramework\Http\StrictContentSecurityPolicy;
class MainController extends Controller
{
private $settings = null;
//@config OC\AppConfig
private $config;
private $aria2Opts;
private $l10n;
public function __construct($appName, IRequest $request, $UserId, IL10N $IL10N, IRootFolder $rootFolder, Aria2 $aria2)
{
parent::__construct($appName, $request);
$this->appName = $appName;
$this->uid = $UserId;
$this->l10n = $IL10N;
$this->rootFolder = $rootFolder;
OC_Util::setupFS();
$this->urlGenerator = \OC::$server->getURLGenerator();
$this->aria2 = $aria2;
$this->aria2->init();
$this->youtube = new Youtube();
$this->dbconn = new DBConn();
}
/**
* @NoAdminRequired
* @NoCSRFRequired
*/
public function Index()
{
// $str = \OC::$server->getDatabaseConnection()->getInner()->getPrefix();
//$config = \OC::$server->getAppConfig();
OC_Util::addScript($this->appName, 'app');
OC_Util::addStyle($this->appName, 'style');
OC_Util::addStyle($this->appName, 'table');
$params = array();
$params['aria2_running'] = $this->aria2->isRunning();
$params['aria2_installed'] = $this->aria2->isInstalled();
$params['youtube_installed'] = $this->youtube->isInstalled();
$response = new TemplateResponse($this->appName, 'Index', $params);
$csp = new StrictContentSecurityPolicy();
$csp->allowEvalScript();
$csp->allowInlineStyle();
$response->setContentSecurityPolicy($csp);
return $response;
}
public function newDownload()
{
$params = array();
$inputValue = trim($this->request->getParam('form_input_text'));
$type = trim($this->request->getParam('type'));
if ($type == 'ytdl') {
$yt = $this->youtube;
if (!$yt->isInstalled()) {
try {
$filename = Helper::getFileName($yt->installUrl());
$this->aria2->setDownloadDir($this->dataDir . "/bin");
$resp = $this->Save($yt->installUrl(), $filename);
return new JSONResponse($resp);
} catch (\Exception $e) {
return new JSONResponse(['error' => $e->getMessage()]);
}
return new JSONResponse(['error' => $this->l10n->t("Youtube-dl NOT installed!")]);
}
if (Helper::isGetUrlSite($inputValue)) {
if ($data = $yt->forceIPV4()->getDownloadUrl($inputValue)) {
$this->Save($data['url'], $data['filename']);
return new JSONResponse(['yt' => $data]);
} else {
return new JSONResponse(['error' => $this->l10n->t("failed to get any url!")]);
}
} else {
$yt->setDownloadDir($this->realDownloadDir);
return new JSONResponse(['yt' => $yt->download($inputValue)]);
}
} else if ($type === 'search') {
$data = torrentSearch::go($inputValue);
$resp['title'] = ['title', 'seeders', 'info', 'actions'];
$resp['row'] = $data;
return new JSONResponse($resp);
}
$filename = Helper::getFileName($inputValue);
$resp = $this->Save($inputValue, $filename);
return new JSONResponse($resp);
}
private function Save($url, $filename = null)
{
if (isset($filename)) {
$this->aria2->setFileName($filename);
}
//$this->aria2->setDownloadDir("/tmp/downloads");
$result = $this->aria2->addUri([$url]);
$gid = $result['result'];
if (!is_string($gid)) {
return ['error' => 'Failed to add download task! ' . $result['error']];
} else {
$data = [
'uid' => $this->uid,
'gid' => $gid,
'type' => 1,
'filename' => $filename ?? 'unknown',
'timestamp' => time(),
'data' => serialize(['link' => $url]),
];
$this->dbconn->save($data);
}
return ['gid' => $gid, 'file' => $filename, 'result' => $gid];
}
}

83
lib/Controller/SettingsController.php Normal file
View File

@@ -0,0 +1,83 @@
<?php
namespace OCA\NcDownloader\Controller;
use OCA\NcDownloader\Tools\Helper;
use OCA\NcDownloader\Tools\Settings;
use OCP\AppFramework\Controller;
use OCP\AppFramework\Http\JSONResponse;
use OCP\IRequest;
use OC_Util;
class SettingsController extends Controller
{
/*@ OC\AppFramework\Http\Request*/
//private $request;
//@config OC\AppConfig
private $config;
public function __construct($AppName, IRequest $Request, $UserId) //, IL10N $L10N)
{
parent::__construct($AppName, $Request);
$this->UserId = $UserId;
//$this->L10N = $L10N;
$this->settings = new Settings($UserId);
//$this->config = \OC::$server->getAppConfig();
}
/**
* @NoAdminRequired
* @NoCSRFRequired
*/
public function personal()
{
$params = $this->request->getParams();
foreach ($params as $key => $value) {
if (substr($key, 0, 1) == '_') {
continue;
}
$this->save($key, $value);
}
}
public function aria2Get()
{
$data = json_decode($this->settings->get("custom_aria2_settings"));
return new JSONResponse($data);
}
public function admin()
{
$this->settings->setType($this->settings::SYSTEM);
$params = $this->request->getParams();
foreach ($params as $key => $value) {
if (substr($key, 0, 1) == '_') {
continue;
}
$this->save($key, $value);
}
}
public function aria2Save()
{
$params = $this->request->getParams();
$data = Helper::filterData($params, Helper::aria2Options());
$this->settings->save("custom_aria2_settings", json_encode($data));
}
public function aria2Delete()
{
$saved = json_decode($this->settings->get("custom_aria2_settings"),1);
$params = $this->request->getParams();
$data = Helper::filterData($params, Helper::aria2Options());
foreach ($data as $key => $value) {
unset($saved[$key]);
}
$this->settings->save("custom_aria2_settings", json_encode($saved));
return new JSONResponse($saved);
}
public function save($key, $value)
{
$this->settings->save($key, $value);
}
}