coolify/app/Livewire/Project/Shared/EnvironmentVariable/All.php

<?php

namespace App\Livewire\Project\Shared\EnvironmentVariable;

use App\Models\EnvironmentVariable;
use Livewire\Component;
use Visus\Cuid2\Cuid2;
use Illuminate\Support\Str;

class All extends Component
{
    public $resource;
    public bool $showPreview = false;
    public ?string $modalId = null;
    public ?string $variables = null;
    public ?string $variablesPreview = null;
    public string $view = 'normal';
    protected $listeners = [
        'refreshEnvs',
        'saveKey' => 'submit',
    ];
    public function mount()
    {
        $resourceClass = get_class($this->resource);
        $resourceWithPreviews = ['App\Models\Application'];
        $simpleDockerfile = !is_null(data_get($this->resource, 'dockerfile'));
        if (Str::of($resourceClass)->contains($resourceWithPreviews) && !$simpleDockerfile) {
            $this->showPreview = true;
        }
        $this->modalId = new Cuid2(7);
        $this->getDevView();
    }
    public function getDevView()
    {
        $this->variables = $this->resource->environment_variables->map(function ($item) {
            if ($item->is_shown_once) {
                return "$item->key=(locked secret)";
            }
            if ($item->is_multiline) {
                return "$item->key=(multiline, edit in normal view)";
            }
            return "$item->key=$item->value";
        })->sort()->join('
');
        if ($this->showPreview) {
            $this->variablesPreview = $this->resource->environment_variables_preview->map(function ($item) {
                if ($item->is_shown_once) {
                    return "$item->key=(locked secret)";
                }
                if ($item->is_multiline) {
                    return "$item->key=(multiline, edit in normal view)";
                }
                return "$item->key=$item->value";
            })->sort()->join('
');
        }
    }
    public function switch()
    {
        $this->view = $this->view === 'normal' ? 'dev' : 'normal';
    }
    public function saveVariables($isPreview)
    {
        if ($isPreview) {
            $variables = parseEnvFormatToArray($this->variablesPreview);
            $this->resource->environment_variables_preview()->whereNotIn('key', array_keys($variables))->delete();
        } else {
            $variables = parseEnvFormatToArray($this->variables);
            $this->resource->environment_variables()->whereNotIn('key', array_keys($variables))->delete();
        }
        foreach ($variables as $key => $variable) {
            if ($isPreview) {
                $found = $this->resource->environment_variables_preview()->where('key', $key)->first();
            } else {
                $found = $this->resource->environment_variables()->where('key', $key)->first();
            }
            if ($found) {
                if ($found->is_shown_once || $found->is_multiline) {
                    continue;
                }
                $found->value = $variable;
                if (str($found->value)->startsWith('{{') && str($found->value)->endsWith('}}')) {
                    $type = str($found->value)->after("{{")->before(".")->value;
                    if (!collect(SHARED_VARIABLE_TYPES)->contains($type)) {
                        $this->dispatch('error', 'Invalid  shared variable type.', "Valid types are: team, project, environment.");
                        return;
                    }
                }
                $found->save();
                continue;
            } else {
                $environment = new EnvironmentVariable();
                $environment->key = $key;
                $environment->value = $variable;
                if (str($environment->value)->startsWith('{{') && str($environment->value)->endsWith('}}')) {
                    $type = str($environment->value)->after("{{")->before(".")->value;
                    if (!collect(SHARED_VARIABLE_TYPES)->contains($type)) {
                        $this->dispatch('error', 'Invalid  shared variable type.', "Valid types are: team, project, environment.");
                        return;
                    }
                }
                $environment->is_build_time = false;
                $environment->is_multiline = false;
                $environment->is_preview = $isPreview ? true : false;
                switch ($this->resource->type()) {
                    case 'application':
                        $environment->application_id = $this->resource->id;
                        break;
                    case 'standalone-postgresql':
                        $environment->standalone_postgresql_id = $this->resource->id;
                        break;
                    case 'standalone-redis':
                        $environment->standalone_redis_id = $this->resource->id;
                        break;
                    case 'standalone-mongodb':
                        $environment->standalone_mongodb_id = $this->resource->id;
                        break;
                    case 'standalone-mysql':
                        $environment->standalone_mysql_id = $this->resource->id;
                        break;
                    case 'standalone-mariadb':
                        $environment->standalone_mariadb_id = $this->resource->id;
                        break;
                    case 'standalone-keydb':
                        $environment->standalone_keydb_id = $this->resource->id;
                        break;
                    case 'standalone-dragonfly':
                        $environment->standalone_dragonfly_id = $this->resource->id;
                        break;
                    case 'standalone-clickhouse':
                        $environment->standalone_clickhouse_id = $this->resource->id;
                        break;
                    case 'service':
                        $environment->service_id = $this->resource->id;
                        break;
                }
                $environment->save();
            }
        }
        if ($isPreview) {
            $this->dispatch('success', 'Preview environment variables updated.');
        } else {
            $this->dispatch('success', 'Environment variables updated.');
        }
        $this->refreshEnvs();
    }
    public function refreshEnvs()
    {
        $this->resource->refresh();
        $this->getDevView();
    }

    public function submit($data)
    {
        try {
            $found = $this->resource->environment_variables()->where('key', $data['key'])->first();
            if ($found) {
                $this->dispatch('error', 'Environment variable already exists.');
                return;
            }
            $environment = new EnvironmentVariable();
            $environment->key = $data['key'];
            $environment->value = $data['value'];
            $environment->is_build_time = $data['is_build_time'];
            $environment->is_multiline = $data['is_multiline'];
            $environment->is_literal = $data['is_literal'];
            $environment->is_preview = $data['is_preview'];

            switch ($this->resource->type()) {
                case 'application':
                    $environment->application_id = $this->resource->id;
                    break;
                case 'standalone-postgresql':
                    $environment->standalone_postgresql_id = $this->resource->id;
                    break;
                case 'standalone-redis':
                    $environment->standalone_redis_id = $this->resource->id;
                    break;
                case 'standalone-mongodb':
                    $environment->standalone_mongodb_id = $this->resource->id;
                    break;
                case 'standalone-mysql':
                    $environment->standalone_mysql_id = $this->resource->id;
                    break;
                case 'standalone-mariadb':
                    $environment->standalone_mariadb_id = $this->resource->id;
                    break;
                case 'standalone-keydb':
                    $environment->standalone_keydb_id = $this->resource->id;
                    break;
                case 'standalone-dragonfly':
                    $environment->standalone_dragonfly_id = $this->resource->id;
                    break;
                case 'standalone-clickhouse':
                    $environment->standalone_clickhouse_id = $this->resource->id;
                    break;
                case 'service':
                    $environment->service_id = $this->resource->id;
                    break;
            }
            $environment->save();
            $this->refreshEnvs();
            $this->dispatch('success', 'Environment variable added.');
        } catch (\Throwable $e) {
            return handleError($e, $this);
        }
    }
}