Add endpoint needed for recieving and processing activity

This commit is contained in:
DaneEveritt 2022-07-09 17:45:38 -04:00
parent 9b8479e85d
commit 1eee55b27c
No known key found for this signature in database
GPG key ID: EEA66103B3D71F53
3 changed files with 170 additions and 0 deletions

View file

@ -0,0 +1,115 @@
<?php
namespace Pterodactyl\Http\Controllers\Api\Remote;
use Exception;
use Carbon\Carbon;
use Pterodactyl\Models\User;
use Webmozart\Assert\Assert;
use Pterodactyl\Models\Server;
use Pterodactyl\Models\ActivityLog;
use Illuminate\Support\Facades\Log;
use Pterodactyl\Models\ActivityLogSubject;
use Pterodactyl\Http\Controllers\Controller;
use Pterodactyl\Http\Requests\Api\Remote\ActivityEventRequest;
class ActivityProcessingController extends Controller
{
public function __invoke(ActivityEventRequest $request)
{
$tz = Carbon::now()->getTimezone();
/** @var \Pterodactyl\Models\Node $node */
$node = $request->attributes->get('node');
$servers = $node->servers()->whereIn('uuid', $request->servers())->get()->keyBy('uuid');
$users = User::query()->whereIn('uuid', $request->users())->get()->keyBy('uuid');
clock()->log($request->input('data'));
$logs = [];
foreach ($request->input('data') as $datum) {
/** @var \Pterodactyl\Models\Server|null $server */
$server = $servers->get($datum['server']);
if (is_null($server) || is_null($event = $this->event($datum['event']))) {
continue;
}
try {
$when = Carbon::createFromFormat(
Carbon::RFC3339,
preg_replace('/(\.\d+)Z$/', 'Z', $datum['timestamp']),
'UTC'
);
} catch (Exception $exception) {
Log::warning($exception, ['timestamp' => $datum['timestamp']]);
// If we cannot parse the value for some reason don't blow up this request, just go ahead
// and log the event with the current time, and set the metadata value to have the original
// timestamp that was provided.
$when = Carbon::now();
$datum['metadata'] = array_merge($datum['metadata'] ?? [], ['original_timestamp' => $datum['timestamp']]);
}
$log = [
'ip' => empty($datum['ip']) ? '127.0.0.1' : $datum['ip'],
'event' => $event,
'properties' => json_encode($datum['metadata'] ?? []),
// We have to change the time to the current timezone due to the way Laravel is handling
// the date casting internally. If we just leave it in UTC it ends up getting double-cast
// and the time is way off.
'timestamp' => $when->setTimezone($tz),
];
if ($user = $users->get($datum['user'])) {
$log['actor_id'] = $user->id;
$log['actor_type'] = $user->getMorphClass();
}
if (!isset($logs[$datum['server']])) {
$logs[$datum['server']] = [];
}
$logs[$datum['server']][] = $log;
}
foreach ($logs as $key => $data) {
Assert::isInstanceOf($server = $servers->get($key), Server::class);
$batch = [];
foreach ($data as $datum) {
$id = ActivityLog::insertGetId($datum);
$batch[] = [
'activity_log_id' => $id,
'subject_id' => $server->id,
'subject_type' => $server->getMorphClass(),
];
}
ActivityLogSubject::insert($batch);
}
}
/**
* Takes an event from Wings and converts it into the expected event type on
* the Panel. If no matching event type can be deduced, null is returned and
* the event won't be logged.
*/
protected function event(string $input): ?string
{
switch ($input) {
case 'console_command':
return 'server:console.command';
case 'power_start':
return 'server:power.start';
case 'power_stop':
return 'server:power.stop';
case 'power_restart':
return 'server:power.restart';
case 'power_kill':
return 'server:power.kill';
default:
return null;
}
}
}

View file

@ -0,0 +1,54 @@
<?php
namespace Pterodactyl\Http\Requests\Api\Remote;
use Illuminate\Support\Collection;
use Illuminate\Foundation\Http\FormRequest;
class ActivityEventRequest extends FormRequest
{
public function authorize(): bool
{
return true;
}
public function rules(): array
{
return [
'data' => ['required', 'array'],
'data.*' => ['array'],
'data.*.user' => ['present', 'uuid'],
'data.*.server' => ['required', 'uuid'],
'data.*.event' => ['required', 'string'],
'data.*.metadata' => ['present', 'nullable', 'array'],
'data.*.ip' => ['present', 'ip'],
'data.*.timestamp' => ['required', 'string'],
];
}
/**
* Returns all of the unique server UUIDs that were recieved in this request.
*
* @return string[]
*/
public function servers(): array
{
return Collection::make($this->input('data'))->pluck('server')->unique()->toArray();
}
/**
* Returns all of the unique user UUIDs that were submitted in this request.
*
* @return string[]
*/
public function users(): array
{
return Collection::make($this->input('data'))
->filter(function ($value) {
return !empty($value['user']);
})
->pluck('user')
->unique()
->toArray();
}
}

View file

@ -8,6 +8,7 @@ Route::post('/sftp/auth', Remote\SftpAuthenticationController::class);
Route::get('/servers', [Remote\Servers\ServerDetailsController::class, 'list']);
Route::post('/servers/reset', [Remote\Servers\ServerDetailsController::class, 'resetState']);
Route::post('/activity', Remote\ActivityProcessingController::class);
Route::group(['prefix' => '/servers/{uuid}'], function () {
Route::get('/', Remote\Servers\ServerDetailsController::class);