diff options
| -rwxr-xr-x | src/bin/roadrunner | 122 |
1 files changed, 96 insertions, 26 deletions
diff --git a/src/bin/roadrunner b/src/bin/roadrunner index 56a99886..3b521c9b 100755 --- a/src/bin/roadrunner +++ b/src/bin/roadrunner @@ -52,7 +52,7 @@ use Symfony\Component\Console\Input\InputInterface; use Symfony\Component\Console\Output\OutputInterface; use Symfony\Component\Console\Question\ConfirmationQuestion; -class RoadRunnerCLIHelper +class RRHelper { /** * Returns version of RoadRunner based on build.sh file @@ -68,7 +68,7 @@ class RoadRunnerCLIHelper $line = fgets($fileResource, 4096); $matches = []; if (preg_match("/^RR_VERSION=(.*)/", $line, $matches)) { - return $matches[1]; + return trim($matches[1]); } } fclose($fileResource); @@ -99,18 +99,93 @@ class RoadRunnerCLIHelper } /** + * @return string + * @throws Exception + */ + public static function getSignature(): string + { + return 'roadrunner-' . self::getVersion() . '-' . self::getOSType() . '-amd64'; + } + + /** * Returns generated URL to zip file on GitHub with binary file * * @return string URL * @throws Exception */ - public static function getBinaryDownloadUrl() + public static function getBinaryDownloadUrl(): string + { + $ext = '.zip'; + if (self::getOSType() == 'linux') { + $ext = '.tar.gz'; + } + + // todo: support linux + return 'https://github.com/spiral/roadrunner/releases/download/v' + . static::getVersion() . '/' . self::getSignature() + . $ext; + } + + /** + * Extracts the roadrunner RR binary into given location. + * + * @param string $archive + * @param string $target + * @throws Exception + */ + public static function extractBinary(string $archive, string $target) + { + if (self::getOSType() !== 'linux') { + self::extractZIP($archive, $target); + } else { + self::extractTAR($archive, $target); + } + } + + /** + * @param string $archive + * @param string $target + * @throws Exception + */ + protected function extractZIP(string $archive, string $target) + { + $zip = new ZipArchive(); + $zip->open($archive); + + $name = self::getSignature() . '/rr'; + if (self::getOSType() == 'windows') { + $name .= '.exe'; + } + + $stream = $zip->getStream($name); + if (!is_resource($stream)) { + return; + } + + $to = fopen($target, 'w'); + stream_copy_to_stream($stream, $to); + fclose($to); + + $zip->close(); + } + + /** + * @param string $archive + * @param string $target + * @throws Exception + */ + protected function extractTAR(string $archive, string $target) { - return 'https://github.com/spiral/roadrunner/releases/download/v' . static::getVersion() . '/roadrunner-' . static::getVersion() . '-' . static::getOSType() . '-amd64.zip'; + $arch = new PharData($archive); + $arch->extractTo('./', self::getSignature() . '/rr'); + + copy('./' . self::getSignature() . '/rr', $target); + unlink('./' . self::getSignature() . '/rr'); + rmdir('./' . self::getSignature()); } } -(new Application('RoadRunner', RoadRunnerCLIHelper::getVersion())) +(new Application('RoadRunner', RRHelper::getVersion())) ->register('get') ->setDescription("Install or update RoadRunner binaries in specified folder (current folder by default)") ->addOption('location', 'l', InputArgument::OPTIONAL, 'destination folder', '.') @@ -118,6 +193,9 @@ class RoadRunnerCLIHelper $output->writeln('<info>Updating binary file of RoadRunner</info>'); $finalFile = $input->getOption('location') . DIRECTORY_SEPARATOR . 'rr'; + if (RRHelper::getOSType() == 'windows') { + $finalFile .= '.exe'; + } if (is_file($finalFile)) { $output->writeln('<error>RoadRunner binary file already exists!</error>'); @@ -129,18 +207,19 @@ class RoadRunnerCLIHelper } } - $output->writeln('<info>Downloading RoadRunner archive for ' . RoadRunnerCLIHelper::getOSType() . '</info>'); + $output->writeln('<info>Downloading RoadRunner archive for <fg=cyan>' . ucfirst(RRHelper::getOSType()) . '</fg=cyan></info>'); $progressBar = new ProgressBar($output); $progressBar->setFormat('verbose'); - $progressBar->start(); $zipFileName = tempnam('.', "rr_zip"); $zipFile = fopen($zipFileName, "w+"); $curlResource = curl_init(); - curl_setopt($curlResource, CURLOPT_URL, RoadRunnerCLIHelper::getBinaryDownloadUrl()); + + curl_setopt($curlResource, CURLOPT_URL, RRHelper::getBinaryDownloadUrl()); curl_setopt($curlResource, CURLOPT_RETURNTRANSFER, true); curl_setopt($curlResource, CURLOPT_BINARYTRANSFER, true); + curl_setopt($curlResource, CURLOPT_SSL_VERIFYPEER, false); curl_setopt($curlResource, CURLOPT_FOLLOWLOCATION, true); curl_setopt($curlResource, CURLOPT_FILE, $zipFile); curl_setopt($curlResource, CURLOPT_PROGRESSFUNCTION, @@ -148,6 +227,11 @@ class RoadRunnerCLIHelper if ($download_size == 0) { return; } + + if ($progressBar->getStartTime() === 0) { + $progressBar->start(); + } + $progressBar->setFormat('[%bar%] %percent:3s%% %elapsed:6s%/%estimated:-6s% ' . intval($download_size / 1024) . 'KB'); $progressBar->setMaxSteps($download_size); $progressBar->setProgress($downloaded); @@ -161,29 +245,15 @@ class RoadRunnerCLIHelper $progressBar->finish(); $output->writeln(""); - $output->writeln('<info>Unpacking ' . basename(RoadRunnerCLIHelper::getBinaryDownloadUrl()) . '</info>'); + $output->writeln('<info>Unpacking <comment>' . basename(RRHelper::getBinaryDownloadUrl()) . '</comment></info>'); - $zipArchive = new ZipArchive(); - $zipArchive->open($zipFileName); - $fileStreamFromZip = $zipArchive->getStream( - 'roadrunner-' . RoadRunnerCLIHelper::getVersion() . '-' . RoadRunnerCLIHelper::getOSType() . '-amd64/rr' - ); - $finalFileResource = fopen($finalFile, 'w'); + RRHelper::extractBinary($zipFileName, $finalFile); + unlink($zipFileName); - if (!$fileStreamFromZip) { + if (!file_exists($finalFile) || filesize($finalFile) === 0) { throw new Exception('Unable to extract the file.'); } - while (!feof($fileStreamFromZip)) { - fwrite($finalFileResource, fread($fileStreamFromZip, 8192)); - } - - fclose($fileStreamFromZip); - fclose($finalFileResource); - $zipArchive->extractTo('.', []); - $zipArchive->close(); - unlink($zipFileName); - chmod($finalFile, 0755); $output->writeln('<info>Binary file updated!</info>'); }) |