Move all logic to the VideoDownload class
This commit is contained in:
parent
76360f8ce0
commit
26388ea1b4
2 changed files with 114 additions and 84 deletions
|
@ -14,6 +14,7 @@ namespace Alltube;
|
||||||
|
|
||||||
use Symfony\Component\Process\Process;
|
use Symfony\Component\Process\Process;
|
||||||
use Symfony\Component\Process\ProcessBuilder;
|
use Symfony\Component\Process\ProcessBuilder;
|
||||||
|
use Chain\Chain;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Main class
|
* Main class
|
||||||
|
@ -113,4 +114,114 @@ class VideoDownload
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public function getFilename($url, $format = null)
|
||||||
|
{
|
||||||
|
$this->procBuilder->setArguments(
|
||||||
|
array(
|
||||||
|
'--get-filename',
|
||||||
|
$url
|
||||||
|
)
|
||||||
|
);
|
||||||
|
if (isset($format)) {
|
||||||
|
$this->procBuilder->add('-f '.$format);
|
||||||
|
}
|
||||||
|
$process = $this->procBuilder->getProcess();
|
||||||
|
$process->run();
|
||||||
|
if (!$process->isSuccessful()) {
|
||||||
|
throw new \Exception($process->getErrorOutput());
|
||||||
|
} else {
|
||||||
|
return $process->getOutput();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public function getAudioFilename($url, $format = null)
|
||||||
|
{
|
||||||
|
return html_entity_decode(
|
||||||
|
pathinfo(
|
||||||
|
$this->getFilename($url, $format),
|
||||||
|
PATHINFO_FILENAME
|
||||||
|
).'.mp3',
|
||||||
|
ENT_COMPAT,
|
||||||
|
'ISO-8859-1'
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
public function getConversionProcess($url, $format)
|
||||||
|
{
|
||||||
|
if (!shell_exec('which '.$this->config->avconv)) {
|
||||||
|
throw(new \Exception('Can\'t find avconv or ffmpeg'));
|
||||||
|
}
|
||||||
|
|
||||||
|
$video = $this->getJSON($url, $format);
|
||||||
|
|
||||||
|
//Vimeo needs a correct user-agent
|
||||||
|
ini_set(
|
||||||
|
'user_agent',
|
||||||
|
$video->http_headers->{'User-Agent'}
|
||||||
|
);
|
||||||
|
$avconvProc = ProcessBuilder::create(
|
||||||
|
array(
|
||||||
|
$this->config->avconv,
|
||||||
|
'-v', 'quiet',
|
||||||
|
'-i', '-',
|
||||||
|
'-f', 'mp3',
|
||||||
|
'-vn',
|
||||||
|
'pipe:1'
|
||||||
|
)
|
||||||
|
);
|
||||||
|
|
||||||
|
if (parse_url($video->url, PHP_URL_SCHEME) == 'rtmp') {
|
||||||
|
if (!shell_exec('which '.$this->config->rtmpdump)) {
|
||||||
|
throw(new \Exception('Can\'t find rtmpdump'));
|
||||||
|
}
|
||||||
|
$builder = new ProcessBuilder(
|
||||||
|
array(
|
||||||
|
$this->config->rtmpdump,
|
||||||
|
'-q',
|
||||||
|
'-r',
|
||||||
|
$video->url,
|
||||||
|
'--pageUrl', $video->webpage_url
|
||||||
|
)
|
||||||
|
);
|
||||||
|
if (isset($video->player_url)) {
|
||||||
|
$builder->add('--swfVfy');
|
||||||
|
$builder->add($video->player_url);
|
||||||
|
}
|
||||||
|
if (isset($video->flash_version)) {
|
||||||
|
$builder->add('--flashVer');
|
||||||
|
$builder->add($video->flash_version);
|
||||||
|
}
|
||||||
|
if (isset($video->play_path)) {
|
||||||
|
$builder->add('--playpath');
|
||||||
|
$builder->add($video->play_path);
|
||||||
|
}
|
||||||
|
foreach ($video->rtmp_conn as $conn) {
|
||||||
|
$builder->add('--conn');
|
||||||
|
$builder->add($conn);
|
||||||
|
}
|
||||||
|
$chain = new Chain($builder->getProcess());
|
||||||
|
$chain->add('|', $avconvProc);
|
||||||
|
} else {
|
||||||
|
if (!shell_exec('which curl')) {
|
||||||
|
throw(new \Exception('Can\'t find curl'));
|
||||||
|
}
|
||||||
|
$chain = new Chain(
|
||||||
|
ProcessBuilder::create(
|
||||||
|
array_merge(
|
||||||
|
array(
|
||||||
|
'curl',
|
||||||
|
'--silent',
|
||||||
|
'--location',
|
||||||
|
'--user-agent', $video->http_headers->{'User-Agent'},
|
||||||
|
$video->url
|
||||||
|
),
|
||||||
|
$this->config->curl_params
|
||||||
|
)
|
||||||
|
)
|
||||||
|
);
|
||||||
|
$chain->add('|', $avconvProc);
|
||||||
|
}
|
||||||
|
return popen($chain->getProcess()->getCommandLine(), 'r');
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -14,8 +14,6 @@ namespace Alltube\Controller;
|
||||||
|
|
||||||
use Alltube\VideoDownload;
|
use Alltube\VideoDownload;
|
||||||
use Alltube\Config;
|
use Alltube\Config;
|
||||||
use Symfony\Component\Process\ProcessBuilder;
|
|
||||||
use Chain\Chain;
|
|
||||||
use Slim\Http\Stream;
|
use Slim\Http\Stream;
|
||||||
use Slim\Http\Request;
|
use Slim\Http\Request;
|
||||||
use Slim\Http\Response;
|
use Slim\Http\Response;
|
||||||
|
@ -122,96 +120,17 @@ class FrontController
|
||||||
$url = $this->download->getURL($params["url"], 'mp3[protocol^=http]');
|
$url = $this->download->getURL($params["url"], 'mp3[protocol^=http]');
|
||||||
return $response->withRedirect($url);
|
return $response->withRedirect($url);
|
||||||
} catch (\Exception $e) {
|
} catch (\Exception $e) {
|
||||||
$video = $this->download->getJSON($params["url"], 'bestaudio/best');
|
|
||||||
|
|
||||||
if (!shell_exec('which '.$this->config->avconv)) {
|
|
||||||
throw(new \Exception('Can\'t find avconv or ffmpeg'));
|
|
||||||
}
|
|
||||||
|
|
||||||
$avconvProc = ProcessBuilder::create(
|
|
||||||
array(
|
|
||||||
$this->config->avconv,
|
|
||||||
'-v', 'quiet',
|
|
||||||
'-i', '-',
|
|
||||||
'-f', 'mp3',
|
|
||||||
'-vn',
|
|
||||||
'pipe:1'
|
|
||||||
)
|
|
||||||
);
|
|
||||||
|
|
||||||
//Vimeo needs a correct user-agent
|
|
||||||
ini_set(
|
|
||||||
'user_agent',
|
|
||||||
$video->http_headers->{'User-Agent'}
|
|
||||||
);
|
|
||||||
|
|
||||||
$response = $response->withHeader(
|
$response = $response->withHeader(
|
||||||
'Content-Disposition',
|
'Content-Disposition',
|
||||||
'attachment; filename="'.
|
'attachment; filename="'.
|
||||||
html_entity_decode(
|
$this->download->getAudioFilename($params["url"], 'bestaudio/best').'"'
|
||||||
pathinfo(
|
|
||||||
$video->_filename,
|
|
||||||
PATHINFO_FILENAME
|
|
||||||
).'.mp3',
|
|
||||||
ENT_COMPAT,
|
|
||||||
'ISO-8859-1'
|
|
||||||
).'"'
|
|
||||||
);
|
);
|
||||||
$response = $response->withHeader('Content-Type', 'audio/mpeg');
|
$response = $response->withHeader('Content-Type', 'audio/mpeg');
|
||||||
|
|
||||||
if (parse_url($video->url, PHP_URL_SCHEME) == 'rtmp') {
|
|
||||||
if (!shell_exec('which '.$this->config->rtmpdump)) {
|
|
||||||
throw(new \Exception('Can\'t find rtmpdump'));
|
|
||||||
}
|
|
||||||
$builder = new ProcessBuilder(
|
|
||||||
array(
|
|
||||||
$this->config->rtmpdump,
|
|
||||||
'-q',
|
|
||||||
'-r',
|
|
||||||
$video->url,
|
|
||||||
'--pageUrl', $video->webpage_url
|
|
||||||
)
|
|
||||||
);
|
|
||||||
if (isset($video->player_url)) {
|
|
||||||
$builder->add('--swfVfy');
|
|
||||||
$builder->add($video->player_url);
|
|
||||||
}
|
|
||||||
if (isset($video->flash_version)) {
|
|
||||||
$builder->add('--flashVer');
|
|
||||||
$builder->add($video->flash_version);
|
|
||||||
}
|
|
||||||
if (isset($video->play_path)) {
|
|
||||||
$builder->add('--playpath');
|
|
||||||
$builder->add($video->play_path);
|
|
||||||
}
|
|
||||||
foreach ($video->rtmp_conn as $conn) {
|
|
||||||
$builder->add('--conn');
|
|
||||||
$builder->add($conn);
|
|
||||||
}
|
|
||||||
$chain = new Chain($builder->getProcess());
|
|
||||||
$chain->add('|', $avconvProc);
|
|
||||||
} else {
|
|
||||||
if (!shell_exec('which curl')) {
|
|
||||||
throw(new \Exception('Can\'t find curl'));
|
|
||||||
}
|
|
||||||
$chain = new Chain(
|
|
||||||
ProcessBuilder::create(
|
|
||||||
array_merge(
|
|
||||||
array(
|
|
||||||
'curl',
|
|
||||||
'--silent',
|
|
||||||
'--location',
|
|
||||||
'--user-agent', $video->http_headers->{'User-Agent'},
|
|
||||||
$video->url
|
|
||||||
),
|
|
||||||
$this->config->curl_params
|
|
||||||
)
|
|
||||||
)
|
|
||||||
);
|
|
||||||
$chain->add('|', $avconvProc);
|
|
||||||
}
|
|
||||||
if ($request->isGet()) {
|
if ($request->isGet()) {
|
||||||
$response = $response->withBody(new Stream(popen($chain->getProcess()->getCommandLine(), 'r')));
|
$process = $this->download->getConversionProcess($params["url"], 'bestaudio/best');
|
||||||
|
$response = $response->withBody(new Stream($process));
|
||||||
}
|
}
|
||||||
return $response;
|
return $response;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue