wip

app/Server/Factory.php

@@ -6,15 +6,21 @@ use App\Contracts\ConnectionManager as ConnectionManagerContract;
 use App\Contracts\SubdomainGenerator;
 use App\HttpServer\HttpServer;
 use App\Server\Connections\ConnectionManager;
+use App\Server\Http\Controllers\Admin\DeleteUsersController;
+use App\Server\Http\Controllers\Admin\ListUsersController;
+use App\Server\Http\Controllers\Admin\StoreUsersController;
 use App\Server\Http\Controllers\ControlMessageController;
 use App\Server\Http\Controllers\TunnelMessageController;
 use App\Server\SubdomainGenerator\RandomSubdomainGenerator;
+use Clue\React\SQLite\DatabaseInterface;
 use Ratchet\Http\Router;
 use Ratchet\Server\IoServer;
 use Ratchet\WebSocket\WsServer;
 use React\Socket\Server;
 use React\EventLoop\LoopInterface;
 use React\EventLoop\Factory as LoopFactory;
+use Symfony\Component\Finder\Finder;
+use Symfony\Component\Finder\SplFileInfo;
 use Symfony\Component\Routing\Matcher\UrlMatcher;
 use Symfony\Component\Routing\RequestContext;
 use Symfony\Component\Routing\Route;
@@ -34,9 +40,13 @@ class Factory
     /** @var \React\EventLoop\LoopInterface */
     protected $loop;
 
+    /** @var RouteCollection */
+    protected $routes;
+
     public function __construct()
     {
         $this->loop = LoopFactory::create();
+        $this->routes = new RouteCollection();
     }
 
     public function setHost(string $host)
@@ -67,25 +77,42 @@ class Factory
         return $this;
     }
 
-    protected function getRoutes(): RouteCollection
+    protected function addExposeRoutes()
     {
-        $routes = new RouteCollection();
-
-        $routes->add('control',
+        $this->routes->add('control',
             new Route('/__expose_control__', [
                 '_controller' => new WsServer(app(ControlMessageController::class))
             ], [], [], null, [], []
             )
         );
 
-        $routes->add('tunnel',
+        $this->routes->add('tunnel',
             new Route('/{__catchall__}', [
                 '_controller' => app(TunnelMessageController::class),
             ], [
                 '__catchall__' => '.*'
             ]));
+    }
 
-        return $routes;
+    protected function addAdminRoutes()
+    {
+        $this->routes->add('admin.users.index',
+            new Route('/expose/users', [
+                '_controller' => app(ListUsersController::class),
+            ], [], [], null, [], ['GET'])
+        );
+
+        $this->routes->add('admin.users.store',
+            new Route('/expose/users', [
+                '_controller' => app(StoreUsersController::class),
+            ], [], [], null, [], ['POST'])
+        );
+
+        $this->routes->add('admin.users.delete',
+            new Route('/expose/users/delete/{id}', [
+                '_controller' => app(DeleteUsersController::class),
+            ], [], [], null, [], ['DELETE'])
+        );
     }
 
     protected function bindConfiguration()
@@ -117,9 +144,17 @@ class Factory
 
         $this->bindSubdomainGenerator();
 
+        $this->bindDatabase();
+
+        $this->ensureDatabaseIsInitialized();
+
         $this->bindConnectionManager();
 
-        $urlMatcher = new UrlMatcher($this->getRoutes(), new RequestContext);
+        $this->addAdminRoutes();
+
+        $this->addExposeRoutes();
+
+        $urlMatcher = new UrlMatcher($this->routes, new RequestContext);
 
         $router = new Router($urlMatcher);
 
@@ -128,4 +163,36 @@ class Factory
         return new IoServer($http, $socket, $this->loop);
     }
 
+    protected function bindDatabase()
+    {
+        app()->singleton(DatabaseInterface::class, function() {
+            $factory = new \Clue\React\SQLite\Factory($this->loop);
+            return $factory->openLazy(base_path('database/expose.db'));
+        });
+    }
+
+    protected function ensureDatabaseIsInitialized()
+    {
+        /** @var DatabaseInterface $db */
+        $db = app(DatabaseInterface::class);
+
+        $migrations = (new Finder())
+            ->files()
+            ->ignoreDotFiles(true)
+            ->in(database_path('migrations'))
+            ->name('*.sql');
+
+        /** @var SplFileInfo $migration */
+        foreach ($migrations as $migration) {
+            $db->exec($migration->getContents());
+        }
+    }
+
+    public function validateAuthTokens(bool $validate)
+    {
+        dump($validate);
+
+        return $this;
+    }
+
 }

app/Server/Http/Controllers/Admin/DeleteUsersController.php

@@ -0,0 +1,38 @@
+<?php
+
+namespace App\Server\Http\Controllers\Admin;
+
+use App\HttpServer\Controllers\PostController;
+use Clue\React\SQLite\DatabaseInterface;
+use Clue\React\SQLite\Result;
+use GuzzleHttp\Psr7\Response;
+use Illuminate\Http\Request;
+use Illuminate\Support\Facades\Validator;
+use Illuminate\Support\Str;
+use Ratchet\ConnectionInterface;
+use Twig\Environment;
+use Twig\Loader\ArrayLoader;
+use function GuzzleHttp\Psr7\str;
+use function GuzzleHttp\Psr7\stream_for;
+
+class DeleteUsersController extends PostController
+{
+    protected $keepConnectionOpen = true;
+
+    /** @var DatabaseInterface */
+    protected $database;
+
+    public function __construct(DatabaseInterface $database)
+    {
+        $this->database = $database;
+    }
+
+    public function handle(Request $request, ConnectionInterface $httpConnection)
+    {
+        $this->database->query("DELETE FROM users WHERE id = :id", ['id' => $request->id])
+                ->then(function (Result $result) use ($httpConnection) {
+                    $httpConnection->send(respond_json(['deleted' => true], 200));
+                    $httpConnection->close();
+                });
+    }
+}

app/Server/Http/Controllers/Admin/ListUsersController.php

@@ -0,0 +1,53 @@
+<?php
+
+namespace App\Server\Http\Controllers\Admin;
+
+use App\HttpServer\Controllers\PostController;
+use Clue\React\SQLite\DatabaseInterface;
+use Clue\React\SQLite\Result;
+use GuzzleHttp\Psr7\Response;
+use Illuminate\Http\Request;
+use Ratchet\ConnectionInterface;
+use Twig\Environment;
+use Twig\Loader\ArrayLoader;
+use function GuzzleHttp\Psr7\str;
+use function GuzzleHttp\Psr7\stream_for;
+
+class ListUsersController extends PostController
+{
+    protected $keepConnectionOpen = true;
+
+    /** @var DatabaseInterface */
+    protected $database;
+
+    public function __construct(DatabaseInterface $database)
+    {
+        $this->database = $database;
+    }
+
+    public function handle(Request $request, ConnectionInterface $httpConnection)
+    {
+        $this->database->query('SELECT * FROM users ORDER by created_at DESC')->then(function (Result $result) use ($httpConnection) {
+            $httpConnection->send(
+                respond_html($this->getView(['users' => $result->rows]))
+            );
+
+            $httpConnection->close();
+        }, function (\Exception $exception) use ($httpConnection) {
+            $httpConnection->send(respond_html('Something went wrong: '.$exception->getMessage(), 500));
+
+            $httpConnection->close();
+        });
+    }
+
+    protected function getView(array $data)
+    {
+        $twig = new Environment(
+            new ArrayLoader([
+                'template' => file_get_contents(base_path('resources/views/admin/users/index.twig')),
+            ])
+        );
+
+        return stream_for($twig->render('template', $data));
+    }
+}

app/Server/Http/Controllers/Admin/StoreUsersController.php

@@ -0,0 +1,62 @@
+<?php
+
+namespace App\Server\Http\Controllers\Admin;
+
+use App\HttpServer\Controllers\PostController;
+use Clue\React\SQLite\DatabaseInterface;
+use Clue\React\SQLite\Result;
+use GuzzleHttp\Psr7\Response;
+use Illuminate\Http\Request;
+use Illuminate\Support\Facades\Validator;
+use Illuminate\Support\Str;
+use Ratchet\ConnectionInterface;
+use Twig\Environment;
+use Twig\Loader\ArrayLoader;
+use function GuzzleHttp\Psr7\str;
+use function GuzzleHttp\Psr7\stream_for;
+
+class StoreUsersController extends PostController
+{
+    protected $keepConnectionOpen = true;
+
+    /** @var DatabaseInterface */
+    protected $database;
+
+    public function __construct(DatabaseInterface $database)
+    {
+        $this->database = $database;
+    }
+
+    public function handle(Request $request, ConnectionInterface $httpConnection)
+    {
+        $validator = Validator::make($request->all(), [
+            'name' => 'required',
+        ], [
+            'required' => 'The :attribute field is required.',
+        ]);
+
+        if ($validator->fails()) {
+            $httpConnection->send(respond_json(['errors' => $validator->getMessageBag()], 401));
+            $httpConnection->close();
+
+            return;
+        }
+
+        $insertData = [
+            'name' => $request->get('name'),
+            'auth_token' => (string)Str::uuid()
+        ];
+
+        $this->database->query("
+            INSERT INTO users (name, auth_token, created_at)
+            VALUES (:name, :auth_token, DATETIME('now'))
+        ", $insertData)
+                ->then(function (Result $result) use ($httpConnection) {
+                    $this->database->query("SELECT * FROM users WHERE id = :id", ['id' => $result->insertId])
+                        ->then(function (Result $result) use ($httpConnection) {
+                            $httpConnection->send(respond_json(['user' => $result->rows[0]], 200));
+                            $httpConnection->close();
+                        });
+                });
+    }
+}

app/Server/Http/Controllers/ControlMessageController.php

@@ -3,6 +3,9 @@
 namespace App\Server\Http\Controllers;
 
 use App\Contracts\ConnectionManager;
+use App\HttpServer\QueryParameters;
+use Clue\React\SQLite\DatabaseInterface;
+use Clue\React\SQLite\Result;
 use stdClass;
 use Ratchet\ConnectionInterface;
 use Ratchet\MessageComponentInterface;
@@ -13,9 +16,21 @@ class ControlMessageController implements MessageComponentInterface
     /** @var ConnectionManager */
     protected $connectionManager;
 
-    public function __construct(ConnectionManager $connectionManager)
+    /** @var DatabaseInterface */
+    protected $database;
+
+    public function __construct(ConnectionManager $connectionManager, DatabaseInterface $database)
     {
         $this->connectionManager = $connectionManager;
+        $this->database = $database;
+    }
+
+    /**
+     * @inheritDoc
+     */
+    function onOpen(ConnectionInterface $connection)
+    {
+        $this->verifyAuthToken($connection);
     }
 
     /**
@@ -80,14 +95,6 @@ class ControlMessageController implements MessageComponentInterface
         ]);
     }
 
-    /**
-     * @inheritDoc
-     */
-    function onOpen(ConnectionInterface $conn)
-    {
-        //
-    }
-
     /**
      * @inheritDoc
      */
@@ -95,4 +102,21 @@ class ControlMessageController implements MessageComponentInterface
     {
         //
     }
+
+    protected function verifyAuthToken(ConnectionInterface $connection)
+    {
+        $authToken = QueryParameters::create($connection->httpRequest)->get('authToken');
+
+        $this->database
+            ->query("SELECT * FROM users WHERE auth_token = :token", ['token' => $authToken])
+            ->then(function (Result $result) use ($connection) {
+                if (count($result->rows) === 0) {
+                    $connection->send(json_encode([
+                        'event' => 'authenticationFailed',
+                        'data' => []
+                    ]));
+                    $connection->close();
+                }
+        });
+    }
 }