Move CallerInfo lookup into a utility class

In preparation for parallelizing the three call-blocking lookup
operations, move the caller info/contact photo lookup out of the call
class.

Bug: 26883888

Change-Id: I9e00b21a3c5ce5c725fa91ec229f104e7c65c81e
diff --git a/src/com/android/server/telecom/CallerInfoLookupHelper.java b/src/com/android/server/telecom/CallerInfoLookupHelper.java
new file mode 100644
index 0000000..263f859
--- /dev/null
+++ b/src/com/android/server/telecom/CallerInfoLookupHelper.java
@@ -0,0 +1,216 @@
+/*
+ * Copyright (C) 2016 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.server.telecom;
+
+import android.content.Context;
+import android.graphics.Bitmap;
+import android.graphics.drawable.Drawable;
+import android.net.Uri;
+import android.os.Handler;
+import android.os.Looper;
+import android.text.TextUtils;
+
+import com.android.internal.annotations.VisibleForTesting;
+import com.android.internal.telephony.CallerInfo;
+import com.android.internal.telephony.CallerInfoAsyncQuery;
+
+import java.io.InputStream;
+import java.util.HashMap;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.Map;
+
+public class CallerInfoLookupHelper {
+    public interface OnQueryCompleteListener {
+        /**
+         * Called when the query returns with the caller info
+         * @param info
+         * @return true if the value should be cached, false otherwise.
+         */
+        void onCallerInfoQueryComplete(Uri handle, CallerInfo info);
+        void onContactPhotoQueryComplete(Uri handle, CallerInfo info);
+    }
+
+    private static class CallerInfoQueryInfo {
+        public CallerInfo callerInfo;
+        public List<OnQueryCompleteListener> listeners;
+        public boolean imageQueryPending = false;
+
+        public CallerInfoQueryInfo() {
+            listeners = new LinkedList<>();
+        }
+    }
+    private final Map<Uri, CallerInfoQueryInfo> mQueryEntries = new HashMap<>();
+
+    private final CallerInfoAsyncQueryFactory mCallerInfoAsyncQueryFactory;
+    private final ContactsAsyncHelper mContactsAsyncHelper;
+    private final Context mContext;
+    private final Object mLock = new Object();
+    private final Handler mHandler = new Handler(Looper.getMainLooper());
+
+    public CallerInfoLookupHelper(Context context,
+            CallerInfoAsyncQueryFactory callerInfoAsyncQueryFactory,
+            ContactsAsyncHelper contactsAsyncHelper) {
+        mCallerInfoAsyncQueryFactory = callerInfoAsyncQueryFactory;
+        mContactsAsyncHelper = contactsAsyncHelper;
+        mContext = context;
+    }
+
+    public void startLookup(final Uri handle, OnQueryCompleteListener listener) {
+        if (handle == null) {
+            return;
+        }
+
+        final String number = handle.getSchemeSpecificPart();
+        if (TextUtils.isEmpty(number)) {
+            return;
+        }
+
+        synchronized (mLock) {
+            if (mQueryEntries.containsKey(handle)) {
+                CallerInfoQueryInfo info = mQueryEntries.get(handle);
+                if (info.callerInfo != null) {
+                    Log.i(this, "Caller info already exists for handle %s; using cached value",
+                            Log.piiHandle(handle));
+                    listener.onCallerInfoQueryComplete(handle, info.callerInfo);
+                    if (!info.imageQueryPending && (info.callerInfo.cachedPhoto != null ||
+                            info.callerInfo.cachedPhotoIcon != null)) {
+                        listener.onContactPhotoQueryComplete(handle, info.callerInfo);
+                    } else if (info.imageQueryPending) {
+                        Log.i(this, "There is a previously incomplete query for handle %s. " +
+                                "Adding to listeners for this query.", Log.piiHandle(handle));
+                        info.listeners.add(listener);
+                    }
+                } else {
+                    Log.i(this, "There is a previously incomplete query for handle %s. Adding to " +
+                            "listeners for this query.", Log.piiHandle(handle));
+                    info.listeners.add(listener);
+                    return;
+                }
+            } else {
+                CallerInfoQueryInfo info = new CallerInfoQueryInfo();
+                info.listeners.add(listener);
+                mQueryEntries.put(handle, info);
+            }
+        }
+
+        mHandler.post(new Runnable("CILH.sL") {
+            @Override
+            public void loggedRun() {
+                Session continuedSession = Log.createSubsession();
+                try {
+                    CallerInfoAsyncQuery query = mCallerInfoAsyncQueryFactory.startQuery(
+                            0, mContext, number,
+                            makeCallerInfoQueryListener(handle), continuedSession);
+                    if (query == null) {
+                        Log.w(this, "Lookup failed for %s.", Log.piiHandle(handle));
+                        Log.cancelSubsession(continuedSession);
+                    }
+                } catch (Throwable t) {
+                    Log.cancelSubsession(continuedSession);
+                    throw t;
+                }
+            }
+        }.prepare());
+    }
+
+    private CallerInfoAsyncQuery.OnQueryCompleteListener makeCallerInfoQueryListener(
+            final Uri handle) {
+        return (token, cookie, ci) -> {
+            synchronized (mLock) {
+                Log.continueSession((Session) cookie, "CILH.oQC");
+                try {
+                    if (mQueryEntries.containsKey(handle)) {
+                        CallerInfoQueryInfo info = mQueryEntries.get(handle);
+                        for (OnQueryCompleteListener l : info.listeners) {
+                            l.onCallerInfoQueryComplete(handle, ci);
+                        }
+                        if (ci.contactDisplayPhotoUri == null) {
+                            mQueryEntries.remove(handle);
+                        } else {
+                            info.callerInfo = ci;
+                            info.imageQueryPending = true;
+                            startPhotoLookup(handle, ci.contactDisplayPhotoUri);
+                        }
+                    } else {
+                        Log.i(CallerInfoLookupHelper.this, "CI query for handle %s has completed," +
+                                " but there are no listeners left.", handle);
+                    }
+                } finally {
+                    Log.endSession();
+                }
+            }
+        };
+    }
+
+    private void startPhotoLookup(final Uri handle, final Uri contactPhotoUri) {
+        mHandler.post(new Runnable("CILH.sPL") {
+            @Override
+            public void loggedRun() {
+                Session continuedSession = Log.createSubsession();
+                try {
+                    mContactsAsyncHelper.startObtainPhotoAsync(
+                            0, mContext, contactPhotoUri,
+                            makeContactPhotoListener(handle), continuedSession);
+                } catch (Throwable t) {
+                    Log.cancelSubsession(continuedSession);
+                    throw t;
+                }
+            }
+        }.prepare());
+    }
+
+    private ContactsAsyncHelper.OnImageLoadCompleteListener makeContactPhotoListener(
+            final Uri handle) {
+        return (token, photo, photoIcon, cookie) -> {
+            synchronized (mLock) {
+                Log.continueSession((Session) cookie, "CLIH.oILC");
+                try {
+                    if (mQueryEntries.containsKey(handle)) {
+                        CallerInfoQueryInfo info = mQueryEntries.get(handle);
+                        if (info.callerInfo == null) {
+                            Log.w(CallerInfoLookupHelper.this, "Photo query finished, but the " +
+                                    "CallerInfo object previously looked up was not cached.");
+                            return;
+                        }
+                        info.callerInfo.cachedPhoto = photo;
+                        info.callerInfo.cachedPhotoIcon = photoIcon;
+                        for (OnQueryCompleteListener l : info.listeners) {
+                            l.onContactPhotoQueryComplete(handle, info.callerInfo);
+                        }
+                        mQueryEntries.remove(handle);
+                    } else {
+                        Log.i(CallerInfoLookupHelper.this, "Photo query for handle %s has" +
+                                " completed, but there are no listeners left.", handle);
+                    }
+                } finally {
+                    Log.endSession();
+                }
+            }
+        };
+    }
+
+    @VisibleForTesting
+    public Map<Uri, CallerInfoQueryInfo> getCallerInfoEntries() {
+        return mQueryEntries;
+    }
+
+    @VisibleForTesting
+    public Handler getHandler() {
+        return mHandler;
+    }
+}