Implement attachment downloading

Author: carlbennett

src/controllers/Attachment/Download.php

@@ -0,0 +1,54 @@
+<?php
+
+namespace BNETDocs\Controllers\Attachment;
+
+use \BNETDocs\Libraries\Attachment;
+use \BNETDocs\Libraries\Cache;
+use \BNETDocs\Libraries\Common;
+use \BNETDocs\Libraries\Controller;
+use \BNETDocs\Libraries\Database;
+use \BNETDocs\Libraries\DatabaseDriver;
+use \BNETDocs\Libraries\Exceptions\AttachmentNotFoundException;
+use \BNETDocs\Libraries\Exceptions\UnspecifiedViewException;
+use \BNETDocs\Libraries\Router;
+use \BNETDocs\Models\Attachment\Download as DownloadModel;
+use \BNETDocs\Views\Attachment\DownloadRaw as DownloadRawView;
+
+class Download extends Controller {
+
+  public function run(Router &$router) {
+    switch ($router->getRequestPathExtension()) {
+      case "":
+        $view = new DownloadRawView();
+      break;
+      default:
+        throw new UnspecifiedViewException();
+    }
+
+    $data = $router->getRequestQueryArray();
+    $id   = (isset($data["id"]) ? (int) $data["id"] : null);
+
+    $model                = new DownloadModel();
+    $model->attachment_id = $id;
+
+    try {
+      $model->attachment = new Attachment($id);
+    } catch (AttachmentNotFoundException $e) {
+      $model->attachment = null;
+    }
+
+    ob_start();
+    $view->render($model);
+    $router->setResponseCode(($model->attachment ? 200 : 404));
+    $router->setResponseTTL(31536000); // 1 year
+    $router->setResponseHeader("Content-Type", $view->getMimeType());
+    if ($model->extra_headers) {
+      foreach ($model->extra_headers as $name => $value) {
+        $router->setResponseHeader($name, $value);
+      }
+    }
+    $router->setResponseContent(ob_get_contents());
+    ob_end_clean();
+  }
+
+}

src/libraries/Router.php

@@ -3,6 +3,7 @@
 namespace BNETDocs\Libraries;
 
 use \BNETDocs\Controllers\API\Comment as APICommentController;
+use \BNETDocs\Controllers\Attachment\Download as AttachmentDownloadController;
 use \BNETDocs\Controllers\Credits as CreditsController;
 use \BNETDocs\Controllers\Document\Index as DocumentIndexController;
 use \BNETDocs\Controllers\Document\Popular as DocumentPopularController;
@@ -254,6 +255,17 @@ public function route(Pair &$redirect = null) {
             }
             $controller = new RedirectController($url, $code);
           break;
+          case "attachment":
+            switch ($subpath) {
+              case "download":
+                $controller = new AttachmentDownloadController();
+              break;
+              default:
+                throw new ControllerNotFoundException(
+                  $path . "/" . $subpath
+                );
+            }
+          break;
           case "api":
             switch ($subpath) {
               case "comment":

src/models/Attachment/Download.php

@@ -0,0 +1,20 @@
+<?php
+
+namespace BNETDocs\Models\Attachment;
+
+use \BNETDocs\Libraries\Model;
+
+class Download extends Model {
+
+  public $attachment;
+  public $attachment_id;
+  public $extra_headers;
+
+  public function __construct() {
+    parent::__construct();
+    $this->attachment    = null;
+    $this->attachment_id = null;
+    $this->extra_headers = null;
+  }
+
+}

src/views/Attachment/DownloadRaw.php

@@ -0,0 +1,32 @@
+<?php
+
+namespace BNETDocs\Views\Attachment;
+
+use \BNETDocs\Libraries\Common;
+use \BNETDocs\Libraries\Exceptions\IncorrectModelException;
+use \BNETDocs\Libraries\Model;
+use \BNETDocs\Libraries\View;
+use \BNETDocs\Models\Attachment\Download as DownloadModel;
+use \DateTime;
+
+class DownloadRaw extends View {
+
+  public function getMimeType() {
+    return "application/octet-stream";
+  }
+
+  public function render(Model &$model) {
+    if (!$model instanceof DownloadModel) {
+      throw new IncorrectModelException();
+    }
+    $model->extra_headers = [
+      "Content-Disposition" => "attachment;filename=\""
+        . $model->attachment->getFilename() . "\"",
+      "Last-Modified" => $model->attachment->getCreatedDateTime()->format(
+        DateTime::RFC1123
+      )
+    ];
+    echo $model->attachment->getContent();
+  }
+
+}