Major refactor in router.zig
This commit is contained in:
parent
789e9062b2
commit
b2430b333c
1 changed files with 214 additions and 168 deletions
382
src/router.zig
382
src/router.zig
|
|
@ -70,54 +70,117 @@ const RouteSegment = union(enum) {
|
|||
param: []const u8,
|
||||
};
|
||||
|
||||
fn RouteWithContext(comptime Context: type) type {
|
||||
return struct {
|
||||
const Self = @This();
|
||||
pub const Handler = fn (Context) void;
|
||||
// convention: return HttpError IFF a status you can't handle happens.
|
||||
// If status line/headers were written, always return void
|
||||
const HttpError = error{Http404};
|
||||
|
||||
path_segments: []const RouteSegment,
|
||||
method: http.Method,
|
||||
handler: Handler,
|
||||
|
||||
pub fn bind(method: http.Method, comptime path: []const u8, handler: Handler) Self {
|
||||
return .{ .method = method, .path_segments = splitRoutePath(path), .handler = handler };
|
||||
}
|
||||
|
||||
fn matchesPath(self: *const Self, request_path: []const u8) bool {
|
||||
var request_segments = PathIter.from(request_path);
|
||||
|
||||
for (self.path_segments) |route_seg| {
|
||||
const request_seg = request_segments.next() orelse return false;
|
||||
switch (route_seg) {
|
||||
.literal => |lit| {
|
||||
if (!ciutf8.eql(lit, request_seg)) return false;
|
||||
},
|
||||
.param => {},
|
||||
}
|
||||
}
|
||||
|
||||
return request_segments.next() == null;
|
||||
}
|
||||
};
|
||||
fn Route(comptime Context: type) type {
|
||||
return fn (Context, http.Method, []const u8) HttpError!void;
|
||||
}
|
||||
|
||||
pub fn Router(comptime Context: type) type {
|
||||
/// `makeRoute` takes a route definition and a handler of the form `fn(<Context>, <Params>) HttpError`
|
||||
/// where `Params` is a struct containing one field of type `[]const u8` for each path parameter
|
||||
///
|
||||
/// Arguments:
|
||||
/// method: The HTTP method to match
|
||||
/// path: The path spec to match against. Path segments beginning with a `:` will cause the rest of
|
||||
/// the segment to be treated as the name of a path parameter
|
||||
/// handler: The code to execute on route match. This must be a function of form `fn(<Context>, <Params>) HttpError!void`
|
||||
///
|
||||
/// Implicit Arguments:
|
||||
/// Context: the type of a user-supplied Context that is passed through the route. typically `http.Context` but
|
||||
/// made generic for ease of testing. There are no restrictions on this type
|
||||
/// Params: the type of a struct representing the path parameters expressed in `<path>`. This must be
|
||||
/// a struct, with a one-one map between fields and path parameters. Each field must be of type
|
||||
/// `[]const u8` and it must have the same name as a single path parameter.
|
||||
///
|
||||
/// Returns:
|
||||
/// A new route function of type `fn(<Context>, http.Method, []const u8) ?HttpError`. When called,
|
||||
/// this function will test the provided values against its specification. If they match, then
|
||||
/// this function will parse path parameters and <handler> will be called with the supplied
|
||||
/// context and params. If they do not match, this function will return null
|
||||
///
|
||||
/// Example:
|
||||
/// route(.GET, "/user/:id/followers", struct{
|
||||
/// fn getFollowers(ctx: http.Context, params: struct{ id: []const u8 } HttpError { ... }
|
||||
/// ).getFollowers)
|
||||
///
|
||||
fn makeRoute(
|
||||
comptime method: http.Method,
|
||||
comptime path: []const u8,
|
||||
comptime handler: anytype,
|
||||
) return_type: {
|
||||
const handler_info = @typeInfo(@TypeOf(handler));
|
||||
if (handler_info != .Fn) @compileError("Route expects a function");
|
||||
break :return_type Route(@typeInfo(@TypeOf(handler)).Fn.args[0].arg_type.?);
|
||||
} {
|
||||
const handler_args = @typeInfo(@TypeOf(handler)).Fn.args;
|
||||
if (handler_args.len != 2) @compileError("handler function must have signature fn(Context, Params) HttpError");
|
||||
if (@typeInfo(handler_args[1].arg_type.?) != .Struct) @compileError("Params in handler(Context, Params) must be struct");
|
||||
|
||||
const Context = handler_args[0].arg_type.?;
|
||||
const Params = handler_args[1].arg_type.?;
|
||||
|
||||
const params_fields = std.meta.fields(Params);
|
||||
var params_field_used = [_]bool{false} ** std.meta.fields(Params).len;
|
||||
const segments = splitRoutePath(path);
|
||||
for (segments) |seg| {
|
||||
if (seg == .param) {
|
||||
const found = for (params_fields) |f, i| {
|
||||
if (std.mem.eql(u8, seg.param, f.name)) {
|
||||
params_field_used[i] = true;
|
||||
break true;
|
||||
}
|
||||
} else false;
|
||||
|
||||
if (!found) @compileError("Params does not contain " ++ seg.param ++ " field");
|
||||
}
|
||||
}
|
||||
|
||||
for (params_fields) |f, i| {
|
||||
if (f.field_type != []const u8) @compileError("Params fields must be []const u8");
|
||||
if (!params_field_used[i]) @compileError("Params field " ++ f.name ++ " not found in path");
|
||||
}
|
||||
|
||||
return struct {
|
||||
fn func(ctx: Context, req_method: http.Method, req_path: []const u8) HttpError!void {
|
||||
if (req_method != method) return error.Http404;
|
||||
|
||||
var params: Params = undefined;
|
||||
var req_segments = PathIter.from(req_path);
|
||||
inline for (segments) |seg| {
|
||||
const req_seg = req_segments.next() orelse return error.Http404;
|
||||
var match = switch (seg) {
|
||||
.literal => |literal| ciutf8.eql(literal, req_seg),
|
||||
.param => |param| blk: {
|
||||
@field(params, param) = req_seg;
|
||||
break :blk true;
|
||||
},
|
||||
};
|
||||
|
||||
if (!match) return error.Http404;
|
||||
}
|
||||
|
||||
if (req_segments.next() != null) return error.Http404;
|
||||
|
||||
return handler(ctx, params);
|
||||
}
|
||||
}.func;
|
||||
}
|
||||
|
||||
pub fn Router(comptime Context: type, comptime routes: []const Route(Context)) type {
|
||||
return struct {
|
||||
const Self = @This();
|
||||
|
||||
pub const Route = RouteWithContext(Context);
|
||||
|
||||
routes: []const Route,
|
||||
route_404: Route.Handler,
|
||||
|
||||
pub fn dispatch(self: *const Self, method: http.Method, path: []const u8, ctx: Context) void {
|
||||
for (self.routes) |r| {
|
||||
if (method == r.method and r.matchesPath(path)) {
|
||||
return r.handler(ctx);
|
||||
}
|
||||
pub fn dispatch(_: *const Self, method: http.Method, path: []const u8, ctx: Context) HttpError!void {
|
||||
for (routes) |r| {
|
||||
return r(ctx, method, path) catch |err| switch (err) {
|
||||
error.Http404 => continue,
|
||||
else => err,
|
||||
};
|
||||
}
|
||||
|
||||
return self.route_404(ctx);
|
||||
return error.Http404;
|
||||
}
|
||||
};
|
||||
}
|
||||
|
|
@ -175,34 +238,47 @@ const _tests = struct {
|
|||
}, segments);
|
||||
}
|
||||
|
||||
test "RouteWithContext(T).matchesPath" {
|
||||
const R = RouteWithContext(Context);
|
||||
const r = R.bind(.GET, "/ab/cd", dummyHandler);
|
||||
|
||||
try std.testing.expectEqual(true, r.matchesPath("ab///cd////"));
|
||||
try std.testing.expectEqual(true, r.matchesPath("//ab///cd"));
|
||||
try std.testing.expectEqual(true, r.matchesPath("ab/cd"));
|
||||
try std.testing.expectEqual(true, r.matchesPath("/ab/cd"));
|
||||
try std.testing.expectEqual(false, r.matchesPath("/a/b/c/d"));
|
||||
try std.testing.expectEqual(false, r.matchesPath("/aa/aa"));
|
||||
try std.testing.expectEqual(false, r.matchesPath(""));
|
||||
}
|
||||
|
||||
fn CallTracker(comptime _uniq: anytype, comptime next: fn (Context) void) type {
|
||||
fn CallTracker(comptime _uniq: anytype, comptime next: anytype) type {
|
||||
_ = _uniq;
|
||||
|
||||
var ctx_type: type = undefined;
|
||||
var args_type: type = undefined;
|
||||
switch (@typeInfo(@TypeOf(next))) {
|
||||
.Fn => |func| {
|
||||
if (func.args.len != 2) @compileError("next() must take 2 arguments");
|
||||
|
||||
ctx_type = func.args[0].arg_type.?;
|
||||
args_type = func.args[1].arg_type.?;
|
||||
//if (@typeInfo(Args) != .Struct) @compileError("second argument to next() must be struct");
|
||||
},
|
||||
else => @compileError("next must be function"),
|
||||
}
|
||||
|
||||
const Context = ctx_type;
|
||||
const Args = args_type;
|
||||
|
||||
return struct {
|
||||
var calls: u32 = 0;
|
||||
var last_arg: ?Context = null;
|
||||
|
||||
fn func(ctx: Context) void {
|
||||
var last_ctx: ?Context = null;
|
||||
var last_args: ?Args = null;
|
||||
|
||||
fn func(ctx: Context, args: Args) !void {
|
||||
calls += 1;
|
||||
last_arg = ctx;
|
||||
return next(ctx);
|
||||
last_ctx = ctx;
|
||||
last_args = args;
|
||||
return next(ctx, args);
|
||||
}
|
||||
|
||||
fn expectCalledOnceWith(expected: Context) !void {
|
||||
fn expectCalledOnceWith(exp_ctx: Context, exp_args: Args) !void {
|
||||
try std.testing.expectEqual(@as(u32, 1), calls);
|
||||
try std.testing.expectEqual(expected, last_arg.?);
|
||||
try std.testing.expectEqual(exp_ctx, last_ctx.?);
|
||||
inline for (std.meta.fields(Args)) |f| {
|
||||
try std.testing.expectEqualStrings(
|
||||
@field(exp_args, f.name),
|
||||
@field(last_args.?, f.name),
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
fn expectNotCalled() !void {
|
||||
|
|
@ -211,172 +287,142 @@ const _tests = struct {
|
|||
|
||||
fn reset() void {
|
||||
calls = 0;
|
||||
last_arg = null;
|
||||
last_ctx = null;
|
||||
last_args = null;
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
const Context = u32;
|
||||
fn dummyHandler(_: Context) void {}
|
||||
|
||||
test "Router(T).dispatch" {
|
||||
const mock_a = CallTracker(.{}, dummyHandler);
|
||||
const mock_b = CallTracker(.{}, dummyHandler);
|
||||
const mock_404 = CallTracker(.{}, dummyHandler);
|
||||
|
||||
const R = Router(Context).Route;
|
||||
const routes = [_]R{
|
||||
R.bind(.GET, "/a", mock_a.func),
|
||||
R.bind(.GET, "/b", mock_b.func),
|
||||
};
|
||||
|
||||
const router = Router(Context){ .routes = &routes, .route_404 = mock_404.func };
|
||||
router.dispatch(.GET, "/a", 10);
|
||||
try mock_a.expectCalledOnceWith(10);
|
||||
try mock_b.expectNotCalled();
|
||||
try mock_404.expectNotCalled();
|
||||
mock_a.reset();
|
||||
|
||||
router.dispatch(.GET, "/b", 0);
|
||||
try mock_a.expectNotCalled();
|
||||
try mock_b.expectCalledOnceWith(0);
|
||||
try mock_404.expectNotCalled();
|
||||
const TestContext = u32;
|
||||
const DummyArgs = struct {};
|
||||
fn dummyHandler(comptime Args: type) type {
|
||||
comptime {
|
||||
return struct {
|
||||
fn func(_: TestContext, _: Args) HttpError!void {}
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
test "Router(T).dispatch 404" {
|
||||
const mock_a = CallTracker(.{}, dummyHandler);
|
||||
const mock_b = CallTracker(.{}, dummyHandler);
|
||||
const mock_404 = CallTracker(.{}, dummyHandler);
|
||||
test "Router(T).dispatch" {
|
||||
const mock_a = CallTracker(.{}, dummyHandler(DummyArgs).func);
|
||||
const mock_b = CallTracker(.{}, dummyHandler(DummyArgs).func);
|
||||
|
||||
const R = Router(Context).Route;
|
||||
const routes = [_]R{
|
||||
R.bind(.GET, "/a", mock_a.func),
|
||||
R.bind(.GET, "/b", mock_b.func),
|
||||
const routes = comptime [_]Route(TestContext){
|
||||
makeRoute(.GET, "/a", mock_a.func),
|
||||
makeRoute(.GET, "/b", mock_b.func),
|
||||
};
|
||||
|
||||
const router = Router(Context){ .routes = &routes, .route_404 = mock_404.func };
|
||||
router.dispatch(.GET, "/c", 10);
|
||||
try mock_a.expectNotCalled();
|
||||
const router = Router(TestContext, &routes){};
|
||||
_ = try router.dispatch(.GET, "/a", 10);
|
||||
try mock_a.expectCalledOnceWith(10, .{});
|
||||
try mock_b.expectNotCalled();
|
||||
try mock_404.expectCalledOnceWith(10);
|
||||
mock_404.reset();
|
||||
mock_a.reset();
|
||||
|
||||
router.dispatch(.POST, "/a", 10);
|
||||
_ = try router.dispatch(.GET, "/b", 0);
|
||||
try mock_a.expectNotCalled();
|
||||
try mock_b.expectNotCalled();
|
||||
try mock_404.expectCalledOnceWith(10);
|
||||
try mock_b.expectCalledOnceWith(0, .{});
|
||||
mock_b.reset();
|
||||
|
||||
try std.testing.expectError(error.Http404, router.dispatch(.GET, "/c", 0));
|
||||
}
|
||||
|
||||
test "Router(T).dispatch same path different methods" {
|
||||
const mock_get = CallTracker(.{}, dummyHandler);
|
||||
const mock_post = CallTracker(.{}, dummyHandler);
|
||||
const mock_404 = CallTracker(.{}, dummyHandler);
|
||||
const mock_get = CallTracker(.{}, dummyHandler(DummyArgs).func);
|
||||
const mock_post = CallTracker(.{}, dummyHandler(DummyArgs).func);
|
||||
|
||||
const R = Router(Context).Route;
|
||||
const routes = [_]R{
|
||||
R.bind(.GET, "/a", mock_get.func),
|
||||
R.bind(.POST, "/a", mock_post.func),
|
||||
const routes = comptime [_]Route(TestContext){
|
||||
makeRoute(.GET, "/a", mock_get.func),
|
||||
makeRoute(.POST, "/a", mock_post.func),
|
||||
};
|
||||
|
||||
const router = Router(Context){ .routes = &routes, .route_404 = mock_404.func };
|
||||
router.dispatch(.GET, "/a", 10);
|
||||
try mock_get.expectCalledOnceWith(10);
|
||||
const router = Router(TestContext, &routes){};
|
||||
_ = try router.dispatch(.GET, "/a", 10);
|
||||
try mock_get.expectCalledOnceWith(10, .{});
|
||||
try mock_post.expectNotCalled();
|
||||
try mock_404.expectNotCalled();
|
||||
mock_get.reset();
|
||||
|
||||
router.dispatch(.POST, "/a", 10);
|
||||
_ = try router.dispatch(.POST, "/a", 10);
|
||||
try mock_get.expectNotCalled();
|
||||
try mock_post.expectCalledOnceWith(10);
|
||||
try mock_404.expectNotCalled();
|
||||
try mock_post.expectCalledOnceWith(10, .{});
|
||||
}
|
||||
|
||||
test "Router(T).dispatch route under subpath" {
|
||||
const mock_a = CallTracker(.{}, dummyHandler);
|
||||
const mock_b = CallTracker(.{}, dummyHandler);
|
||||
const mock_404 = CallTracker(.{}, dummyHandler);
|
||||
const mock_a = CallTracker(.{}, dummyHandler(DummyArgs).func);
|
||||
const mock_b = CallTracker(.{}, dummyHandler(DummyArgs).func);
|
||||
|
||||
const R = Router(Context).Route;
|
||||
const routes = [_]R{
|
||||
R.bind(.GET, "/a", mock_a.func),
|
||||
R.bind(.GET, "/a/b", mock_b.func),
|
||||
const routes = comptime [_]Route(TestContext){
|
||||
makeRoute(.GET, "/a", mock_a.func),
|
||||
makeRoute(.GET, "/a/b", mock_b.func),
|
||||
};
|
||||
|
||||
const router = Router(Context){ .routes = &routes, .route_404 = mock_404.func };
|
||||
router.dispatch(.GET, "/a", 10);
|
||||
try mock_a.expectCalledOnceWith(10);
|
||||
const router = Router(TestContext, &routes){};
|
||||
_ = try router.dispatch(.GET, "/a", 10);
|
||||
try mock_a.expectCalledOnceWith(10, .{});
|
||||
try mock_b.expectNotCalled();
|
||||
try mock_404.expectNotCalled();
|
||||
mock_a.reset();
|
||||
|
||||
router.dispatch(.GET, "/a/b", 11);
|
||||
_ = try router.dispatch(.GET, "/a/b", 11);
|
||||
try mock_a.expectNotCalled();
|
||||
try mock_b.expectCalledOnceWith(11);
|
||||
try mock_404.expectNotCalled();
|
||||
try mock_b.expectCalledOnceWith(11, .{});
|
||||
}
|
||||
|
||||
test "Router(T).dispatch case-insensitive route" {
|
||||
const mock_a = CallTracker(.{}, dummyHandler);
|
||||
const mock_404 = CallTracker(.{}, dummyHandler);
|
||||
const mock_a = CallTracker(.{}, dummyHandler(DummyArgs).func);
|
||||
|
||||
const R = Router(Context).Route;
|
||||
const routes = [_]R{
|
||||
R.bind(.GET, "/test/a", mock_a.func),
|
||||
const routes = comptime [_]Route(TestContext){
|
||||
makeRoute(.GET, "/test/a", mock_a.func),
|
||||
};
|
||||
|
||||
const router = Router(Context){ .routes = &routes, .route_404 = mock_404.func };
|
||||
router.dispatch(.GET, "/TEST/A", 10);
|
||||
try mock_a.expectCalledOnceWith(10);
|
||||
try mock_404.expectNotCalled();
|
||||
const router = Router(TestContext, &routes){};
|
||||
_ = try router.dispatch(.GET, "/TEST/A", 10);
|
||||
try mock_a.expectCalledOnceWith(10, .{});
|
||||
mock_a.reset();
|
||||
|
||||
router.dispatch(.GET, "/TesT/a", 11);
|
||||
try mock_a.expectCalledOnceWith(11);
|
||||
try mock_404.expectNotCalled();
|
||||
_ = try router.dispatch(.GET, "/TesT/a", 11);
|
||||
try mock_a.expectCalledOnceWith(11, .{});
|
||||
}
|
||||
|
||||
test "Router(T).dispatch redundant /" {
|
||||
const mock_a = CallTracker(.{}, dummyHandler);
|
||||
const mock_404 = CallTracker(.{}, dummyHandler);
|
||||
const mock_a = CallTracker(.{}, dummyHandler(DummyArgs).func);
|
||||
|
||||
const R = Router(Context).Route;
|
||||
const routes = [_]R{
|
||||
R.bind(.GET, "/test/a", mock_a.func),
|
||||
const routes = comptime [_]Route(TestContext){
|
||||
makeRoute(.GET, "/test/a", mock_a.func),
|
||||
};
|
||||
|
||||
const router = Router(Context){ .routes = &routes, .route_404 = mock_404.func };
|
||||
router.dispatch(.GET, "/test//a", 10);
|
||||
try mock_a.expectCalledOnceWith(10);
|
||||
try mock_404.expectNotCalled();
|
||||
const router = Router(TestContext, &routes){};
|
||||
_ = try router.dispatch(.GET, "/test//a", 10);
|
||||
try mock_a.expectCalledOnceWith(10, .{});
|
||||
mock_a.reset();
|
||||
|
||||
router.dispatch(.GET, "/test/a/", 11);
|
||||
try mock_a.expectCalledOnceWith(11);
|
||||
try mock_404.expectNotCalled();
|
||||
_ = try router.dispatch(.GET, "//test///////////a////", 11);
|
||||
try mock_a.expectCalledOnceWith(11, .{});
|
||||
mock_a.reset();
|
||||
|
||||
router.dispatch(.GET, "test/a/", 12);
|
||||
try mock_a.expectCalledOnceWith(12);
|
||||
try mock_404.expectNotCalled();
|
||||
_ = try router.dispatch(.GET, "test/a", 12);
|
||||
try mock_a.expectCalledOnceWith(12, .{});
|
||||
}
|
||||
|
||||
test "Router(T).dispatch with variables" {
|
||||
const mock_a = CallTracker(.{}, dummyHandler);
|
||||
const mock_404 = CallTracker(.{}, dummyHandler);
|
||||
const mock_a = CallTracker(.{}, dummyHandler(struct { id: []const u8 }).func);
|
||||
const mock_b = CallTracker(.{}, dummyHandler(struct { a_id: []const u8, b_id: []const u8 }).func);
|
||||
|
||||
const R = Router(Context).Route;
|
||||
const routes = [_]R{
|
||||
R.bind(.GET, "/test/:id/abcd", mock_a.func),
|
||||
const routes = comptime [_]Route(TestContext){
|
||||
makeRoute(.GET, "/test/:id/abcd", mock_a.func),
|
||||
makeRoute(.GET, "/test/:a_id/abcd/:b_id", mock_b.func),
|
||||
};
|
||||
|
||||
const router = Router(Context){ .routes = &routes, .route_404 = mock_404.func };
|
||||
router.dispatch(.GET, "/test/lskdjflsdjfksld/abcd", 10);
|
||||
try mock_a.expectCalledOnceWith(10);
|
||||
try mock_404.expectNotCalled();
|
||||
const router = Router(TestContext, &routes){};
|
||||
_ = try router.dispatch(.GET, "/test/xyz/abcd", 10);
|
||||
try mock_a.expectCalledOnceWith(10, .{ .id = "xyz" });
|
||||
try mock_b.expectNotCalled();
|
||||
mock_a.reset();
|
||||
|
||||
router.dispatch(.GET, "/test//abcd", 10);
|
||||
try std.testing.expectError(error.Http404, router.dispatch(.GET, "/test//abcd", 10));
|
||||
try mock_a.expectNotCalled();
|
||||
try mock_404.expectCalledOnceWith(10);
|
||||
try mock_b.expectNotCalled();
|
||||
|
||||
_ = try router.dispatch(.GET, "/test/xyz/abcd/zyx", 10);
|
||||
try mock_a.expectNotCalled();
|
||||
try mock_b.expectCalledOnceWith(10, .{ .a_id = "xyz", .b_id = "zyx" });
|
||||
}
|
||||
};
|
||||
|
|
|
|||
Loading…
Reference in a new issue