From 1feee6aa9a6ef15875a3d32dfd9443123e627a61 Mon Sep 17 00:00:00 2001
From: "Pfeil, Scott Robert" <scott.pfeil3@verizonwireless.com>
Date: Mon, 13 Jan 2020 10:48:35 -0500
Subject: [PATCH] three layer model fix

---
 .../ThreeLayerPageTemplateModel.swift         | 42 +++++++++++++++++--
 1 file changed, 39 insertions(+), 3 deletions(-)

diff --git a/MVMCoreUI/Models/Template/ThreeLayerPageTemplateModel.swift b/MVMCoreUI/Models/Template/ThreeLayerPageTemplateModel.swift
index b4604fbc..5d4d1b25 100644
--- a/MVMCoreUI/Models/Template/ThreeLayerPageTemplateModel.swift
+++ b/MVMCoreUI/Models/Template/ThreeLayerPageTemplateModel.swift
@@ -17,7 +17,43 @@ import Foundation
 
     public var isAtomicTabs: Bool?
 
-    public var header: MoleculeStackModel?
-    public var middle: MoleculeStackModel?
-    public var footer: MoleculeStackModel?
+    public var header: MoleculeProtocol?
+    public var middle: MoleculeProtocol?
+    public var footer: MoleculeProtocol?
+    
+    public init(pageType: String, header: MoleculeProtocol?, middle: MoleculeProtocol?, footer: MoleculeProtocol?) {
+        self.pageType = pageType
+        self.header = header
+        self.middle = middle
+        self.footer = footer
+    }
+
+    enum CodingKeys: String, CodingKey {
+        case pageType
+        case screenHeading
+        case header
+        case footer
+        case middle
+        case isAtomicTabs
+    }
+
+    required public init(from decoder: Decoder) throws {
+        let typeContainer = try decoder.container(keyedBy: CodingKeys.self)
+        pageType = try typeContainer.decode(String.self, forKey: .pageType)
+        screenHeading = try typeContainer.decodeIfPresent(String.self, forKey: .screenHeading)
+        isAtomicTabs = try typeContainer.decodeIfPresent(Bool.self, forKey: .isAtomicTabs)
+        header = try typeContainer.decodeMoleculeIfPresent(codingKey: .header)
+        header = try typeContainer.decodeMoleculeIfPresent(codingKey: .middle)
+        footer = try typeContainer.decodeMoleculeIfPresent(codingKey: .footer)
+    }
+
+    public func encode(to encoder: Encoder) throws {
+        var container = encoder.container(keyedBy: CodingKeys.self)
+        try container.encode(pageType, forKey: .pageType)
+        try container.encodeIfPresent(screenHeading, forKey: .screenHeading)
+        try container.encodeIfPresent(isAtomicTabs, forKey: .isAtomicTabs)
+        try container.encodeModelIfPresent(header, forKey: .header)
+        try container.encodeModelIfPresent(header, forKey: .middle)
+        try container.encodeModelIfPresent(footer, forKey: .footer)
+    }
 }