rr composer bin moved from src

1 files changed, 292 insertions, 0 deletions

diff --git a/bin/rr b/bin/rr
new file mode 100644
index 00000000..71670ff4
--- /dev/null
+++ b/bin/rr
@@ -0,0 +1,292 @@
+#!/usr/bin/env php
+<?php
+/**
+ * RoadRunner
+ * High-performance PHP process supervisor and load balancer written in Go
+ *
+ * This file responsive for cli commands
+ */
+declare(strict_types=1);
+
+foreach ([
+             __DIR__ . '/../../../autoload.php',
+             __DIR__ . '/../vendor/autoload.php',
+             __DIR__ . '/vendor/autoload.php'
+         ] as $file) {
+    if (file_exists($file)) {
+        define('RR_COMPOSER_INSTALL', $file);
+
+        break;
+    }
+}
+
+unset($file);
+
+if (!defined('RR_COMPOSER_INSTALL')) {
+    fwrite(
+        STDERR,
+        'You need to set up the project dependencies using Composer:' . PHP_EOL . PHP_EOL .
+        '    composer install' . PHP_EOL . PHP_EOL .
+        'You can learn all about Composer on https://getcomposer.org/.' . PHP_EOL
+    );
+
+    die(1);
+}
+
+if (RRHelper::getOSType() !== 'linux' && !class_exists('ZipArchive')) {
+    fwrite(STDERR, 'Extension `php-zip` is required.' . PHP_EOL);
+    die(1);
+}
+
+if (!function_exists('curl_init')) {
+    fwrite(STDERR, 'Extension `php-curl` is required.' . PHP_EOL);
+    die(1);
+}
+
+require RR_COMPOSER_INSTALL;
+
+use Symfony\Component\Console\Application;
+use Symfony\Component\Console\Helper\ProgressBar;
+use Symfony\Component\Console\Input\InputArgument;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Console\Question\ConfirmationQuestion;
+
+class RRHelper
+{
+    /**
+     * Returns version of RoadRunner based on build.sh file
+     *
+     * @return string Version of RoadRunner
+     * @throws Exception
+     */
+    public static function getVersion(): string
+    {
+        $file = __DIR__ . DIRECTORY_SEPARATOR . '..' . DIRECTORY_SEPARATOR . 'build.sh';
+        $fileResource = fopen($file, 'r') or die(1);
+        while (!feof($fileResource)) {
+            $line = fgets($fileResource, 4096);
+            $matches = [];
+            if (preg_match("/^RR_VERSION=(.*)/", $line, $matches)) {
+                return trim($matches[1]);
+            }
+        }
+        fclose($fileResource);
+        throw new Exception("Can't find version of RoadRunner");
+    }
+
+    /**
+     * Returns OS Type for filename
+     *
+     * @return string OS Type
+     */
+    public static function getOSType(): string
+    {
+        switch (PHP_OS) {
+            case 'Darwin':
+                return 'darwin';
+            case 'Linux':
+                return 'linux';
+            case 'FreeBSD':
+                return 'freebsd';
+            case 'WIN32':
+            case 'WINNT':
+            case 'Windows':
+                return 'windows';
+            default:
+                return 'linux';
+        }
+    }
+
+    /**
+     * @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(): string
+    {
+        $ext = '.zip';
+        if (self::getOSType() == 'linux') {
+            $ext = '.tar.gz';
+        }
+
+        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 static 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 static function extractTAR(string $archive, string $target)
+    {
+        $arch = new PharData($archive);
+        $arch->extractTo('./', self::getSignature() . '/rr');
+
+        copy('./' . self::getSignature() . '/rr', $target);
+        unlink('./' . self::getSignature() . '/rr');
+        rmdir('./' . self::getSignature());
+    }
+}
+
+(new Application('RoadRunner', RRHelper::getVersion()))
+    ->register('get-binary')
+    ->setDescription("Install or update RoadRunner binaries in specified folder (current folder by default)")
+    ->addOption('location', 'l', InputArgument::OPTIONAL, 'destination folder', '.')
+    ->setCode(function (InputInterface $input, OutputInterface $output) {
+        $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>');
+            $helper = $this->getHelper('question');
+            $question = new ConfirmationQuestion('Do you want overwrite it? [Y/n] ');
+
+            if (!$helper->ask($input, $output, $question)) {
+                return;
+            }
+        }
+
+        $output->writeln('<info>Downloading RoadRunner archive for <fg=cyan>' . ucfirst(RRHelper::getOSType()) . '</fg=cyan></info>');
+
+        $progressBar = new ProgressBar($output);
+        $progressBar->setFormat('verbose');
+
+        $zipFileName = tempnam('.', "rr_zip");
+        if (RRHelper::getOSType() == 'linux') {
+            $zipFileName .= '.tar.gz';
+        }
+
+        $zipFile = fopen($zipFileName, "w+");
+        $curlResource = curl_init();
+
+        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,
+            function ($resource, $download_size, $downloaded, $upload_size, $uploaded) use (&$progressBar, $output) {
+                if ($download_size == 0) {
+                    return;
+                }
+
+                if ($progressBar->getStartTime() === 0) {
+                    $progressBar->start();
+                }
+
+                if ($progressBar->getMaxSteps() != $download_size) {
+                    /**
+                     * Workaround for symfony < 4.1.x, for example PHP 7.0 will use 3.x
+                     * feature #26449 Make ProgressBar::setMaxSteps public (ostrolucky)
+                     */
+                    $progressBar = new ProgressBar($output, $download_size);
+                }
+
+                $progressBar->setFormat('[%bar%] %percent:3s%% %elapsed:6s%/%estimated:-6s% ' . intval($download_size / 1024) . 'KB');
+                $progressBar->setProgress($downloaded);
+            });
+        curl_setopt($curlResource, CURLOPT_NOPROGRESS, false); // needed to make progress function work
+        curl_setopt($curlResource, CURLOPT_HEADER, 0);
+        curl_exec($curlResource);
+        curl_close($curlResource);
+        fclose($zipFile);
+
+        $progressBar->finish();
+        $output->writeln("");
+
+        $output->writeln('<info>Unpacking <comment>' . basename(RRHelper::getBinaryDownloadUrl()) . '</comment></info>');
+
+        RRHelper::extractBinary($zipFileName, $finalFile);
+        unlink($zipFileName);
+
+        if (!file_exists($finalFile) || filesize($finalFile) === 0) {
+            throw new Exception('Unable to extract the file.');
+        }
+
+        chmod($finalFile, 0755);
+        $output->writeln('<info>Binary file updated!</info>');
+    })
+    ->getApplication()
+    ->register("init-config")
+    ->setDescription("Inits default .rr.yaml config in specified folder (current folder by default)")
+    ->addOption('location', 'l', InputArgument::OPTIONAL, 'destination folder', '.')
+    ->setCode(function (InputInterface $input, OutputInterface $output) {
+        if (is_file($input->getOption('location') . DIRECTORY_SEPARATOR . '.rr.yaml')) {
+            $output->writeln('<error>Config file already exists!</error>');
+            $helper = $this->getHelper('question');
+            $question = new ConfirmationQuestion('Do you want overwrite it? [Y/n] ');
+
+            if (!$helper->ask($input, $output, $question)) {
+                return;
+            }
+        }
+
+        copy(
+            __DIR__ . DIRECTORY_SEPARATOR . '..' . DIRECTORY_SEPARATOR . '..' . DIRECTORY_SEPARATOR . '.rr.yaml',
+            $input->getOption('location') . DIRECTORY_SEPARATOR . '.rr.yaml'
+        );
+        $output->writeln('<info>Config file created!</info>');
+    })
+    ->getApplication()
+    ->run();