789 lines
29 KiB
Zig
789 lines
29 KiB
Zig
/// Middlewares are types with a method of type:
|
|
/// fn handle(
|
|
/// self: @This(),
|
|
/// request: *http.Request(< some type >),
|
|
/// response: *http.Response(< some type >),
|
|
/// context: anytype,
|
|
/// next_handler: anytype,
|
|
/// ) !void
|
|
///
|
|
/// If a middleware returns error.RouteMismatch, then it is assumed that the handler
|
|
/// did not apply to the request, and this is used by routing implementations to
|
|
/// determine when to stop attempting to match a route.
|
|
///
|
|
/// Terminal middlewares that are not implemented using other middlewares should
|
|
/// only accept a `void` value for `next_handler`.
|
|
const std = @import("std");
|
|
const util = @import("util");
|
|
const http = @import("./lib.zig");
|
|
const urlencode = @import("./urlencode.zig");
|
|
const json_utils = @import("./json.zig");
|
|
|
|
/// Takes an iterable of middlewares and chains them together.
|
|
pub fn apply(middlewares: anytype) Apply(@TypeOf(middlewares)) {
|
|
return applyInternal(middlewares, std.meta.fields(@TypeOf(middlewares)));
|
|
}
|
|
|
|
/// Helper function for the return type of `apply()`
|
|
pub fn Apply(comptime Middlewares: type) type {
|
|
return ApplyInternal(std.meta.fields(Middlewares));
|
|
}
|
|
|
|
fn ApplyInternal(comptime fields: []const std.builtin.Type.StructField) type {
|
|
if (fields.len == 0) return void;
|
|
|
|
return HandlerList(
|
|
fields[0].field_type,
|
|
ApplyInternal(fields[1..]),
|
|
);
|
|
}
|
|
|
|
fn applyInternal(middlewares: anytype, comptime fields: []const std.builtin.Type.StructField) ApplyInternal(fields) {
|
|
if (fields.len == 0) return {};
|
|
return .{
|
|
.first = @field(middlewares, fields[0].name),
|
|
.next = applyInternal(middlewares, fields[1..]),
|
|
};
|
|
}
|
|
|
|
pub fn HandlerList(comptime First: type, comptime Next: type) type {
|
|
return struct {
|
|
first: First,
|
|
next: Next,
|
|
|
|
pub fn handle(
|
|
self: @This(),
|
|
req: anytype,
|
|
res: anytype,
|
|
ctx: anytype,
|
|
next: void,
|
|
) !void {
|
|
_ = next;
|
|
return self.first.handle(req, res, ctx, self.next);
|
|
}
|
|
};
|
|
}
|
|
|
|
test "apply" {
|
|
var count: usize = 0;
|
|
const NoOp = struct {
|
|
ptr: *usize,
|
|
fn handle(self: @This(), req: anytype, res: anytype, ctx: anytype, next: anytype) !void {
|
|
self.ptr.* += 1;
|
|
if (@TypeOf(next) != void) return next.handle(req, res, ctx, {});
|
|
}
|
|
};
|
|
|
|
const middlewares = .{
|
|
NoOp{ .ptr = &count },
|
|
NoOp{ .ptr = &count },
|
|
NoOp{ .ptr = &count },
|
|
NoOp{ .ptr = &count },
|
|
};
|
|
try std.testing.expectEqual(
|
|
Apply(@TypeOf(middlewares)),
|
|
HandlerList(NoOp, HandlerList(NoOp, HandlerList(NoOp, HandlerList(NoOp, void)))),
|
|
);
|
|
|
|
try apply(middlewares).handle(.{}, .{}, .{}, {});
|
|
try std.testing.expectEqual(count, 4);
|
|
}
|
|
|
|
test "injectContextValue - chained" {
|
|
try apply(.{
|
|
injectContextValue("abcd", @as(usize, 5)),
|
|
injectContextValue("efgh", @as(usize, 10)),
|
|
injectContextValue("ijkl", @as(usize, 15)),
|
|
ExpectContext(.{ .abcd = 5, .efgh = 10, .ijkl = 15 }){},
|
|
}).handle(.{}, .{}, .{}, {});
|
|
}
|
|
|
|
fn AddUniqueField(comptime Lhs: type, comptime N: usize, comptime name: [N]u8, comptime Val: type) type {
|
|
const Ctx = @Type(.{ .Struct = .{
|
|
.layout = .Auto,
|
|
.fields = std.meta.fields(Lhs) ++ &[_]std.builtin.Type.StructField{
|
|
.{
|
|
.name = &name,
|
|
.field_type = Val,
|
|
.alignment = if (@sizeOf(Val) != 0) @alignOf(Val) else 0,
|
|
.default_value = null,
|
|
.is_comptime = false,
|
|
},
|
|
},
|
|
.decls = &.{},
|
|
.is_tuple = false,
|
|
} });
|
|
return Ctx;
|
|
}
|
|
|
|
fn AddField(comptime Lhs: type, comptime name: []const u8, comptime Val: type) type {
|
|
return AddUniqueField(Lhs, name.len, name[0..].*, Val);
|
|
}
|
|
|
|
fn addField(lhs: anytype, comptime name: []const u8, val: anytype) AddField(@TypeOf(lhs), name, @TypeOf(val)) {
|
|
var result: AddField(@TypeOf(lhs), name, @TypeOf(val)) = undefined;
|
|
inline for (std.meta.fields(@TypeOf(lhs))) |f| @field(result, f.name) = @field(lhs, f.name);
|
|
@field(result, name) = val;
|
|
return result;
|
|
}
|
|
|
|
test "addField" {
|
|
const expect = std.testing.expect;
|
|
const eql = std.meta.eql;
|
|
|
|
try expect(eql(addField(.{}, "abcd", 5), .{ .abcd = 5 }));
|
|
try expect(eql(addField(.{ .abcd = 5 }, "efgh", 10), .{ .abcd = 5, .efgh = 10 }));
|
|
try expect(eql(
|
|
addField(addField(.{}, "abcd", 5), "efgh", 10),
|
|
.{ .abcd = 5, .efgh = 10 },
|
|
));
|
|
}
|
|
|
|
/// Adds a single value to the context object
|
|
pub fn InjectContextValue(comptime name: []const u8, comptime V: type) type {
|
|
return struct {
|
|
val: V,
|
|
pub fn handle(self: @This(), req: anytype, res: anytype, ctx: anytype, next: anytype) !void {
|
|
return next.handle(req, res, addField(ctx, name, self.val), {});
|
|
}
|
|
};
|
|
}
|
|
|
|
pub fn injectContextValue(comptime name: []const u8, val: anytype) InjectContextValue(name, @TypeOf(val)) {
|
|
return .{ .val = val };
|
|
}
|
|
|
|
test "InjectContextValue" {
|
|
try injectContextValue("abcd", @as(usize, 5))
|
|
.handle(.{}, .{}, .{}, ExpectContext(.{ .abcd = 5 }){});
|
|
try injectContextValue("abcd", @as(usize, 5))
|
|
.handle(.{}, .{}, .{ .efgh = @as(usize, 10) }, ExpectContext(.{ .abcd = 5, .efgh = 10 }){});
|
|
}
|
|
|
|
fn expectDeepEquals(expected: anytype, actual: anytype) !void {
|
|
const E = @TypeOf(expected);
|
|
const A = @TypeOf(actual);
|
|
if (E == void) return std.testing.expect(A == void);
|
|
try std.testing.expect(std.meta.fields(E).len == std.meta.fields(A).len);
|
|
inline for (std.meta.fields(E)) |f| {
|
|
const e = @field(expected, f.name);
|
|
const a = @field(actual, f.name);
|
|
if (comptime std.meta.trait.isZigString(f.field_type)) {
|
|
try std.testing.expectEqualStrings(a, e);
|
|
} else {
|
|
try std.testing.expectEqual(a, e);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Helper for testing purposes
|
|
fn ExpectContext(comptime val: anytype) type {
|
|
return struct {
|
|
pub fn handle(_: @This(), _: anytype, _: anytype, ctx: anytype, _: void) !void {
|
|
try expectDeepEquals(val, ctx);
|
|
}
|
|
};
|
|
}
|
|
fn expectContext(comptime val: anytype) ExpectContext(val) {
|
|
return .{};
|
|
}
|
|
|
|
/// Catches any errors returned by the `next` chain, and passes them via context
|
|
/// to an error handler if one occurs
|
|
pub fn CatchErrors(comptime ErrorHandler: type) type {
|
|
return struct {
|
|
error_handler: ErrorHandler,
|
|
pub fn handle(self: @This(), req: anytype, res: anytype, ctx: anytype, next: anytype) !void {
|
|
return next.handle(req, res, ctx, {}) catch |err| {
|
|
return self.error_handler.handle(
|
|
req,
|
|
res,
|
|
addField(ctx, "err", err),
|
|
next,
|
|
);
|
|
};
|
|
}
|
|
};
|
|
}
|
|
|
|
pub fn catchErrors(error_handler: anytype) CatchErrors(@TypeOf(error_handler)) {
|
|
return .{ .error_handler = error_handler };
|
|
}
|
|
|
|
/// Default error handler for CatchErrors, logs the error and outputs responds with a 500 if
|
|
/// a response has not been written yet
|
|
pub const default_error_handler = struct {
|
|
fn handle(_: @This(), req: anytype, res: anytype, ctx: anytype, _: anytype) !void {
|
|
const should_log = !@import("builtin").is_test;
|
|
if (should_log) std.log.err("Error {} on uri {s}", .{ ctx.err, req.uri });
|
|
|
|
// Tell the server to close the connection after this request
|
|
res.should_close = true;
|
|
|
|
var buf: [1024]u8 = undefined;
|
|
var fba = std.heap.FixedBufferAllocator.init(&buf);
|
|
var headers = http.Fields.init(fba.allocator());
|
|
if (!res.was_opened) {
|
|
var stream = res.open(.internal_server_error, &headers) catch return;
|
|
defer stream.close();
|
|
stream.finish() catch {};
|
|
}
|
|
}
|
|
}{};
|
|
|
|
test "CatchErrors" {
|
|
const TestResponse = struct {
|
|
should_close: bool = false,
|
|
was_opened: bool = false,
|
|
|
|
test_should_open: bool,
|
|
const TestStream = struct {
|
|
fn close(_: *@This()) void {}
|
|
fn finish(_: *@This()) !void {}
|
|
};
|
|
|
|
fn open(self: *@This(), status: http.Status, _: *http.Fields) !TestStream {
|
|
self.was_opened = true;
|
|
if (!self.test_should_open) return error.ResponseOpenedTwice;
|
|
try std.testing.expectEqual(status, .internal_server_error);
|
|
return .{};
|
|
}
|
|
};
|
|
|
|
const middleware_list = apply(.{
|
|
catchErrors(default_error_handler),
|
|
struct {
|
|
fn handle(_: @This(), _: anytype, _: anytype, _: anytype, _: anytype) !void {
|
|
return error.SomeError;
|
|
}
|
|
}{},
|
|
});
|
|
|
|
var response = TestResponse{ .test_should_open = true };
|
|
try middleware_list.handle(.{ .uri = "abcd" }, &response, .{}, {});
|
|
try std.testing.expect(response.should_close);
|
|
|
|
// Test that it doesn't open a response if one was already opened
|
|
response = TestResponse{ .test_should_open = false, .was_opened = true };
|
|
try middleware_list.handle(.{ .uri = "abcd" }, &response, .{}, {});
|
|
try std.testing.expect(response.should_close);
|
|
}
|
|
|
|
/// Takes the request uri provided and splits it into "path", "query_string", and "fragment_string"
|
|
/// parts, which are placed into context.
|
|
const SplitUri = struct {
|
|
pub fn handle(_: @This(), req: anytype, res: anytype, ctx: anytype, next: anytype) !void {
|
|
var frag_split = std.mem.split(u8, req.uri, "#");
|
|
const without_fragment = frag_split.first();
|
|
const fragment = frag_split.rest();
|
|
|
|
var query_split = std.mem.split(u8, without_fragment, "?");
|
|
const path = query_split.first();
|
|
const query = query_split.rest();
|
|
|
|
const new_ctx = addField(
|
|
addField(
|
|
addField(ctx, "path", path),
|
|
"query_string",
|
|
query,
|
|
),
|
|
"fragment_string",
|
|
fragment,
|
|
);
|
|
|
|
return next.handle(
|
|
req,
|
|
res,
|
|
new_ctx,
|
|
{},
|
|
);
|
|
}
|
|
};
|
|
pub const split_uri = SplitUri{};
|
|
|
|
test "split_uri" {
|
|
const testCase = struct {
|
|
fn func(uri: []const u8, ctx: anytype, expected: anytype) !void {
|
|
const v = apply(.{
|
|
split_uri,
|
|
expectContext(expected),
|
|
});
|
|
try v.handle(.{ .uri = uri }, .{}, ctx, {});
|
|
}
|
|
}.func;
|
|
|
|
try testCase("/", .{}, .{ .path = "/", .query_string = "", .fragment_string = "" });
|
|
try testCase("", .{}, .{ .path = "", .query_string = "", .fragment_string = "" });
|
|
try testCase("/path", .{}, .{ .path = "/path", .query_string = "", .fragment_string = "" });
|
|
try testCase("?abcd=1234", .{}, .{ .path = "", .query_string = "abcd=1234", .fragment_string = "" });
|
|
try testCase("#abcd", .{}, .{ .path = "", .query_string = "", .fragment_string = "abcd" });
|
|
try testCase("/abcd/efgh?query=no#frag", .{}, .{ .path = "/abcd/efgh", .query_string = "query=no", .fragment_string = "frag" });
|
|
}
|
|
|
|
/// Routes a request between the provided routes.
|
|
///
|
|
/// CURRENTLY: Does not do this intelligently, all routing is handled by the routes themselves.
|
|
/// TODO: Consider implementing this with a hashmap?
|
|
pub fn Router(comptime Routes: type) type {
|
|
return struct {
|
|
routes: Routes,
|
|
|
|
pub fn handle(self: @This(), req: anytype, res: anytype, ctx: anytype, next: void) !void {
|
|
_ = next;
|
|
|
|
inline for (self.routes) |r| {
|
|
if (r.handle(req, res, ctx, {})) |_|
|
|
// success
|
|
return
|
|
else |err| switch (err) {
|
|
error.RouteMismatch => {},
|
|
else => return err,
|
|
}
|
|
}
|
|
|
|
return error.RouteMismatch;
|
|
}
|
|
};
|
|
}
|
|
pub fn router(routes: anytype) Router(@TypeOf(routes)) {
|
|
return Router(@TypeOf(routes)){ .routes = routes };
|
|
}
|
|
|
|
pub const PathIter = struct {
|
|
is_first: bool,
|
|
iter: std.mem.SplitIterator(u8),
|
|
|
|
pub fn from(path: []const u8) PathIter {
|
|
return .{ .is_first = true, .iter = std.mem.split(u8, path, "/") };
|
|
}
|
|
|
|
pub fn next(self: *PathIter) ?[]const u8 {
|
|
defer self.is_first = false;
|
|
while (self.iter.next()) |it| if (it.len != 0) {
|
|
return it;
|
|
};
|
|
|
|
if (self.is_first) return self.iter.rest();
|
|
|
|
return null;
|
|
}
|
|
|
|
pub fn first(self: *PathIter) []const u8 {
|
|
std.debug.assert(self.is_first);
|
|
return self.next().?;
|
|
}
|
|
|
|
pub fn rest(self: *PathIter) []const u8 {
|
|
return self.iter.rest();
|
|
}
|
|
};
|
|
|
|
test "PathIter" {
|
|
const testCase = struct {
|
|
fn case(path: []const u8, segments: []const []const u8) !void {
|
|
var iter = PathIter.from(path);
|
|
for (segments) |s| {
|
|
try std.testing.expectEqualStrings(s, iter.next() orelse return error.TestExpectedEqual);
|
|
}
|
|
try std.testing.expect(iter.next() == null);
|
|
}
|
|
}.case;
|
|
|
|
try testCase("", &.{""});
|
|
try testCase("*", &.{"*"});
|
|
try testCase("/", &.{""});
|
|
try testCase("/ab/cd", &.{ "ab", "cd" });
|
|
try testCase("/ab/cd/", &.{ "ab", "cd" });
|
|
try testCase("/ab/cd//", &.{ "ab", "cd" });
|
|
try testCase("ab", &.{"ab"});
|
|
try testCase("/ab", &.{"ab"});
|
|
try testCase("ab/", &.{"ab"});
|
|
try testCase("ab//ab//", &.{ "ab", "ab" });
|
|
}
|
|
|
|
// helper function for doing route analysis
|
|
fn pathMatches(route: []const u8, path: []const u8) bool {
|
|
var path_iter = PathIter.from(path);
|
|
var route_iter = PathIter.from(route);
|
|
while (route_iter.next()) |route_segment| {
|
|
const path_segment = path_iter.next() orelse return false;
|
|
if (route_segment.len > 0 and route_segment[0] == ':') {
|
|
// Route Argument
|
|
if (path_segment.len == 0) return false;
|
|
} else {
|
|
if (!std.ascii.eqlIgnoreCase(route_segment, path_segment)) return false;
|
|
}
|
|
}
|
|
if (path_iter.next() != null) return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
/// Handler that either calls its next middleware parameter or returns error.RouteMismatch
|
|
/// depending on if the request matches the described route.
|
|
/// Must be below `split_uri` on the middleware list.
|
|
///
|
|
/// Format:
|
|
/// Each route segment can be either a literal string or an argument. Literal strings
|
|
/// must match exactly in order to constitute a matching route. Arguments must begin with
|
|
/// the character ':', with the remainer of the segment referring to the name of the argument.
|
|
/// Argument values must be nonempty.
|
|
///
|
|
/// For example, the route "/abc/:foo/def" would match "/abc/x/def" or "/abc/blahblah/def" but
|
|
/// not "/abc//def".
|
|
pub const Route = struct {
|
|
pub const Desc = struct {
|
|
path: []const u8,
|
|
method: http.Method,
|
|
};
|
|
|
|
desc: Desc,
|
|
|
|
fn applies(self: @This(), req: anytype, ctx: anytype) bool {
|
|
if (self.desc.method != req.method) return false;
|
|
|
|
const eff_path = if (@hasField(@TypeOf(ctx), "path"))
|
|
ctx.path
|
|
else
|
|
std.mem.sliceTo(req.uri, '?');
|
|
|
|
return pathMatches(self.desc.path, eff_path);
|
|
}
|
|
|
|
pub fn handle(self: @This(), req: anytype, res: anytype, ctx: anytype, next: anytype) !void {
|
|
return if (self.applies(req, ctx))
|
|
next.handle(req, res, ctx, {})
|
|
else
|
|
error.RouteMismatch;
|
|
}
|
|
};
|
|
|
|
test "route" {
|
|
const testCase = struct {
|
|
fn func(should_match: bool, route: Route.Desc, method: http.Method, path: []const u8) !void {
|
|
const no_op = struct {
|
|
fn handle(_: anytype, _: anytype, _: anytype, _: anytype, _: anytype) !void {}
|
|
}{};
|
|
const result = (Route{ .desc = route }).handle(.{ .method = method }, .{}, .{ .path = path }, no_op);
|
|
try if (should_match) result else std.testing.expectError(error.RouteMismatch, result);
|
|
}
|
|
}.func;
|
|
|
|
try testCase(true, .{ .method = .GET, .path = "/" }, .GET, "/");
|
|
try testCase(true, .{ .method = .GET, .path = "/" }, .GET, "");
|
|
try testCase(true, .{ .method = .GET, .path = "/abcd" }, .GET, "/abcd");
|
|
try testCase(true, .{ .method = .GET, .path = "/abcd" }, .GET, "abcd");
|
|
try testCase(true, .{ .method = .POST, .path = "/" }, .POST, "/");
|
|
try testCase(true, .{ .method = .POST, .path = "/" }, .POST, "");
|
|
try testCase(true, .{ .method = .POST, .path = "/abcd" }, .POST, "/abcd");
|
|
try testCase(true, .{ .method = .POST, .path = "/abcd" }, .POST, "abcd");
|
|
try testCase(true, .{ .method = .POST, .path = "/abcd/efgh" }, .POST, "abcd/efgh");
|
|
try testCase(true, .{ .method = .GET, .path = "/abcd/:arg" }, .GET, "abcd/efgh");
|
|
try testCase(true, .{ .method = .GET, .path = "/abcd/:arg/xyz" }, .GET, "abcd/efgh/xyz");
|
|
|
|
try testCase(false, .{ .method = .POST, .path = "/" }, .GET, "/");
|
|
try testCase(false, .{ .method = .GET, .path = "/abcd" }, .GET, "");
|
|
try testCase(false, .{ .method = .GET, .path = "/" }, .GET, "/abcd");
|
|
try testCase(false, .{ .method = .POST, .path = "/abcd/efgh" }, .POST, "efgh");
|
|
try testCase(false, .{ .method = .GET, .path = "/abcd/:arg" }, .GET, "/abcd/");
|
|
try testCase(false, .{ .method = .GET, .path = "/abcd/:arg/xyz" }, .GET, "abcd/efgh/");
|
|
try testCase(false, .{ .method = .GET, .path = "/abcd/:arg/xyz" }, .GET, "abcd/efgh/xyz/foo");
|
|
}
|
|
|
|
/// Mounts a router subtree under a given path. Middlewares further down on the list
|
|
/// are called with the path prefix specified by `route` removed from the path.
|
|
/// Must be below `split_uri` on the middleware list.
|
|
pub fn Mount(comptime route: []const u8) type {
|
|
return struct {
|
|
pub fn handle(_: @This(), req: anytype, res: anytype, ctx: anytype, next: anytype) !void {
|
|
var path_iter = PathIter.from(ctx.path);
|
|
comptime var route_iter = PathIter.from(route);
|
|
var path_unused: []const u8 = ctx.path;
|
|
|
|
inline while (comptime route_iter.next()) |route_segment| {
|
|
if (comptime route_segment.len == 0) continue;
|
|
const path_segment = path_iter.next() orelse return error.RouteMismatch;
|
|
path_unused = path_iter.rest();
|
|
if (comptime route_segment[0] == ':') {
|
|
@compileLog("Argument segments cannot be mounted");
|
|
// Route Argument
|
|
} else {
|
|
if (!std.ascii.eqlIgnoreCase(route_segment, path_segment)) return error.RouteMismatch;
|
|
}
|
|
}
|
|
|
|
var new_ctx = ctx;
|
|
new_ctx.path = path_unused;
|
|
return next.handle(req, res, new_ctx, {});
|
|
}
|
|
};
|
|
}
|
|
pub fn mount(comptime route: []const u8) Mount(route) {
|
|
return .{};
|
|
}
|
|
|
|
test "mount" {
|
|
const testCase = struct {
|
|
fn func(comptime base: []const u8, request: []const u8, comptime expected: ?[]const u8) !void {
|
|
const result = mount(base).handle(.{}, .{}, addField(.{}, "path", request), expectContext(.{ .path = expected orelse "" }));
|
|
try if (expected != null) result else std.testing.expectError(error.RouteMismatch, result);
|
|
}
|
|
}.func;
|
|
try testCase("/api/", "/api/", "");
|
|
try testCase("/api/", "/api/abcd", "abcd");
|
|
try testCase("/api/", "/api/abcd/efgh", "abcd/efgh");
|
|
try testCase("/api/", "/api/abcd/efgh/", "abcd/efgh/");
|
|
try testCase("/api/v0", "/api/v0/call", "call");
|
|
|
|
try testCase("/api/", "/web/abcd/efgh/", null);
|
|
try testCase("/api/", "/", null);
|
|
try testCase("/api/", "/ap", null);
|
|
try testCase("/api/v0", "/api/v1/", null);
|
|
}
|
|
|
|
fn parseArgsFromPath(comptime route: []const u8, comptime Args: type, path: []const u8) !Args {
|
|
var args: Args = undefined;
|
|
var path_iter = PathIter.from(path);
|
|
comptime var route_iter = PathIter.from(route);
|
|
inline while (comptime route_iter.next()) |route_segment| {
|
|
const path_segment = path_iter.next() orelse return error.RouteMismatch;
|
|
if (route_segment.len > 0 and route_segment[0] == ':') {
|
|
// route segment is an argument segment
|
|
if (path_segment.len == 0) return error.RouteMismatch;
|
|
const A = @TypeOf(@field(args, route_segment[1..]));
|
|
@field(args, route_segment[1..]) = try parseArgFromPath(A, path_segment);
|
|
} else {
|
|
if (!std.ascii.eqlIgnoreCase(route_segment, path_segment)) return error.RouteMismatch;
|
|
}
|
|
}
|
|
|
|
if (path_iter.next() != null) return error.RouteMismatch;
|
|
|
|
return args;
|
|
}
|
|
|
|
fn parseArgFromPath(comptime T: type, segment: []const u8) !T {
|
|
if (T == []const u8) return segment;
|
|
if (comptime std.meta.trait.isContainer(T) and std.meta.trait.hasFn("parse")(T)) return T.parse(segment);
|
|
if (comptime std.meta.trait.is(.Int)(T)) return std.fmt.parseInt(T, segment, 0);
|
|
|
|
@compileError("Unsupported Type " ++ @typeName(T));
|
|
}
|
|
|
|
/// Parse arguments directly the request path.
|
|
/// Must be placed after a `split_uri` middleware in order to get `path` from context.
|
|
///
|
|
/// Route arguments are specified in the same format as for Route. The name of the argument
|
|
/// refers to the field name in Args that the argument will be parsed to.
|
|
///
|
|
/// This currently works with arguments of 3 different types:
|
|
/// - integers
|
|
/// - []const u8,
|
|
/// - anything with a function of the form:
|
|
/// * T.parse([]const u8) Error!T
|
|
/// * This function cannot hold a reference to the passed string once it appears
|
|
///
|
|
/// Example:
|
|
/// ParsePathArgs("/:id/foo/:name/byrank/:rank", struct {
|
|
/// id: util.Uuid,
|
|
/// name: []const u8,
|
|
/// rank: u32,
|
|
/// })
|
|
/// Would parse a path of "/00000000-0000-0000-0000-000000000000/foo/jaina/byrank/3" into
|
|
/// .{ .id = try Uuid.parse("00000000-0000-0000-0000-000000000000"), .name = "jaina", .rank = 3 }
|
|
pub fn ParsePathArgs(comptime route: []const u8, comptime Args: type) type {
|
|
return struct {
|
|
pub fn handle(_: @This(), req: anytype, res: anytype, ctx: anytype, next: anytype) !void {
|
|
if (Args == void) return next.handle(req, res, addField(ctx, "args", {}), {});
|
|
return next.handle(
|
|
req,
|
|
res,
|
|
addField(ctx, "args", try parseArgsFromPath(route, Args, ctx.path)),
|
|
{},
|
|
);
|
|
}
|
|
};
|
|
}
|
|
pub fn parsePathArgs(comptime route: []const u8, comptime Args: type) ParsePathArgs(route, Args) {
|
|
return .{};
|
|
}
|
|
|
|
test "ParsePathArgs" {
|
|
const testCase = struct {
|
|
fn func(comptime route: []const u8, comptime Args: type, path: []const u8, expected: anytype) !void {
|
|
const check = struct {
|
|
expected: @TypeOf(expected),
|
|
path: []const u8,
|
|
fn handle(self: @This(), _: anytype, _: anytype, ctx: anytype, _: void) !void {
|
|
try expectDeepEquals(self.expected, ctx.args);
|
|
try std.testing.expectEqualStrings(self.path, ctx.path);
|
|
}
|
|
}{ .expected = expected, .path = path };
|
|
try parsePathArgs(route, Args).handle(.{}, .{}, .{ .path = path }, check);
|
|
}
|
|
}.func;
|
|
|
|
try testCase("/", void, "/", {});
|
|
try testCase("/:id", struct { id: usize }, "/3", .{ .id = 3 });
|
|
try testCase("/:str", struct { str: []const u8 }, "/abcd", .{ .str = "abcd" });
|
|
try testCase("/:id/xyz/:str", struct { id: usize, str: []const u8 }, "/3/xyz/abcd", .{ .id = 3, .str = "abcd" });
|
|
try testCase("/:id", struct { id: util.Uuid }, "/" ++ util.Uuid.nil.toCharArray(), .{ .id = util.Uuid.nil });
|
|
|
|
try std.testing.expectError(error.RouteMismatch, testCase("/:id", struct { id: usize }, "/", .{}));
|
|
try std.testing.expectError(error.RouteMismatch, testCase("/abcd/:id", struct { id: usize }, "/123", .{}));
|
|
try std.testing.expectError(error.RouteMismatch, testCase("/:id", struct { id: usize }, "/3/id/blahblah", .{ .id = 3 }));
|
|
try std.testing.expectError(error.InvalidCharacter, testCase("/:id", struct { id: usize }, "/xyz", .{}));
|
|
}
|
|
|
|
const BaseContentType = enum {
|
|
json,
|
|
url_encoded,
|
|
octet_stream,
|
|
multipart_formdata,
|
|
|
|
other,
|
|
};
|
|
|
|
fn parseBodyFromRequest(comptime T: type, content_type: ?[]const u8, reader: anytype, alloc: std.mem.Allocator) !T {
|
|
// Use json by default for now for testing purposes
|
|
const eff_type = content_type orelse "application/json";
|
|
const parser_type = matchContentType(eff_type);
|
|
const buf = try reader.readAllAlloc(alloc, 1 << 16);
|
|
defer alloc.free(buf);
|
|
|
|
switch (parser_type) {
|
|
.octet_stream, .json => {
|
|
const body = try json_utils.parse(T, buf, alloc);
|
|
defer json_utils.parseFree(body, alloc);
|
|
|
|
return try util.deepClone(alloc, body);
|
|
},
|
|
.url_encoded => return urlencode.parse(alloc, T, buf) catch |err| switch (err) {
|
|
//error.NoQuery => error.NoBody,
|
|
else => err,
|
|
},
|
|
.multipart_formdata => {
|
|
const param_string = std.mem.split(u8, eff_type, ";").rest();
|
|
const params = urlencode.parse(alloc, struct {
|
|
boundary: []const u8,
|
|
}, param_string) catch |err| return switch (err) {
|
|
//error.NoQuery => error.MissingBoundary,
|
|
else => err,
|
|
};
|
|
defer util.deepFree(alloc, params);
|
|
|
|
unreachable;
|
|
//try @import("./multipart.zig").parseFormData(params.boundary, reader, alloc);
|
|
},
|
|
else => return error.UnsupportedMediaType,
|
|
}
|
|
}
|
|
|
|
// figure out what base parser to use
|
|
fn matchContentType(hdr: []const u8) BaseContentType {
|
|
const trimmed = std.mem.sliceTo(hdr, ';');
|
|
if (std.ascii.eqlIgnoreCase(trimmed, "application/x-www-form-urlencoded")) return .url_encoded;
|
|
if (std.ascii.eqlIgnoreCase(trimmed, "application/json")) return .json;
|
|
if (std.ascii.endsWithIgnoreCase(trimmed, "+json")) return .json;
|
|
if (std.ascii.eqlIgnoreCase(trimmed, "application/octet-stream")) return .octet_stream;
|
|
if (std.ascii.eqlIgnoreCase(trimmed, "multipart/form-data")) return .multipart_formdata;
|
|
|
|
return .other;
|
|
}
|
|
|
|
/// Parses a set of body arguments from the request body based on the request's Content-Type
|
|
/// header.
|
|
///
|
|
/// The exact method for parsing depends partially on the Content-Type. json types are preferred
|
|
/// TODO: Need tests for this, including various Content-Type values
|
|
pub fn ParseBody(comptime Body: type) type {
|
|
return struct {
|
|
pub fn handle(_: @This(), req: anytype, res: anytype, ctx: anytype, next: anytype) !void {
|
|
const content_type = req.headers.get("Content-Type");
|
|
if (Body == void) {
|
|
if (content_type != null) return error.UnexpectedBody;
|
|
const new_ctx = addField(ctx, "body", {});
|
|
//if (true) @compileError("bug");
|
|
return next.handle(req, res, new_ctx, {});
|
|
}
|
|
|
|
var stream = req.body orelse return error.NoBody;
|
|
const body = try parseBodyFromRequest(Body, content_type, stream.reader(), ctx.allocator);
|
|
defer util.deepFree(ctx.allocator, body);
|
|
|
|
return next.handle(
|
|
req,
|
|
res,
|
|
addField(ctx, "body", body),
|
|
{},
|
|
);
|
|
}
|
|
};
|
|
}
|
|
pub fn parseBody(comptime Body: type) ParseBody(Body) {
|
|
return .{};
|
|
}
|
|
|
|
test "parseBodyFromRequest" {
|
|
const testCase = struct {
|
|
fn case(content_type: []const u8, body: []const u8, expected: anytype) !void {
|
|
var stream = std.io.StreamSource{ .const_buffer = std.io.fixedBufferStream(body) };
|
|
const result = try parseBodyFromRequest(@TypeOf(expected), content_type, stream.reader(), std.testing.allocator);
|
|
defer util.deepFree(std.testing.allocator, result);
|
|
|
|
try util.testing.expectDeepEqual(expected, result);
|
|
}
|
|
}.case;
|
|
|
|
const Struct = struct {
|
|
id: usize,
|
|
};
|
|
try testCase("application/json", "{\"id\": 3}", Struct{ .id = 3 });
|
|
try testCase("application/x-www-form-urlencoded", "id=3", Struct{ .id = 3 });
|
|
|
|
//try testCase("multipart/form-data; ",
|
|
//\\
|
|
//, Struct{ .id = 3 });
|
|
}
|
|
|
|
test "parseBody" {
|
|
const Struct = struct {
|
|
foo: []const u8,
|
|
};
|
|
const body =
|
|
\\{"foo": "bar"}
|
|
;
|
|
var stream = std.io.StreamSource{ .const_buffer = std.io.fixedBufferStream(body) };
|
|
var headers = http.Fields.init(std.testing.allocator);
|
|
defer headers.deinit();
|
|
|
|
try parseBody(Struct).handle(
|
|
.{ .body = @as(?std.io.StreamSource, stream), .headers = headers },
|
|
.{},
|
|
.{ .allocator = std.testing.allocator },
|
|
struct {
|
|
fn handle(_: anytype, _: anytype, _: anytype, ctx: anytype, _: void) !void {
|
|
try util.testing.expectDeepEqual(Struct{ .foo = "bar" }, ctx.body);
|
|
}
|
|
}{},
|
|
);
|
|
}
|
|
|
|
/// Parses query parameters as defined in query.zig
|
|
pub fn ParseQueryParams(comptime QueryParams: type) type {
|
|
return struct {
|
|
pub fn handle(_: @This(), req: anytype, res: anytype, ctx: anytype, next: anytype) !void {
|
|
if (QueryParams == void) return next.handle(req, res, addField(ctx, "query_params", {}), {});
|
|
const query = try urlencode.parse(ctx.allocator, QueryParams, ctx.query_string);
|
|
defer util.deepFree(ctx.allocator, query);
|
|
|
|
return next.handle(
|
|
req,
|
|
res,
|
|
addField(ctx, "query_params", query),
|
|
{},
|
|
);
|
|
}
|
|
};
|
|
}
|