const std = @import("std");
const cbor = @import("cbor");
const tp = @import("thespian");
const root = @import("soft_root").root;
const command = @import("command");

const tui = @import("../../tui.zig");
pub const Type = @import("palette.zig").Create(@This());

pub const label = "Search commands";
pub const name = "󱊒 command";
pub const description = "command";

pub const Entry = struct {
    label: []const u8,
    name: []const u8,
    hint: []const u8,
    id: command.ID,
    used_time: i64,
};

pub fn load_entries(palette: *Type) !usize {
    const hints = if (tui.input_mode()) |m| m.keybind_hints else @panic("no keybind hints");
    var longest_description: usize = 0;
    var longest_total: usize = 0;
    for (command.commands.items) |cmd_| if (cmd_) |p| {
        if (p.meta.description.len > 0) {
            const hint = hints.get(p.name) orelse "";
            longest_description = @max(longest_description, p.meta.description.len);
            longest_total = @max(longest_total, p.meta.description.len + hint.len + 1);
            (try palette.entries.addOne(palette.allocator)).* = .{
                .label = if (p.meta.description.len > 0) p.meta.description else p.name,
                .name = p.name,
                .hint = hint,
                .id = p.id,
                .used_time = 0,
            };
        }
    };
    return longest_total - longest_description;
}

pub fn add_menu_entry(palette: *Type, entry: *Entry, matches: ?[]const usize) !void {
    var value: std.Io.Writer.Allocating = .init(palette.allocator);
    defer value.deinit();
    const writer = &value.writer;
    try cbor.writeValue(writer, entry.label);
    try cbor.writeValue(writer, entry.hint);
    try cbor.writeValue(writer, matches orelse &[_]usize{});
    try cbor.writeValue(writer, entry.id);
    try palette.menu.add_item_with_handler(value.written(), select);
    palette.items += 1;
}

fn select(menu: **Type.MenuType, button: *Type.ButtonType, _: Type.Pos) void {
    var unused: []const u8 = undefined;
    var command_id: command.ID = undefined;
    var iter = button.opts.label;
    if (!(cbor.matchString(&iter, &unused) catch false)) return;
    if (!(cbor.matchString(&iter, &unused) catch false)) return;
    var len = cbor.decodeArrayHeader(&iter) catch return;
    while (len > 0) : (len -= 1)
        cbor.skipValue(&iter) catch break;
    if (!(cbor.matchValue(&iter, cbor.extract(&command_id)) catch false)) return;
    update_used_time(menu.*.opts.ctx, command_id);
    tp.self_pid().send(.{ "cmd", "exit_overlay_mode" }) catch |e| menu.*.opts.ctx.logger.err("navigate", e);
    tp.self_pid().send(.{ "cmd", command_id, .{} }) catch |e| menu.*.opts.ctx.logger.err("navigate", e);
}

fn sort_by_used_time(palette: *Type) void {
    const less_fn = struct {
        fn less_fn(_: void, lhs: Entry, rhs: Entry) bool {
            return lhs.used_time > rhs.used_time;
        }
    }.less_fn;
    std.mem.sort(Entry, palette.entries.items, {}, less_fn);
}

fn update_used_time(palette: *Type, id: command.ID) void {
    set_used_time(palette, id, std.time.milliTimestamp());
    write_state(palette) catch {};
}

fn set_used_time(palette: *Type, id: command.ID, used_time: i64) void {
    for (palette.entries.items) |*cmd_| if (cmd_.id == id) {
        cmd_.used_time = used_time;
        return;
    };
}

fn write_state(palette: *Type) !void {
    var state_file_buffer: [std.fs.max_path_bytes]u8 = undefined;
    const state_file = try std.fmt.bufPrint(&state_file_buffer, "{s}/{s}", .{ try root.get_state_dir(), "commands" });
    var file = try std.fs.createFileAbsolute(state_file, .{ .truncate = true });
    defer file.close();
    var buf: [4096]u8 = undefined;
    var file_writer = file.writer(&buf);
    const writer = &file_writer.interface;
    defer writer.flush() catch {};

    for (palette.entries.items) |cmd_| {
        if (cmd_.used_time == 0) continue;
        try cbor.writeArrayHeader(writer, 2);
        try cbor.writeValue(writer, cmd_.name);
        try cbor.writeValue(writer, cmd_.used_time);
    }
}

pub fn restore_state(palette: *Type) !void {
    var state_file_buffer: [std.fs.max_path_bytes]u8 = undefined;
    const state_file = try std.fmt.bufPrint(&state_file_buffer, "{s}/{s}", .{ try root.get_state_dir(), "commands" });
    const a = std.heap.c_allocator;
    var file = std.fs.openFileAbsolute(state_file, .{ .mode = .read_only }) catch |e| switch (e) {
        error.FileNotFound => return,
        else => return e,
    };
    defer file.close();
    const stat = try file.stat();
    var buffer = try a.alloc(u8, @intCast(stat.size));
    defer a.free(buffer);
    const size = try file.readAll(buffer);
    const data = buffer[0..size];

    defer sort_by_used_time(palette);
    var name_: []const u8 = undefined;
    var used_time: i64 = undefined;
    var iter: []const u8 = data;
    while (cbor.matchValue(&iter, .{
        tp.extract(&name_),
        tp.extract(&used_time),
    }) catch |e| switch (e) {
        error.TooShort => return,
        else => return e,
    }) {
        const id = command.get_id(name_) orelse continue;
        set_used_time(palette, id, used_time);
    }
}