347 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			347 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
 | 
						|
namespace App\Actions\Database;
 | 
						|
 | 
						|
use App\Helpers\SslHelper;
 | 
						|
use App\Models\SslCertificate;
 | 
						|
use App\Models\StandalonePostgresql;
 | 
						|
use Lorisleiva\Actions\Concerns\AsAction;
 | 
						|
use Symfony\Component\Yaml\Yaml;
 | 
						|
 | 
						|
class StartPostgresql
 | 
						|
{
 | 
						|
    use AsAction;
 | 
						|
 | 
						|
    public StandalonePostgresql $database;
 | 
						|
 | 
						|
    public array $commands = [];
 | 
						|
 | 
						|
    public array $init_scripts = [];
 | 
						|
 | 
						|
    public string $configuration_dir;
 | 
						|
 | 
						|
    public string $volume_configuration_dir;
 | 
						|
 | 
						|
    private ?SslCertificate $ssl_certificate = null;
 | 
						|
 | 
						|
    public function handle(StandalonePostgresql $database)
 | 
						|
    {
 | 
						|
        $this->database = $database;
 | 
						|
        $container_name = $this->database->uuid;
 | 
						|
        $this->configuration_dir = database_configuration_dir().'/'.$container_name;
 | 
						|
        $this->volume_configuration_dir = $this->configuration_dir;
 | 
						|
        if (isDev()) {
 | 
						|
            $this->volume_configuration_dir = '/var/lib/docker/volumes/coolify_dev_coolify_data/_data/databases/'.$container_name;
 | 
						|
        }
 | 
						|
 | 
						|
        $this->commands = [
 | 
						|
            "echo 'Starting database.'",
 | 
						|
            "echo 'Creating directories.'",
 | 
						|
            "mkdir -p $this->configuration_dir",
 | 
						|
            "mkdir -p $this->configuration_dir/docker-entrypoint-initdb.d/",
 | 
						|
            "echo 'Directories created successfully.'",
 | 
						|
        ];
 | 
						|
 | 
						|
        if (! $this->database->enable_ssl) {
 | 
						|
            $this->commands[] = "rm -rf $this->configuration_dir/ssl";
 | 
						|
 | 
						|
            $this->database->sslCertificates()->delete();
 | 
						|
 | 
						|
            $this->database->fileStorages()
 | 
						|
                ->where('resource_type', $this->database->getMorphClass())
 | 
						|
                ->where('resource_id', $this->database->id)
 | 
						|
                ->get()
 | 
						|
                ->filter(function ($storage) {
 | 
						|
                    return in_array($storage->mount_path, [
 | 
						|
                        '/var/lib/postgresql/certs/server.crt',
 | 
						|
                        '/var/lib/postgresql/certs/server.key',
 | 
						|
                    ]);
 | 
						|
                })
 | 
						|
                ->each(function ($storage) {
 | 
						|
                    $storage->delete();
 | 
						|
                });
 | 
						|
        } else {
 | 
						|
            $this->commands[] = "echo 'Setting up SSL for this database.'";
 | 
						|
            $this->commands[] = "mkdir -p $this->configuration_dir/ssl";
 | 
						|
 | 
						|
            $server = $this->database->destination->server;
 | 
						|
            $caCert = SslCertificate::where('server_id', $server->id)->where('is_ca_certificate', true)->first();
 | 
						|
 | 
						|
            if (! $caCert) {
 | 
						|
                $server->generateCaCertificate();
 | 
						|
                $caCert = SslCertificate::where('server_id', $server->id)->where('is_ca_certificate', true)->first();
 | 
						|
            }
 | 
						|
 | 
						|
            if (! $caCert) {
 | 
						|
                $this->dispatch('error', 'No CA certificate found for this database. Please generate a CA certificate for this server in the server/advanced page.');
 | 
						|
 | 
						|
                return;
 | 
						|
            }
 | 
						|
 | 
						|
            $this->ssl_certificate = $this->database->sslCertificates()->first();
 | 
						|
 | 
						|
            if (! $this->ssl_certificate) {
 | 
						|
                $this->commands[] = "echo 'No SSL certificate found, generating new SSL certificate for this database.'";
 | 
						|
                $this->ssl_certificate = SslHelper::generateSslCertificate(
 | 
						|
                    commonName: $this->database->uuid,
 | 
						|
                    resourceType: $this->database->getMorphClass(),
 | 
						|
                    resourceId: $this->database->id,
 | 
						|
                    serverId: $server->id,
 | 
						|
                    caCert: $caCert->ssl_certificate,
 | 
						|
                    caKey: $caCert->ssl_private_key,
 | 
						|
                    configurationDir: $this->configuration_dir,
 | 
						|
                    mountPath: '/var/lib/postgresql/certs',
 | 
						|
                );
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        $persistent_storages = $this->generate_local_persistent_volumes();
 | 
						|
        $persistent_file_volumes = $this->database->fileStorages()->get();
 | 
						|
        $volume_names = $this->generate_local_persistent_volumes_only_volume_names();
 | 
						|
        $environment_variables = $this->generate_environment_variables();
 | 
						|
        $this->generate_init_scripts();
 | 
						|
        $this->add_custom_conf();
 | 
						|
 | 
						|
        $docker_compose = [
 | 
						|
            'services' => [
 | 
						|
                $container_name => [
 | 
						|
                    'image' => $this->database->image,
 | 
						|
                    'container_name' => $container_name,
 | 
						|
                    'environment' => $environment_variables,
 | 
						|
                    'restart' => RESTART_MODE,
 | 
						|
                    'networks' => [
 | 
						|
                        $this->database->destination->network,
 | 
						|
                    ],
 | 
						|
                    'labels' => defaultDatabaseLabels($this->database)->toArray(),
 | 
						|
                    'healthcheck' => [
 | 
						|
                        'test' => [
 | 
						|
                            'CMD-SHELL',
 | 
						|
                            "psql -U {$this->database->postgres_user} -d {$this->database->postgres_db} -c 'SELECT 1' || exit 1",
 | 
						|
                        ],
 | 
						|
                        'interval' => '5s',
 | 
						|
                        'timeout' => '5s',
 | 
						|
                        'retries' => 10,
 | 
						|
                        'start_period' => '5s',
 | 
						|
                    ],
 | 
						|
                    'mem_limit' => $this->database->limits_memory,
 | 
						|
                    'memswap_limit' => $this->database->limits_memory_swap,
 | 
						|
                    'mem_swappiness' => $this->database->limits_memory_swappiness,
 | 
						|
                    'mem_reservation' => $this->database->limits_memory_reservation,
 | 
						|
                    'cpus' => (float) $this->database->limits_cpus,
 | 
						|
                    'cpu_shares' => $this->database->limits_cpu_shares,
 | 
						|
                ],
 | 
						|
            ],
 | 
						|
            'networks' => [
 | 
						|
                $this->database->destination->network => [
 | 
						|
                    'external' => true,
 | 
						|
                    'name' => $this->database->destination->network,
 | 
						|
                    'attachable' => true,
 | 
						|
                ],
 | 
						|
            ],
 | 
						|
        ];
 | 
						|
 | 
						|
        if (filled($this->database->limits_cpuset)) {
 | 
						|
            data_set($docker_compose, "services.{$container_name}.cpuset", $this->database->limits_cpuset);
 | 
						|
        }
 | 
						|
 | 
						|
        if ($this->database->destination->server->isLogDrainEnabled() && $this->database->isLogDrainEnabled()) {
 | 
						|
            $docker_compose['services'][$container_name]['logging'] = generate_fluentd_configuration();
 | 
						|
        }
 | 
						|
 | 
						|
        if (count($this->database->ports_mappings_array) > 0) {
 | 
						|
            $docker_compose['services'][$container_name]['ports'] = $this->database->ports_mappings_array;
 | 
						|
        }
 | 
						|
 | 
						|
        $docker_compose['services'][$container_name]['volumes'] ??= [];
 | 
						|
 | 
						|
        if (count($persistent_storages) > 0) {
 | 
						|
            $docker_compose['services'][$container_name]['volumes'] = array_merge(
 | 
						|
                $docker_compose['services'][$container_name]['volumes'],
 | 
						|
                $persistent_storages
 | 
						|
            );
 | 
						|
        }
 | 
						|
 | 
						|
        if (count($persistent_file_volumes) > 0) {
 | 
						|
            $docker_compose['services'][$container_name]['volumes'] = array_merge(
 | 
						|
                $docker_compose['services'][$container_name]['volumes'],
 | 
						|
                $persistent_file_volumes->map(function ($item) {
 | 
						|
                    return "$item->fs_path:$item->mount_path";
 | 
						|
                })->toArray()
 | 
						|
            );
 | 
						|
        }
 | 
						|
 | 
						|
        if (count($volume_names) > 0) {
 | 
						|
            $docker_compose['volumes'] = $volume_names;
 | 
						|
        }
 | 
						|
 | 
						|
        if (count($this->init_scripts) > 0) {
 | 
						|
            foreach ($this->init_scripts as $init_script) {
 | 
						|
                $docker_compose['services'][$container_name]['volumes'] = array_merge(
 | 
						|
                    $docker_compose['services'][$container_name]['volumes'],
 | 
						|
                    [[
 | 
						|
                        'type' => 'bind',
 | 
						|
                        'source' => $init_script,
 | 
						|
                        'target' => '/docker-entrypoint-initdb.d/'.basename($init_script),
 | 
						|
                        'read_only' => true,
 | 
						|
                    ]]
 | 
						|
                );
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        $command = ['postgres'];
 | 
						|
 | 
						|
        if (filled($this->database->postgres_conf)) {
 | 
						|
            $docker_compose['services'][$container_name]['volumes'] = array_merge(
 | 
						|
                $docker_compose['services'][$container_name]['volumes'],
 | 
						|
                [[
 | 
						|
                    'type' => 'bind',
 | 
						|
                    'source' => $this->volume_configuration_dir.'/custom-postgres.conf',
 | 
						|
                    'target' => '/etc/postgresql/postgresql.conf',
 | 
						|
                    'read_only' => true,
 | 
						|
                ]]
 | 
						|
            );
 | 
						|
            $command = array_merge($command, ['-c', 'config_file=/etc/postgresql/postgresql.conf']);
 | 
						|
        }
 | 
						|
 | 
						|
        if ($this->database->enable_ssl) {
 | 
						|
            $command = array_merge($command, [
 | 
						|
                '-c', 'ssl=on',
 | 
						|
                '-c', 'ssl_cert_file=/var/lib/postgresql/certs/server.crt',
 | 
						|
                '-c', 'ssl_key_file=/var/lib/postgresql/certs/server.key',
 | 
						|
            ]);
 | 
						|
        }
 | 
						|
 | 
						|
        // Add custom docker run options
 | 
						|
        $docker_run_options = convertDockerRunToCompose($this->database->custom_docker_run_options);
 | 
						|
        $docker_compose = generateCustomDockerRunOptionsForDatabases($docker_run_options, $docker_compose, $container_name, $this->database->destination->network);
 | 
						|
 | 
						|
        if (count($command) > 1) {
 | 
						|
            $docker_compose['services'][$container_name]['command'] = $command;
 | 
						|
        }
 | 
						|
 | 
						|
        $docker_compose = Yaml::dump($docker_compose, 10);
 | 
						|
        $this->commands[] = [
 | 
						|
            'transfer_file' => [
 | 
						|
                'content' => $docker_compose,
 | 
						|
                'destination' => "$this->volume_configuration_dir/docker-compose.yml",
 | 
						|
            ],
 | 
						|
        ];
 | 
						|
        $readme = generate_readme_file($this->database->name, now());
 | 
						|
        $this->commands[] = "echo '{$readme}' > $this->configuration_dir/README.md";
 | 
						|
        $this->commands[] = "echo 'Pulling {$database->image} image.'";
 | 
						|
        $this->commands[] = "docker compose -f $this->configuration_dir/docker-compose.yml pull";
 | 
						|
        $this->commands[] = "docker compose -f $this->configuration_dir/docker-compose.yml up -d";
 | 
						|
        if ($this->database->enable_ssl) {
 | 
						|
            $this->commands[] = executeInDocker($this->database->uuid, "chown {$this->database->postgres_user}:{$this->database->postgres_user} /var/lib/postgresql/certs/server.key /var/lib/postgresql/certs/server.crt");
 | 
						|
        }
 | 
						|
        $this->commands[] = "echo 'Database started.'";
 | 
						|
 | 
						|
        return remote_process($this->commands, $database->destination->server, callEventOnFinish: 'DatabaseStatusChanged');
 | 
						|
    }
 | 
						|
 | 
						|
    private function generate_local_persistent_volumes()
 | 
						|
    {
 | 
						|
        $local_persistent_volumes = [];
 | 
						|
        foreach ($this->database->persistentStorages as $persistentStorage) {
 | 
						|
            if ($persistentStorage->host_path !== '' && $persistentStorage->host_path !== null) {
 | 
						|
                $local_persistent_volumes[] = $persistentStorage->host_path.':'.$persistentStorage->mount_path;
 | 
						|
            } else {
 | 
						|
                $volume_name = $persistentStorage->name;
 | 
						|
                $local_persistent_volumes[] = $volume_name.':'.$persistentStorage->mount_path;
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        return $local_persistent_volumes;
 | 
						|
    }
 | 
						|
 | 
						|
    private function generate_local_persistent_volumes_only_volume_names()
 | 
						|
    {
 | 
						|
        $local_persistent_volumes_names = [];
 | 
						|
        foreach ($this->database->persistentStorages as $persistentStorage) {
 | 
						|
            if ($persistentStorage->host_path) {
 | 
						|
                continue;
 | 
						|
            }
 | 
						|
            $name = $persistentStorage->name;
 | 
						|
            $local_persistent_volumes_names[$name] = [
 | 
						|
                'name' => $name,
 | 
						|
                'external' => false,
 | 
						|
            ];
 | 
						|
        }
 | 
						|
 | 
						|
        return $local_persistent_volumes_names;
 | 
						|
    }
 | 
						|
 | 
						|
    private function generate_environment_variables()
 | 
						|
    {
 | 
						|
        $environment_variables = collect();
 | 
						|
        foreach ($this->database->runtime_environment_variables as $env) {
 | 
						|
            $environment_variables->push("$env->key=$env->real_value");
 | 
						|
        }
 | 
						|
 | 
						|
        if ($environment_variables->filter(fn ($env) => str($env)->contains('POSTGRES_USER'))->isEmpty()) {
 | 
						|
            $environment_variables->push("POSTGRES_USER={$this->database->postgres_user}");
 | 
						|
        }
 | 
						|
        if ($environment_variables->filter(fn ($env) => str($env)->contains('PGUSER'))->isEmpty()) {
 | 
						|
            $environment_variables->push("PGUSER={$this->database->postgres_user}");
 | 
						|
        }
 | 
						|
 | 
						|
        if ($environment_variables->filter(fn ($env) => str($env)->contains('POSTGRES_PASSWORD'))->isEmpty()) {
 | 
						|
            $environment_variables->push("POSTGRES_PASSWORD={$this->database->postgres_password}");
 | 
						|
        }
 | 
						|
 | 
						|
        if ($environment_variables->filter(fn ($env) => str($env)->contains('POSTGRES_DB'))->isEmpty()) {
 | 
						|
            $environment_variables->push("POSTGRES_DB={$this->database->postgres_db}");
 | 
						|
        }
 | 
						|
 | 
						|
        add_coolify_default_environment_variables($this->database, $environment_variables, $environment_variables);
 | 
						|
 | 
						|
        return $environment_variables->all();
 | 
						|
    }
 | 
						|
 | 
						|
    private function generate_init_scripts()
 | 
						|
    {
 | 
						|
        $this->commands[] = "rm -rf $this->configuration_dir/docker-entrypoint-initdb.d/*";
 | 
						|
 | 
						|
        if (blank($this->database->init_scripts) || count($this->database->init_scripts) === 0) {
 | 
						|
            return;
 | 
						|
        }
 | 
						|
 | 
						|
        foreach ($this->database->init_scripts as $init_script) {
 | 
						|
            $filename = data_get($init_script, 'filename');
 | 
						|
            $content = data_get($init_script, 'content');
 | 
						|
            $this->commands[] = [
 | 
						|
                'transfer_file' => [
 | 
						|
                    'content' => $content,
 | 
						|
                    'destination' => "$this->configuration_dir/docker-entrypoint-initdb.d/{$filename}",
 | 
						|
                ],
 | 
						|
            ];
 | 
						|
            $this->init_scripts[] = "$this->configuration_dir/docker-entrypoint-initdb.d/{$filename}";
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    private function add_custom_conf()
 | 
						|
    {
 | 
						|
        $filename = 'custom-postgres.conf';
 | 
						|
        $config_file_path = "$this->configuration_dir/$filename";
 | 
						|
 | 
						|
        if (blank($this->database->postgres_conf)) {
 | 
						|
            $this->commands[] = "rm -f $config_file_path";
 | 
						|
 | 
						|
            return;
 | 
						|
        }
 | 
						|
 | 
						|
        $content = $this->database->postgres_conf;
 | 
						|
        if (! str($content)->contains('listen_addresses')) {
 | 
						|
            $content .= "\nlisten_addresses = '*'";
 | 
						|
            $this->database->postgres_conf = $content;
 | 
						|
            $this->database->save();
 | 
						|
        }
 | 
						|
        $this->commands[] = [
 | 
						|
            'transfer_file' => [
 | 
						|
                'content' => $content,
 | 
						|
                'destination' => $config_file_path,
 | 
						|
            ],
 | 
						|
        ];
 | 
						|
    }
 | 
						|
}
 |