From d8812a94556d51b778b7095128c4b2fe85b0043b Mon Sep 17 00:00:00 2001
From: Yozone Wang <wangyaozh@gmail.com>
Date: Mon, 6 Nov 2017 16:05:31 +0800
Subject: [PATCH] Make header's font size customizable

---
 .../TextKit/ParagraphProperty/Header.swift    | 22 +++++++++++--------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/Aztec/Classes/TextKit/ParagraphProperty/Header.swift b/Aztec/Classes/TextKit/ParagraphProperty/Header.swift
index 618fd0218..cb73034d6 100644
--- a/Aztec/Classes/TextKit/ParagraphProperty/Header.swift
+++ b/Aztec/Classes/TextKit/ParagraphProperty/Header.swift
@@ -19,16 +19,20 @@ open class Header: ParagraphProperty {
         case h5 = 5
         case h6 = 6
 
+        public static var fontSizeMap: [HeaderType: Float] = {
+            return [
+                .h1: 36,
+                .h2: 24,
+                .h3: 21,
+                .h4: 16,
+                .h5: 14,
+                .h6: 11,
+                .none: Constants.defaultFontSize
+                ]
+        }()
+
         public var fontSize: Float {
-            switch self {
-            case .none: return Constants.defaultFontSize
-            case .h1: return 36
-            case .h2: return 24
-            case .h3: return 21
-            case .h4: return 16
-            case .h5: return 14
-            case .h6: return 11
-            }
+            return HeaderType.fontSizeMap[self] ?? Constants.defaultFontSize
         }
     }