LibreMail/app/src/test/kotlin/org/libremail/mail/GraphSenderTest.kt

// SPDX-License-Identifier: GPL-3.0-or-later
package org.libremail.mail

import org.json.JSONObject
import org.junit.Test
import org.libremail.domain.model.OutgoingMessage
import java.io.File
import kotlin.test.assertEquals
import kotlin.test.assertFalse
import kotlin.test.assertTrue

class GraphSenderTest {

    private fun message(to: String, cc: String = "", subject: String = "Hi", body: String = "Body") =
        OutgoingMessage(accountId = "outlook:me@example.com", to = to, cc = cc, subject = subject, body = body)

    @Test
    fun `payload carries subject, body and parsed recipients`() {
        val json = JSONObject(buildSendMailPayload(message(to = "a@x.com, b@y.com", cc = "c@z.com"), emptyList()))
        assertTrue(json.getBoolean("saveToSentItems"))

        val msg = json.getJSONObject("message")
        assertEquals("Hi", msg.getString("subject"))
        assertEquals("Text", msg.getJSONObject("body").getString("contentType"))
        assertEquals("Body", msg.getJSONObject("body").getString("content"))

        val to = msg.getJSONArray("toRecipients")
        assertEquals(2, to.length())
        assertEquals("a@x.com", to.getJSONObject(0).getJSONObject("emailAddress").getString("address"))
        assertEquals("b@y.com", to.getJSONObject(1).getJSONObject("emailAddress").getString("address"))

        val cc = msg.getJSONArray("ccRecipients")
        assertEquals(1, cc.length())
        assertEquals("c@z.com", cc.getJSONObject(0).getJSONObject("emailAddress").getString("address"))
    }

    @Test
    fun `payload uses the HTML content type when a formatted body is present`() {
        val message = OutgoingMessage(
            accountId = "outlook:me@example.com",
            to = "a@x.com",
            subject = "Hi",
            body = "Hello world",
            bodyHtml = "<p>Hello <b>world</b></p>",
        )
        val body = JSONObject(buildSendMailPayload(message, emptyList()))
            .getJSONObject("message").getJSONObject("body")
        assertEquals("HTML", body.getString("contentType"))
        assertEquals("<p>Hello <b>world</b></p>", body.getString("content"))
    }

    @Test
    fun `payload falls back to plain text when there is no HTML body`() {
        val body = JSONObject(buildSendMailPayload(message(to = "a@x.com"), emptyList()))
            .getJSONObject("message").getJSONObject("body")
        assertEquals("Text", body.getString("contentType"))
    }

    @Test
    fun `recipients parse RFC822 display names into bare addresses`() {
        val json = JSONObject(
            buildSendMailPayload(
                message(to = "John Doe <john@example.com>", cc = "\"Doe, Jane\" <jane@example.com>"),
                emptyList(),
            ),
        )
        val msg = json.getJSONObject("message")

        val to = msg.getJSONArray("toRecipients")
        assertEquals(1, to.length())
        val toAddress = to.getJSONObject(0).getJSONObject("emailAddress")
        assertEquals("john@example.com", toAddress.getString("address"))
        assertEquals("John Doe", toAddress.getString("name"))

        // The comma inside the quoted display name must not be treated as an address separator.
        val cc = msg.getJSONArray("ccRecipients")
        assertEquals(1, cc.length())
        assertEquals("jane@example.com", cc.getJSONObject(0).getJSONObject("emailAddress").getString("address"))
    }

    @Test
    fun `payload carries bcc recipients when present`() {
        val json = JSONObject(
            buildSendMailPayload(message(to = "a@x.com").copy(bcc = "hidden@z.com, more@z.com"), emptyList()),
        )
        val bcc = json.getJSONObject("message").getJSONArray("bccRecipients")
        assertEquals(2, bcc.length())
        assertEquals("hidden@z.com", bcc.getJSONObject(0).getJSONObject("emailAddress").getString("address"))
        assertEquals("more@z.com", bcc.getJSONObject(1).getJSONObject("emailAddress").getString("address"))
    }

    @Test
    fun `payload omits cc and bcc when blank and encodes attachments as base64`() {
        val file = File.createTempFile("graph-att", ".txt").apply { writeText("hello") }
        try {
            val msg = JSONObject(buildSendMailPayload(message(to = "a@x.com"), listOf(SendableAttachment(file))))
                .getJSONObject("message")
            assertFalse(msg.has("ccRecipients"))
            assertFalse(msg.has("bccRecipients"))

            val attachments = msg.getJSONArray("attachments")
            assertEquals(1, attachments.length())
            val attachment = attachments.getJSONObject(0)
            assertEquals("#microsoft.graph.fileAttachment", attachment.getString("@odata.type"))
            assertEquals(file.name, attachment.getString("name"))
            assertEquals("aGVsbG8=", attachment.getString("contentBytes")) // base64("hello")
            // A plain attachment carries no inline markers.
            assertFalse(attachment.has("isInline"))
            assertFalse(attachment.has("contentId"))
        } finally {
            file.delete()
        }
    }

    @Test
    fun `an inline image attachment is marked isInline with its contentId`() {
        val image = File.createTempFile("graph-inline", ".png").apply { writeText("PNGDATA") }
        try {
            val message = message(to = "a@x.com").copy(bodyHtml = "<p><img src=\"cid:logo@libremail\"></p>")
            val inline = SendableAttachment(image, contentId = "logo@libremail", isInline = true)
            val attachment = JSONObject(buildSendMailPayload(message, listOf(inline)))
                .getJSONObject("message").getJSONArray("attachments").getJSONObject(0)
            assertEquals(image.name, attachment.getString("name"))
            assertTrue(attachment.getBoolean("isInline"))
            assertEquals("logo@libremail", attachment.getString("contentId"))
        } finally {
            image.delete()
        }
    }

    @Test
    fun `an inline image contentId is stripped of control characters in the payload`() {
        val image = File.createTempFile("graph-inline", ".png").apply { writeText("PNGDATA") }
        try {
            val message = message(to = "a@x.com").copy(bodyHtml = "<p><img src=\"cid:logo@libremail\"></p>")
            // A crafted content id with CR/LF and a control char; only the printable text must remain.
            val inline = SendableAttachment(image, contentId = "logo@libremail\r\nevil", isInline = true)
            val contentId = JSONObject(buildSendMailPayload(message, listOf(inline)))
                .getJSONObject("message").getJSONArray("attachments").getJSONObject(0)
                .getString("contentId")
            assertEquals("logo@libremailevil", contentId)
            assertFalse('\r' in contentId || '\n' in contentId, "contentId=$contentId")
        } finally {
            image.delete()
        }
    }
}