From fb98b1892d668e1666c7eff9716a27b2fb964441 Mon Sep 17 00:00:00 2001
From: Dane Everitt <dane@daneeveritt.com>
Date: Sat, 23 Jan 2021 16:07:49 -0800
Subject: [PATCH] Add simple logic to download and unpack the archive

---
 app/Console/Commands/UpgradeCommand.php | 33 ++++++++++++++++++++++---
 1 file changed, 29 insertions(+), 4 deletions(-)

diff --git a/app/Console/Commands/UpgradeCommand.php b/app/Console/Commands/UpgradeCommand.php
index 9ca38515d..6da06038c 100644
--- a/app/Console/Commands/UpgradeCommand.php
+++ b/app/Console/Commands/UpgradeCommand.php
@@ -10,11 +10,13 @@ use Symfony\Component\Console\Helper\ProgressBar;
 
 class UpgradeCommand extends Command
 {
-    /** @var string */
-    protected $signature = 'p:upgrade {--force}';
+    protected const DEFAULT_URL = 'https://github.com/pterodactyl/panel/releases/latest/download/panel.tar.gz';
 
     /** @var string */
-    protected $description = 'Executes the commands necessary for getting Pterodactyl operational after installing new files.';
+    protected $signature = 'p:upgrade {--url=} {--skip-download}';
+
+    /** @var string */
+    protected $description = 'Downloads a new archive for Pterodactyl from GitHub and then executes the normal upgrade commands.';
 
     /**
      * Executes an upgrade command which will run through all of our standard
@@ -28,16 +30,39 @@ class UpgradeCommand extends Command
      */
     public function handle()
     {
+        $skipDownload = $this->option('skip-download');
+
+        if (!$skipDownload) {
+            $this->output->warning('This command does not verify the integrity of downloaded assets. Please ensure that you trust the download source before continuing. If you do not wish to download an archive, please indicate that using the --skip-download flag, or answering "no" to the question below.');
+            $this->output->comment('Download Source (set with --url=):');
+            $this->line($this->option('url') ?? self::DEFAULT_URL);
+        }
+
         if ($this->input->isInteractive()) {
+            if (!$skipDownload) {
+                $skipDownload = !$this->confirm('Would you like to download and unpack the archive files for the latest version?', true);
+            }
+
             if (!$this->confirm('Are you sure you want to run the upgrade process for your Panel?')) {
                 return;
             }
         }
 
         ini_set('output_buffering', 0);
-        $bar = $this->output->createProgressBar(8);
+        $bar = $this->output->createProgressBar($skipDownload ? 8 : 9);
         $bar->start();
 
+        if (!$skipDownload) {
+            $this->withProgress($bar, function () {
+                $url = $this->option('url') ?? self::DEFAULT_URL;
+                $this->line("\$upgrader> curl -L \"$url\" | tar -xzv");
+                $process = Process::fromShellCommandline("curl -L \"$url\" | tar -xzvf");
+                $process->run(function ($type, $buffer) {
+                    $this->{$type === Process::ERR ? 'error' : 'line'}($buffer);
+                });
+            });
+        }
+
         $this->withProgress($bar, function () {
             $this->line('$upgrader> php artisan down');
             $this->call('down');