misc_pterodactyl-panel/app/Jobs/Schedule/RunTaskJob.php

155 lines
4.8 KiB
PHP
Raw Normal View History

2017-09-17 04:10:00 +00:00
<?php
namespace Pterodactyl\Jobs\Schedule;
use Exception;
use Carbon\Carbon;
2017-09-17 04:10:00 +00:00
use Pterodactyl\Jobs\Job;
use InvalidArgumentException;
use Illuminate\Container\Container;
2017-09-17 04:10:00 +00:00
use Illuminate\Queue\SerializesModels;
use Illuminate\Queue\InteractsWithQueue;
use Illuminate\Contracts\Queue\ShouldQueue;
use Illuminate\Foundation\Bus\DispatchesJobs;
use Pterodactyl\Repositories\Eloquent\TaskRepository;
use Pterodactyl\Repositories\Wings\DaemonPowerRepository;
use Pterodactyl\Repositories\Wings\DaemonCommandRepository;
2017-09-17 04:10:00 +00:00
use Pterodactyl\Contracts\Repository\TaskRepositoryInterface;
2017-09-30 16:45:24 +00:00
use Pterodactyl\Services\DaemonKeys\DaemonKeyProviderService;
2017-09-17 04:10:00 +00:00
use Pterodactyl\Contracts\Repository\ScheduleRepositoryInterface;
class RunTaskJob extends Job implements ShouldQueue
{
use DispatchesJobs, InteractsWithQueue, SerializesModels;
/**
* @var int
*/
2017-09-27 03:16:26 +00:00
public $schedule;
2017-09-17 04:10:00 +00:00
/**
* @var int
*/
public $task;
/**
* @var \Pterodactyl\Repositories\Eloquent\TaskRepository
2017-09-17 04:10:00 +00:00
*/
protected $taskRepository;
/**
* RunTaskJob constructor.
*
* @param int $task
* @param int $schedule
*/
public function __construct(int $task, int $schedule)
2017-09-17 04:10:00 +00:00
{
$this->queue = config('pterodactyl.queues.standard');
2017-09-17 04:10:00 +00:00
$this->task = $task;
$this->schedule = $schedule;
}
/**
* Run the job and send actions to the daemon running the server.
*
* @param \Pterodactyl\Repositories\Wings\DaemonCommandRepository $commandRepository
2019-09-06 04:32:57 +00:00
* @param \Pterodactyl\Services\DaemonKeys\DaemonKeyProviderService $keyProviderService
* @param \Pterodactyl\Repositories\Wings\DaemonPowerRepository $powerRepository
* @param \Pterodactyl\Repositories\Eloquent\TaskRepository $taskRepository
2017-09-17 04:10:00 +00:00
*
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/
public function handle(
DaemonCommandRepository $commandRepository,
2017-09-30 16:45:24 +00:00
DaemonKeyProviderService $keyProviderService,
DaemonPowerRepository $powerRepository,
TaskRepository $taskRepository
2017-09-17 04:10:00 +00:00
) {
$this->taskRepository = $taskRepository;
$task = $this->taskRepository->getTaskForJobProcess($this->task);
$server = $task->getRelation('server');
2017-09-17 04:10:00 +00:00
// Do not process a task that is not set to active.
if (! $task->getRelation('schedule')->is_active) {
$this->markTaskNotQueued();
$this->markScheduleComplete();
return;
}
2018-05-13 14:50:56 +00:00
// Perform the provided task against the daemon.
2017-09-17 04:10:00 +00:00
switch ($task->action) {
case 'power':
$powerRepository->setServer($server)->send($task->payload);
2017-09-17 04:10:00 +00:00
break;
case 'command':
$commandRepository->setServer($server)->send($task->payload);
2017-09-17 04:10:00 +00:00
break;
default:
throw new InvalidArgumentException('Cannot run a task that points to a non-existent action.');
2017-09-17 04:10:00 +00:00
}
$this->markTaskNotQueued();
$this->queueNextTask($task->sequence_id);
}
/**
* Handle a failure while sending the action to the daemon or otherwise processing the job.
*
* @param \Exception|null $exception
2017-09-17 04:10:00 +00:00
*/
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.
*
* @param int $sequence
*
* @throws \Pterodactyl\Exceptions\Model\DataValidationException
* @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
*/
private function queueNextTask($sequence)
{
$nextTask = $this->taskRepository->getNextTask($this->schedule, $sequence);
if (is_null($nextTask)) {
$this->markScheduleComplete();
return;
}
$this->taskRepository->update($nextTask->id, ['is_queued' => true]);
$this->dispatch((new self($nextTask->id, $this->schedule))->delay($nextTask->time_offset));
}
/**
* Marks the parent schedule as being complete.
*/
private function markScheduleComplete()
{
Container::getInstance()
->make(ScheduleRepositoryInterface::class)
->withoutFreshModel()
->update($this->schedule, [
'is_processing' => false,
'last_run_at' => Carbon::now()->toDateTimeString(),
]);
2017-09-17 04:10:00 +00:00
}
/**
* Mark a specific task as no longer being queued.
*/
private function markTaskNotQueued()
{
Container::getInstance()
->make(TaskRepositoryInterface::class)
->update($this->task, ['is_queued' => false]);
2017-09-17 04:10:00 +00:00
}
}