/* * Twidere - Twitter client for Android * * Copyright (C) 2012-2017 Mariotaku Lee * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ package org.mariotaku.twidere.adapter import android.content.Context import android.support.v4.util.ArrayMap import android.support.v7.widget.RecyclerView import android.view.LayoutInflater import android.view.ViewGroup import com.bumptech.glide.RequestManager import org.mariotaku.twidere.R import org.mariotaku.twidere.adapter.iface.IItemCountsAdapter import org.mariotaku.twidere.adapter.iface.ILoadMoreSupportAdapter import org.mariotaku.twidere.adapter.iface.ILoadMoreSupportAdapter.Companion.ITEM_VIEW_TYPE_LOAD_INDICATOR import org.mariotaku.twidere.exception.UnsupportedCountIndexException import org.mariotaku.twidere.model.ItemCounts import org.mariotaku.twidere.model.ParcelableUser import org.mariotaku.twidere.model.UserKey import org.mariotaku.twidere.view.holder.LoadIndicatorViewHolder import org.mariotaku.twidere.view.holder.SelectableUserViewHolder class SelectableUsersAdapter( context: Context, requestManager: RequestManager ) : LoadMoreSupportAdapter(context, requestManager), IItemCountsAdapter { val ITEM_VIEW_TYPE_USER = 2 override val itemCounts: ItemCounts = ItemCounts(3) private val inflater: LayoutInflater = LayoutInflater.from(context) private val checkedState: MutableMap = ArrayMap() private val lockedState: MutableMap = ArrayMap() var itemCheckedListener: ((Int, Boolean) -> Boolean)? = null var data: List? = null set(value) { field = value updateItemCounts() notifyDataSetChanged() } init { setHasStableIds(true) } override fun getItemCount(): Int { return itemCounts.itemCount } override fun onCreateViewHolder(parent: ViewGroup, viewType: Int): RecyclerView.ViewHolder { when (viewType) { ITEM_VIEW_TYPE_USER -> { val view = inflater.inflate(R.layout.list_item_simple_user, parent, false) val holder = SelectableUserViewHolder(view, this) return holder } ITEM_VIEW_TYPE_LOAD_INDICATOR -> { val view = inflater.inflate(R.layout.list_item_load_indicator, parent, false) return LoadIndicatorViewHolder(view) } } throw IllegalStateException("Unknown view type " + viewType) } override fun onBindViewHolder(holder: RecyclerView.ViewHolder, position: Int) { when (holder.itemViewType) { ITEM_VIEW_TYPE_USER -> { bindUser(holder as SelectableUserViewHolder, position) } } } override fun getItemViewType(position: Int): Int { val countIndex = itemCounts.getItemCountIndex(position) when (countIndex) { ITEM_TYPE_START_INDICATOR, ITEM_TYPE_END_INDICATOR -> return ITEM_VIEW_TYPE_LOAD_INDICATOR ITEM_TYPE_USER -> return ITEM_VIEW_TYPE_USER else -> throw UnsupportedCountIndexException(countIndex, position) } } override fun getItemId(position: Int): Long { val countIndex = itemCounts.getItemCountIndex(position) return when (countIndex) { ITEM_TYPE_START_INDICATOR, ITEM_TYPE_END_INDICATOR -> (countIndex.toLong() shl 32) ITEM_TYPE_USER -> (countIndex.toLong() shl 32) or getUser(position).hashCode().toLong() else -> throw UnsupportedCountIndexException(countIndex, position) } } private fun bindUser(holder: SelectableUserViewHolder, position: Int) { holder.displayUser(getUser(position)) } private fun updateItemCounts() { val position = loadMoreIndicatorPosition itemCounts[ITEM_TYPE_START_INDICATOR] = if (position and ILoadMoreSupportAdapter.START != 0L) 1 else 0 itemCounts[ITEM_TYPE_USER] = userCount itemCounts[ITEM_TYPE_END_INDICATOR] = if (position and ILoadMoreSupportAdapter.END != 0L) 1 else 0 } fun getUser(position: Int): ParcelableUser { return data!![position - itemCounts.getItemStartPosition(1)] } val userStartIndex: Int get() { val position = loadMoreIndicatorPosition var start = 0 if (position and ILoadMoreSupportAdapter.START != 0L) { start += 1 } return start } fun getUserKey(position: Int): UserKey { return getUser(position).key } val userCount: Int get() = data?.size ?: 0 fun removeUserAt(position: Int): Boolean { val data = this.data as? MutableList ?: return false val dataPosition = position - userStartIndex if (dataPosition < 0 || dataPosition >= userCount) return false data.removeAt(dataPosition) notifyItemRemoved(position) return true } fun setUserAt(position: Int, user: ParcelableUser): Boolean { val data = this.data as? MutableList ?: return false val dataPosition = position - userStartIndex if (dataPosition < 0 || dataPosition >= userCount) return false data[dataPosition] = user notifyItemChanged(position) return true } fun findPosition(accountKey: UserKey, userKey: UserKey): Int { if (data == null) return RecyclerView.NO_POSITION for (i in userStartIndex until userStartIndex + userCount) { val user = data!![i] if (accountKey == user.account_key && userKey == user.key) { return i } } return RecyclerView.NO_POSITION } val checkedCount: Int get() { return data?.count { it.key !in lockedState && checkedState[it.key] ?: false } ?: 0 } fun setItemChecked(position: Int, value: Boolean) { val userKey = getUserKey(position) setCheckState(userKey, value) if (itemCheckedListener?.invoke(position, value) == false) { setCheckState(userKey, !value) notifyItemChanged(position) } } fun clearCheckState() { checkedState.clear() } fun setCheckState(userKey: UserKey, value: Boolean) { checkedState[userKey] = value } fun clearLockedState() { lockedState.clear() } fun setLockedState(userKey: UserKey, locked: Boolean) { lockedState[userKey] = locked } fun removeLockedState(userKey: UserKey) { lockedState.remove(userKey) } fun isItemChecked(position: Int): Boolean { return checkedState[getUserKey(position)] ?: false } fun isItemChecked(userKey: UserKey): Boolean { return checkedState[userKey] ?: false } fun getLockedState(position: Int): Boolean { return lockedState[getUserKey(position)] ?: false } fun getLockedState(userKey: UserKey): Boolean { return lockedState[userKey] ?: false } fun isItemLocked(position: Int): Boolean { return getUserKey(position) in lockedState } fun isItemLocked(userKey: UserKey): Boolean { return userKey in lockedState } fun clearSelection() { checkedState.clear() } companion object { const val ITEM_TYPE_START_INDICATOR = 0 const val ITEM_TYPE_USER = 1 const val ITEM_TYPE_END_INDICATOR = 2 } }