tidying up
This commit is contained in:
huangjx
266
lib/Ytdl/Ytdl.php
Normal file
266
lib/Ytdl/Ytdl.php
Normal file
@@ -0,0 +1,266 @@
|
||||
<?php
|
||||
|
||||
namespace OCA\NCDownloader\Ytdl;
|
||||
|
||||
use OCA\NCDownloader\Tools\Helper;
|
||||
use OCA\NCDownloader\Ytdl\Helper as YtdHelper;
|
||||
use Symfony\Component\Process\Process;
|
||||
|
||||
class Ytdl
|
||||
{
|
||||
public $audioOnly = 0;
|
||||
public $audioFormat = 'm4a', $videoFormat = null;
|
||||
//path in nextcloud fs
|
||||
public $dbDlPath = null;
|
||||
private $format = 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best';
|
||||
private $options = [];
|
||||
private $downloadDir;
|
||||
private $timeout = 60 * 60 * 10; //10 hours
|
||||
private $outTpl = "%(id)s-%(title).64s.%(ext)s";
|
||||
private $defaultDir = "/tmp/downloads";
|
||||
private $env = [];
|
||||
private $bin;
|
||||
|
||||
public function __construct(array $options)
|
||||
{
|
||||
$options += ['downloadDir' => '/tmp/downloads', 'settings' => []];
|
||||
$this->init($options);
|
||||
}
|
||||
public function init(array $options)
|
||||
{
|
||||
extract($options);
|
||||
if (isset($binary) && $this->isExecutable($binary)) {
|
||||
$this->bin = $binary;
|
||||
} else {
|
||||
$this->bin = __DIR__ . "/../../bin/yt-dlp"; //Helper::findBinaryPath('ytdl', __DIR__ . "/../../bin/yt-dlp");
|
||||
}
|
||||
if ($this->isInstalled() && !$this->isExecutable()) {
|
||||
chmod($this->bin, 0744);
|
||||
}
|
||||
$this->setDownloadDir($downloadDir);
|
||||
if (!empty($settings)) {
|
||||
foreach ($settings as $key => $value) {
|
||||
if (empty($value)) {
|
||||
$this->addOption($key, true);
|
||||
} else {
|
||||
$this->setOption($key, $value, true);
|
||||
}
|
||||
}
|
||||
}
|
||||
if (empty($lang = getenv('LANG')) || strpos(strtolower($lang), 'c.utf-8') === false) {
|
||||
$lang = 'C.UTF-8';
|
||||
}
|
||||
$this->setEnv('LANG', $lang);
|
||||
$this->addOption("--no-mtime");
|
||||
$this->addOption('--ignore-errors');
|
||||
|
||||
if (($index = $this->hasOption('--output')) !== false) {
|
||||
$this->outTpl = $this->options[$index + 1];
|
||||
unset($this->options[$index]);
|
||||
unset($this->options[$index + 1]);
|
||||
}
|
||||
}
|
||||
|
||||
public function setEnv($key, $val)
|
||||
{
|
||||
$this->env[$key] = $val;
|
||||
}
|
||||
|
||||
public function audioMode()
|
||||
{
|
||||
if (Helper::ffmpegInstalled()) {
|
||||
$this->addOption('--prefer-ffmpeg');
|
||||
// $this->addOption('--add-metadata');
|
||||
// $this->setOption('--metadata-from-title', "%(artist)s-%(title).64s");
|
||||
$this->addOption('--extract-audio');
|
||||
} else {
|
||||
$this->audioFormat = "m4a";
|
||||
}
|
||||
/*$pos = strrpos($this->outTpl, '.');
|
||||
$this->outTpl = substr($this->outTpl, 0, $pos) . "." . $this->audioFormat;
|
||||
$this->outTpl = "/%(id)s-%(title)s.m4a";*/
|
||||
$this->setAudioFormat($this->audioFormat);
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function setAudioQuality($value = 0)
|
||||
{
|
||||
$this->setOption('--audio-quality', $value);
|
||||
}
|
||||
|
||||
public function setAudioFormat($format)
|
||||
{
|
||||
$this->setOption('--audio-format', $format);
|
||||
}
|
||||
|
||||
public function setVideoFormat($format)
|
||||
{
|
||||
//$this->videoFormat = $format;
|
||||
$this->setOption('--recode-video', $format);
|
||||
}
|
||||
|
||||
public function GetUrlOnly()
|
||||
{
|
||||
$this->addOption('--get-filename');
|
||||
$this->addOption('--get-url');
|
||||
return $this;
|
||||
}
|
||||
|
||||
public static function create($options)
|
||||
{
|
||||
return new self($options);
|
||||
}
|
||||
|
||||
public function setDownloadDir($dir)
|
||||
{
|
||||
$this->downloadDir = rtrim($dir, '/');
|
||||
}
|
||||
|
||||
public function getDownloadDir()
|
||||
{
|
||||
return $this->downloadDir;
|
||||
}
|
||||
|
||||
public function prependOption(string $option)
|
||||
{
|
||||
array_unshift($this->options, $option);
|
||||
}
|
||||
|
||||
public function download($url)
|
||||
{
|
||||
if ($this->audioOnly) {
|
||||
$this->audioMode();
|
||||
} else {
|
||||
if (Helper::ffmpegInstalled() && $this->videoFormat) {
|
||||
$this->setOption('--format', 'bestvideo+bestaudio/best');
|
||||
$this->setVideoFormat($this->videoFormat);
|
||||
} else {
|
||||
$this->setOption('--format', $this->format);
|
||||
}
|
||||
}
|
||||
$this->helper = YtdHelper::create();
|
||||
$this->downloadDir = $this->downloadDir ?? $this->defaultDir;
|
||||
$this->setOption("--output", $this->downloadDir . "/" . $this->outTpl);
|
||||
$this->setUrl($url);
|
||||
$this->prependOption($this->bin);
|
||||
$process = new Process($this->options, null, $this->env);
|
||||
$process->setTimeout($this->timeout);
|
||||
$data = ['link' => $url, 'path' => $this->dbDlPath];
|
||||
if ($this->audioOnly) {
|
||||
$data['ext'] = $this->audioFormat;
|
||||
} else {
|
||||
$data['ext'] = $this->videoFormat;
|
||||
}
|
||||
$process->run(function ($type, $buffer) use ($data, $process) {
|
||||
if (Process::ERR === $type) {
|
||||
$this->onError($buffer);
|
||||
} else {
|
||||
$data['pid'] = $process->getPid();
|
||||
$this->onOutput($buffer, $data);
|
||||
}
|
||||
});
|
||||
if ($process->isSuccessful()) {
|
||||
$this->helper->updateStatus(Helper::STATUS['COMPLETE']);
|
||||
return ['message' => $this->helper->file ?? $process->getErrorOutput()];
|
||||
}
|
||||
return ['error' => $process->getErrorOutput()];
|
||||
}
|
||||
|
||||
private function onError($buffer)
|
||||
{
|
||||
$this->helper->log($buffer);
|
||||
}
|
||||
|
||||
public function onOutput($buffer, $extra)
|
||||
{
|
||||
$this->helper->run($buffer, $extra);
|
||||
}
|
||||
public function getDownloadUrl($url)
|
||||
{
|
||||
$this->setUrl($url);
|
||||
$this->GetUrlOnly();
|
||||
$this->buildCMD();
|
||||
exec($this->cmd, $output, $returnCode);
|
||||
if (count($output) === 1) {
|
||||
return ['url' => reset($output)];
|
||||
}
|
||||
list($url, $filename) = $output;
|
||||
$filename = Helper::cleanString($filename);
|
||||
return ['url' => $url, 'filename' => Helper::clipFilename($filename)];
|
||||
}
|
||||
|
||||
public function setUrl($url)
|
||||
{
|
||||
$this->prependOption($url);
|
||||
//$index = array_search('-i', $this->options);
|
||||
//array_splice($this->options, $index + 1, 0, $url);
|
||||
}
|
||||
public function setOption($key, $value, $hyphens = false)
|
||||
{
|
||||
$this->addOption($key, $hyphens);
|
||||
$this->addOption($value, false);
|
||||
return $this;
|
||||
}
|
||||
public function addOption(String $option, $hyphens = false)
|
||||
{
|
||||
if ($hyphens && substr($option, 0, 2) !== '--') {
|
||||
$option = "--" . $option;
|
||||
}
|
||||
array_push($this->options, $option);
|
||||
}
|
||||
|
||||
protected function hasOption($name)
|
||||
{
|
||||
return array_search($name, $this->options);
|
||||
}
|
||||
|
||||
public function forceIPV4()
|
||||
{
|
||||
$this->addOption('force-ipv4', true);
|
||||
return $this;
|
||||
}
|
||||
|
||||
private function buildCMD()
|
||||
{
|
||||
$this->cmd = $this->bin; //. " 2>&1";
|
||||
|
||||
foreach ($this->options as $option) {
|
||||
$this->cmd .= " " . $option;
|
||||
}
|
||||
}
|
||||
public function isInstalled()
|
||||
{
|
||||
return @is_file($this->bin);
|
||||
}
|
||||
public function isExecutable()
|
||||
{
|
||||
return @is_executable($this->bin);
|
||||
}
|
||||
|
||||
public function isReadable()
|
||||
{
|
||||
return @is_readable($this->bin);
|
||||
}
|
||||
|
||||
public function getBin()
|
||||
{
|
||||
return $this->bin;
|
||||
}
|
||||
public function install()
|
||||
{
|
||||
$url = $this->installUrl();
|
||||
$file = __DIR__ . "/../../bin/yt-dlp2";
|
||||
try {
|
||||
Helper::Download($url, $file);
|
||||
chmod($file, 0744);
|
||||
} catch (\Exception $e) {
|
||||
return $e->getMessage();
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
public function installUrl()
|
||||
{
|
||||
return "https://github.com/shiningw/ncdownloader-bin/raw/master/yt-dlp";
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user