1 files changed, 13 insertions, 21 deletions

diff --git a/src/web/_model/error.php b/src/web/_model/error.php
index 0a08fdd..11b56f9 100644
--- a/src/web/_model/error.php
+++ b/src/web/_model/error.php
@@ -1,30 +1,22 @@
 <?php /* Copyright (c) 2024 Freya Murphy */
 class Error_model extends Model {
 
-	function __construct($load) {
-		parent::__construct($load);
-	}
-
-	private function get_msg(&$data) {
-		if (!array_key_exists('code', $_GET)) {
-			http_response_code(500);
-			$data['msg'] = ucfirst(lang('error'));
-			$data['title'] = '500';
-		} else {
-			$code = $_GET['code'];
-			http_response_code($code);
-			$data['title'] = $code;
-			$msg = ucfirst(lang('error_' . $code, FALSE));
-			if (!$msg) {
-				$msg = ucfirst(lang('error'));
-			}
-			$data['msg'] = $msg;
+	private function get_msg(&$data, int $code)
+	{
+		http_response_code($code);
+		$data['title'] = $code;
+		$msg = ucfirst(lang('error_' . $code, FALSE));
+		if (!$msg) {
+			$msg = ucfirst(lang('error'));
 		}
+		$data['msg'] = $msg;
+
 	}
 
-	public function get_data(): ?array {
-		$data = parent::get_data();
-		$this->get_msg($data);
+	public function get_data(int $code): array
+	{
+		$data = parent::get_base_data('error');
+		$this->get_msg($data, $code);
 		return $data;
 	}
 }