Sync to ub-tv-killing at 6f6e46557accb62c9548e4177d6005aa944dbf33

Change-Id: I873644d6d9d0110c981ef6075cb4019c16bbb94b

1 files changed, 104 insertions, 0 deletions

diff --git a/src/com/android/tv/dvr/ui/DvrStopSeriesRecordingFragment.java b/src/com/android/tv/dvr/ui/DvrStopSeriesRecordingFragment.java
new file mode 100644
index 00000000..feaa2357
--- /dev/null
+++ b/src/com/android/tv/dvr/ui/DvrStopSeriesRecordingFragment.java
@@ -0,0 +1,104 @@
+/*
+ * 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.tv.dvr.ui;
+
+import android.app.Activity;
+import android.graphics.drawable.Drawable;
+import android.os.Bundle;
+import android.support.annotation.NonNull;
+import android.support.v17.leanback.widget.GuidanceStylist;
+import android.support.v17.leanback.widget.GuidedAction;
+import android.view.LayoutInflater;
+import android.view.View;
+import android.view.ViewGroup;
+
+import com.android.tv.ApplicationSingletons;
+import com.android.tv.R;
+import com.android.tv.TvApplication;
+import com.android.tv.dvr.DvrDataManager;
+import com.android.tv.dvr.DvrManager;
+import com.android.tv.dvr.ScheduledRecording;
+import com.android.tv.dvr.SeriesRecording;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * A fragment which asks the user to stop series recording.
+ */
+public class DvrStopSeriesRecordingFragment extends DvrGuidedStepFragment {
+    /**
+     * Key for the series recording to be stopped.
+     */
+    public static final String KEY_SERIES_RECORDING = "key_series_recoridng";
+
+    private static final int ACTION_STOP_SERIES_RECORDING = 1;
+
+    private SeriesRecording mSeriesRecording;
+
+    @Override
+    public View onCreateView(LayoutInflater inflater, ViewGroup container, Bundle savedInstanceState) {
+        mSeriesRecording = getArguments().getParcelable(KEY_SERIES_RECORDING);
+        return super.onCreateView(inflater, container, savedInstanceState);
+    }
+
+    @NonNull
+    @Override
+    public GuidanceStylist.Guidance onCreateGuidance(Bundle savedInstanceState) {
+        String title = getString(R.string.dvr_series_schedules_stop_dialog_title);
+        String description = getString(R.string.dvr_series_schedules_stop_dialog_description);
+        Drawable icon = getContext().getDrawable(R.drawable.ic_dvr_delete);
+        return new GuidanceStylist.Guidance(title, description, null, icon);
+    }
+
+    @Override
+    public void onCreateActions(@NonNull List<GuidedAction> actions, Bundle savedInstanceState) {
+        Activity activity = getActivity();
+        actions.add(new GuidedAction.Builder(activity)
+                .id(ACTION_STOP_SERIES_RECORDING)
+                .title(R.string.dvr_series_schedules_stop_dialog_action_stop)
+                .build());
+        actions.add(new GuidedAction.Builder(activity)
+                .clickAction(GuidedAction.ACTION_ID_CANCEL)
+                .build());
+    }
+
+    @Override
+    public void onGuidedActionClicked(GuidedAction action) {
+        if (action.getId() == ACTION_STOP_SERIES_RECORDING) {
+            ApplicationSingletons singletons = TvApplication.getSingletons(getContext());
+            DvrManager dvrManager = singletons.getDvrManager();
+            DvrDataManager dataManager = singletons.getDvrDataManager();
+            List<ScheduledRecording> toDelete = new ArrayList<>();
+            for (ScheduledRecording r : dataManager.getAvailableScheduledRecordings()) {
+                if (r.getSeriesRecordingId() == mSeriesRecording.getId()) {
+                    if (r.getState() == ScheduledRecording.STATE_RECORDING_NOT_STARTED) {
+                        toDelete.add(r);
+                    } else {
+                        dvrManager.stopRecording(r);
+                    }
+                }
+            }
+            if (!toDelete.isEmpty()) {
+                dvrManager.forceRemoveScheduledRecording(ScheduledRecording.toArray(toDelete));
+            }
+            dvrManager.updateSeriesRecording(SeriesRecording.buildFrom(mSeriesRecording)
+                    .setState(SeriesRecording.STATE_SERIES_STOPPED).build());
+        }
+        dismissDialog();
+    }
+}