package org.meshbay.client

import org.json.JSONObject
import org.junit.Assert.assertEquals
import org.junit.Assert.assertFalse
import org.junit.Assert.assertNull
import org.junit.Assert.assertTrue
import org.junit.Test
import org.meshbay.client.drive.DriveFile
import org.meshbay.client.drive.DriveKind
import org.meshbay.client.drive.DrivePlan
import org.meshbay.client.drive.WhatsApp
import java.util.TimeZone

class WhatsAppTest {
    private val tree = "content://tree/wa"

    private fun file(sub: String, name: String, size: Long = 10) =
        DriveFile(tree, "primary:${WhatsApp.FOLDER}/$sub/$name", sub, name, size, 1000, "application/octet-stream")

    @Test fun `only WhatsApp's own folder can be chosen`() {
        assertNull(WhatsApp.refusal("primary:Android/media/com.whatsapp/WhatsApp"))
        assertNull(WhatsApp.refusal("primary:Android/media/com.whatsapp.w4b/WhatsApp Business"))
        assertEquals("not_whatsapp", WhatsApp.refusal("primary:Android/media/com.whatsapp"))
        assertEquals("not_whatsapp", WhatsApp.refusal("primary:Documents"))
        assertEquals("not_whatsapp", WhatsApp.refusal("primary:Android/media/com.whatsapp/WhatsApp/Media"))
    }

    @Test fun `the chats as they are now, not the dated copies of earlier weeks`() {
        assertTrue(WhatsApp.include(file("Databases", "msgstore.db.crypt14"), false))
        assertTrue(WhatsApp.include(file("Databases", "msgstore-increment-1.db.crypt14"), false))
        assertFalse(WhatsApp.include(file("Databases", "msgstore-2026-10-09.1.db.crypt14"), false))
        assertFalse(WhatsApp.include(file("Databases", "msgstore-increment-3-2026-10-09.1.db.crypt14"), false))
        assertTrue(WhatsApp.include(file("Backups", "wa.db.crypt14"), false))
        assertTrue(WhatsApp.include(file("Backups/Stickers", "s.webp"), false))
    }

    @Test fun `media only when asked for, and nothing else of the folder`() {
        val photo = file("Media/WhatsApp Images", "IMG-20261010-WA0001.jpg")
        assertFalse(WhatsApp.include(photo, false))
        assertTrue(WhatsApp.include(photo, true))
        assertFalse(WhatsApp.include(file(".Shared", "x"), true))
        assertTrue(DriveKind.WHATSAPP.include(photo, JSONObject().put("media", true)))
    }

    @Test fun `what a restore needs together is the chats and the backups, not the media`() {
        assertTrue(WhatsApp.inRestoreSet(file("Databases", "msgstore.db.crypt14")))
        assertTrue(WhatsApp.inRestoreSet(file("Backups", "wa.db.crypt14")))
        assertFalse(WhatsApp.inRestoreSet(file("Media/WhatsApp Video", "v.mp4")))
    }

    @Test fun `on the node the folders are WhatsApp's own, with no extra level`() {
        val out = DrivePlan.plan(listOf(file("Databases", "msgstore.db.crypt14")), "Backups/bob-whatsapp",
                                 mapOf(tree to ""), { null }, emptySet(), TimeZone.getTimeZone("UTC")) { _, _ -> false }
        assertEquals("Backups/bob-whatsapp/Databases", out.single().dir)
    }
}