代码优化

galaxy/galaxy/Classes/Modules/Home(首页)/C/YHHomeWebViewController.swift

@@ -43,10 +43,11 @@ class YHHomeWebViewController: YHBaseViewController, WKUIDelegate, WKNavigationD
         gk_navBarAlpha = 1.0
         gk_navLineHidden = false
         gk_navTitle = vcTitle
-        if !(vcTitle == "积分中心" || vcTitle == "预约银河湾") {
-            let rightButtonItem = UIBarButtonItem(image: UIImage(named: "share_item")?.withRenderingMode(.alwaysOriginal), style: .plain, target: self, action: #selector(share))
-            gk_navRightBarButtonItem = rightButtonItem
-            gk_navItemRightSpace = 16
+        
+        if (vcTitle == "香港身份测评码") {
+            addRightItems()
+        } else {
+            printLog("通过ID的逻辑来处理")
         }
         view.backgroundColor = .white
         viewModel = YHHomeWebViewModel()
@@ -210,6 +211,11 @@ class YHHomeWebViewController: YHBaseViewController, WKUIDelegate, WKNavigationD
                     if operationFlag {
                         self.block?(model)
                     }
+                    
+                    
+                    if model.img_url.isEmpty == false {
+                        self.addRightItems()
+                    }
                 }
             }
         }
@@ -356,3 +362,19 @@ class YHHomeWebViewController: YHBaseViewController, WKUIDelegate, WKNavigationD
         getData()
     }
 }
+
+
+extension YHHomeWebViewController {
+    
+    func addRightItems() {
+        let rightButtonItem = UIBarButtonItem(image: UIImage(named: "share_item")?.withRenderingMode(.alwaysOriginal), style: .plain, target: self, action: #selector(share))
+        gk_navRightBarButtonItem = rightButtonItem
+        gk_navItemRightSpace = 16
+    }
+    
+    func removeRightItems() {
+        gk_navRightBarButtonItems = []
+    }
+}
+
+