Mercury-IM/app/src/main/java/org/mercury_im/messenger/ui/chat/ChatRecyclerViewAdapter.java

package org.mercury_im.messenger.ui.chat;

import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.TextView;

import androidx.annotation.NonNull;
import androidx.recyclerview.widget.RecyclerView;

import org.mercury_im.messenger.R;
import org.mercury_im.messenger.persistence.model.MessageModel;

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

public class ChatRecyclerViewAdapter extends RecyclerView.Adapter<ChatRecyclerViewAdapter.ChatItemViewHolder> {

    private List<MessageModel> messages = new ArrayList<>();

    public ChatRecyclerViewAdapter() {

    }

    public void updateMessages(List<MessageModel> messages) {
        this.messages.clear();
        this.messages.addAll(messages);
        notifyDataSetChanged();
    }

    @NonNull
    @Override
    public ChatItemViewHolder onCreateViewHolder(@NonNull ViewGroup parent, int viewType) {
        return new ChatItemViewHolder(LayoutInflater.from(parent.getContext()).inflate(viewType, parent, false));
    }

    @Override
    public void onBindViewHolder(@NonNull ChatItemViewHolder holder, int position) {
        MessageModel message = messages.get(position);
        holder.body.setText(message.getBody());
    }

    @Override
    public int getItemViewType(int position) {
        if (messages.get(position).isIncoming()) {
            return R.layout.view_message_text_in;
        } else {
            return R.layout.view_message_text_out;
        }
    }

    @Override
    public int getItemCount() {
        return messages.size();
    }

    public class ChatItemViewHolder extends RecyclerView.ViewHolder {

        private TextView body;

        public ChatItemViewHolder(@NonNull View itemView) {
            super(itemView);

            body = itemView.findViewById(R.id.msg_body);
        }
    }
}