1 files changed, 9 insertions, 8 deletions
diff --git a/app/src/main/kotlin/com/pitchedapps/frost/fragments/WebFragments.kt b/app/src/main/kotlin/com/pitchedapps/frost/fragments/WebFragments.kt
index 2740a36f..cdeea064 100644
--- a/app/src/main/kotlin/com/pitchedapps/frost/fragments/WebFragments.kt
+++ b/app/src/main/kotlin/com/pitchedapps/frost/fragments/WebFragments.kt
@@ -1,26 +1,27 @@
 package com.pitchedapps.frost.fragments
 
 import com.pitchedapps.frost.R
+import com.pitchedapps.frost.facebook.FbItem
 import com.pitchedapps.frost.views.FrostWebView
 import com.pitchedapps.frost.web.FrostWebViewClient
 import com.pitchedapps.frost.web.FrostWebViewClientMenu
 
 /**
  * Created by Allan Wang on 27/12/17.
+ *
+ * Basic webfragment
+ * Do not extend as this is always a fallback
  */
-open class WebFragment : BaseFragment() {
+class WebFragment : BaseFragment() {
 
     override val layoutRes: Int = R.layout.view_content_web
 
     /**
      * Given a webview, output a client
      */
-    open fun client(web: FrostWebView) = FrostWebViewClient(web)
-
-}
-
-class WebFragmentMenu : WebFragment() {
-
-    override fun client(web: FrostWebView) = FrostWebViewClientMenu(web)
+    fun client(web: FrostWebView) = when (baseEnum) {
+        FbItem.MENU -> FrostWebViewClientMenu(web)
+        else -> FrostWebViewClient(web)
+    }
 
 }
 \ No newline at end of file