const base64Enc = std.base64.Base64Encoder.init(std.base64.standard_alphabet_chars, '=');
const base64Dec = std.base64.Base64Decoder.init(std.base64.standard_alphabet_chars, '=');

var rand: ?Random = null;

const max_message_size = 2048;

pub fn init() !void {
    var prng = Random.DefaultPrng.init(blk: {
        var seed: u64 = undefined;
        try posix.getrandom(mem.asBytes(&seed));
        break :blk seed;
    });
    rand = prng.random();
    try network.init();
}

pub fn deinit() void {
    network.deinit();
}

fn broadcastSaprusMessage(msg_bytes: []align(@alignOf(SaprusMessage)) u8, udp_port: u16) !void {
    if (false) {
        var foo: gcat.nic.RawSocket = try .init("enp7s0"); // /proc/net/dev
        defer foo.deinit();
    }
    const msg: *SaprusMessage = try .bytesAsValue(msg_bytes);
    try msg.networkFromNativeEndian();
    defer msg.nativeFromNetworkEndian() catch unreachable;

    var sock = try network.Socket.create(.ipv4, .udp);
    defer sock.close();

    try sock.setBroadcast(true);

    // Bind to 0.0.0.0:0
    const bind_addr = network.EndPoint{
        .address = network.Address{ .ipv4 = network.Address.IPv4.any },
        .port = 0,
    };

    const dest_addr = network.EndPoint{
        .address = network.Address{ .ipv4 = network.Address.IPv4.broadcast },
        .port = udp_port,
    };

    try sock.bind(bind_addr);

    std.debug.print("{x}\n", .{msg_bytes});

    _ = try sock.sendTo(dest_addr, msg_bytes);
}

pub fn sendRelay(payload: []const u8, dest: [4]u8) !void {
    var buf: [max_message_size]u8 align(@alignOf(SaprusMessage)) = undefined;
    const msg_bytes = buf[0..try SaprusMessage.calcSize(
        .relay,
        base64Enc.calcSize(payload.len),
    )];
    const msg: *SaprusMessage = .init(.relay, msg_bytes);

    const relay = (try msg.getSaprusTypePayload()).relay;
    relay.dest = dest;
    _ = base64Enc.encode(relay.getPayload(), payload);

    try broadcastSaprusMessage(msg_bytes, 8888);
}

fn randomPort() u16 {
    var p: u16 = 0;
    if (rand) |r| {
        p = r.intRangeAtMost(u16, 1024, 65000);
    } else unreachable;

    return p;
}

pub fn sendInitialConnection(payload: []const u8, initial_port: u16) !*SaprusMessage {
    const dest_port = randomPort();
    var buf: [max_message_size]u8 align(@alignOf(SaprusMessage)) = undefined;
    const msg_bytes = buf[0..try SaprusMessage.calcSize(
        .connection,
        base64Enc.calcSize(payload.len),
    )];
    const msg: *SaprusMessage = .init(.connection, msg_bytes);

    const connection = (try msg.getSaprusTypePayload()).connection;
    connection.src_port = initial_port;
    connection.dest_port = dest_port;
    _ = base64Enc.encode(connection.getPayload(), payload);

    try broadcastSaprusMessage(msg_bytes, 8888);

    return msg;
}

pub fn connect(payload: []const u8) !?SaprusConnection {
    const initial_port = randomPort();

    var initial_conn_res: ?*SaprusMessage = null;

    var sock = try network.Socket.create(.ipv4, .udp);
    defer sock.close();

    // Bind to 255.255.255.255:8888
    const bind_addr = network.EndPoint{
        .address = network.Address{ .ipv4 = network.Address.IPv4.broadcast },
        .port = 8888,
    };

    // timeout 1s
    try sock.setReadTimeout(1 * std.time.us_per_s);
    try sock.bind(bind_addr);

    const msg = try sendInitialConnection(payload, initial_port);

    var response_buf: [max_message_size]u8 align(@alignOf(SaprusMessage)) = undefined;
    _ = try sock.receive(&response_buf); // Ignore message that I sent.
    const len = try sock.receive(&response_buf);

    std.debug.print("response bytes: {x}\n", .{response_buf[0..len]});
    initial_conn_res = try .networkBytesAsValue(response_buf[0..len]);

    // Complete handshake after awaiting response
    try broadcastSaprusMessage(msg.asBytes(), randomPort());

    if (false) {
        return initial_conn_res.?;
    }
    return null;
}

const SaprusMessage = @import("message.zig").Message;
const SaprusConnection = @import("Connection.zig");

const std = @import("std");
const Random = std.Random;
const posix = std.posix;
const mem = std.mem;

const network = @import("network");
const gcat = @import("gatorcat");