Notification.kt for dashbuddy

package cloud.trotter.dashbuddy.bubble // You can choose your package structure

import android.app.Notification
import android.app.PendingIntent
import android.content.Context
import android.util.Log
import androidx.core.app.NotificationCompat
import androidx.core.app.Person
import androidx.core.content.LocusIdCompat
import androidx.core.graphics.drawable.IconCompat
import java.util.Date

/**
 * Helper object to create bubble-specific notifications.
 */
object Notification {

    /**
     * Creates a notification configured for display as a bubble.
     *
     * @param context The application context.
     * @param channelId The ID of the notification channel to use.
     * @param senderPerson The Person object representing the sender of the message (e.g., "DashBuddy").
     * @param shortcutId The ID of the shortcut this bubble is associated with.
     * @param bubbleIcon The icon to display for the bubble and notification.
     * @param messageText The text content of the message.
     * @param contentIntent The PendingIntent to launch when the bubble is tapped.
     * @param locusId Optional LocusIdCompat to link the notification to app state.
     * @param desiredHeight The desired height of the bubble's expanded view.
     * @param suppressNotification True to suppress the fly-out notification and only show the bubble.
     * @param autoExpandBubble True to have the bubble auto-expand when it first appears.
     * @return A configured Notification object.
     */
    fun create(
        context: Context,
        channelId: String,
        senderPerson: Person,
        shortcutId: String,
        bubbleIcon: IconCompat,
        messageText: String,
        messagingStyle: NotificationCompat.MessagingStyle,
        contentIntent: PendingIntent,
        locusId: LocusIdCompat? = null, // Added LocusIdCompat parameter
        desiredHeight: Int = 600, // Standard height for bubbles
        suppressNotification: Boolean = true,
        autoExpandBubble: Boolean = true
    ): Notification {
        Log.d("BubbleNotificationHelper", "Creating messaging bubble notification with message: '$messageText', Locus ID: $locusId")

        // 1. Create the BubbleMetadata
        val bubbleMetadataBuilder = NotificationCompat.BubbleMetadata.Builder(contentIntent, bubbleIcon)
            .setDesiredHeight(desiredHeight)
            .setSuppressNotification(suppressNotification)
            .setAutoExpandBubble(autoExpandBubble)

        // For API 30+, associate LocusId with BubbleMetadata if available
        // Although setLocusId is not directly on BubbleMetadata.Builder,
        // the LocusId on the Notification itself is what matters most for system context.

        val bubbleMetadata = bubbleMetadataBuilder.build()

        // 2. Create the MessagingStyle for the notification content
        messagingStyle.addMessage(
                NotificationCompat.MessagingStyle.Message(
                    messageText,
                    Date().time,
                    senderPerson
                )
            )

        // 3. Build the Notification
        val builder = NotificationCompat.Builder(context, channelId)
            .setSmallIcon(bubbleIcon)
            .setContentTitle(senderPerson.name)
            .setContentText(messageText)
            .setShortcutId(shortcutId)
            .setCategory(NotificationCompat.CATEGORY_MESSAGE)
            .addPerson(senderPerson)
            .setStyle(messagingStyle)
            .setBubbleMetadata(bubbleMetadata)
            .setContentIntent(contentIntent)
            .setShowWhen(true)

        // Set LocusId on the notification builder if provided (for API 29+)
        if (locusId != null) {
            builder.setLocusId(locusId)
        }

        return builder. Build()
    }
}