bitchat/bitchatTests/Protocols/BitchatFilePacketTests.swift

import BitFoundation
import XCTest
@testable import bitchat

final class BitchatFilePacketTests: XCTestCase {

    func testRoundTripPreservesFields() throws {
        let content = Data((0..<4096).map { UInt8($0 % 251) })
        let packet = BitchatFilePacket(
            fileName: "sample.jpg",
            fileSize: UInt64(content.count),
            mimeType: "image/jpeg",
            content: content
        )

        guard let encoded = packet.encode() else {
            return XCTFail("Failed to encode file packet")
        }
        guard let decoded = BitchatFilePacket.decode(encoded) else {
            return XCTFail("Failed to decode file packet")
        }

        XCTAssertEqual(decoded.fileName, packet.fileName)
        XCTAssertEqual(decoded.fileSize, packet.fileSize)
        XCTAssertEqual(decoded.mimeType, packet.mimeType)
        XCTAssertEqual(decoded.content, packet.content)
    }

    func testDecodeFallsBackToContentSizeWhenFileSizeMissing() throws {
        let content = Data(repeating: 0x7F, count: 1024)
        let packet = BitchatFilePacket(
            fileName: nil,
            fileSize: nil,
            mimeType: nil,
            content: content
        )

        guard let encoded = packet.encode() else {
            return XCTFail("Failed to encode file packet")
        }
        guard let decoded = BitchatFilePacket.decode(encoded) else {
            return XCTFail("Failed to decode file packet")
        }

        XCTAssertEqual(decoded.fileSize, UInt64(content.count))
        XCTAssertEqual(decoded.content, content)
    }

    func testDecodeSupportsLegacyEightByteFileSizeTLV() throws {
        let content = Data([0x01, 0x02, 0x03, 0x04])
        var data = Data()

        data.append(0x02)
        data.append(contentsOf: [0x00, 0x08])
        data.append(contentsOf: [0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00])
        data.append(0x04)
        data.append(contentsOf: [0x00, 0x00, 0x00, 0x04])
        data.append(content)

        let decoded = try XCTUnwrap(BitchatFilePacket.decode(data))
        XCTAssertEqual(decoded.fileSize, 256)
        XCTAssertEqual(decoded.content, content)
    }

    func testDecodeUsesContentCountWhenFileSizeTLVIsMissing() throws {
        let content = Data([0xAA, 0xBB, 0xCC])
        var data = Data()

        data.append(0x04)
        data.append(contentsOf: [0x00, 0x00, 0x00, 0x03])
        data.append(content)

        let decoded = try XCTUnwrap(BitchatFilePacket.decode(data))
        XCTAssertEqual(decoded.fileSize, UInt64(content.count))
        XCTAssertEqual(decoded.content, content)
    }

    func testPrivateMediaMessageIdentityConvergesAcrossPeerIDAliases() throws {
        let senderKey = Data(repeating: 0x11, count: 32)
        let recipientKey = Data(repeating: 0x22, count: 32)
        let senderStable = PeerID(hexData: senderKey)
        let recipientStable = PeerID(hexData: recipientKey)
        let fileName = "img_20260725_105708_1CC2760D-76AA-40C3-8013-C7FAA6C2EF99.jpg"

        let senderID = try XCTUnwrap(PrivateMediaMessageIdentity.stableID(
            senderPeerID: senderStable.toShort(),
            recipientPeerID: PeerID(str: "mesh:\(recipientStable.toShort().bare)"),
            fileName: fileName
        ))
        let receiverID = try XCTUnwrap(PrivateMediaMessageIdentity.stableID(
            senderPeerID: senderStable,
            recipientPeerID: recipientStable.toShort(),
            fileName: fileName
        ))

        XCTAssertEqual(senderID, receiverID)
        XCTAssertTrue(senderID.hasPrefix("media-"))
        XCTAssertEqual(senderID.count, 38)
        XCTAssertTrue(PrivateMediaMessageIdentity.isStableID(senderID))
        XCTAssertFalse(PrivateMediaMessageIdentity.isStableID("media-\(String(repeating: "A", count: 32))"))
        XCTAssertFalse(PrivateMediaMessageIdentity.isStableID("media-\(String(repeating: "a", count: 31))"))
        XCTAssertFalse(PrivateMediaMessageIdentity.isStableID(UUID().uuidString))
    }

    func testPrivateMediaMessageIdentitySeparatesDirectionAndFilename() throws {
        let alice = PeerID(str: "0011223344556677")
        let bob = PeerID(str: "8899aabbccddeeff")
        let firstName = "voice_20260725_105708_11111111-1111-1111-1111-111111111111.m4a"
        let secondName = "voice_20260725_105709_22222222-2222-2222-2222-222222222222.m4a"
        let first = try XCTUnwrap(PrivateMediaMessageIdentity.stableID(
            senderPeerID: alice,
            recipientPeerID: bob,
            fileName: firstName
        ))

        XCTAssertNotEqual(first, PrivateMediaMessageIdentity.stableID(
            senderPeerID: bob,
            recipientPeerID: alice,
            fileName: firstName
        ))
        XCTAssertNotEqual(first, PrivateMediaMessageIdentity.stableID(
            senderPeerID: alice,
            recipientPeerID: bob,
            fileName: secondName
        ))
        XCTAssertNil(PrivateMediaMessageIdentity.stableID(
            senderPeerID: alice,
            recipientPeerID: bob,
            fileName: nil
        ))
        XCTAssertNil(PrivateMediaMessageIdentity.stableID(
            senderPeerID: alice,
            recipientPeerID: bob,
            fileName: "photo.jpg"
        ))
        XCTAssertNil(PrivateMediaMessageIdentity.stableID(
            senderPeerID: alice,
            recipientPeerID: bob,
            fileName: "img_11111111-1111-1111-1111-111111111111.pdf"
        ))
        XCTAssertNotNil(PrivateMediaMessageIdentity.stableID(
            senderPeerID: alice,
            recipientPeerID: bob,
            fileName: "voice_0011223344556677.m4a"
        ))
    }

    func testPrivateMediaMessageIdentityMatchesVersionOneGoldenVector() {
        XCTAssertEqual(
            PrivateMediaMessageIdentity.stableID(
                senderPeerID: PeerID(str: "0011223344556677"),
                recipientPeerID: PeerID(str: "8899aabbccddeeff"),
                fileName: "img_20260725_105708_1CC2760D-76AA-40C3-8013-C7FAA6C2EF99.jpg"
            ),
            "media-910bd42c65060ab76bb6406f220c4516"
        )
    }
}