Photo Picker dialog: Make toolbar behave more like mocks.

chrome/android/java/src/org/chromium/chrome/browser/photo_picker/PickerCategoryView.java

Index: chrome/android/java/src/org/chromium/chrome/browser/photo_picker/PickerCategoryView.java
diff --git a/chrome/android/java/src/org/chromium/chrome/browser/photo_picker/PickerCategoryView.java b/chrome/android/java/src/org/chromium/chrome/browser/photo_picker/PickerCategoryView.java
index 62e4d5828d73b824792fe94a146dde5cef4f1036..0583814a71d9b0d1ae54e61e3bf355fcc497c6a7 100644
--- a/chrome/android/java/src/org/chromium/chrome/browser/photo_picker/PickerCategoryView.java
+++ b/chrome/android/java/src/org/chromium/chrome/browser/photo_picker/PickerCategoryView.java
@@ -9,10 +9,9 @@ import android.content.Context;
 import android.graphics.Rect;
 import android.support.v7.widget.GridLayoutManager;
 import android.support.v7.widget.RecyclerView;
-import android.support.v7.widget.Toolbar.OnMenuItemClickListener;
 import android.view.LayoutInflater;
-import android.view.MenuItem;
 import android.view.View;
+import android.widget.Button;
 import android.widget.RelativeLayout;
 
 import org.chromium.chrome.R;
@@ -29,7 +28,7 @@ import java.util.List;
  */
 public class PickerCategoryView extends RelativeLayout
         implements FileEnumWorkerTask.FilesEnumeratedCallback, RecyclerView.RecyclerListener,
-                   DecoderServiceHost.ServiceReadyCallback, OnMenuItemClickListener {
+                   DecoderServiceHost.ServiceReadyCallback, View.OnClickListener {
     // The dialog that owns us.
     private PhotoPickerDialog mDialog;
 
@@ -105,9 +104,13 @@ public class PickerCategoryView extends RelativeLayout
 
         mPickerAdapter = new PickerAdapter(this);
         mRecyclerView = mSelectableListLayout.initializeRecyclerView(mPickerAdapter);
-        mSelectableListLayout.initializeToolbar(R.layout.photo_picker_toolbar, mSelectionDelegate,
-                R.string.photo_picker_select_images, null, R.id.photo_picker_normal_menu_group,
-                R.id.photo_picker_selection_mode_menu_group, R.color.default_primary_color, this);
+        PhotoPickerToolbar toolbar = (PhotoPickerToolbar) mSelectableListLayout.initializeToolbar(
+                R.layout.photo_picker_toolbar, mSelectionDelegate,
+                R.string.photo_picker_select_images, null, 0, 0, R.color.default_primary_color,
+                null);
+        toolbar.setBackButtonClickListener(this);
+        Button doneButton = (Button) toolbar.findViewById(R.id.done);
+        doneButton.setOnClickListener(this);
 
         Rect appRect = new Rect();
         ((Activity) context).getWindow().getDecorView().getWindowVisibleDisplayFrame(appRect);
@@ -178,20 +181,17 @@ public class PickerCategoryView extends RelativeLayout
         }
     }
 
-    // OnMenuItemClickListener:
+    // OnClickListener:
 
     @Override
-    public boolean onMenuItemClick(MenuItem item) {
-        if (item.getItemId() == R.id.close_menu_id) {
-            mListener.onPickerUserAction(PhotoPickerListener.Action.CANCEL, null);
-            mDialog.dismiss();
-            return true;
-        } else if (item.getItemId() == R.id.selection_mode_done_menu_id) {
+    public void onClick(View view) {
+        if (view.getId() == R.id.done) {
             notifyPhotosSelected();
-            mDialog.dismiss();
-            return true;
+        } else {
+            mListener.onPickerUserAction(PhotoPickerListener.Action.CANCEL, null);
         }
-        return false;
+
+        mDialog.dismiss();
     }
 
     /**