Update MediaSession callback

Callback and TransportControlsCallback were merged.

Change-Id: I5b918316a192af70d7ff661f447ee24857be2eca
diff --git a/src/com/android/telecomm/HeadsetMediaButton.java b/src/com/android/telecomm/HeadsetMediaButton.java
index cca89b9..9fd26fc 100644
--- a/src/com/android/telecomm/HeadsetMediaButton.java
+++ b/src/com/android/telecomm/HeadsetMediaButton.java
@@ -41,14 +41,16 @@
 
     private final MediaSession.Callback mSessionCallback = new MediaSession.Callback() {
         @Override
-        public void onMediaButtonEvent(Intent intent) {
+        public boolean onMediaButtonEvent(Intent intent) {
             KeyEvent event = (KeyEvent) intent.getParcelableExtra(Intent.EXTRA_KEY_EVENT);
             Log.v(this, "SessionCallback.onMediaButton()...  event = %s.", event);
             if ((event != null) && (event.getKeyCode() == KeyEvent.KEYCODE_HEADSETHOOK)) {
                 Log.v(this, "SessionCallback: HEADSETHOOK");
                 boolean consumed = handleHeadsetHook(event);
                 Log.v(this, "==> handleHeadsetHook(): consumed = %b.", consumed);
+                return consumed;
             }
+            return true;
         }
     };
 
@@ -62,7 +64,7 @@
         // Create a MediaSession but don't enable it yet. This is a
         // replacement for MediaButtonReceiver
         mSession = new MediaSession(context, HeadsetMediaButton.class.getSimpleName());
-        mSession.addCallback(mSessionCallback);
+        mSession.setCallback(mSessionCallback);
         mSession.setFlags(MediaSession.FLAG_EXCLUSIVE_GLOBAL_PRIORITY
                 | MediaSession.FLAG_HANDLES_MEDIA_BUTTONS);
         mSession.setPlaybackToLocal(AUDIO_ATTRIBUTES);