1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
const httpz = @import("httpz");
const std = @import("std");
const zqlite = @import("zqlite");
const common = @import("common.zig");
const static = @import("static.zig");
const users_repo = @import("../repos/users_repo.zig");
pub const RouteData = struct {
is_public: bool,
};
pub const Handler = struct {
conn: zqlite.Conn,
secure_cookies: bool,
pub fn dispatch(self: *Handler, action: httpz.Action(*common.Env), req: *httpz.Request, res: *httpz.Response) !void {
var user: ?users_repo.User = null;
if (!is_route_public(req)) {
const cookies = req.cookies();
const login_token = cookies.get("token") orelse return common.ServiceError.Forbidden;
user = try users_repo.get_user(res.arena, self.conn, login_token) orelse return common.ServiceError.Forbidden;
}
var env = common.Env{ .conn = self.conn, .secure_cookies = self.secure_cookies, .user = user };
try action(&env, req, res);
}
pub fn notFound(handler: *Handler, req: *httpz.Request, res: *httpz.Response) !void {
const path = req.url.path;
if (path.len >= 5 and std.mem.eql(u8, path[0..5], "/api/")) {
return common.ServiceError.NotFound;
} else {
// non API route, let client router take care of that
var env = common.Env{ .conn = handler.conn, .secure_cookies = handler.secure_cookies, .user = null };
try static.index(&env, req, res);
}
}
pub fn uncaughtError(_: *Handler, req: *httpz.Request, res: *httpz.Response, err: anyerror) void {
switch (err) {
common.ServiceError.BadRequest => error_response(res, 400, "Bad Request"),
common.ServiceError.NotFound => error_response(res, 404, "Not Found"),
common.ServiceError.Forbidden => error_response(res, 403, "Forbidden"),
else => {
std.debug.print("Internal Server Error at {s}: {}\n", .{ req.url.path, err });
error_response(res, 500, "Internal Server Error");
},
}
}
};
fn is_route_public(req: *httpz.Request) bool {
if (req.route_data) |rd| {
const route_data: *const RouteData = @ptrCast(@alignCast(rd));
return route_data.is_public;
} else {
return false;
}
}
fn error_response(res: *httpz.Response, code: u16, message: []const u8) void {
res.status = code;
res.json(.{ .message = message }, .{}) catch {
res.body = message;
};
}
|
