From 04e431dae5a1a63f91edde06adeaba00b2d0698f Mon Sep 17 00:00:00 2001
From: Rostislav Wolny <rosta@mailpoet.com>
Date: Wed, 5 Aug 2020 15:34:06 +0200
Subject: [PATCH] Add form template ability to transform to form entity

[MAILPOET-2810]
---
 lib/Form/Templates/Template.php           | 10 ++++++++++
 lib/Form/Templates/TemplateRepository.php |  6 +-----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/lib/Form/Templates/Template.php b/lib/Form/Templates/Template.php
index ec6443dfbf..006ca936e3 100644
--- a/lib/Form/Templates/Template.php
+++ b/lib/Form/Templates/Template.php
@@ -2,6 +2,8 @@
 
 namespace MailPoet\Form\Templates;
 
+use MailPoet\Entities\FormEntity;
+
 abstract class Template {
   abstract public function getName(): string;
 
@@ -91,4 +93,12 @@ abstract class Template {
 }
 EOL;
   }
+
+  public function toFormEntity(): FormEntity {
+    $formEntity = new FormEntity($this->getName());
+    $formEntity->setBody($this->getBody());
+    $formEntity->setSettings($this->getSettings());
+    $formEntity->setStyles($this->getStyles());
+    return $formEntity;
+  }
 }
diff --git a/lib/Form/Templates/TemplateRepository.php b/lib/Form/Templates/TemplateRepository.php
index 30f5a74eba..ab6223b5cb 100644
--- a/lib/Form/Templates/TemplateRepository.php
+++ b/lib/Form/Templates/TemplateRepository.php
@@ -25,11 +25,7 @@ class TemplateRepository {
     }
     /** @var Template $template */
     $template = new $this->templates[$templateId]();
-    $formEntity = new FormEntity($template->getName());
-    $formEntity->setBody($template->getBody());
-    $formEntity->setSettings($template->getSettings());
-    $formEntity->setStyles($template->getStyles());
-    return $formEntity;
+    return $template->toFormEntity();
   }
 
   /**