/*
ChatRoomsAdapter.java
Copyright (C) 2017  Belledonne Communications, Grenoble, France

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 2
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, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

package org.linphone.chat;

import android.content.Context;
import android.graphics.Bitmap;
import android.graphics.BitmapFactory;
import android.graphics.Typeface;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.CheckBox;
import android.widget.ImageView;
import android.widget.TextView;

import org.linphone.LinphoneManager;
import org.linphone.LinphoneUtils;
import org.linphone.R;
import org.linphone.activities.LinphoneActivity;
import org.linphone.contacts.ContactsManager;
import org.linphone.contacts.LinphoneContact;
import org.linphone.core.Address;
import org.linphone.core.ChatMessage;
import org.linphone.core.ChatMessageListener;
import org.linphone.core.ChatMessageListenerStub;
import org.linphone.core.ChatRoom;
import org.linphone.core.ChatRoomCapabilities;
import org.linphone.core.ChatRoomListenerStub;
import org.linphone.core.ChatRoomSecurityLevel;
import org.linphone.core.EventLog;
import org.linphone.core.PresenceActivity;
import org.linphone.core.PresenceModel;
import org.linphone.core.ZrtpPeerStatus;
import org.linphone.ui.ListSelectionAdapter;
import org.linphone.ui.ListSelectionHelper;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

import static android.text.format.DateUtils.isToday;
import static org.linphone.LinphoneUtils.getZrtpStatus;
import static org.linphone.LinphoneUtils.hasContentFileSharing;
import static org.linphone.LinphoneUtils.isUs;

public class ChatRoomsAdapter extends ListSelectionAdapter {

	private class ChatRoomViewHolder {
		public TextView lastMessageSenderView;
		public TextView lastMessageView;
		public TextView date;
		public TextView displayName;
		public TextView unreadMessages;
		public CheckBox delete;
		public ImageView contactPicture;
		public ImageView messageStatus;
		public ImageView filesharing;

		public ChatRoomViewHolder(View view) {
			lastMessageSenderView = view.findViewById(R.id.lastMessageSender);
			lastMessageView = view.findViewById(R.id.lastMessage);
			date = view.findViewById(R.id.date);
			displayName = view.findViewById(R.id.sipUri);
			unreadMessages = view.findViewById(R.id.unreadMessages);
			delete = view.findViewById(R.id.delete_chatroom);
			contactPicture = view.findViewById(R.id.contact_picture);
			messageStatus = view.findViewById(R.id.lastMessageStatus);
			filesharing = view.findViewById(R.id.filesharing);
		}
	}

	private Context mContext;
	private List<ChatRoom> mRooms;
	private LayoutInflater mLayoutInflater;
	private Bitmap mDefaultBitmap, mDefaultGroupBitmap;
	private ChatRoomListenerStub mListener;
	private ChatMessageListenerStub mChatMessageListener;

	public ChatRoomsAdapter(Context context, ListSelectionHelper helper, LayoutInflater inflater) {
		super(helper);
		mContext = context;
		mLayoutInflater = inflater;
		mRooms = new ArrayList<>();
		mDefaultBitmap = BitmapFactory.decodeResource(context.getResources(), R.drawable.avatar_big_secure1);
		mDefaultGroupBitmap = BitmapFactory.decodeResource(context.getResources(), R.drawable.avatar_group_small_unregistered);

		mListener = new ChatRoomListenerStub() {
			@Override
			public void onSubjectChanged(ChatRoom cr, EventLog eventLog) {
				ChatRoomViewHolder holder = (ChatRoomViewHolder) cr.getUserData();
				holder.displayName.setText(cr.getSubject());
				refresh();
			}
		};

		mChatMessageListener = new ChatMessageListenerStub() {
			@Override
			public void onMsgStateChanged(ChatMessage msg, ChatMessage.State state) {
				super.onMsgStateChanged(msg, state);
				refresh();
			}
		};
	}

	public void refresh() {
		mRooms = new ArrayList<>(Arrays.asList(LinphoneManager.getLc().getChatRooms()));
		Collections.sort(mRooms, new Comparator<ChatRoom>() {
			public int compare(ChatRoom cr1, ChatRoom cr2) {
				long timeDiff = cr1.getLastUpdateTime() - cr2.getLastUpdateTime();
				if (timeDiff > 0) return -1;
				else if (timeDiff == 0) return 0;
				return 1;
			}
		});
		notifyDataSetChanged();
	}

	public void clear() {
		for (ChatRoom room : mRooms) {
			room.removeListener(mListener);
		}
		mRooms.clear();
	}

	/**
	 * Adapter's methods
	 */

	@Override
	public int getCount() {
		return mRooms.size();
	}

	@Override
	public Object getItem(int position) {
		return mRooms.get(position);
	}

	@Override
	public long getItemId(int position) {
		return position;
	}

	@Override
	public View getView(final int position, View convertView, ViewGroup viewGroup) {
		View view;
		ChatRoomViewHolder holder;

		if (convertView != null) {
			view = convertView;
			holder = (ChatRoomViewHolder) view.getTag();
		} else {
			view = mLayoutInflater.inflate(R.layout.chatlist_cell, viewGroup, false);
			holder = new ChatRoomViewHolder(view);
			view.setTag(holder);
		}

		ChatRoom chatRoom = mRooms.get(position);
		Address remoteAddress = chatRoom.getPeerAddress();
		Address contactAddress = remoteAddress;

		if (chatRoom.getLastMessageInHistory() != null && chatRoom.getLastMessageInHistory().isOutgoing()) {
			chatRoom.getLastMessageInHistory().setListener(mChatMessageListener);
		}

		if (chatRoom.hasCapability(ChatRoomCapabilities.OneToOne.toInt()) && chatRoom.getParticipants().length > 0) {
			contactAddress = chatRoom.getParticipants()[0].getAddress();
		}

		if (chatRoom.hasCapability(ChatRoomCapabilities.Conference.toInt()) && chatRoom.getState() == ChatRoom.State.Created) { // Only set for state Created otherwise it will conflict with removal listener
			chatRoom.addListener(mListener);
			chatRoom.setUserData(holder);
		}

		int unreadMessagesCount = LinphoneManager.getInstance().getUnreadCountForChatRoom(chatRoom);
		ChatMessage lastMessage = chatRoom.getLastMessageInHistory();
		holder.lastMessageView.setText("");
		holder.lastMessageSenderView.setText("");
		holder.date.setText(LinphoneUtils.timestampToHumanDate(mContext, chatRoom.getLastUpdateTime(), ((LinphoneUtils.isToday(chatRoom.getLastUpdateTime())) ? R.string.today_date_format2 : R.string.messages_list_date_format), false));

		if (lastMessage != null) {
			boolean hasContent = hasContentFileSharing(lastMessage.getContents());
			holder.lastMessageView.setText(lastMessage.getTextContent());
			if (lastMessage.getFileTransferInformation() != null || lastMessage.getExternalBodyUrl() != null || hasContent) {
				holder.filesharing.setVisibility(View.VISIBLE);
			} else if (lastMessage.getTextContent() != null && lastMessage.getTextContent().length() > 0) {
				holder.filesharing.setVisibility(View.GONE);
			}

			if (lastMessage.isOutgoing()) {
				holder.messageStatus.setVisibility(View.VISIBLE);
				holder.lastMessageSenderView.setVisibility(View.GONE);
				if (lastMessage.getState().toInt() == ChatMessage.State.Delivered.toInt() ||
						lastMessage.getState().toInt() == ChatMessage.State.DeliveredToUser.toInt() ||
						lastMessage.getState().toInt() == ChatMessage.State.FileTransferDone.toInt()) {
					holder.messageStatus.setImageResource(R.drawable.message_delivered);
				} else if (lastMessage.getState().toInt() == ChatMessage.State.Displayed.toInt()) {
					holder.messageStatus.setImageResource(R.drawable.message_read);
				} else if (lastMessage.getState().toInt() == ChatMessage.State.Idle.toInt() ||
						lastMessage.getState().toInt() == ChatMessage.State.InProgress.toInt()) {
					holder.messageStatus.setImageResource(R.drawable.chat_message_inprogress);
				} else {
					holder.messageStatus.setImageResource(R.drawable.message_undelivered);
				}
			} else {
				holder.messageStatus.setVisibility(View.GONE);
			}
			if (!chatRoom.hasCapability(ChatRoomCapabilities.OneToOne.toInt())){
				Address lastMessageSenderAddress = lastMessage.getFromAddress();
				LinphoneContact contact = ContactsManager.getInstance().findContactFromAddress(lastMessageSenderAddress);
				holder.lastMessageSenderView.setVisibility(View.VISIBLE);
				holder.messageStatus.setVisibility(View.GONE);
				if (contact != null) {
					holder.lastMessageSenderView.setText(contact.getFullName() + mContext.getString(R.string.separator));
				} else {
					if (lastMessage.getFromAddress() != null && isUs(LinphoneManager.getLc().getDefaultProxyConfig(), lastMessage.getFromAddress().getUsername())) {
						holder.lastMessageSenderView.setText(mContext.getString(R.string.you) + mContext.getString(R.string.separator));
					} else {
						holder.lastMessageSenderView.setText(LinphoneUtils.getAddressDisplayName(lastMessageSenderAddress) + mContext.getString(R.string.separator));
					}
				}
			}
		}

		holder.displayName.setSelected(true); // For animation
		holder.contactPicture.setImageBitmap(mDefaultBitmap);

		if (chatRoom.hasCapability(ChatRoomCapabilities.OneToOne.toInt())) {
			LinphoneContact contact;
			if (chatRoom.getParticipants().length > 0) {
				contact = ContactsManager.getInstance().findContactFromAddress(chatRoom.getParticipants()[0].getAddress());
				if (contact != null) {
					holder.displayName.setText(contact.getFullName());
					//LinphoneUtils.setThumbnailPictureFromUri(LinphoneActivity.instance(), holder.contactPicture, contact.getThumbnailUri());
				} else {
					holder.displayName.setText(LinphoneUtils.getAddressDisplayName(chatRoom.getParticipants()[0].getAddress()));
				}
			} else {
				contact = ContactsManager.getInstance().findContactFromAddress(contactAddress);
				if (contact != null) {
					holder.displayName.setText(contact.getFullName());
					//LinphoneUtils.setThumbnailPictureFromUri(LinphoneActivity.instance(), holder.contactPicture, contact.getThumbnailUri());
				} else {
					holder.displayName.setText(LinphoneUtils.getAddressDisplayName(contactAddress));
				}
			}
			ChatRoomSecurityLevel level = chatRoom.getSecurityLevel();
			if (LinphoneManager.getLc().limeV2Enabled()) {
				if (level == ChatRoomSecurityLevel.Safe) {
					holder.contactPicture.setImageResource(R.drawable.avatar_small_secure2);
				} else if (level == ChatRoomSecurityLevel.Unsafe) {
					holder.contactPicture.setImageResource(R.drawable.avatar_small_unsecure);
				} else if (level == ChatRoomSecurityLevel.Encrypted) {
					holder.contactPicture.setImageResource(R.drawable.avatar_small_secure1);
				} else {
					ZrtpPeerStatus zrtpStatus = getZrtpStatus(LinphoneManager.getLc(), contactAddress.asStringUriOnly());
					if (zrtpStatus == ZrtpPeerStatus.Valid) {
						holder.contactPicture.setImageResource(R.drawable.avatar_medium_secure2);
					} else if (zrtpStatus == ZrtpPeerStatus.Invalid) {
						holder.contactPicture.setImageResource(R.drawable.avatar_medium_unsecure);
					} else {
						if (!ContactsManager.getInstance().isContactPresenceDisabled() && contact != null && contact.getFriend() != null) {
							PresenceModel presenceModel = contact.getFriend().getPresenceModel();
							if (presenceModel != null) {
								holder.contactPicture.setImageResource(R.drawable.avatar_medium_secure1);
							} else {
								holder.contactPicture.setImageResource(R.drawable.avatar_medium_unregistered);
							}
						} else {
							holder.contactPicture.setImageResource(R.drawable.avatar_medium_unregistered);
						}
					}
				}
			}
		} else {
			holder.displayName.setText(chatRoom.getSubject());
			ChatRoomSecurityLevel level = chatRoom.getSecurityLevel();
			if (level == ChatRoomSecurityLevel.Safe) {
				holder.contactPicture.setImageResource(R.drawable.avatar_group_small_secure2);
			} else if (level == ChatRoomSecurityLevel.Unsafe) {
				holder.contactPicture.setImageResource(R.drawable.avatar_group_small_unsecure);
			} else if (level == ChatRoomSecurityLevel.Encrypted) {
				holder.contactPicture.setImageResource(R.drawable.avatar_group_small_secure1);
			} else {
				holder.contactPicture.setImageResource(R.drawable.avatar_group_small_unregistered);
			}
		}

		if (unreadMessagesCount > 0) {
			holder.unreadMessages.setVisibility(View.VISIBLE);
			holder.unreadMessages.setText(String.valueOf(unreadMessagesCount));
			if (unreadMessagesCount > 99) {
				holder.unreadMessages.setTextSize(12);
			}
			holder.unreadMessages.setVisibility(View.VISIBLE);
			holder.displayName.setTypeface(null, Typeface.BOLD);
		} else {
			holder.unreadMessages.setVisibility(View.GONE);
			holder.displayName.setTypeface(null, Typeface.NORMAL);
		}

		if (isEditionEnabled()) {
			view.setOnClickListener(new View.OnClickListener() {
				@Override
				public void onClick(View v) {
					ChatRoomViewHolder holder = (ChatRoomViewHolder)v.getTag();
					holder.delete.setChecked(!holder.delete.isChecked());
					notifyDataSetInvalidated();
				}
			});
			holder.unreadMessages.setVisibility(View.GONE);
			holder.delete.setOnCheckedChangeListener(null);
			holder.delete.setVisibility(View.VISIBLE);
			holder.delete.setChecked(getSelectedItemsPosition().contains(position));
			holder.delete.setTag(position);
			holder.delete.setOnCheckedChangeListener(getDeleteListener());
			holder.date.setVisibility(View.GONE);
			if (getSelectedItemsPosition().contains(position)) {
				view.setBackgroundColor(0x26ff6600);
			} else {
				view.setBackgroundColor(0xffffff);
			}
		} else {
			view.setBackgroundColor(0xffffff);
			view.setOnClickListener(new View.OnClickListener() {
				@Override
				public void onClick(View v) {
					ChatRoom chatRoom = mRooms.get(position);
					LinphoneActivity.instance().goToChat(chatRoom.getPeerAddress().asString(), null);
				}
			});
			holder.delete.setVisibility(isEditionEnabled() ? View.VISIBLE : View.GONE);
			holder.date.setVisibility(View.VISIBLE);
		}
		return view;
	}
}