1 files changed, 111 insertions, 5 deletions

diff --git a/src/com/android/tv/dvr/ui/DvrGuidedStepFragment.java b/src/com/android/tv/dvr/ui/DvrGuidedStepFragment.java
index d26e6836..ab852e10 100644
--- a/src/com/android/tv/dvr/ui/DvrGuidedStepFragment.java
+++ b/src/com/android/tv/dvr/ui/DvrGuidedStepFragment.java
@@ -16,24 +16,43 @@
 
 package com.android.tv.dvr.ui;
 
+import android.app.Activity;
 import android.app.DialogFragment;
 import android.content.Context;
 import android.os.Bundle;
-import android.support.v17.leanback.app.GuidedStepFragment;
+import android.support.v17.leanback.widget.GuidanceStylist;
 import android.support.v17.leanback.widget.GuidedAction;
 import android.support.v17.leanback.widget.VerticalGridView;
 import android.view.LayoutInflater;
 import android.view.View;
 import android.view.ViewGroup;
 
+import com.android.tv.ApplicationSingletons;
 import com.android.tv.MainActivity;
 import com.android.tv.R;
 import com.android.tv.TvApplication;
+import com.android.tv.dialog.HalfSizedDialogFragment.OnActionClickListener;
 import com.android.tv.dialog.SafeDismissDialogFragment;
 import com.android.tv.dvr.DvrManager;
-import com.android.tv.dvr.ui.HalfSizedDialogFragment.OnActionClickListener;
+import com.android.tv.dvr.DvrStorageStatusManager;
+
+import java.util.List;
+
+public abstract class DvrGuidedStepFragment extends TrackedGuidedStepFragment {
+    /**
+     * Action ID for "recording/scheduling the program anyway".
+     */
+    public static final int ACTION_RECORD_ANYWAY = 1;
+    /**
+     * Action ID for "deleting existed recordings".
+     */
+    public static final int ACTION_DELETE_RECORDINGS = 2;
+    /**
+     * Action ID for "cancelling current recording request".
+     */
+    public static final int ACTION_CANCEL_RECORDING = 3;
+    public static final String UNKNOWN_DVR_ACTION = "Unknown DVR Action";
 
-public class DvrGuidedStepFragment extends GuidedStepFragment {
     private DvrManager mDvrManager;
     private OnActionClickListener mOnActionClickListener;
 
@@ -44,7 +63,8 @@ public class DvrGuidedStepFragment extends GuidedStepFragment {
     @Override
     public void onAttach(Context context) {
         super.onAttach(context);
-        mDvrManager = TvApplication.getSingletons(context).getDvrManager();
+        ApplicationSingletons singletons = TvApplication.getSingletons(context);
+        mDvrManager = singletons.getDvrManager();
     }
 
     @Override
@@ -64,13 +84,27 @@ public class DvrGuidedStepFragment extends GuidedStepFragment {
     }
 
     @Override
-    public void onGuidedActionClicked(GuidedAction action) {
+    public void onTrackedGuidedActionClicked(GuidedAction action) {
         if (mOnActionClickListener != null) {
             mOnActionClickListener.onActionClick(action.getId());
         }
         dismissDialog();
     }
 
+    @Override
+    public String getTrackerLabelForGuidedAction(GuidedAction action) {
+        long actionId = action.getId();
+        if (actionId == ACTION_RECORD_ANYWAY) {
+            return "record-anyway";
+        } else if (actionId == ACTION_DELETE_RECORDINGS) {
+            return "delete-recordings";
+        } else if (actionId == ACTION_CANCEL_RECORDING) {
+            return "cancel-recording";
+        } else {
+            return super.getTrackerLabelForGuidedAction(action);
+        }
+    }
+
     protected void dismissDialog() {
         if (getActivity() instanceof MainActivity) {
             SafeDismissDialogFragment currentDialog =
@@ -86,4 +120,76 @@ public class DvrGuidedStepFragment extends GuidedStepFragment {
     protected void setOnActionClickListener(OnActionClickListener listener) {
         mOnActionClickListener = listener;
     }
+
+    /**
+     * The inner guided step fragment for
+     * {@link com.android.tv.dvr.ui.DvrHalfSizedDialogFragment
+     * .DvrNoFreeSpaceErrorDialogFragment}.
+     */
+    public static class DvrNoFreeSpaceErrorFragment extends DvrGuidedStepFragment {
+        @Override
+        public GuidanceStylist.Guidance onCreateGuidance(Bundle savedInstanceState) {
+            return new GuidanceStylist.Guidance(getString(R.string.dvr_error_no_free_space_title),
+                    getString(R.string.dvr_error_no_free_space_description), null, null);
+        }
+
+        @Override
+        public void onCreateActions(List<GuidedAction> actions, Bundle savedInstanceState) {
+            Activity activity = getActivity();
+            actions.add(new GuidedAction.Builder(activity)
+                    .id(ACTION_RECORD_ANYWAY)
+                    .title(R.string.dvr_action_record_anyway)
+                    .build());
+            actions.add(new GuidedAction.Builder(activity)
+                    .id(ACTION_DELETE_RECORDINGS)
+                    .title(R.string.dvr_action_delete_recordings)
+                    .build());
+            actions.add(new GuidedAction.Builder(activity)
+                    .id(ACTION_CANCEL_RECORDING)
+                    .title(R.string.dvr_action_record_cancel)
+                    .build());
+        }
+
+        @Override
+        public String getTrackerPrefix() {
+            return "DvrNoFreeSpaceErrorFragment";
+        }
+    }
+
+    /**
+     * The inner guided step fragment for
+     * {@link com.android.tv.dvr.ui.DvrHalfSizedDialogFragment
+     * .DvrSmallSizedStorageErrorDialogFragment}.
+     */
+    public static class DvrSmallSizedStorageErrorFragment extends DvrGuidedStepFragment {
+        @Override
+        public GuidanceStylist.Guidance onCreateGuidance(Bundle savedInstanceState) {
+            String title = getResources().getString(
+                    R.string.dvr_error_small_sized_storage_title);
+            String description = getResources().getString(
+                    R.string.dvr_error_small_sized_storage_description,
+                    DvrStorageStatusManager.MIN_STORAGE_SIZE_FOR_DVR_IN_BYTES / 1024
+                            / 1024 / 1024);
+            return new GuidanceStylist.Guidance(title, description, null, null);
+        }
+
+        @Override
+        public void onCreateActions(List<GuidedAction> actions, Bundle savedInstanceState) {
+            Activity activity = getActivity();
+            actions.add(new GuidedAction.Builder(activity)
+                    .id(GuidedAction.ACTION_ID_OK)
+                    .title(android.R.string.ok)
+                    .build());
+        }
+
+        @Override
+        public void onTrackedGuidedActionClicked(GuidedAction action) {
+            dismissDialog();
+        }
+
+        @Override
+        public String getTrackerPrefix() {
+            return "DvrSmallSizedStorageErrorFragment";
+        }
+    }
 }
\ No newline at end of file