Replace allocation repository

app/Http/Controllers/Admin/Nodes/NodeViewController.php

@@ -14,7 +14,6 @@ use Pterodactyl\Repositories\Eloquent\ServerRepository;
 use Pterodactyl\Traits\Controllers\JavascriptInjection;
 use Pterodactyl\Services\Helpers\SoftwareVersionService;
 use Pterodactyl\Repositories\Eloquent\LocationRepository;
-use Pterodactyl\Repositories\Eloquent\AllocationRepository;
 
 class NodeViewController extends Controller
 {
@@ -24,7 +23,6 @@ class NodeViewController extends Controller
      * NodeViewController constructor.
      */
     public function __construct(
-        private AllocationRepository $allocationRepository,
         private LocationRepository $locationRepository,
         private NodeRepository $repository,
         private ServerRepository $serverRepository,

app/Http/Controllers/Admin/NodesController.php

@@ -23,7 +23,6 @@ use Pterodactyl\Contracts\Repository\ServerRepositoryInterface;
 use Pterodactyl\Http\Requests\Admin\Node\AllocationFormRequest;
 use Pterodactyl\Services\Allocations\AllocationDeletionService;
 use Pterodactyl\Contracts\Repository\LocationRepositoryInterface;
-use Pterodactyl\Contracts\Repository\AllocationRepositoryInterface;
 use Pterodactyl\Http\Requests\Admin\Node\AllocationAliasFormRequest;
 
 class NodesController extends Controller
@@ -34,7 +33,6 @@ class NodesController extends Controller
     public function __construct(
         protected AlertsMessageBag $alert,
         protected AllocationDeletionService $allocationDeletionService,
-        protected AllocationRepositoryInterface $allocationRepository,
         protected AssignmentService $assignmentService,
         protected CacheRepository $cache,
         protected NodeCreationService $creationService,
@@ -125,11 +123,12 @@ class NodesController extends Controller
      */
     public function allocationRemoveBlock(Request $request, int $node): RedirectResponse
     {
-        $this->allocationRepository->deleteWhere([
-            ['node_id', '=', $node],
-            ['server_id', '=', null],
-            ['ip', '=', $request->input('ip')],
-        ]);
+        /** @var Node $node */
+        $node = Node::query()->findOrFail($node);
+        $node->allocations()
+            ->where('ip', $request->input('ip'))
+            ->whereNull('server_id')
+            ->delete();
 
         $this->alert->success(trans('admin/node.notices.unallocated_deleted', ['ip' => $request->input('ip')]))
             ->flash();
@@ -140,14 +139,12 @@ class NodesController extends Controller
     /**
      * Sets an alias for a specific allocation on a node.
      *
-     * @throws \Pterodactyl\Exceptions\Model\DataValidationException
-     * @throws \Pterodactyl\Exceptions\Repository\RecordNotFoundException
      */
     public function allocationSetAlias(AllocationAliasFormRequest $request): \Symfony\Component\HttpFoundation\Response
     {
-        $this->allocationRepository->update($request->input('allocation_id'), [
-            'ip_alias' => (empty($request->input('alias'))) ? null : $request->input('alias'),
-        ]);
+        $allocation = Allocation::query()->findOrFail($request->input('allocation_id'));
+        $alias = (empty($request->input('alias'))) ? null : $request->input('alias');
+        $allocation->update(['ip_alias' => $alias]);
 
         return response('', 204);
     }

app/Http/Controllers/Admin/Servers/ServerTransferController.php

@@ -3,15 +3,16 @@
 namespace Pterodactyl\Http\Controllers\Admin\Servers;
 
 use Illuminate\Http\Request;
+use Pterodactyl\Models\Node;
 use Pterodactyl\Models\Server;
-use Prologue\Alerts\AlertsMessageBag;
+use Pterodactyl\Models\Allocation;
 use Illuminate\Http\RedirectResponse;
+use Prologue\Alerts\AlertsMessageBag;
 use Pterodactyl\Models\ServerTransfer;
 use Pterodactyl\Http\Controllers\Controller;
 use Pterodactyl\Services\Servers\TransferService;
 use Pterodactyl\Repositories\Eloquent\NodeRepository;
 use Pterodactyl\Repositories\Wings\DaemonConfigurationRepository;
-use Pterodactyl\Contracts\Repository\AllocationRepositoryInterface;
 
 class ServerTransferController extends Controller
 {
@@ -20,7 +21,6 @@ class ServerTransferController extends Controller
      */
     public function __construct(
         private AlertsMessageBag $alert,
-        private AllocationRepositoryInterface $allocationRepository,
         private NodeRepository $nodeRepository,
         private TransferService $transferService,
         private DaemonConfigurationRepository $daemonConfigurationRepository
@@ -87,7 +87,12 @@ class ServerTransferController extends Controller
         $allocations = $additional_allocations;
         $allocations[] = $allocation_id;
 
-        $unassigned = $this->allocationRepository->getUnassignedAllocationIds($node_id);
+        /** @var Node $node */
+        $node = Node::query()->findOrFail($node_id);
+        $unassigned = $node->allocations()
+            ->whereNull('server_id')
+            ->pluck('id')
+            ->toArray();
 
         $updateIds = [];
         foreach ($allocations as $allocation) {
@@ -99,7 +104,9 @@ class ServerTransferController extends Controller
         }
 
         if (!empty($updateIds)) {
-            $this->allocationRepository->updateWhereIn('id', $updateIds, ['server_id' => $server->id]);
+            Allocation::query()
+                ->whereIn('id', $updateIds)
+                ->update(['server_id' => $server->id]);
         }
     }
 }

app/Http/Controllers/Admin/ServersController.php

@@ -30,7 +30,6 @@ use Pterodactyl\Services\Databases\DatabaseManagementService;
 use Illuminate\Contracts\Config\Repository as ConfigRepository;
 use Pterodactyl\Contracts\Repository\ServerRepositoryInterface;
 use Pterodactyl\Contracts\Repository\DatabaseRepositoryInterface;
-use Pterodactyl\Contracts\Repository\AllocationRepositoryInterface;
 use Pterodactyl\Services\Servers\ServerConfigurationStructureService;
 use Pterodactyl\Http\Requests\Admin\Servers\Databases\StoreServerDatabaseRequest;
 
@@ -41,7 +40,6 @@ class ServersController extends Controller
      */
     public function __construct(
         protected AlertsMessageBag $alert,
-        protected AllocationRepositoryInterface $allocationRepository,
         protected BuildModificationService $buildModificationService,
         protected ConfigRepository $config,
         protected DaemonServerRepository $daemonServerRepository,