1 files changed, 117 insertions, 0 deletions
diff --git a/java/com/android/dialer/blocking/Blocking.java b/java/com/android/dialer/blocking/Blocking.java
new file mode 100644
index 000000000..e86d0a6ac
--- /dev/null
+++ b/java/com/android/dialer/blocking/Blocking.java
@@ -0,0 +1,117 @@
+/*
+ * Copyright (C) 2018 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License
+ */
+
+package com.android.dialer.blocking;
+
+import android.content.ContentValues;
+import android.content.Context;
+import android.provider.BlockedNumberContract.BlockedNumbers;
+import android.support.annotation.Nullable;
+import android.telephony.PhoneNumberUtils;
+import com.android.dialer.common.database.Selection;
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.common.util.concurrent.ListeningExecutorService;
+
+/** Blocks and unblocks number. */
+public final class Blocking {
+
+  private Blocking() {}
+
+  /**
+   * Thrown when blocking cannot be performed because dialer is not the default dialer, or the
+   * current user is not a primary user.
+   *
+   * <p>Blocking is only allowed on the primary user (the first user added). Primary user cannot be
+   * easily checked because {@link
+   * android.provider.BlockedNumberContract#canCurrentUserBlockNumbers(Context)} is a slow IPC, and
+   * UserManager.isPrimaryUser() is a system API. Since secondary users are rare cases this class
+   * choose to ignore the check and let callers handle the failure later.
+   */
+  public static final class BlockingFailedException extends Exception {
+    BlockingFailedException(Throwable cause) {
+      super(cause);
+    }
+  }
+
+  /**
+   * Block a number.
+   *
+   * @param countryIso the current location used to guess the country code of the number if not
+   *     available. If {@code null} and {@code number} does not have a country code, only the
+   *     original number will be blocked.
+   * @throws BlockingFailedException in the returned future if the operation failed.
+   */
+  public static ListenableFuture<Void> block(
+      Context context,
+      ListeningExecutorService executorService,
+      String number,
+      @Nullable String countryIso) {
+    return executorService.submit(
+        () -> {
+          ContentValues values = new ContentValues();
+          values.put(BlockedNumbers.COLUMN_ORIGINAL_NUMBER, number);
+          String e164Number = PhoneNumberUtils.formatNumberToE164(number, countryIso);
+          if (e164Number != null) {
+            values.put(BlockedNumbers.COLUMN_E164_NUMBER, e164Number);
+          }
+          try {
+            context.getContentResolver().insert(BlockedNumbers.CONTENT_URI, values);
+          } catch (SecurityException e) {
+            throw new BlockingFailedException(e);
+          }
+          return null;
+        });
+  }
+
+  /**
+   * Unblock a number.
+   *
+   * @param countryIso the current location used to guess the country code of the number if not
+   *     available. If {@code null} and {@code number} does not have a country code, only the
+   *     original number will be unblocked.
+   * @throws BlockingFailedException in the returned future if the operation failed.
+   */
+  public static ListenableFuture<Void> unblock(
+      Context context,
+      ListeningExecutorService executorService,
+      String number,
+      @Nullable String countryIso) {
+    return executorService.submit(
+        () -> {
+          Selection selection =
+              Selection.column(BlockedNumbers.COLUMN_ORIGINAL_NUMBER).is("=", number);
+          String e164Number = PhoneNumberUtils.formatNumberToE164(number, countryIso);
+          if (e164Number != null) {
+            selection =
+                selection
+                    .buildUpon()
+                    .or(Selection.column(BlockedNumbers.COLUMN_E164_NUMBER).is("=", e164Number))
+                    .build();
+          }
+          try {
+            context
+                .getContentResolver()
+                .delete(
+                    BlockedNumbers.CONTENT_URI,
+                    selection.getSelection(),
+                    selection.getSelectionArgs());
+          } catch (SecurityException e) {
+            throw new BlockingFailedException(e);
+          }
+          return null;
+        });
+  }
+}