优化代码

Author: cuizhen

anypermission/src/main/java/per/goweii/anypermission/AnyPermission.java

@@ -3,12 +3,10 @@
 import android.app.Activity;
 import android.content.Context;
 import android.net.Uri;
-import android.os.Build;
 import android.support.annotation.NonNull;
 import android.support.v4.app.Fragment;
 
 import com.yanzhenjie.permission.AndPermission;
-import com.yanzhenjie.permission.option.Option;
 import com.yanzhenjie.permission.runtime.Permission;
 
 import java.io.File;
@@ -21,8 +19,7 @@
  */
 public class AnyPermission {
 
-    private final Context mContext;
-    private final Option mOption;
+    private final ContextHolder mContextHolder;
 
     public static AnyPermission with(@NonNull final Context context) {
         return new AnyPermission(context);
@@ -41,60 +38,52 @@ public static AnyPermission with(@NonNull final android.app.Fragment fragment) {
     }
 
     private AnyPermission(final Context context){
-        mContext = context;
-        mOption = AndPermission.with(context);
+        mContextHolder = new ContextHolder(context);
     }
 
     private AnyPermission(final Activity activity){
-        mContext = activity;
-        mOption = AndPermission.with(activity);
+        mContextHolder = new ContextHolder(activity);
     }
 
     private AnyPermission(final Fragment fragment){
-        mContext = fragment.getContext();
-        mOption = AndPermission.with(fragment);
+        mContextHolder = new ContextHolder(fragment);
     }
 
     private AnyPermission(final android.app.Fragment fragment){
-        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
-            mContext = fragment.getContext();
-        } else {
-            mContext = fragment.getActivity();
-        }
-        mOption = AndPermission.with(fragment);
+        mContextHolder = new ContextHolder(fragment);
     }
 
     public String name(String permission) {
-        return Permission.transformText(mContext, permission).get(0);
+        return Permission.transformText(mContextHolder.getContext(), permission).get(0);
     }
 
     public Uri fileUri(File file) {
-        return AndPermission.getFileUri(mContext, file);
+        return AndPermission.getFileUri(mContextHolder.getContext(), file);
     }
 
     public RuntimeRequester runtime(int requestCodeWhenGoSetting) {
-        return new RuntimeRequester(mOption, mContext, requestCodeWhenGoSetting);
+        return new RuntimeRequester(mContextHolder.getOption(), mContextHolder.getContext(), requestCodeWhenGoSetting);
     }
 
     public InstallRequester install(File apkFile) {
-        return new InstallRequester(mOption, apkFile);
+        return new InstallRequester(mContextHolder.getOption(), apkFile);
     }
 
     public OverlayRequester overlay() {
-        return new OverlayRequester(mOption);
+        return new OverlayRequester(mContextHolder.getOption());
     }
 
     @Deprecated
     public SettingRequester setting() {
-        return new SettingRequester(mOption);
+        return new SettingRequester(mContextHolder.getOption());
     }
 
     public NotificationShowRequester notificationShow() {
-        return new NotificationShowRequester(mOption);
+        return new NotificationShowRequester(mContextHolder.getOption());
     }
 
     public NotificationAccessRequester notificationAccess() {
-        return new NotificationAccessRequester(mOption);
+        return new NotificationAccessRequester(mContextHolder.getOption());
     }
 
 }

anypermission/src/main/java/per/goweii/anypermission/ContextHolder.java

@@ -0,0 +1,82 @@
+package per.goweii.anypermission;
+
+import android.app.Activity;
+import android.content.Context;
+import android.os.Build;
+import android.support.annotation.NonNull;
+import android.support.v4.app.Fragment;
+
+import com.yanzhenjie.permission.AndPermission;
+import com.yanzhenjie.permission.option.Option;
+
+/**
+ * 描述：
+ *
+ * @author Cuizhen
+ * @date 2019/3/19
+ */
+public class ContextHolder {
+
+    private final Context mContext;
+    private final Activity mActivity;
+    private final Fragment mSupportFragment;
+    private final android.app.Fragment mFragment;
+
+    ContextHolder(@NonNull final Context context) {
+        mContext = context;
+        mActivity = null;
+        mSupportFragment = null;
+        mFragment = null;
+    }
+
+    ContextHolder(@NonNull final Activity activity) {
+        mContext = null;
+        mActivity = activity;
+        mSupportFragment = null;
+        mFragment = null;
+    }
+
+    ContextHolder(@NonNull final Fragment fragment) {
+        mContext = null;
+        mActivity = null;
+        mSupportFragment = fragment;
+        mFragment = null;
+    }
+
+    ContextHolder(@NonNull final android.app.Fragment fragment) {
+        mContext = null;
+        mActivity = null;
+        mSupportFragment = null;
+        mFragment = fragment;
+    }
+
+    public Option getOption(){
+        if (mContext != null) {
+            return AndPermission.with(mContext);
+        } else if (mActivity != null) {
+            return AndPermission.with(mActivity);
+        } else if (mSupportFragment != null) {
+            return AndPermission.with(mSupportFragment);
+        } else if (mFragment != null) {
+            return AndPermission.with(mFragment);
+        }
+        return null;
+    }
+
+    public Context getContext(){
+        if (mContext != null) {
+            return mContext;
+        } else if (mActivity != null) {
+            return mActivity;
+        } else if (mSupportFragment != null) {
+            return mSupportFragment.getContext();
+        } else if (mFragment != null) {
+            if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
+                return mFragment.getContext();
+            } else {
+                return mFragment.getActivity();
+            }
+        }
+        return null;
+    }
+}