From edf7edc9c101e607e2fa1c5f48f82ccbf2999708 Mon Sep 17 00:00:00 2001
From: Dave Mc Nicoll <info@mcnd.ca>
Date: Mon, 10 Feb 2020 15:18:26 -0500
Subject: [PATCH] - Added a custom 'context' if needed, called from
 getContext().

---
 src/Method/FormHandler.php          | 11 +++++++++--
 src/Method/Message/ErrorMessage.php | 14 ++++++++++++++
 2 files changed, 23 insertions(+), 2 deletions(-)
 create mode 100644 src/Method/Message/ErrorMessage.php

diff --git a/src/Method/FormHandler.php b/src/Method/FormHandler.php
index 00c5166..7edea9c 100644
--- a/src/Method/FormHandler.php
+++ b/src/Method/FormHandler.php
@@ -8,7 +8,7 @@ class FormHandler {
     # return new class($this->request, new FormContext(), $form) {
 
     public bool $sent = false;
-    
+
     public FormContext $context;
 
     protected ServerRequestInterface $request;
@@ -20,7 +20,14 @@ class FormHandler {
         $this->request = $request;
         $this->sent = $this->requestSent();
         $this->form = $form;
-        $this->context = $context ?: new FormContext($request);
+
+        if ( $context ) {
+            $this->context = $context;
+        }
+        else {
+            $this->context = method_exists($form, 'getContext') ? $form->getContext($request) : new FormContext($request);
+        }
+
         $this->context->formSent = $this->sent;
         $this->initialize();
     }
diff --git a/src/Method/Message/ErrorMessage.php b/src/Method/Message/ErrorMessage.php
new file mode 100644
index 0000000..799fe66
--- /dev/null
+++ b/src/Method/Message/ErrorMessage.php
@@ -0,0 +1,14 @@
+<?php
+
+namespace Picea\Ui\Method\Message;
+
+use Picea\Ui\Metohd\FormMessage;
+
+class ErrorMessage implements FormMessage {
+
+    public function isError() : bool
+    {
+        return true;
+    }
+
+}