From e35a14939be8f83bfd8c3aee60036f066975318f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Halil=20=C4=B0brahim=20Kalkan?= <hi_kalkan@yahoo.com>
Date: Tue, 1 May 2018 14:02:12 +0300
Subject: [PATCH] Refactor AbpButtonTagHelperServiceBase

---
 .../Button/AbpButtonTagHelperServiceBase.cs         | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/Volo.Abp.AspNetCore.Mvc.UI.Bootstrap/TagHelpers/Button/AbpButtonTagHelperServiceBase.cs b/src/Volo.Abp.AspNetCore.Mvc.UI.Bootstrap/TagHelpers/Button/AbpButtonTagHelperServiceBase.cs
index b293b36c78..cd5c22be86 100644
--- a/src/Volo.Abp.AspNetCore.Mvc.UI.Bootstrap/TagHelpers/Button/AbpButtonTagHelperServiceBase.cs
+++ b/src/Volo.Abp.AspNetCore.Mvc.UI.Bootstrap/TagHelpers/Button/AbpButtonTagHelperServiceBase.cs
@@ -42,7 +42,18 @@ namespace Volo.Abp.AspNetCore.Mvc.UI.Bootstrap.TagHelpers.Button
                 return;
             }
 
-            output.Content.AppendHtml($"<i class=\"fa fa-{TagHelper.Icon}\"></i> ");
+            output.Content.AppendHtml($"<i class=\"{GetIconClass(context, output)}\"></i> ");
+        }
+
+        protected virtual string GetIconClass(TagHelperContext context, TagHelperOutput output)
+        {
+            switch (TagHelper.IconType)
+            {
+                case FontIconType.FontAwesome:
+                    return "fa fa-" + TagHelper.Icon;
+                default:
+                    return TagHelper.Icon;
+            }
         }
 
         protected virtual void AddText(TagHelperContext context, TagHelperOutput output)