fixed youtube-dl issues

2021-09-11 15:07:01 +08:00
parent bf539e90b7
commit cba9d22cc8
3 changed files with 53 additions and 38 deletions
--- a/lib/Tools/Youtube.php
+++ b/lib/Tools/Youtube.php
@@ -0,0 +1,137 @@
+<?php
+namespace OCA\NCDownloader\Tools;
+
+use OCA\NCDownloader\Tools\Helper;
+use Symfony\Component\Process\Exception\ProcessFailedException;
+use Symfony\Component\Process\Process;
+
+class Youtube
+{
+    private $ipv4Only;
+    private $audioOnly = 0;
+    private $audioFormat, $videoFormat = 'mp4';
+    private $options = [];
+    private $downloadDir;
+    public function __construct($config)
+    {
+        $config += ['downloadDir' => '/tmp/downloads'];
+        $this->bin = Helper::findBinaryPath('youtube-dl');
+        $this->setDownloadDir($config['downloadDir']);
+    }
+
+    public function GetUrlOnly()
+    {
+        $this->addOption('--get-filename');
+        $this->addOption('--get-url');
+        return $this;
+    }
+
+    public static function create()
+    {
+        return new self();
+    }
+
+    public function setDownloadDir($dir)
+    {
+        $this->downloadDir = rtrim($dir, '/');
+    }
+
+    public function getDownloadDir()
+    {
+        return $this->getDownloadDir;
+    }
+
+    public function prependOption($option)
+    {
+        array_unshift($this->options, $option);
+    }
+
+    public function download($url)
+    {
+        $this->downloadDir = $this->downloadDir ?? "/tmp/downloads";
+        $this->prependOption($this->downloadDir . "/%(id)s-%(title)s.%(ext)s");
+        $this->prependOption("-o");
+        $this->setUrl($url);
+        $this->prependOption($this->bin);
+        // $this->buildCMD();
+        $process = new Process($this->options);
+        //the maximum time required to download the file
+        $process->setTimeout(60*60*15);
+        try {
+            $process->mustRun();
+            $output = $process->getOutput();
+        } catch (ProcessFailedException $exception) {
+            $output = $exception->getMessage();
+        }
+        return $output;
+    }
+
+    public function getDownloadUrl($url)
+    {
+        $this->setUrl($url);
+        $this->GetUrlOnly();
+        //$process = new Process($this->options);
+        $this->buildCMD();
+        exec($this->cmd, $output, $returnCode);
+        if (count($output) === 1) {
+            return ['url' => reset($output)];
+        }
+        list($url, $filename) = $output;
+        return ['url' => $url, 'filename' => Helper::cleanString($filename)];
+    }
+
+    public function setUrl($url)
+    {
+        $this->addOption('-i');
+        $this->addOption($url);
+        //$index = array_search('-i', $this->options);
+        //array_splice($this->options, $index + 1, 0, $url);
+    }
+
+    public function addOption($option)
+    {
+        array_push($this->options, $option);
+    }
+
+    public function forceIPV4()
+    {
+        $this->addOption('-4');
+        return $this;
+    }
+
+    public function setAudioFormat($format)
+    {
+        $this->audioFormat = $format;
+    }
+
+    public function setvideoFormat($format)
+    {
+        $this->videoFormat = $format;
+    }
+
+    private function buildCMD()
+    {
+        $this->cmd = $this->bin; //. " 2>&1";
+
+        foreach ($this->options as $option) {
+            $this->cmd .= " " . $option;
+        }
+
+    }
+    public function isInstalled()
+    {
+        return (bool) isset($this->bin);
+    }
+    public static function install()
+    {
+        $url = $this->installUrl();
+        $path = \OC::$server->getSystemConfig()->getValue('datadirectory');
+        Helper::Download($url, $path . "/youtube-dl");
+    }
+
+    public function installUrl()
+    {
+        return "https://yt-dl.org/downloads/latest/youtube-dl";
+    }
+
+}