src/com/android/tv/menu/ChannelsRowAdapter.java

/*
 * Copyright (C) 2015 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.menu;

import android.content.Context;
import android.content.Intent;
import android.media.tv.TvInputInfo;
import android.view.View;

import com.android.tv.ApplicationSingletons;
import com.android.tv.R;
import com.android.tv.TvApplication;
import com.android.tv.analytics.Tracker;
import com.android.tv.common.feature.CommonFeatures;
import com.android.tv.data.Channel;
import com.android.tv.dvr.DvrDataManager;
import com.android.tv.recommendation.Recommender;
import com.android.tv.util.SetupUtils;
import com.android.tv.util.TvInputManagerHelper;

import java.util.ArrayList;
import java.util.List;

/**
 * An adapter of the Channels row.
 */
public class ChannelsRowAdapter extends ItemListRowView.ItemListAdapter<Channel> {
    private static final String TAG = "ChannelsRowAdapter";

    // There are four special cards: guide, setup, dvr, applink.
    private static final int SIZE_OF_VIEW_TYPE = 5;

    private final Context mContext;
    private final Tracker mTracker;
    private final Recommender mRecommender;
    private final DvrDataManager mDvrDataManager;
    private final int mMaxCount;
    private final int mMinCount;
    private final int[] mViewType = new int[SIZE_OF_VIEW_TYPE];

    private final View.OnClickListener mGuideOnClickListener = new View.OnClickListener() {
        @Override
        public void onClick(View view) {
            mTracker.sendMenuClicked(R.string.channels_item_program_guide);
            getMainActivity().getOverlayManager().showProgramGuide();
        }
    };

    private final View.OnClickListener mSetupOnClickListener = new View.OnClickListener() {
        @Override
        public void onClick(View view) {
            mTracker.sendMenuClicked(R.string.channels_item_setup);
            getMainActivity().getOverlayManager().showSetupFragment();
        }
    };

    private final View.OnClickListener mDvrOnClickListener = new View.OnClickListener() {
        @Override
        public void onClick(View view) {
            mTracker.sendMenuClicked(R.string.channels_item_dvr);
            getMainActivity().getOverlayManager().showDvrManager();
        }
    };

    private final View.OnClickListener mAppLinkOnClickListener = new View.OnClickListener() {
        @Override
        public void onClick(View view) {
            mTracker.sendMenuClicked(R.string.channels_item_app_link);
            Intent intent = ((AppLinkCardView) view).getIntent();
            if (intent != null) {
                getMainActivity().startActivitySafe(intent);
            }
        }
    };

    private final View.OnClickListener mChannelOnClickListener = new View.OnClickListener() {
        @Override
        public void onClick(View view) {
            // Always send the label "Channels" because the channel ID or name or number might be
            // sensitive.
            mTracker.sendMenuClicked(R.string.menu_title_channels);
            getMainActivity().tuneToChannel((Channel) view.getTag());
            getMainActivity().hideOverlaysForTune();
        }
    };

    public ChannelsRowAdapter(Context context, Recommender recommender,
            int minCount, int maxCount) {
        super(context);
        mContext = context;
        ApplicationSingletons appSingletons = TvApplication.getSingletons(context);
        mTracker = appSingletons.getTracker();
        if (CommonFeatures.DVR.isEnabled(context)) {
            mDvrDataManager = appSingletons.getDvrDataManager();
        } else {
            mDvrDataManager = null;
        }
        mRecommender = recommender;
        mMinCount = minCount;
        mMaxCount = maxCount;
    }

    @Override
    public int getItemViewType(int position) {
        if (position >= SIZE_OF_VIEW_TYPE) {
            return R.layout.menu_card_channel;
        }
        return mViewType[position];
    }

    @Override
    protected int getLayoutResId(int viewType) {
        return viewType;
    }

    @Override
    public void onBindViewHolder(MyViewHolder viewHolder, int position) {
        int viewType = getItemViewType(position);
        if (viewType == R.layout.menu_card_guide) {
            viewHolder.itemView.setOnClickListener(mGuideOnClickListener);
        } else if (viewType == R.layout.menu_card_setup) {
            viewHolder.itemView.setOnClickListener(mSetupOnClickListener);
        } else if (viewType == R.layout.menu_card_app_link) {
            viewHolder.itemView.setOnClickListener(mAppLinkOnClickListener);
        } else if (viewType == R.layout.menu_card_dvr) {
            viewHolder.itemView.setOnClickListener(mDvrOnClickListener);
            SimpleCardView view = (SimpleCardView) viewHolder.itemView;
            view.setText(R.string.channels_item_dvr);
        } else {
            viewHolder.itemView.setTag(getItemList().get(position));
            viewHolder.itemView.setOnClickListener(mChannelOnClickListener);
        }
        super.onBindViewHolder(viewHolder, position);
    }

    @Override
    public void update() {
        List<Channel> channelList = new ArrayList<>();
        Channel dummyChannel = new Channel.Builder().build();
        // For guide item
        channelList.add(dummyChannel);
        // For setup item
        TvInputManagerHelper inputManager = TvApplication.getSingletons(mContext)
                .getTvInputManagerHelper();
        boolean showSetupCard = SetupUtils.getInstance(mContext).hasNewInput(inputManager);
        Channel currentChannel = getMainActivity().getCurrentChannel();
        boolean showAppLinkCard = currentChannel != null
                && currentChannel.getAppLinkType(mContext) != Channel.APP_LINK_TYPE_NONE
                // Sometimes applicationInfo can be null. b/28932537
                && inputManager.getTvInputAppInfo(currentChannel.getInputId()) != null;
        boolean showDvrCard = false;
        if (mDvrDataManager != null) {
            for (TvInputInfo info : inputManager.getTvInputInfos(true, true)) {
                if (info.canRecord()) {
                    showDvrCard = true;
                    break;
                }
            }
        }

        mViewType[0] = R.layout.menu_card_guide;
        int index = 1;
        if (showSetupCard) {
            channelList.add(dummyChannel);
            mViewType[index++] = R.layout.menu_card_setup;
        }
        if (showDvrCard) {
            channelList.add(dummyChannel);
            mViewType[index++] = R.layout.menu_card_dvr;
        }
        if (showAppLinkCard) {
            channelList.add(currentChannel);
            mViewType[index++] = R.layout.menu_card_app_link;
        }
        for ( ; index < mViewType.length; ++index) {
            mViewType[index] = R.layout.menu_card_channel;
        }
        channelList.addAll(getRecentChannels());
        setItemList(channelList);
    }

    private List<Channel> getRecentChannels() {
        List<Channel> channelList = new ArrayList<>();
        for (Channel channel : mRecommender.recommendChannels(mMaxCount)) {
            if (channel.isBrowsable()) {
                channelList.add(channel);
            }
        }
        int count = channelList.size();
        // If the number of recommended channels is not enough, add more from the recent channel
        // list.
        if (count < mMinCount) {
            for (long channelId : getMainActivity().getRecentChannels()) {
                Channel channel = mRecommender.getChannel(channelId);
                if (channel == null || channelList.contains(channel)
                        || !channel.isBrowsable()) {
                   continue;
                }
                channelList.add(channel);
                if (++count >= mMinCount) {
                    break;
                }
            }
        }
        return channelList;
    }
}