const std = @import("std");
const io = std.io;
const network = @import("zig-network/network.zig");
const dns = @import("zig-dns/src/dns.zig");

pub fn main() !void {
    var myallo = std.heap.GeneralPurposeAllocator(.{}){};
    const allocator = myallo.allocator();

    try network.init();
    defer network.deinit();
    const sock = try network.connectToHost(allocator, "2a09::", 53, .udp);
    defer sock.close();
    const writer = sock.writer();

    const message = try dns.createQuery(allocator, "www.afnic.fr", .AAAA);
    defer message.deinit();

    var message_bytes = try message.to_bytes(allocator);
    try writer.writeAll(message_bytes);

    var recv = [_]u8{0} ** 1024;
    const recv_size = try sock.receive(&recv);

    const response = try dns.Message.from_bytes(allocator, recv[0..recv_size]);
    defer response.deinit();

    std.debug.print("Response:\n{any}\n", .{response});
}