From 459e5b09c038ae3c2eac55dede9ca9fdf2fe72f9 Mon Sep 17 00:00:00 2001
From: Kosh <kosh20111@gmail.com>
Date: Sat, 13 May 2017 11:23:07 +0800
Subject: [PATCH] this commit improves wrap_code menuItem.

---
 .../repos/code/prettifier/ViewerFragment.java     | 15 +++++++++------
 app/src/main/res/menu/wrap_menu_option.xml        |  2 +-
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/app/src/main/java/com/fastaccess/ui/modules/repos/code/prettifier/ViewerFragment.java b/app/src/main/java/com/fastaccess/ui/modules/repos/code/prettifier/ViewerFragment.java
index 9383d64c..43947838 100644
--- a/app/src/main/java/com/fastaccess/ui/modules/repos/code/prettifier/ViewerFragment.java
+++ b/app/src/main/java/com/fastaccess/ui/modules/repos/code/prettifier/ViewerFragment.java
@@ -63,7 +63,6 @@ public class ViewerFragment extends BaseFragment<ViewerMvp.View, ViewerPresenter
     @Override public void onSetMdText(@NonNull String text, String baseUrl) {
         stateLayout.hideProgress();
         webView.setVisibility(View.VISIBLE);
-        Logger.e(!getPresenter().isRepo());
         webView.setGithubContent(text, baseUrl);
         getActivity().supportInvalidateOptionsMenu();
     }
@@ -145,20 +144,24 @@ public class ViewerFragment extends BaseFragment<ViewerMvp.View, ViewerPresenter
                 onSetCode(getPresenter().downloadedStream());
             }
         }
+        getActivity().supportInvalidateOptionsMenu();
     }
 
     @Override public void onCreateOptionsMenu(Menu menu, MenuInflater inflater) {
+        super.onCreateOptionsMenu(menu, inflater);
         inflater.inflate(R.menu.wrap_menu_option, menu);
         menu.findItem(R.id.wrap).setVisible(false);
-        super.onCreateOptionsMenu(menu, inflater);
     }
 
     @Override public void onPrepareOptionsMenu(Menu menu) {
-        MenuItem menuItem = menu.findItem(R.id.wrap);
-        if (!getPresenter().isMarkDown() || !getPresenter().isRepo()) {
-            menuItem.setVisible(true).setCheckable(isWrap);
-        }
         super.onPrepareOptionsMenu(menu);
+        MenuItem menuItem = menu.findItem(R.id.wrap);
+        Logger.e(getPresenter().isMarkDown() || getPresenter().isRepo() || getPresenter().isImage());
+        if (getPresenter().isMarkDown() || getPresenter().isRepo() || getPresenter().isImage()) {
+            menuItem.setVisible(false);
+        } else {
+            menuItem.setVisible(true).setCheckable(true).setChecked(isWrap);
+        }
     }
 
     @Override public boolean onOptionsItemSelected(MenuItem item) {
diff --git a/app/src/main/res/menu/wrap_menu_option.xml b/app/src/main/res/menu/wrap_menu_option.xml
index df8b33aa..ba0dd6b2 100644
--- a/app/src/main/res/menu/wrap_menu_option.xml
+++ b/app/src/main/res/menu/wrap_menu_option.xml
@@ -5,7 +5,7 @@
     <item
         android:id="@+id/wrap"
         android:checkable="true"
-        android:visible="false"
+        android:checked="false"
         android:title="@string/wrap_content"
         app:showAsAction="never"/>
 </menu>
\ No newline at end of file