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

var read_buffer: [4096]u8 = undefined;
var write_buffer: [4096]u8 = undefined;

pub fn main() !void {
    const allocator = std.heap.page_allocator;
    const args = try std.process.argsAlloc(allocator);
    defer std.process.argsFree(allocator, args);

    if (args.len != 3) {
        std.debug.print("Usage: {s} <input_file> <output_file>\n", .{args[0]});
        return;
    }

    const input_file = args[1];
    const output_file = args[2];

    var input_stream = try std.fs.cwd().openFile(input_file, .{});
    defer input_stream.close();

    var output_stream = try std.fs.cwd().createFile(output_file, .{ .truncate = true });
    defer output_stream.close();

    var image = try zigimg.Image.fromFile(allocator, input_stream, &read_buffer);
    defer image.deinit(allocator);

    try image.writeToFilePath(allocator, output_file, &write_buffer, .{ .jpeg = .{} });

    std.debug.print("Image successfully written to {s}\n", .{output_file});

    // Test: Try to read the JPEG back to verify it's valid
    std.debug.print("Testing round-trip: reading back the JPEG...\n", .{});
    var read_file = try std.fs.cwd().openFile(output_file, .{});
    defer read_file.close();
    var read_back = try zigimg.Image.fromFile(allocator, read_file, &read_buffer);
    defer read_back.deinit(allocator);
    std.debug.print("Successfully read back JPEG: {d}x{d}, format: {s}\n", .{ read_back.width, read_back.height, @tagName(read_back.pixels) });
}