1 files changed, 16 insertions, 5 deletions
diff --git a/src/http/serve.zig b/src/http/serve.zig
index 587f4a3..036eecd 100644
--- a/src/http/serve.zig
+++ b/src/http/serve.zig
@@ -1,20 +1,30 @@
 const std = @import("std");
 
+const Response = @import("Response.zig");
 const RouteSet = @import("RouteSet.zig");
 
+pub const Options = struct {
+    address: std.Io.net.IpAddress,
+    headers: []const std.http.Header = &.{},
+};
+
 pub fn serve(
     gpa: std.mem.Allocator,
     comptime routes: RouteSet,
-    address: std.Io.net.IpAddress,
+    options: Options,
 ) !void {
     var threaded: std.Io.Threaded = .init(gpa, .{});
     var group: std.Io.Group = .init;
     const io = threaded.io();
-    var server = try address.listen(io, .{ .reuse_address = true });
+    var server = try options.address.listen(io, .{ .reuse_address = true });
+
+    const send_options: Response.SendOptions = .{
+        .headers = options.headers,
+    };
 
     while (true) {
         const stream = try server.accept(io);
-        group.async(io, handle, .{ io, gpa, stream, &routes });
+        group.async(io, handle, .{ io, gpa, stream, &routes, send_options });
     }
 }
 
@@ -23,6 +33,7 @@ pub fn handle(
     gpa: std.mem.Allocator,
     stream: std.Io.net.Stream,
     routes: *const RouteSet,
+    send_options: Response.SendOptions,
 ) void {
     var read_buffer: [1024 * 8]u8 = undefined;
     var write_buffer: [1024 * 8]u8 = undefined;
@@ -44,8 +55,8 @@ pub fn handle(
     std.debug.print("{s} {s}\n", .{ @tagName(request.head.method), request.head.target });
 
     if (routes.getInterface(request.head.target)) |interface| {
-        const message = interface.handle(gpa, arena.allocator(), io, &request);
-        message.send(io, &request) catch return;
+        const response = interface.handle(gpa, arena.allocator(), io, &request);
+        response.send(io, arena.allocator(), &request, send_options) catch return;
     } else {
         request.respond("", .{ .status = .not_found }) catch return;
     }