122 lines
		
	
	
		
			4.9 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			122 lines
		
	
	
		
			4.9 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
 | 
						|
namespace App\Livewire\Project\Database;
 | 
						|
 | 
						|
use Livewire\Component;
 | 
						|
use App\Models\Server;
 | 
						|
use Illuminate\Support\Facades\Storage;
 | 
						|
 | 
						|
class Import extends Component
 | 
						|
{
 | 
						|
    public bool $unsupported = false;
 | 
						|
    public $resource;
 | 
						|
    public $parameters;
 | 
						|
    public $containers;
 | 
						|
    public bool $scpInProgress = false;
 | 
						|
    public bool $importRunning = false;
 | 
						|
 | 
						|
    public ?string $filename = null;
 | 
						|
    public ?string $filesize = null;
 | 
						|
    public bool $isUploading = false;
 | 
						|
    public int $progress = 0;
 | 
						|
    public bool $error = false;
 | 
						|
 | 
						|
    public Server $server;
 | 
						|
    public string $container;
 | 
						|
    public array $importCommands = [];
 | 
						|
    public string $postgresqlRestoreCommand = 'pg_restore -U $POSTGRES_USER -d $POSTGRES_DB';
 | 
						|
    public string $mysqlRestoreCommand = 'mysql -u $MYSQL_USER -p$MYSQL_PASSWORD $MYSQL_DATABASE';
 | 
						|
    public string $mariadbRestoreCommand = 'mariadb -u $MARIADB_USER -p$MARIADB_PASSWORD $MARIADB_DATABASE';
 | 
						|
    public string $mongodbRestoreCommand = 'mongorestore --authenticationDatabase=admin --username $MONGO_INITDB_ROOT_USERNAME --password $MONGO_INITDB_ROOT_PASSWORD --uri mongodb://localhost:27017 --gzip --archive=';
 | 
						|
 | 
						|
    public function getListeners()
 | 
						|
    {
 | 
						|
        $userId = auth()->user()->id;
 | 
						|
        return [
 | 
						|
            "echo-private:user.{$userId},DatabaseStatusChanged" => '$refresh',
 | 
						|
        ];
 | 
						|
    }
 | 
						|
    public function mount()
 | 
						|
    {
 | 
						|
        $this->parameters = get_route_parameters();
 | 
						|
        $this->getContainers();
 | 
						|
    }
 | 
						|
 | 
						|
    public function getContainers()
 | 
						|
    {
 | 
						|
        $this->containers = collect();
 | 
						|
        if (!data_get($this->parameters, 'database_uuid')) {
 | 
						|
            abort(404);
 | 
						|
        }
 | 
						|
        $resource = getResourceByUuid($this->parameters['database_uuid'], data_get(auth()->user()->currentTeam(), 'id'));
 | 
						|
        if (is_null($resource)) {
 | 
						|
            abort(404);
 | 
						|
        }
 | 
						|
        $this->resource = $resource;
 | 
						|
        $this->server = $this->resource->destination->server;
 | 
						|
        $this->container = $this->resource->uuid;
 | 
						|
        if (str(data_get($this, 'resource.status'))->startsWith('running')) {
 | 
						|
            $this->containers->push($this->container);
 | 
						|
        }
 | 
						|
 | 
						|
        if (
 | 
						|
            $this->resource->getMorphClass() == 'App\Models\StandaloneRedis' ||
 | 
						|
            $this->resource->getMorphClass() == 'App\Models\StandaloneKeydb' ||
 | 
						|
            $this->resource->getMorphClass() == 'App\Models\StandaloneDragonfly' ||
 | 
						|
            $this->resource->getMorphClass() == 'App\Models\StandaloneClickhouse'
 | 
						|
        ) {
 | 
						|
            $this->unsupported = true;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    public function runImport()
 | 
						|
    {
 | 
						|
 | 
						|
        if ($this->filename == '') {
 | 
						|
            $this->dispatch('error', 'Please select a file to import.');
 | 
						|
            return;
 | 
						|
        }
 | 
						|
        try {
 | 
						|
            $uploadedFilename = "upload/{$this->resource->uuid}/restore";
 | 
						|
            $path = Storage::path($uploadedFilename);
 | 
						|
            if (!Storage::exists($uploadedFilename)) {
 | 
						|
                $this->dispatch('error', 'The file does not exist or has been deleted.');
 | 
						|
                return;
 | 
						|
            }
 | 
						|
            $tmpPath = '/tmp/' . basename($uploadedFilename);
 | 
						|
            instant_scp($path, $tmpPath, $this->server);
 | 
						|
            Storage::delete($uploadedFilename);
 | 
						|
            $this->importCommands[] = "docker cp {$tmpPath} {$this->container}:{$tmpPath}";
 | 
						|
 | 
						|
            switch ($this->resource->getMorphClass()) {
 | 
						|
                case 'App\Models\StandaloneMariadb':
 | 
						|
                    $this->importCommands[] = "docker exec {$this->container} sh -c '{$this->mariadbRestoreCommand} < {$tmpPath}'";
 | 
						|
                    $this->importCommands[] = "rm {$tmpPath}";
 | 
						|
                    break;
 | 
						|
                case 'App\Models\StandaloneMysql':
 | 
						|
                    $this->importCommands[] = "docker exec {$this->container} sh -c '{$this->mysqlRestoreCommand} < {$tmpPath}'";
 | 
						|
                    $this->importCommands[] = "rm {$tmpPath}";
 | 
						|
                    break;
 | 
						|
                case 'App\Models\StandalonePostgresql':
 | 
						|
                    $this->importCommands[] = "docker exec {$this->container} sh -c '{$this->postgresqlRestoreCommand} {$tmpPath}'";
 | 
						|
                    $this->importCommands[] = "rm {$tmpPath}";
 | 
						|
                    break;
 | 
						|
                case 'App\Models\StandaloneMongodb':
 | 
						|
                    $this->importCommands[] = "docker exec {$this->container} sh -c '{$this->mongodbRestoreCommand}{$tmpPath}'";
 | 
						|
                    $this->importCommands[] = "rm {$tmpPath}";
 | 
						|
                    break;
 | 
						|
            }
 | 
						|
 | 
						|
            $this->importCommands[] = "docker exec {$this->container} sh -c 'rm {$tmpPath}'";
 | 
						|
            $this->importCommands[] = "docker exec {$this->container} sh -c 'echo \"Import finished with exit code $?\"'";
 | 
						|
 | 
						|
            if (!empty($this->importCommands)) {
 | 
						|
                $activity = remote_process($this->importCommands, $this->server, ignore_errors: true);
 | 
						|
                $this->dispatch('activityMonitor', $activity->id);
 | 
						|
            }
 | 
						|
        } catch (\Throwable $e) {
 | 
						|
            return handleError($e, $this);
 | 
						|
        }
 | 
						|
    }
 | 
						|
}
 |