From 58523b0000ada9554093702dcd47e0e87b470d82 Mon Sep 17 00:00:00 2001
From: Andras Bacsai <andras.bacsai@gmail.com>
Date: Tue, 16 Apr 2024 21:22:57 +0200
Subject: [PATCH] Refactor code to add sudo prefix for certain commands

---
 app/Traits/ExecuteRemoteCommand.php |  9 ++++++++-
 bootstrap/helpers/shared.php        | 23 ++++++++++++++++++++++-
 2 files changed, 30 insertions(+), 2 deletions(-)

diff --git a/app/Traits/ExecuteRemoteCommand.php b/app/Traits/ExecuteRemoteCommand.php
index 529dacd7a..0a83e9f09 100644
--- a/app/Traits/ExecuteRemoteCommand.php
+++ b/app/Traits/ExecuteRemoteCommand.php
@@ -34,7 +34,14 @@ public function execute_remote_command(...$commands)
             $ignore_errors = data_get($single_command, 'ignore_errors', false);
             $append = data_get($single_command, 'append', true);
             $this->save = data_get($single_command, 'save');
-
+            if ($this->server->isNonRoot()) {
+                if (str($command)->startsWith('docker exec')) {
+                    $command = str($command)->replace('docker exec', 'sudo docker exec');
+                } else {
+                    $command = parseLineForSudo($command, $this->server);
+                }
+            }
+            ray($command);
             $remote_command = generateSshCommand($this->server, $command);
             $process = Process::timeout(3600)->idleTimeout(3600)->start($remote_command, function (string $type, string $output) use ($command, $hidden, $customType, $append) {
                 $output = Str::of($output)->trim();
diff --git a/bootstrap/helpers/shared.php b/bootstrap/helpers/shared.php
index 582674685..dae9542f6 100644
--- a/bootstrap/helpers/shared.php
+++ b/bootstrap/helpers/shared.php
@@ -1951,7 +1951,8 @@ function check_domain_usage(ServiceApplication|Application|null $resource = null
     }
 }
 
-function parseCommandsByLineForSudo(Collection $commands, Server $server): array {
+function parseCommandsByLineForSudo(Collection $commands, Server $server): array
+{
     $commands = $commands->map(function ($line) {
         if (!str($line)->startSwith('cd') && !str($line)->startSwith('command')) {
             return "sudo $line";
@@ -1979,3 +1980,23 @@ function parseCommandsByLineForSudo(Collection $commands, Server $server): array
 
     return $commands->toArray();
 }
+function parseLineForSudo(string $command, Server $server): string
+{
+    if (!str($command)->startSwith('cd') && !str($command)->startSwith('command')) {
+        $command = "sudo $command";
+    }
+    if (Str::startsWith($command, 'sudo mkdir -p')) {
+        $command =  "$command && sudo chown -R $server->user:$server->user " . Str::after($command, 'sudo mkdir -p') . ' && sudo chmod -R o-rwx ' . Str::after($command, 'sudo mkdir -p');
+    }
+    if (str($command)->contains('$(') || str($command)->contains('`')) {
+        $command = str($command)->replace('$(', '$(sudo ')->replace('`', '`sudo ')->value();
+    }
+    if (str($command)->contains('||')) {
+        $command = str($command)->replace('||', '|| sudo ')->value();
+    }
+    if (str($command)->contains('&&')) {
+        $command = str($command)->replace('&&', '&& sudo ')->value();
+    }
+
+    return $command;
+}