Add new activity logging code to replace audit log

This commit is contained in:
DaneEveritt 2022-05-28 15:36:26 -04:00
parent c14c7b436e
commit 5bb66a00d8
No known key found for this signature in database
GPG key ID: EEA66103B3D71F53
11 changed files with 534 additions and 0 deletions

View file

@ -0,0 +1,62 @@
<?php
namespace Pterodactyl\Services\Activity;
use Ramsey\Uuid\Uuid;
class AcitvityLogBatchService
{
protected int $transaction = 0;
protected ?string $uuid = null;
/**
* Returns the UUID of the batch, or null if there is not a batch currently
* being executed.
*/
public function uuid(): ?string
{
return $this->uuid;
}
/**
* Starts a new batch transaction. If there is already a transaction present
* this will be nested.
*/
public function start(): void
{
if ($this->transaction === 0) {
$this->uuid = Uuid::uuid4()->toString();
}
++$this->transaction;
}
/**
* Ends a batch transaction, if this is the last transaction in the stack
* the UUID will be cleared out.
*/
public function end(): void
{
$this->transaction = max(0, $this->transaction - 1);
if ($this->transaction === 0) {
$this->uuid = null;
}
}
/**
* Executes the logic provided within the callback in the scope of an activity
* log batch transaction.
*
* @param \Closure $callback
* @return mixed
*/
public function transaction(\Closure $callback)
{
$this->start();
$result = $callback($this->uuid());
$this->end();
return $result;
}
}

View file

@ -0,0 +1,177 @@
<?php
namespace Pterodactyl\Services\Activity;
use Illuminate\Support\Collection;
use Pterodactyl\Models\ActivityLog;
use Illuminate\Contracts\Auth\Factory;
use Illuminate\Database\Eloquent\Model;
use Illuminate\Database\ConnectionInterface;
class ActivityLogService
{
protected ?ActivityLog $activity = null;
protected Factory $manager;
protected ConnectionInterface $connection;
protected AcitvityLogBatchService $batch;
protected ActivityLogTargetableService $targetable;
public function __construct(
Factory $manager,
AcitvityLogBatchService $batch,
ActivityLogTargetableService $targetable,
ConnectionInterface $connection
) {
$this->manager = $manager;
$this->batch = $batch;
$this->targetable = $targetable;
$this->connection = $connection;
}
/**
* Sets the activity logger as having been caused by an anonymous
* user type.
*/
public function anonymous(): self
{
$this->getActivity()->actor_id = null;
$this->getActivity()->actor_type = null;
$this->getActivity()->setRelation('actor', null);
return $this;
}
/**
* Sets the action for this activity log.
*/
public function event(string $action): self
{
$this->getActivity()->event = $action;
return $this;
}
/**
* Set the description for this activity.
*/
public function withDescription(?string $description): self
{
$this->getActivity()->description = $description;
return $this;
}
/**
* Sets the subject model instance.
*/
public function withSubject(Model $subject): self
{
$this->getActivity()->subject()->associate($subject);
return $this;
}
/**
* Sets the actor model instance.
*/
public function withActor(Model $actor): self
{
$this->getActivity()->actor()->associate($actor);
return $this;
}
/**
* Sets the custom properties for the activity log instance.
*
* @param \Illuminate\Support\Collection|array $properties
*/
public function withProperties($properties): self
{
$this->getActivity()->properties = Collection::make($properties);
return $this;
}
/**
* Sets a custom property on the activty log instance.
*
* @param mixed $value
*/
public function withProperty(string $key, $value): self
{
$this->getActivity()->properties = $this->getActivity()->properties->put($key, $value);
return $this;
}
/**
* Logs an activity log entry with the set values and then returns the
* model instance to the caller.
*/
public function log(string $description): ActivityLog
{
$this->withDescription($description);
$activity = $this->activity;
$activity->save();
$this->activity = null;
return $activity;
}
/**
* Executes the provided callback within the scope of a database transaction
* and will only save the activity log entry if everything else succesfully
* settles.
*
* @return mixed
*
* @throws \Throwable
*/
public function transaction(\Closure $callback, string $description = null)
{
if (!is_null($description)) {
$this->withDescription($description);
}
return $this->connection->transaction(function () use ($callback) {
$response = $callback($activity = $this->getActivity());
$activity->save();
$this->activity = null;
return $response;
});
}
/**
* Returns the current activity log instance.
*/
protected function getActivity(): ActivityLog
{
if ($this->activity) {
return $this->activity;
}
$this->activity = new ActivityLog([
'batch_uuid' => $this->batch->uuid(),
'properties' => Collection::make([]),
]);
if ($subject = $this->targetable->subject()) {
$this->withSubject($subject);
}
if ($actor = $this->targetable->actor()) {
$this->withActor($actor);
} elseif ($user = $this->manager->guard()->user()) {
if ($user instanceof Model) {
$this->withActor($user);
}
}
return $this->activity;
}
}

View file

@ -0,0 +1,47 @@
<?php
namespace Pterodactyl\Services\Activity;
use InvalidArgumentException;
use Illuminate\Database\Eloquent\Model;
class ActivityLogTargetableService
{
protected ?Model $actor = null;
protected ?Model $subject = null;
public function setActor(Model $actor): void
{
if (!is_null($this->actor)) {
throw new InvalidArgumentException('Cannot call ' . __METHOD__ . ' when an actor is already set on the instance.');
}
$this->actor = $actor;
}
public function setSubject(Model $subject): void
{
if (!is_null($this->subject)) {
throw new InvalidArgumentException('Cannot call ' . __METHOD__ . ' when a target is already set on the instance.');
}
$this->subject = $subject;
}
public function actor(): ?Model
{
return $this->actor;
}
public function subject(): ?Model
{
return $this->subject;
}
public function reset(): void
{
$this->actor = null;
$this->subject = null;
}
}