import org.xbill.DNS.DClass
import org.xbill.DNS.Message
import org.xbill.DNS.Name
import org.xbill.DNS.SimpleResolver
import org.xbill.DNS.TSIG
import org.xbill.DNS.TXTRecord
import org.xbill.DNS.Update
import org.xbill.DNS.io.IoClientFactory
import org.xbill.DNS.io.TcpIoClient
import org.xbill.DNS.io.UdpIoClient
import java.util.concurrent.CompletableFuture


fun main() {
    val update = Update(Name.fromConstantString("zone.example.com."))

    repeat(2000) { i ->
        val record = TXTRecord(
            Name.fromConstantString("name-$i.zone.example.com."),
            DClass.IN,
            900,
            "a",
        )
        update.absent(record.name, record.type)
        update.add(record)
    }

    val resolver = SimpleResolver().apply {
        tsigKey = TSIG(TSIG.HMAC_SHA384, "zone.example.com.", "c2VjcmU=")
        ioClientFactory = object : IoClientFactory {
            override fun createOrGetTcpClient(): TcpIoClient {
                return TcpIoClient { local, remote, query, data, timeout ->
                    println("Sending data of length: ${data.size}")
                    println("Is data > max message size (${Message.MAXLENGTH})? ${data.size > Message.MAXLENGTH}")
                    CompletableFuture.failedFuture(Exception())
                }
            }

            override fun createOrGetUdpClient(): UdpIoClient {
                TODO("Not yet implemented")
            }
        }
    }

    resolver.send(update)
}