const std = @import("std");
const zterm = @import("zterm");

const App = zterm.App(union(enum) {});
const Key = zterm.Key;

const log = std.log.scoped(.example);

// TODO: maybe inlining the functions should be done as well to reduce the call
//       stacks (might be important for more complex structures)

pub const ExampleElement = struct {
    pub fn element(this: *@This()) App.Element {
        return .{
            .ptr = this,
            .vtable = &.{
                .handle = handle,
                .content = content,
            },
        };
    }

    // example function to render contents for a `Container`
    fn content(ctx: *anyopaque, cells: []zterm.Cell, size: zterm.Size) !void {
        _ = ctx;
        std.debug.assert(cells.len == @as(usize, size.cols) * @as(usize, size.rows));

        // NOTE: error should only be returned here in case an in-recoverable exception has occured
        const row = size.rows / 2;
        const col = size.cols / 2 -| 3;

        for (0..5) |c| {
            cells[(row * size.cols) + col + c].style.fg = .black;
            cells[(row * size.cols) + col + c].cp = '-';
        }
    }

    // example function to handle events for a `Container`
    fn handle(ctx: *anyopaque, event: App.Event) !void {
        _ = ctx;
        switch (event) {
            .init => log.debug(".init event", .{}),
            else => {},
        }
    }
};

pub fn main() !void {
    errdefer |err| log.err("Application Error: {any}", .{err});

    // TODO: maybe create own allocator as some sort of arena allocator to have consistent memory usage
    var gpa: std.heap.GeneralPurposeAllocator(.{}) = .init;
    defer {
        const deinit_status = gpa.deinit();
        if (deinit_status == .leak) {
            log.err("memory leak", .{});
        }
    }
    const allocator = gpa.allocator();

    var app: App = .init;
    var renderer = zterm.Renderer.Buffered.init(allocator);
    defer renderer.deinit();

    var element_wrapper = ExampleElement{};
    const element = element_wrapper.element();

    var container = try App.Container.init(allocator, .{
        .border = .{ .separator = .{ .enabled = true } },
        .layout = .{
            .gap = 2,
            .padding = .all(5),
            .direction = .horizontal,
        },
    }, element);
    var box = try App.Container.init(allocator, .{
        .rectangle = .{ .fill = .blue },
        .layout = .{
            .gap = 1,
            .direction = .vertical,
            .padding = .vertical(1),
        },
    }, .{});
    try box.append(try App.Container.init(allocator, .{
        .rectangle = .{ .fill = .light_green },
    }, element));
    try box.append(try App.Container.init(allocator, .{
        .rectangle = .{ .fill = .light_green },
    }, .{}));
    try box.append(try App.Container.init(allocator, .{
        .rectangle = .{ .fill = .light_green },
    }, element));
    try container.append(box);
    try container.append(try App.Container.init(allocator, .{
        .border = .{
            .color = .light_blue,
            .sides = .vertical,
        },
    }, .{}));
    try container.append(try App.Container.init(allocator, .{
        .rectangle = .{ .fill = .blue },
    }, .{}));
    defer container.deinit(); // also de-initializes the children

    try app.start();
    defer app.stop() catch |err| log.err("Failed to stop application: {any}", .{err});

    // event loop
    while (true) {
        const event = app.nextEvent();
        log.debug("received event: {s}", .{@tagName(event)});

        switch (event) {
            .init => {
                try container.handle(event);
                continue; // do not render
            },
            .quit => break,
            .resize => |size| try renderer.resize(size),
            .key => |key| {
                if (key.matches(.{ .cp = 'q' })) app.quit();

                if (key.matches(.{ .cp = 'n', .mod = .{ .ctrl = true } })) {
                    try app.interrupt();
                    defer app.start() catch @panic("could not start app event loop");
                    var child = std.process.Child.init(&.{"hx"}, allocator);
                    _ = child.spawnAndWait() catch |err| app.postEvent(.{
                        .err = .{
                            .err = err,
                            .msg = "Spawning $EDITOR failed",
                        },
                    });
                }
            },
            // NOTE: errors could be displayed in another container in case one was received, etc. to provide the user with feedback
            .err => |err| log.err("Received {s} with message: {s}", .{ @errorName(err.err), err.msg }),
            else => {},
        }

        // NOTE: returned errors should be propagated back to the application
        container.handle(event) catch |err| app.postEvent(.{
            .err = .{
                .err = err,
                .msg = "Container Event handling failed",
            },
        });
        try renderer.render(@TypeOf(container), &container);
        try renderer.flush();
    }
}