misc_pterodactyl-panel/app/Repositories/TaskRepository.php

164 lines
4.8 KiB
PHP
Raw Normal View History

<?php
/**
* Pterodactyl - Panel
2017-01-24 22:57:08 +00:00
* Copyright (c) 2015 - 2017 Dane Everitt <dane@daneeveritt.com>.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
2016-12-07 22:46:38 +00:00
namespace Pterodactyl\Repositories;
use Cron;
use Validator;
2017-04-15 17:52:43 +00:00
use Pterodactyl\Models\Task;
use Pterodactyl\Models\User;
use Pterodactyl\Models\Server;
use Pterodactyl\Exceptions\DisplayException;
2016-12-07 22:46:38 +00:00
use Pterodactyl\Exceptions\DisplayValidationException;
class TaskRepository
{
2017-03-19 23:36:50 +00:00
/**
* The default values to use for new tasks.
*
* @var array
*/
protected $defaults = [
'year' => '*',
'day_of_week' => '*',
'month' => '*',
'day_of_month' => '*',
'hour' => '*',
'minute' => '*/30',
];
2017-03-19 23:36:50 +00:00
/**
* Task action types.
*
* @var array
*/
protected $actions = [
'command',
'power',
];
2016-09-05 21:13:22 +00:00
/**
* Deletes a given task.
*
2017-03-19 23:36:50 +00:00
* @param int $id
2016-09-05 21:13:22 +00:00
* @return bool
*/
public function delete($id)
{
2017-04-15 17:52:43 +00:00
$task = Task::findOrFail($id);
$task->delete();
2016-09-05 21:13:22 +00:00
}
/**
* Toggles a task active or inactive.
*
2017-03-19 23:36:50 +00:00
* @param int $id
2017-04-15 17:52:43 +00:00
* @return bool
2016-09-05 21:13:22 +00:00
*/
public function toggle($id)
{
2017-04-15 17:52:43 +00:00
$task = Task::findOrFail($id);
2016-09-05 21:13:22 +00:00
2017-04-15 17:52:43 +00:00
$task->active = ! $task->active;
$task->queued = false;
$task->save();
return $task->active;
2016-09-05 21:13:22 +00:00
}
/**
* Create a new scheduled task for a given server.
*
2017-04-15 17:52:43 +00:00
* @param int $server
* @param int $user
2017-03-19 23:36:50 +00:00
* @param array $data
2017-04-15 17:52:43 +00:00
* @return \Pterodactyl\Models\Task
2017-03-19 23:36:50 +00:00
*
* @throws \Pterodactyl\Exceptions\DisplayException
* @throws \Pterodactyl\Exceptions\DisplayValidationException
*/
2017-04-15 17:52:43 +00:00
public function create($server, $user, $data)
{
2017-04-15 17:52:43 +00:00
$server = Server::findOrFail($server);
$user = User::findOrFail($user);
$validator = Validator::make($data, [
'action' => 'string|required',
'data' => 'string|required',
'year' => 'string|sometimes',
'day_of_week' => 'string|sometimes',
'month' => 'string|sometimes',
'day_of_month' => 'string|sometimes',
'hour' => 'string|sometimes',
2016-12-07 22:46:38 +00:00
'minute' => 'string|sometimes',
]);
if ($validator->fails()) {
throw new DisplayValidationException(json_encode($validator->errors()));
}
2016-12-07 22:46:38 +00:00
if (! in_array($data['action'], $this->actions)) {
throw new DisplayException('The action provided is not valid.');
}
$cron = $this->defaults;
foreach ($this->defaults as $setting => $value) {
2016-12-07 22:46:38 +00:00
if (array_key_exists($setting, $data) && ! is_null($data[$setting]) && $data[$setting] !== '') {
$cron[$setting] = $data[$setting];
}
}
// Check that is this a valid Cron Entry
try {
$buildCron = Cron::factory(sprintf('%s %s %s %s %s %s',
$cron['minute'],
$cron['hour'],
$cron['day_of_month'],
$cron['month'],
$cron['day_of_week'],
$cron['year']
));
} catch (\Exception $ex) {
throw $ex;
}
2017-04-15 17:52:43 +00:00
return Task::create([
'user_id' => $user->id,
'server_id' => $server->id,
'active' => 1,
'action' => $data['action'],
'data' => $data['data'],
'queued' => 0,
'year' => $cron['year'],
'day_of_week' => $cron['day_of_week'],
'month' => $cron['month'],
'day_of_month' => $cron['day_of_month'],
'hour' => $cron['hour'],
'minute' => $cron['minute'],
'last_run' => null,
2016-12-07 22:46:38 +00:00
'next_run' => $buildCron->getNextRunDate(),
]);
}
}