136 lines
		
	
	
		
			4.8 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			136 lines
		
	
	
		
			4.8 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
 | 
						|
use App\Actions\RemoteProcess\DispatchRemoteProcess;
 | 
						|
use App\Data\RemoteProcessArgs;
 | 
						|
use App\Enums\ActivityTypes;
 | 
						|
use App\Models\Server;
 | 
						|
use Illuminate\Database\Eloquent\Model;
 | 
						|
use Illuminate\Support\Collection;
 | 
						|
use Illuminate\Support\Facades\Log;
 | 
						|
use Illuminate\Support\Facades\Process;
 | 
						|
use Illuminate\Support\Facades\Storage;
 | 
						|
use Spatie\Activitylog\Contracts\Activity;
 | 
						|
 | 
						|
if (!function_exists('remoteProcess')) {
 | 
						|
    /**
 | 
						|
     * Run a Remote Process, which SSH's asynchronously into a machine to run the command(s).
 | 
						|
     * @TODO Change 'root' to 'coolify' when it's able to run Docker commands without sudo
 | 
						|
     *
 | 
						|
     */
 | 
						|
    function remoteProcess(
 | 
						|
        array           $command,
 | 
						|
        Server          $server,
 | 
						|
        ?string         $deployment_uuid = null,
 | 
						|
        ?Model          $model = null,
 | 
						|
    ): Activity {
 | 
						|
 | 
						|
        $command_string = implode("\n", $command);
 | 
						|
 | 
						|
        // @TODO: Check if the user has access to this server
 | 
						|
        // checkTeam($server->team_id);
 | 
						|
 | 
						|
        $private_key_location = savePrivateKeyForServer($server);
 | 
						|
 | 
						|
        return resolve(DispatchRemoteProcess::class, [
 | 
						|
            'remoteProcessArgs' => new RemoteProcessArgs(
 | 
						|
                server_ip: $server->ip,
 | 
						|
                private_key_location: $private_key_location,
 | 
						|
                deployment_uuid: $deployment_uuid,
 | 
						|
                command: <<<EOT
 | 
						|
                {$command_string}
 | 
						|
                EOT,
 | 
						|
                port: $server->port,
 | 
						|
                user: $server->user,
 | 
						|
                type: $deployment_uuid ? ActivityTypes::DEPLOYMENT->value : ActivityTypes::REMOTE_PROCESS->value,
 | 
						|
                model: $model,
 | 
						|
            ),
 | 
						|
        ])();
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
// function checkTeam(string $team_id)
 | 
						|
// {
 | 
						|
//     $found_team = auth()->user()->teams->pluck('id')->contains($team_id);
 | 
						|
//     if (!$found_team) {
 | 
						|
//         throw new \RuntimeException('You do not have access to this server.');
 | 
						|
//     }
 | 
						|
// }
 | 
						|
 | 
						|
if (!function_exists('savePrivateKeyForServer')) {
 | 
						|
    function savePrivateKeyForServer(Server $server)
 | 
						|
    {
 | 
						|
        $temp_file = 'id.rsa_' . 'root' . '@' . $server->ip;
 | 
						|
        Storage::disk('local')->put($temp_file, $server->privateKey->private_key, 'private');
 | 
						|
        return '/var/www/html/storage/app/' . $temp_file;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
if (!function_exists('generateSshCommand')) {
 | 
						|
    function generateSshCommand(string $private_key_location, string $server_ip, string $user, string $port, string $command, bool $isMux = false)
 | 
						|
    {
 | 
						|
        $delimiter = 'EOF-COOLIFY-SSH';
 | 
						|
        Storage::disk('local')->makeDirectory('.ssh');
 | 
						|
        $ssh_command = "ssh ";
 | 
						|
        if ($isMux) {
 | 
						|
            $ssh_command .= '-o ControlMaster=auto -o ControlPersist=1m -o ControlPath=/var/www/html/storage/app/.ssh/ssh_mux_%h_%p_%r ';
 | 
						|
        }
 | 
						|
        $ssh_command .= "-i {$private_key_location} "
 | 
						|
            . '-o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null '
 | 
						|
            . '-o PasswordAuthentication=no '
 | 
						|
            . '-o RequestTTY=no '
 | 
						|
            . '-o LogLevel=ERROR '
 | 
						|
            . "-p {$port} "
 | 
						|
            . "{$user}@{$server_ip} "
 | 
						|
            . " 'bash -se' << \\$delimiter" . PHP_EOL
 | 
						|
            . $command . PHP_EOL
 | 
						|
            . $delimiter;
 | 
						|
 | 
						|
        return $ssh_command;
 | 
						|
    }
 | 
						|
}
 | 
						|
if (!function_exists('formatDockerCmdOutputToJson')) {
 | 
						|
    function formatDockerCmdOutputToJson($rawOutput): Collection
 | 
						|
    {
 | 
						|
        $outputLines = explode(PHP_EOL, $rawOutput);
 | 
						|
 | 
						|
        return collect($outputLines)
 | 
						|
            ->reject(fn ($line) => empty($line))
 | 
						|
            ->map(fn ($outputLine) => json_decode($outputLine, true, flags: JSON_THROW_ON_ERROR));
 | 
						|
    }
 | 
						|
}
 | 
						|
if (!function_exists('formatDockerLabelsToJson')) {
 | 
						|
    function formatDockerLabelsToJson($rawOutput): Collection
 | 
						|
    {
 | 
						|
        $outputLines = explode(PHP_EOL, $rawOutput);
 | 
						|
 | 
						|
        return collect($outputLines)
 | 
						|
            ->reject(fn ($line) => empty($line))
 | 
						|
            ->map(function ($outputLine) {
 | 
						|
                $outputArray = explode(',', $outputLine);
 | 
						|
                return collect($outputArray)
 | 
						|
                    ->map(function ($outputLine) {
 | 
						|
                        return explode('=', $outputLine);
 | 
						|
                    })
 | 
						|
                    ->mapWithKeys(function ($outputLine) {
 | 
						|
                        return [$outputLine[0] => $outputLine[1]];
 | 
						|
                    });
 | 
						|
            })[0];
 | 
						|
    }
 | 
						|
}
 | 
						|
if (!function_exists('runRemoteCommandSync')) {
 | 
						|
    function runRemoteCommandSync($server, array $command)
 | 
						|
    {
 | 
						|
        $command_string = implode("\n", $command);
 | 
						|
        $private_key_location = savePrivateKeyForServer($server);
 | 
						|
        $ssh_command = generateSshCommand($private_key_location, $server->ip, $server->user, $server->port, $command_string);
 | 
						|
        $process = Process::run($ssh_command);
 | 
						|
        $output = trim($process->output());
 | 
						|
        $exitCode = $process->exitCode();
 | 
						|
        if ($exitCode !== 0) {
 | 
						|
            Log::error($output);
 | 
						|
            throw new \RuntimeException('There was an error running the command.');
 | 
						|
        }
 | 
						|
        return $output;
 | 
						|
    }
 | 
						|
}
 |