From 218a2627701296b3e4b092410a9e7746350d4ea1 Mon Sep 17 00:00:00 2001
From: Aayush Gupta <aayushgupta219@gmail.com>
Date: Tue, 26 Sep 2023 20:33:35 +0530
Subject: [PATCH] AllInOneActivity: Switch to Material date picker

Signed-off-by: Aayush Gupta <aayushgupta219@gmail.com>
---
 .../android/calendar/AllInOneActivity.java    | 40 ++++++++++++-------
 1 file changed, 26 insertions(+), 14 deletions(-)

diff --git a/app/src/main/java/com/android/calendar/AllInOneActivity.java b/app/src/main/java/com/android/calendar/AllInOneActivity.java
index 8a3abbc03..224633338 100644
--- a/app/src/main/java/com/android/calendar/AllInOneActivity.java
+++ b/app/src/main/java/com/android/calendar/AllInOneActivity.java
@@ -26,7 +26,6 @@
 import android.animation.Animator;
 import android.animation.Animator.AnimatorListener;
 import android.animation.ObjectAnimator;
-import android.app.DatePickerDialog;
 import android.content.AsyncQueryHandler;
 import android.content.BroadcastReceiver;
 import android.content.ContentResolver;
@@ -58,7 +57,6 @@
 import android.view.MenuItem;
 import android.view.View;
 import android.view.accessibility.AccessibilityEvent;
-import android.widget.DatePicker;
 import android.widget.LinearLayout;
 import android.widget.RelativeLayout;
 import android.widget.RelativeLayout.LayoutParams;
@@ -89,6 +87,9 @@
 import com.android.calendar.settings.SettingsActivityKt;
 import com.android.calendar.settings.ViewDetailsPreferences;
 import com.android.calendarcommon2.Time;
+import com.google.android.material.datepicker.CalendarConstraints;
+import com.google.android.material.datepicker.MaterialDatePicker;
+import com.google.android.material.datepicker.MaterialPickerOnPositiveButtonClickListener;
 import com.google.android.material.floatingactionbutton.FloatingActionButton;
 import com.google.android.material.navigation.NavigationView;
 
@@ -96,6 +97,7 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Calendar;
+import java.util.Date;
 import java.util.List;
 import java.util.Locale;
 import java.util.TimeZone;
@@ -916,18 +918,19 @@ public boolean onOptionsItemSelected(MenuItem item) {
                 t = todayTime;
             }
 
-            DatePickerDialog.OnDateSetListener datePickerListener = new DatePickerDialog.OnDateSetListener() {
+            MaterialPickerOnPositiveButtonClickListener<Long> materialPickerOnPositiveButtonClickListener = new MaterialPickerOnPositiveButtonClickListener<>() {
+                @Override
+                public void onPositiveButtonClick(Long selection) {
+                    Calendar calendar = Calendar.getInstance();
+                    calendar.setTime(new Date(selection));
 
-                public void onDateSet(DatePicker view, int year, int monthOfYear, int dayOfMonth) {
                     Time selectedTime = new Time(mTimeZone);
                     selectedTime.set(System.currentTimeMillis());  // Needed for recalc function in DayView(time + gmtoff)
-                    selectedTime.setYear(year);
-                    selectedTime.setMonth(monthOfYear);
-                    selectedTime.setDay(dayOfMonth);
+                    selectedTime.setYear(calendar.get(Calendar.YEAR));
+                    selectedTime.setMonth(calendar.get(Calendar.MONTH));
+                    selectedTime.setDay(calendar.get(Calendar.DAY_OF_MONTH));
 
-                    Calendar c = Calendar.getInstance();
-                    c.set(year, monthOfYear, dayOfMonth);
-                    int weekday = c.get(Calendar.DAY_OF_WEEK);
+                    int weekday = calendar.get(Calendar.DAY_OF_WEEK);
                     if (weekday == 1) {
                         selectedTime.setWeekDay(7);
                     } else {
@@ -938,10 +941,19 @@ public void onDateSet(DatePicker view, int year, int monthOfYear, int dayOfMonth
                     mController.sendEvent(this, EventType.GO_TO, selectedTime, null, selectedTime, -1, ViewType.CURRENT, extras, null, null);
                 }
             };
-            DatePickerDialog datePickerDialog = new DatePickerDialog(this, datePickerListener,
-                    t.getYear(), t.getMonth(), t.getDay());
-                    datePickerDialog.getDatePicker().setFirstDayOfWeek(Utils.getFirstDayOfWeekAsCalendar(this));
-                    datePickerDialog.show();
+
+            CalendarConstraints calendarConstraints = new CalendarConstraints.Builder()
+                    .setFirstDayOfWeek(Utils.getFirstDayOfWeekAsCalendar(this))
+                    .build();
+
+            MaterialDatePicker<Long> datePickerDialog = MaterialDatePicker.Builder.datePicker()
+                    .setSelection(Calendar.getInstance().getTimeInMillis())
+                    .setCalendarConstraints(calendarConstraints)
+                    .setTitleText(R.string.goto_date)
+                    .build();
+
+            datePickerDialog.addOnPositiveButtonClickListener(materialPickerOnPositiveButtonClickListener);
+            datePickerDialog.show(getSupportFragmentManager(), "GoTo");
 
         } else if (itemId == R.id.action_hide_controls) {
             mHideControls = !mHideControls;