const std = @import("std");
const os = std.os.linux;
const Drm = @import("../request.zig").Drm;
const Card = @import("../card.zig").Card;

pub const Connection = @import("connection.zig").Connection;
pub const Mode = @import("mode.zig").Mode;

const RawConnector = extern struct {
    encoder_ids: ?*u32,
    modes: ?*Mode,
    prop_ids: ?*u32,
    prop_value_ids: ?*u64,
    count_modes: u32,
    count_props: u32,
    count_encoders: u32,
    encoder_id: u32,
    connector_id: u32,
    connector_type: u32,
    connector_type_id: u32,
    connection: Connection,
    mm_width: u32,
    mm_height: u32,
    subpixel: u32,
    pad: u32,
};

pub const Connector = struct {
    const Self = @This();

    card: *Card,

    encoder_ids: []u32,
    modes: []Mode,
    prop_ids: []u32,
    prop_value_ids: []u64,
    encoder_id: u32,
    connector_id: u32,
    connector_type: u32,
    connector_type_id: u32,
    connection: Connection,
    mm_width: u32,
    mm_height: u32,
    subpixel: u32,
    pad: u32,

    pub fn raw_without_ids(card: *Card, id: u32) !RawConnector {
        var result = std.mem.zeroInit(RawConnector, .{ .connector_id = id });
        try Drm.get_connector.request(card.file.handle, RawConnector, &result);
        return result;
    }

    // NOTE: This function does not take in account
    //       that there might be some hot-plugging going
    //       on. This might have to change in the future.
    pub fn init(card: *Card, id: u32) !Self {
        var raw = try Self.raw_without_ids(card, id);
        const resources = .{
            .encoder_ids = try card.allocator.alloc(u32, raw.count_encoders),
            .modes = try card.allocator.alloc(Mode, raw.count_modes),
            .prop_ids = try card.allocator.alloc(u32, raw.count_props),
            .prop_value_ids = try card.allocator.alloc(u64, raw.count_props),
            .encoder_id = raw.encoder_id,
            .connector_id = raw.connector_id,
            .connector_type = raw.connector_type,
            .connector_type_id = raw.connector_type_id,
            .connection = raw.connection,
            .mm_width = raw.mm_width,
            .mm_height = raw.mm_height,
            .subpixel = raw.subpixel,
            .pad = raw.pad,
            .card = card,
        };

        @memset(resources.encoder_ids, 0);
        @memset(resources.modes, std.mem.zeroes(Mode));
        @memset(resources.prop_ids, 0);
        @memset(resources.prop_value_ids, 0);

        raw.encoder_ids = @ptrCast(resources.encoder_ids);
        raw.modes = @ptrCast(resources.modes);
        raw.prop_ids = @ptrCast(resources.prop_ids);
        raw.prop_value_ids = @ptrCast(resources.prop_value_ids);

        try Drm.get_connector.request(card.file.handle, RawConnector, &raw);

        return resources;
    }

    pub fn deinit(self: *Self) void {
        self.card.allocator.free(self.encoder_ids);
        self.card.allocator.free(self.modes);
        self.card.allocator.free(self.prop_ids);
    }
};