Clean Code

app/src/main/java/com/hippo/ehviewer/ui/EhActivity.java

@@ -19,6 +19,8 @@ package com.hippo.ehviewer.ui;
 import android.os.Bundle;
 import android.support.annotation.Nullable;
 import android.support.v7.app.AppCompatActivity;
+import android.util.Log;
+import android.view.WindowManager;
 
 import com.google.analytics.tracking.android.EasyTracker;
 import com.hippo.ehviewer.EhApplication;
@@ -61,4 +63,15 @@ public abstract class EhActivity extends AppCompatActivity {
             mTrackStarted = false;
         }
     }
+
+    @Override
+    protected void onResume() {
+        super.onResume();
+        if(Settings.getEnabledSecurity()){
+            getWindow().setFlags(WindowManager.LayoutParams.FLAG_SECURE,
+                    WindowManager.LayoutParams.FLAG_SECURE);
+        }else{
+            getWindow().clearFlags(WindowManager.LayoutParams.FLAG_SECURE);
+        }
+    }
 }