1package com.example.llama
 2
 3import android.view.LayoutInflater
 4import android.view.View
 5import android.view.ViewGroup
 6import android.widget.TextView
 7import androidx.recyclerview.widget.RecyclerView
 8
 9data class Message(
10    val id: String,
11    val content: String,
12    val isUser: Boolean
13)
14
15class MessageAdapter(
16    private val messages: List<Message>
17) : RecyclerView.Adapter<RecyclerView.ViewHolder>() {
18
19    companion object {
20        private const val VIEW_TYPE_USER = 1
21        private const val VIEW_TYPE_ASSISTANT = 2
22    }
23
24    override fun getItemViewType(position: Int): Int {
25        return if (messages[position].isUser) VIEW_TYPE_USER else VIEW_TYPE_ASSISTANT
26    }
27
28    override fun onCreateViewHolder(parent: ViewGroup, viewType: Int): RecyclerView.ViewHolder {
29        val layoutInflater = LayoutInflater.from(parent.context)
30        return if (viewType == VIEW_TYPE_USER) {
31            val view = layoutInflater.inflate(R.layout.item_message_user, parent, false)
32            UserMessageViewHolder(view)
33        } else {
34            val view = layoutInflater.inflate(R.layout.item_message_assistant, parent, false)
35            AssistantMessageViewHolder(view)
36        }
37    }
38
39    override fun onBindViewHolder(holder: RecyclerView.ViewHolder, position: Int) {
40        val message = messages[position]
41        if (holder is UserMessageViewHolder || holder is AssistantMessageViewHolder) {
42            val textView = holder.itemView.findViewById<TextView>(R.id.msg_content)
43            textView.text = message.content
44        }
45    }
46
47    override fun getItemCount(): Int = messages.size
48
49    class UserMessageViewHolder(view: View) : RecyclerView.ViewHolder(view)
50    class AssistantMessageViewHolder(view: View) : RecyclerView.ViewHolder(view)
51}