src/com/android/tv/dvr/ui/browse/SeriesRecordingPresenter.java

/*
 * 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.browse;

import android.content.Context;
import android.media.tv.TvInputManager;
import android.text.TextUtils;

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.DvrDataManager.RecordedProgramListener;
import com.android.tv.dvr.DvrDataManager.ScheduledRecordingListener;
import com.android.tv.dvr.DvrManager;
import com.android.tv.dvr.DvrWatchedPositionManager;
import com.android.tv.dvr.DvrWatchedPositionManager.WatchedPositionChangedListener;
import com.android.tv.dvr.data.RecordedProgram;
import com.android.tv.dvr.data.ScheduledRecording;
import com.android.tv.dvr.data.SeriesRecording;

import java.util.List;

/**
 * Presents a {@link SeriesRecording} in {@link DvrBrowseFragment}.
 */
class SeriesRecordingPresenter extends DvrItemPresenter<SeriesRecording> {
    private final DvrDataManager mDvrDataManager;
    private final DvrManager mDvrManager;
    private final DvrWatchedPositionManager mWatchedPositionManager;

    private final class SeriesRecordingViewHolder extends DvrItemViewHolder implements
            WatchedPositionChangedListener, ScheduledRecordingListener, RecordedProgramListener {
        private SeriesRecording mSeriesRecording;
        private RecordingCardView mCardView;
        private DvrDataManager mDvrDataManager;
        private DvrManager mDvrManager;
        private DvrWatchedPositionManager mWatchedPositionManager;

        SeriesRecordingViewHolder(RecordingCardView view, DvrDataManager dvrDataManager,
                DvrManager dvrManager, DvrWatchedPositionManager watchedPositionManager) {
            super(view);
            mCardView = view;
            mDvrDataManager = dvrDataManager;
            mDvrManager = dvrManager;
            mWatchedPositionManager = watchedPositionManager;
        }

        @Override
        public void onWatchedPositionChanged(long recordedProgramId, long positionMs) {
            if (positionMs != TvInputManager.TIME_SHIFT_INVALID_TIME) {
                mWatchedPositionManager.removeListener(this, recordedProgramId);
                updateCardViewContent();
            }
        }

        @Override
        public void onScheduledRecordingAdded(ScheduledRecording... scheduledRecordings) {
            for (ScheduledRecording scheduledRecording : scheduledRecordings) {
                if (scheduledRecording.getSeriesRecordingId() == mSeriesRecording.getId()) {
                    updateCardViewContent();
                    return;
                }
            }
        }

        @Override
        public void onScheduledRecordingRemoved(ScheduledRecording... scheduledRecordings) {
            for (ScheduledRecording scheduledRecording : scheduledRecordings) {
                if (scheduledRecording.getSeriesRecordingId() == mSeriesRecording.getId()) {
                    updateCardViewContent();
                    return;
                }
            }
        }

        @Override
        public void onRecordedProgramsAdded(RecordedProgram... recordedPrograms) {
            boolean needToUpdateCardView = false;
            for (RecordedProgram recordedProgram : recordedPrograms) {
                if (TextUtils.equals(recordedProgram.getSeriesId(),
                        mSeriesRecording.getSeriesId())) {
                    mDvrDataManager.removeScheduledRecordingListener(this);
                    mWatchedPositionManager.addListener(this, recordedProgram.getId());
                    needToUpdateCardView = true;
                }
            }
            if (needToUpdateCardView) {
                updateCardViewContent();
            }
        }

        @Override
        public void onRecordedProgramsRemoved(RecordedProgram... recordedPrograms) {
            boolean needToUpdateCardView = false;
            for (RecordedProgram recordedProgram : recordedPrograms) {
                if (TextUtils.equals(recordedProgram.getSeriesId(),
                        mSeriesRecording.getSeriesId())) {
                    if (mWatchedPositionManager.getWatchedPosition(recordedProgram.getId())
                            == TvInputManager.TIME_SHIFT_INVALID_TIME) {
                        mWatchedPositionManager.removeListener(this, recordedProgram.getId());
                    }
                    needToUpdateCardView = true;
                }
            }
            if (needToUpdateCardView) {
                updateCardViewContent();
            }
        }

        @Override
        public void onRecordedProgramsChanged(RecordedProgram... recordedPrograms) {
            // Do nothing
        }

        @Override
        public void onScheduledRecordingStatusChanged(ScheduledRecording... scheduledRecordings) {
            // Do nothing
        }

        @Override
        protected void onBound(SeriesRecording seriesRecording) {
            mSeriesRecording = seriesRecording;
            mDvrDataManager.addScheduledRecordingListener(this);
            mDvrDataManager.addRecordedProgramListener(this);
            for (RecordedProgram recordedProgram :
                    mDvrDataManager.getRecordedPrograms(mSeriesRecording.getId())) {
                if (mWatchedPositionManager.getWatchedPosition(recordedProgram.getId())
                        == TvInputManager.TIME_SHIFT_INVALID_TIME) {
                    mWatchedPositionManager.addListener(this, recordedProgram.getId());
                }
            }
            updateCardViewContent();
        }

        @Override
        protected void onUnbound() {
            mDvrDataManager.removeScheduledRecordingListener(this);
            mDvrDataManager.removeRecordedProgramListener(this);
            mWatchedPositionManager.removeListener(this);
            getView().reset();
        }

        private void updateCardViewContent() {
            int count = 0;
            int quantityStringID;
            List<RecordedProgram> recordedPrograms =
                    mDvrDataManager.getRecordedPrograms(mSeriesRecording.getId());
            if (recordedPrograms.size() == 0) {
                count = mDvrManager.getAvailableScheduledRecording(mSeriesRecording.getId()).size();
                quantityStringID = R.plurals.dvr_count_scheduled_recordings;
            } else {
                for (RecordedProgram recordedProgram : recordedPrograms) {
                    if (mWatchedPositionManager.getWatchedPosition(recordedProgram.getId())
                            == TvInputManager.TIME_SHIFT_INVALID_TIME) {
                        count++;
                    }
                }
                if (count == 0) {
                    count = recordedPrograms.size();
                    quantityStringID = R.plurals.dvr_count_recordings;
                } else {
                    quantityStringID = R.plurals.dvr_count_new_recordings;
                }
            }
            mCardView.setContent(mCardView.getResources()
                    .getQuantityString(quantityStringID, count, count), null);
        }
    }

    public SeriesRecordingPresenter(Context context) {
        super(context);
        ApplicationSingletons singletons = TvApplication.getSingletons(context);
        mDvrDataManager = singletons.getDvrDataManager();
        mDvrManager = singletons.getDvrManager();
        mWatchedPositionManager = singletons.getDvrWatchedPositionManager();
    }

    @Override
    public DvrItemViewHolder onCreateDvrItemViewHolder() {
        return new SeriesRecordingViewHolder(new RecordingCardView(mContext), mDvrDataManager,
                mDvrManager, mWatchedPositionManager);
    }

    @Override
    public void onBindDvrItemViewHolder(DvrItemViewHolder baseHolder, SeriesRecording series) {
        final SeriesRecordingViewHolder viewHolder = (SeriesRecordingViewHolder) baseHolder;
        final RecordingCardView cardView = viewHolder.getView();
        viewHolder.onBound(series);
        DetailsContent details = DetailsContent.createFromSeriesRecording(mContext, series);
        cardView.setTitle(details.getTitle());
        cardView.setImageUri(details.getLogoImageUri(), details.isUsingChannelLogo());
        cardView.setDetailBackgroundImageUri(details.getBackgroundImageUri());
    }

    @Override
    public void onUnbindViewHolder(ViewHolder viewHolder) {
        ((RecordingCardView) viewHolder.view).reset();
        ((SeriesRecordingViewHolder) viewHolder).onUnbound();
        super.onUnbindViewHolder(viewHolder);
    }
}