mirror of
				https://github.com/pelican-dev/panel.git
				synced 2025-11-04 04:16:52 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			135 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			135 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
 | 
						|
namespace App\Jobs\Schedule;
 | 
						|
 | 
						|
use App\Jobs\Job;
 | 
						|
use Carbon\CarbonImmutable;
 | 
						|
use App\Models\Task;
 | 
						|
use Illuminate\Queue\SerializesModels;
 | 
						|
use Illuminate\Queue\InteractsWithQueue;
 | 
						|
use Illuminate\Contracts\Queue\ShouldQueue;
 | 
						|
use Illuminate\Foundation\Bus\DispatchesJobs;
 | 
						|
use App\Services\Backups\InitiateBackupService;
 | 
						|
use App\Repositories\Daemon\DaemonPowerRepository;
 | 
						|
use App\Exceptions\Http\Connection\DaemonConnectionException;
 | 
						|
 | 
						|
class RunTaskJob extends Job implements ShouldQueue
 | 
						|
{
 | 
						|
    use DispatchesJobs;
 | 
						|
    use InteractsWithQueue;
 | 
						|
    use SerializesModels;
 | 
						|
 | 
						|
    /**
 | 
						|
     * RunTaskJob constructor.
 | 
						|
     */
 | 
						|
    public function __construct(public Task $task, public bool $manualRun = false)
 | 
						|
    {
 | 
						|
        $this->queue = 'standard';
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Run the job and send actions to the daemon running the server.
 | 
						|
     *
 | 
						|
     * @throws \Throwable
 | 
						|
     */
 | 
						|
    public function handle(
 | 
						|
        InitiateBackupService $backupService,
 | 
						|
        DaemonPowerRepository $powerRepository
 | 
						|
    ): void {
 | 
						|
        // Do not process a task that is not set to active, unless it's been manually triggered.
 | 
						|
        if (!$this->task->schedule->is_active && !$this->manualRun) {
 | 
						|
            $this->markTaskNotQueued();
 | 
						|
            $this->markScheduleComplete();
 | 
						|
 | 
						|
            return;
 | 
						|
        }
 | 
						|
 | 
						|
        $server = $this->task->server;
 | 
						|
        // If we made it to this point and the server status is not null it means the
 | 
						|
        // server was likely suspended or marked as reinstalling after the schedule
 | 
						|
        // was queued up. Just end the task right now — this should be a very rare
 | 
						|
        // condition.
 | 
						|
        if (!is_null($server->status)) {
 | 
						|
            $this->failed();
 | 
						|
 | 
						|
            return;
 | 
						|
        }
 | 
						|
 | 
						|
        // Perform the provided task against the daemon.
 | 
						|
        try {
 | 
						|
            switch ($this->task->action) {
 | 
						|
                case Task::ACTION_POWER:
 | 
						|
                    $powerRepository->setServer($server)->send($this->task->payload);
 | 
						|
                    break;
 | 
						|
                case Task::ACTION_COMMAND:
 | 
						|
                    $server->send($this->task->payload);
 | 
						|
                    break;
 | 
						|
                case Task::ACTION_BACKUP:
 | 
						|
                    $backupService->setIgnoredFiles(explode(PHP_EOL, $this->task->payload))->handle($server, null, true);
 | 
						|
                    break;
 | 
						|
                default:
 | 
						|
                    throw new \InvalidArgumentException('Invalid task action provided: ' . $this->task->action);
 | 
						|
            }
 | 
						|
        } catch (\Exception $exception) {
 | 
						|
            // If this isn't a DaemonConnectionException on a task that allows for failures
 | 
						|
            // throw the exception back up the chain so that the task is stopped.
 | 
						|
            if (!($this->task->continue_on_failure && $exception instanceof DaemonConnectionException)) {
 | 
						|
                throw $exception;
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        $this->markTaskNotQueued();
 | 
						|
        $this->queueNextTask();
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Handle a failure while sending the action to the daemon or otherwise processing the job.
 | 
						|
     */
 | 
						|
    public function failed(\Exception $exception = null)
 | 
						|
    {
 | 
						|
        $this->markTaskNotQueued();
 | 
						|
        $this->markScheduleComplete();
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Get the next task in the schedule and queue it for running after the defined period of wait time.
 | 
						|
     */
 | 
						|
    private function queueNextTask()
 | 
						|
    {
 | 
						|
        /** @var \App\Models\Task|null $nextTask */
 | 
						|
        $nextTask = Task::query()->where('schedule_id', $this->task->schedule_id)
 | 
						|
            ->orderBy('sequence_id', 'asc')
 | 
						|
            ->where('sequence_id', '>', $this->task->sequence_id)
 | 
						|
            ->first();
 | 
						|
 | 
						|
        if (is_null($nextTask)) {
 | 
						|
            $this->markScheduleComplete();
 | 
						|
 | 
						|
            return;
 | 
						|
        }
 | 
						|
 | 
						|
        $nextTask->update(['is_queued' => true]);
 | 
						|
 | 
						|
        $this->dispatch((new self($nextTask, $this->manualRun))->delay($nextTask->time_offset));
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Marks the parent schedule as being complete.
 | 
						|
     */
 | 
						|
    private function markScheduleComplete()
 | 
						|
    {
 | 
						|
        $this->task->schedule()->update([
 | 
						|
            'is_processing' => false,
 | 
						|
            'last_run_at' => CarbonImmutable::now()->toDateTimeString(),
 | 
						|
        ]);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Mark a specific task as no longer being queued.
 | 
						|
     */
 | 
						|
    private function markTaskNotQueued()
 | 
						|
    {
 | 
						|
        $this->task->update(['is_queued' => false]);
 | 
						|
    }
 | 
						|
}
 |