diff --git a/database/yggo.mwb b/database/yggo.mwb index e4df957..63fa3d7 100644 Binary files a/database/yggo.mwb and b/database/yggo.mwb differ diff --git a/library/mysql.php b/library/mysql.php index 9a9d3f6..b857838 100644 --- a/library/mysql.php +++ b/library/mysql.php @@ -456,11 +456,11 @@ class MySQL { return $this->_db->lastInsertId(); } - public function updateHostPageSnapDownload(int $hostPageSnapDownloadId, string $storage, int $size) { + public function updateHostPageSnapDownload(int $hostPageSnapDownloadId, string $storage, int $size, mixed $httpCode = NULL) { - $query = $this->_db->prepare('UPDATE `hostPageSnapDownload` SET `storage` = ?, `size` = ? WHERE `hostPageSnapDownloadId` = ? LIMIT 1'); + $query = $this->_db->prepare('UPDATE `hostPageSnapDownload` SET `storage` = ?, `size` = ?, `httpCode` = ? WHERE `hostPageSnapDownloadId` = ? LIMIT 1'); - $query->execute([$storage, $size, $hostPageSnapDownloadId]); + $query->execute([$storage, $size, $httpCode, $hostPageSnapDownloadId]); return $query->rowCount(); } diff --git a/media/db-prototype.png b/media/db-prototype.png index c838292..c58415d 100644 Binary files a/media/db-prototype.png and b/media/db-prototype.png differ diff --git a/public/file.php b/public/file.php index 766b744..f2b7646 100644 --- a/public/file.php +++ b/public/file.php @@ -78,7 +78,7 @@ switch ($type) { $snapSize = (int) @filesize('../storage/snap/' . $snapFile); - $db->updateHostPageSnapDownload($hostPageSnapDownloadId, 'local', $snapSize); + $db->updateHostPageSnapDownload($hostPageSnapDownloadId, 'local', $snapSize, 200); header('Content-Type: application/zip'); header(sprintf('Content-Length: %s', $snapSize)); @@ -96,7 +96,7 @@ switch ($type) { if ($snapSize = $ftp->size($snapFile)) { - $db->updateHostPageSnapDownload($hostPageSnapDownloadId, 'mega', $snapSize); + $db->updateHostPageSnapDownload($hostPageSnapDownloadId, 'mega', $snapSize, 200); header('Content-Type: application/zip'); header(sprintf('Content-Length: %s', $snapSize)); @@ -108,7 +108,7 @@ switch ($type) { } else { - $db->updateHostPageSnapDownload($hostPageSnapDownloadId, 'mega', $snapSize); + $db->updateHostPageSnapDownload($hostPageSnapDownloadId, 'mega', $snapSize, 404); header('HTTP/1.0 404 Not Found'); @@ -117,7 +117,7 @@ switch ($type) { } else { - $db->updateHostPageSnapDownload($hostPageSnapDownloadId, 'mega', $snapSize); + $db->updateHostPageSnapDownload($hostPageSnapDownloadId, 'mega', $snapSize, 404); header('HTTP/1.0 404 Not Found'); @@ -127,7 +127,7 @@ switch ($type) { // Return 404 when file not found } else { - $db->updateHostPageSnapDownload($hostPageSnapDownloadId, 'other', $snapSize); + $db->updateHostPageSnapDownload($hostPageSnapDownloadId, 'other', $snapSize, 404); header('HTTP/1.0 404 Not Found');