1 /*
2  * Copyright (C) 2018 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License
15  */
16 
17 package com.android.dialer.commandline.impl;
18 
19 import android.content.Context;
20 import android.support.annotation.NonNull;
21 import com.android.dialer.DialerPhoneNumber;
22 import com.android.dialer.blocking.Blocking;
23 import com.android.dialer.commandline.Arguments;
24 import com.android.dialer.commandline.Command;
25 import com.android.dialer.common.concurrent.Annotations.BackgroundExecutor;
26 import com.android.dialer.inject.ApplicationContext;
27 import com.android.dialer.phonelookup.PhoneLookupComponent;
28 import com.android.dialer.phonelookup.PhoneLookupInfo;
29 import com.android.dialer.phonelookup.consolidator.PhoneLookupInfoConsolidator;
30 import com.android.dialer.phonenumberproto.DialerPhoneNumberUtil;
31 import com.google.common.collect.ImmutableList;
32 import com.google.common.util.concurrent.Futures;
33 import com.google.common.util.concurrent.ListenableFuture;
34 import com.google.common.util.concurrent.ListeningExecutorService;
35 import com.google.common.util.concurrent.MoreExecutors;
36 import javax.inject.Inject;
37 
38 /** Block or unblock a number. */
39 public class BlockingCommand implements Command {
40 
41   @NonNull
42   @Override
getShortDescription()43   public String getShortDescription() {
44     return "block or unblock numbers";
45   }
46 
47   @NonNull
48   @Override
getUsage()49   public String getUsage() {
50     return "blocking block|unblock|isblocked number\n\n" + "number should be e.164 formatted";
51   }
52 
53   private final Context appContext;
54   private final ListeningExecutorService executorService;
55 
56   @Inject
BlockingCommand( @pplicationContext Context context, @BackgroundExecutor ListeningExecutorService executorService)57   BlockingCommand(
58       @ApplicationContext Context context,
59       @BackgroundExecutor ListeningExecutorService executorService) {
60     this.appContext = context;
61     this.executorService = executorService;
62   }
63 
64   @Override
run(Arguments args)65   public ListenableFuture<String> run(Arguments args) throws IllegalCommandLineArgumentException {
66     if (args.getPositionals().isEmpty()) {
67       return Futures.immediateFuture(getUsage());
68     }
69 
70     String command = args.getPositionals().get(0);
71 
72     if ("block".equals(command)) {
73       String number = args.getPositionals().get(1);
74       return Futures.transform(
75           Blocking.block(appContext, ImmutableList.of(number), null),
76           (unused) -> "blocked " + number,
77           MoreExecutors.directExecutor());
78     }
79 
80     if ("unblock".equals(command)) {
81       String number = args.getPositionals().get(1);
82       return Futures.transform(
83           Blocking.unblock(appContext, ImmutableList.of(number), null),
84           (unused) -> "unblocked " + number,
85           MoreExecutors.directExecutor());
86     }
87 
88     if ("isblocked".equals(command)) {
89       String number = args.getPositionals().get(1);
90       ListenableFuture<DialerPhoneNumber> dialerPhoneNumberFuture =
91           executorService.submit(() -> new DialerPhoneNumberUtil().parse(number, null));
92 
93       ListenableFuture<PhoneLookupInfo> lookupFuture =
94           Futures.transformAsync(
95               dialerPhoneNumberFuture,
96               (dialerPhoneNumber) ->
97                   PhoneLookupComponent.get(appContext)
98                       .compositePhoneLookup()
99                       .lookup(dialerPhoneNumber),
100               executorService);
101 
102       return Futures.transform(
103           lookupFuture,
104           (info) -> new PhoneLookupInfoConsolidator(info).isBlocked() ? "true" : "false",
105           MoreExecutors.directExecutor());
106     }
107 
108     return Futures.immediateFuture(getUsage());
109   }
110 }
111