diff --git a/build.zig b/build.zig index c7efd86..0ccbb6f 100644 --- a/build.zig +++ b/build.zig @@ -34,7 +34,7 @@ pub fn build(b: *std.build.Builder) !void { .{ .name = "hello", .src = "examples/hello/hello.zig" }, .{ .name = "https", .src = "examples/https/https.zig" }, .{ .name = "hello2", .src = "examples/hello2/hello2.zig" }, - .{ .name = "hello3", .src = "examples/hello3/hello3.zig" }, + .{ .name = "simple_router", .src = "examples/simple_router/simple_router.zig" }, .{ .name = "routes", .src = "examples/routes/routes.zig" }, .{ .name = "serve", .src = "examples/serve/serve.zig" }, .{ .name = "hello_json", .src = "examples/hello_json/hello_json.zig" }, diff --git a/examples/hello3/hello3.zig b/examples/simple_router/simple_router.zig similarity index 100% rename from examples/hello3/hello3.zig rename to examples/simple_router/simple_router.zig diff --git a/src/router.zig b/src/router.zig index 387e4da..25886ea 100644 --- a/src/router.zig +++ b/src/router.zig @@ -7,71 +7,54 @@ const RouterError = error{ EmptyPath, }; -// inline closure for RequestFn with self argument -pub inline fn RequestHandler(self: anytype, func: *const fn (@TypeOf(self), zap.Request) void) *const fn (zap.Request) void { - return (opaque { - var hidden_self: @TypeOf(self) = undefined; - var hidden_func: *const fn (@TypeOf(self), zap.Request) void = undefined; - pub fn init(h_self: @TypeOf(self), h_func: *const fn (@TypeOf(self), zap.Request) void) *const @TypeOf(run) { - hidden_self = h_self; - hidden_func = h_func; - return &run; - } +const Self = @This(); - fn run(req: zap.Request) void { - hidden_func(hidden_self, req); - } - }).init(self, func); -} - -pub const RouterOptions = struct { +pub const Options = struct { not_found: ?zap.HttpRequestFn = null, }; -pub const Router = struct { - const Self = @This(); +routes: std.StringHashMap(zap.HttpRequestFn), +not_found: ?zap.HttpRequestFn, - routes: std.StringHashMap(zap.HttpRequestFn), - not_found: ?zap.HttpRequestFn, +pub fn init(allocator: Allocator, options: Options) Self { + return .{ + .routes = std.StringHashMap(zap.HttpRequestFn).init(allocator), - pub fn init(allocator: Allocator, options: RouterOptions) Self { - return .{ - .routes = std.StringHashMap(zap.HttpRequestFn).init(allocator), + .not_found = options.not_found, + }; +} - .not_found = options.not_found, - }; - } +pub fn deinit(self: *Self) void { + self.routes.deinit(); +} - pub fn deinit(self: *Self) void { - self.routes.deinit(); +pub fn handle_func(self: *Self, path: []const u8, h: zap.HttpRequestFn) !void { + if (path.len == 0) { + return RouterError.EmptyPath; } - pub fn handle_func(self: *Self, path: []const u8, h: zap.HttpRequestFn) !void { - if (path.len == 0) { - return RouterError.EmptyPath; - } + const route = self.routes.get(path); - const route = self.routes.get(path); + if (route != null) { + return RouterError.AlreadyExists; + } - if (route != null) { - return RouterError.AlreadyExists; - } + try self.routes.put(path, h); +} - try self.routes.put(path, h); - } +pub fn serve(self: *Self, r: zap.Request) void { + const path = if (r.path) |p| p else "/"; - pub fn serve(self: *Self, r: zap.Request) void { - var route = self.routes.get(r.path.?); + var route = self.routes.get(path); - if (route) |handler| { - handler(r); - } else if (self.not_found) |handler| { - // not found handler - handler(r); - } else { - // default 404 output - r.setStatus(.not_found); - r.sendBody("404 Not Found") catch return; - } + if (route) |handler| { + handler(r); + } else if (self.not_found) |handler| { + // not found handler + handler(r); + } else { + // default 404 output + r.setStatus(.not_found); + r.sendBody("404 Not Found") catch return; } -}; +} diff --git a/src/util.zig b/src/util.zig index d9ff0db..21e9eaf 100644 --- a/src/util.zig +++ b/src/util.zig @@ -1,5 +1,23 @@ const std = @import("std"); const fio = @import("fio.zig"); +const zap = @import("zap.zig"); + +/// capture self for RequestFn signature support +pub inline fn RequestHandler(self: anytype, func: *const fn (@TypeOf(self), zap.Request) void) *const fn (zap.Request) void { + return (opaque { + var hidden_self: @TypeOf(self) = undefined; + var hidden_func: *const fn (@TypeOf(self), zap.Request) void = undefined; + pub fn init(h_self: @TypeOf(self), h_func: *const fn (@TypeOf(self), zap.Request) void) *const @TypeOf(run) { + hidden_self = h_self; + hidden_func = h_func; + return &run; + } + + fn run(req: zap.Request) void { + hidden_func(hidden_self, req); + } + }).init(self, func); +} /// Used internally: convert a FIO object into its string representation. /// note: since this is called from within request functions, we don't make diff --git a/src/zap.zig b/src/zap.zig index 5d9a91f..0414d68 100644 --- a/src/zap.zig +++ b/src/zap.zig @@ -52,7 +52,7 @@ pub const Tls = @import("tls.zig"); /// ``` pub const Endpoint = @import("endpoint.zig"); -pub usingnamespace @import("router.zig"); +pub const Router = @import("router.zig"); pub usingnamespace @import("util.zig"); pub usingnamespace @import("http.zig");