Improved file downloading

Author: DaneEveritt

app/Exceptions/Handler.php

@@ -50,7 +50,7 @@ public function render($request, Exception $e)
             $e = new NotFoundHttpException($e->getMessage(), $e);
         }
 
-        if ($request->isXmlHttpRequest() || $request->ajax() || $request->is('/api/*')) {
+        if ($request->isXmlHttpRequest() || $request->ajax() || $request->is('api/*') || $request->is('remote/*')) {
 
             $exception = 'An exception occured while attempting to perform this action, please try again.';
 

app/Http/Controllers/Remote/RemoteController.php

@@ -0,0 +1,37 @@
+<?php
+
+namespace Pterodactyl\Http\Controllers\Remote;
+
+use Pterodactyl\Models\Download;
+use Pterodactyl\Exceptions\DisplayException;
+
+use Pterodactyl\Http\Controllers\Controller;
+use Illuminate\Http\Request;
+
+class RemoteController extends Controller
+{
+
+    /**
+     * Controller Constructor
+     */
+    public function __construct()
+    {
+        // No middleware for this route.
+    }
+
+    public function postDownload(Request $request) {
+        $download = Download::where('token', $request->input('token', '00'))->first();
+        if (!$download) {
+            return response()->json([
+                'error' => 'An invalid request token was recieved with this request.'
+            ], 403);
+        }
+
+        $download->delete();
+        return response()->json([
+            'path' => $download->path,
+            'server' => $download->server
+        ]);
+    }
+
+}

app/Http/Controllers/Scales/FileController.php

@@ -150,7 +150,7 @@ public function getDownloadFile(Request $request, $uuid, $file)
         $download = new Download;
 
         $download->token = Uuid::generate(4);
-        $download->server = $server->id;
+        $download->server = $server->uuid;
         $download->path = str_replace('../', '', $file);
 
         $download->save();

app/Http/Controllers/Server/ServerController.php

@@ -12,6 +12,6 @@ class VerifyCsrfToken extends BaseVerifier
      * @var array
      */
     protected $except = [
-        //
+        'remote/*',
     ];
 }

app/Http/Middleware/VerifyCsrfToken.php

@@ -0,0 +1,16 @@
+<?php
+
+namespace Pterodactyl\Http\Routes;
+
+use Illuminate\Routing\Router;
+use Request;
+
+class RemoteRoutes {
+
+    public function map(Router $router) {
+        $router->group(['prefix' => 'remote'], function () use ($router) {
+            $router->post('download', [ 'as' => 'remote.download', 'uses' => 'Remote\RemoteController@postDownload' ]);
+        });
+    }
+
+}

app/Http/Routes/RemoteRoutes.php

@@ -0,0 +1,27 @@
+<?php
+
+use Illuminate\Database\Schema\Blueprint;
+use Illuminate\Database\Migrations\Migration;
+
+class DownloadsServerIntToString extends Migration
+{
+    /**
+     * Run the migrations.
+     *
+     * @return void
+     */
+    public function up()
+    {
+        DB::statement('ALTER TABLE `downloads` MODIFY `server` CHAR(36) NOT NULL');
+    }
+
+    /**
+     * Reverse the migrations.
+     *
+     * @return void
+     */
+    public function down()
+    {
+        DB::statement('ALTER TABLE `downloads` MODIFY `server` MEDIUMINT(9) NOT NULL');
+    }
+}