添加线程选择接口

app/src/main/java/com/arialyy/simple/activity/MainActivity.java

@@ -1,10 +1,15 @@
 package com.arialyy.simple.activity;
 
+import android.Manifest;
 import android.content.Intent;
 import android.os.Bundle;
 import android.support.v7.widget.Toolbar;
 import android.view.View;
+import android.widget.Button;
 import butterknife.Bind;
+import com.arialyy.frame.permission.OnPermissionCallback;
+import com.arialyy.frame.permission.PermissionManager;
+import com.arialyy.frame.util.show.T;
 import com.arialyy.simple.R;
 import com.arialyy.simple.base.BaseActivity;
 import com.arialyy.simple.databinding.ActivityMainBinding;
@@ -14,6 +19,8 @@ import com.arialyy.simple.databinding.ActivityMainBinding;
  */
 public class MainActivity extends BaseActivity<ActivityMainBinding> {
   @Bind(R.id.toolbar) Toolbar mBar;
+  @Bind(R.id.single_task) Button mSigleBt;
+  @Bind(R.id.multi_task) Button mMultiBt;
 
   @Override protected int setLayoutId() {
     return R.layout.activity_main;
@@ -23,6 +30,28 @@ public class MainActivity extends BaseActivity<ActivityMainBinding> {
     super.init(savedInstanceState);
     setSupportActionBar(mBar);
     mBar.setTitle("多线程多任务下载");
+    boolean hasPermission = PermissionManager.getInstance()
+        .checkPermission(this, Manifest.permission.WRITE_EXTERNAL_STORAGE);
+    if (hasPermission) {
+      setEnable(true);
+    } else {
+      setEnable(false);
+      PermissionManager.getInstance().requestPermission(this, new OnPermissionCallback() {
+        @Override public void onSuccess(String... permissions) {
+          setEnable(true);
+        }
+
+        @Override public void onFail(String... permissions) {
+          T.showShort(MainActivity.this, "没有文件读写权限");
+          setEnable(false);
+        }
+      }, Manifest.permission.WRITE_EXTERNAL_STORAGE);
+    }
+  }
+
+  private void setEnable(boolean enable) {
+    mSigleBt.setEnabled(enable);
+    mMultiBt.setEnabled(enable);
   }
 
   public void onClick(View view) {