1 const std = @import("std");
2 const lmdb = @import("lmdb");
3 const db = @import("db");
4 const http = @import("http");
9 fn users(txn: lmdb.Txn) !db.Db(UserId, User) {
10 return try db.Db(UserId, User).init(txn, "users");
12 fn user_ids(txn: lmdb.Txn) !db.Db(Username, UserId) {
13 return try db.Db(Username, UserId).init(txn, "user_ids");
15 fn sessions(txn: lmdb.Txn) !db.Db(SessionToken, UserId) {
16 return try db.Db(SessionToken, UserId).init(txn, "sessions");
18 fn posts(txn: lmdb.Txn) !db.Db(PostId, Post) {
19 return try db.Db(PostId, Post).init(txn, "posts");
31 display_name: DisplayName,
32 description: UserDescription,
33 password_hash: PasswordHash,
40 post_lists: PostListList,
61 const SavedPostList = struct {
65 const SavedUserList = struct {
71 const Kind = enum { Up, Down };
78 const Login = struct {
80 session_token: SessionToken,
82 const Name = std.BoundedArray(u8, 32);
83 const UserId = enum(u64) { _ };
84 const PostId = enum(u64) { _ };
85 const Timestamp = i64;
86 const Username = std.BoundedArray(u8, 32);
87 const DisplayName = std.BoundedArray(u8, 64);
88 const UserDescription = std.BoundedArray(u8, 1024);
89 const PasswordHash = std.BoundedArray(u8, 128);
90 const SessionToken = u64;
91 const CookieValue = std.BoundedArray(u8, 128);
92 const PostText = std.BoundedArray(u8, 1024);
93 const PostList = db.Set(PostId);
94 const UserList = db.Set(UserId);
95 const VoteList = db.SetList(UserId, Vote);
96 const PostListList = db.List(SavedPostList);
97 const UserListList = db.List(SavedUserList);
99 fn parse_enum(comptime E: type, buf: []const u8, base: u8) !E {
100 return @enumFromInt(try std.fmt.parseUnsigned(@typeInfo(E).@"enum".tag_type, buf, base));
103 // https://developer.mozilla.org/en-US/docs/Glossary/Percent-encoding
104 fn reencode(comptime T: type, text: []const u8) !T {
105 var result = try T.init(0);
107 const len = @min(text.len, 1024); // TODO: PostText length
110 while (idx < len) : (idx += 1) {
113 try result.append(' ');
114 } else if (c == '%' and idx + 2 < text.len) {
115 const allow = &[_]u8{ 0x26, 0x23, 0x3b, 0x0a };
117 const escaped_value = std.fmt.parseUnsigned(u8, text[idx + 1 .. idx + 3], 16) catch continue;
119 if (escaped_value == 0x0d) {
120 try std.fmt.format(result.writer(), "<br />", .{});
121 } else if (std.mem.indexOfScalar(u8, allow, escaped_value) != null) {
122 try std.fmt.format(result.writer(), "{c}", .{escaped_value});
124 try std.fmt.format(result.writer(), "&#x{x};", .{escaped_value});
129 try result.append(c);
136 fn decode(text: []const u8) !std.BoundedArray(u8, 1024) {
137 var result = try std.BoundedArray(u8, 1024).init(0);
139 const max_len = @min(text.len, 1024); // TODO: PostText length
143 while (len < max_len and idx < text.len) : ({
149 try result.append(' ');
150 } else if (c == '%') {
151 if (idx + 2 < text.len) {
152 try std.fmt.format(result.writer(), "{c}", .{try std.fmt.parseUnsigned(u8, text[idx + 1 .. idx + 3], 16)});
156 try result.append(c);
163 const Chirp = struct {
164 const PostsPerPage = 10;
165 const UsersPerPage = 10;
166 var HashBuffer = std.mem.zeroes([1024 * 1024 * 50]u8);
168 pub fn hash_password(password: []const u8) !PasswordHash {
169 var hash_buffer = try PasswordHash.init(128);
171 // TODO: choose buffer size
172 var alloc = std.heap.FixedBufferAllocator.init(&HashBuffer);
174 // TODO: choose limits
175 const result = try std.crypto.pwhash.argon2.strHash(password, .{
176 .allocator = alloc.allocator(),
177 .params = std.crypto.pwhash.argon2.Params.owasp_2id,
178 }, hash_buffer.slice());
180 try hash_buffer.resize(result.len);
185 pub fn verify_password(password: []const u8, hash: PasswordHash) bool {
186 var alloc = std.heap.FixedBufferAllocator.init(&HashBuffer);
188 if (std.crypto.pwhash.argon2.strVerify(hash.constSlice(), password, .{
189 .allocator = alloc.allocator(),
193 std.debug.print("verify error: {}\n", .{err});
198 pub fn register_user(env: lmdb.Env, username: []const u8, password: []const u8) !bool {
199 const username_array = try Username.fromSlice(username);
200 const display_name = try DisplayName.fromSlice(username);
202 const txn = try env.txn();
203 defer txn.commit() catch |err| {
204 std.debug.print("error registering user: {}\n", .{err});
207 const users = try Db.users(txn);
208 const user_ids = try Db.user_ids(txn);
210 if (try user_ids.has(username_array)) {
213 const user_id = try db.Prng.gen(users.dbi, UserId);
215 try users.put(user_id, User{
217 .name = username_array,
218 .display_name = display_name,
219 .description = try UserDescription.init(0),
220 .password_hash = try hash_password(password),
221 .posts = try PostList.init(txn),
222 .following = try UserList.init(txn),
223 .followers = try UserList.init(txn),
224 .post_lists = try PostListList.init(txn),
225 .feeds = try UserListList.init(txn),
228 try user_ids.put(username_array, user_id);
236 username: []const u8,
237 password: []const u8,
239 const username_array = try Username.fromSlice(username);
241 const txn = try env.txn();
242 defer txn.commit() catch {};
244 const user_ids = try Db.user_ids(txn);
245 const user_id = try user_ids.get(username_array);
246 std.debug.print("user logging in, id: {}\n", .{user_id});
248 const users = try Db.users(txn);
249 const user = try users.get(user_id);
251 if (verify_password(password, user.password_hash)) {
252 const sessions = try Db.sessions(txn);
253 const session_token = try db.Prng.gen(sessions.dbi, SessionToken);
254 try sessions.put(session_token, user_id);
255 return session_token;
257 return error.IncorrectPassword;
261 fn logout_user(env: lmdb.Env, session_token: SessionToken) !void {
262 const txn = try env.txn();
263 defer txn.commit() catch {};
265 const sessions = try Db.sessions(txn);
266 try sessions.del(session_token);
269 fn append_post(env: lmdb.Env, user_id: UserId, post_list: PostList, parent_id: ?PostId, quote_id: ?PostId, text: []const u8) !PostId {
270 var post_id: PostId = undefined;
272 // TODO: do this in one commit
274 var txn: lmdb.Txn = undefined;
278 defer txn.commit() catch {};
280 const posts = try Db.posts(txn);
281 post_id = try db.Prng.gen(posts.dbi, PostId);
283 const decoded_text = try reencode(PostText, text);
284 try posts.put(post_id, Post{
286 .parent_id = parent_id,
287 .quote_id = quote_id,
289 .time = std.time.timestamp(),
290 .votes = try VoteList.init(txn),
291 .comments = try PostList.init(txn),
292 .quotes = try PostList.init(txn),
293 .text = decoded_text,
298 // append to user's posts
300 defer txn.commit() catch {};
302 var posts_view = try post_list.open(txn);
303 try posts_view.append(post_id);
306 if (quote_id != null) {
308 defer txn.commit() catch {};
310 const posts = try Db.posts(txn);
311 const quote_post = try posts.get(quote_id.?);
312 var quotes = try quote_post.quotes.open(txn);
313 try quotes.append(post_id);
319 fn post(env: lmdb.Env, user_id: UserId, text: []const u8) !void {
320 var txn = try env.txn();
321 const users = try Db.users(txn);
322 const user = try users.get(user_id);
325 const post_id = try append_post(env, user_id, user.posts, null, null, text);
329 fn comment(env: lmdb.Env, user_id: UserId, parent_post_id: PostId, text: []const u8) !void {
330 var txn = try env.txn();
331 const users = try Db.users(txn);
332 const user = try users.get(user_id);
334 const posts = try Db.posts(txn);
335 const parent_post = try posts.get(parent_post_id);
338 const post_id = try append_post(env, user_id, parent_post.comments, parent_post_id, null, text);
341 var replies_view = try user.posts.open(txn);
342 try replies_view.append(post_id);
346 fn quote(env: lmdb.Env, user_id: UserId, quote_post_id: PostId, text: []const u8) !void {
347 var txn = try env.txn();
348 const users = try Db.users(txn);
349 const user = try users.get(user_id);
352 const post_id = try append_post(env, user_id, user.posts, null, quote_post_id, text);
356 fn vote(env: lmdb.Env, post_id: PostId, user_id: UserId, kind: Vote.Kind) !void {
357 const txn = try env.txn();
358 defer txn.commit() catch {};
360 const posts = try Db.posts(txn);
362 var p = try posts.get(post_id);
363 var votes_view = try p.votes.open(txn);
367 if (try votes_view.has(user_id)) {
368 const old_vote = try votes_view.get(user_id);
370 add_vote = old_vote.kind != kind;
372 try votes_view.del(user_id);
374 switch (old_vote.kind) {
375 .Up => p.upvotes -= 1,
376 .Down => p.downvotes -= 1,
378 try posts.put(post_id, p);
382 try votes_view.append(user_id, Vote{
384 .time = std.time.timestamp(),
392 try posts.put(post_id, p);
396 fn follow(env: lmdb.Env, user_id: UserId, user_id_to_follow: UserId) !void {
397 const txn = try env.txn();
398 defer txn.commit() catch {};
400 const users = try Db.users(txn);
402 const user = try users.get(user_id);
403 const user_to_follow = try users.get(user_id_to_follow);
405 var user_following = try user.following.open(txn);
406 var user_to_follow_followers = try user_to_follow.followers.open(txn);
408 if ((user_following.has(user_id_to_follow) catch false) and (user_to_follow_followers.has(user_id) catch false)) {
409 try user_following.del(user_id_to_follow);
410 try user_to_follow_followers.del(user_id);
411 } else if (!(user_following.has(user_id_to_follow) catch true) and !(user_to_follow_followers.has(user_id) catch true)) {
412 try user_following.append(user_id_to_follow);
413 try user_to_follow_followers.append(user_id);
415 std.debug.print("Something went wrong when trying to unfollow\n", .{});
419 fn get_session_user_id(env: lmdb.Env, session_token: SessionToken) !UserId {
420 const txn = try env.txn();
423 const sessions = try Db.sessions(txn);
425 return try sessions.get(session_token);
428 fn get_user(env: lmdb.Env, user_id: UserId) !User {
429 const txn = try env.txn();
432 const users = try Db.users(txn);
433 return try users.get(user_id);
440 pub fn Paginate(comptime T: type) type {
442 const Self = @This();
444 const IterateResult = T.Base.View.Iterator.Result;
450 it: T.Base.View.Iterator,
451 starting_idx: ?T.Base.Key,
454 pub fn init(res: *http.Response, view: T.View, per_page: u64) !Self {
455 var it = view.reverse_iterator();
456 if (res.req.get_param("starting_at")) |starting_at_str| {
457 it.idx = try parse_enum(T.Base.Key, starting_at_str, 16);
463 .per_page = per_page,
465 .starting_idx = it.idx,
468 pub fn next(self: *Self) ?IterateResult {
469 if (self.it.next()) |kv| {
470 if (self.count < self.per_page) {
477 pub fn write_navigation(self: *Self) !void {
478 const next_idx = self.it.next();
480 if (self.view.base.head.last != self.starting_idx) {
481 var prev_it = self.view.iterator();
482 prev_it.idx = self.starting_idx.?;
483 var oldest_idx = self.starting_idx.?;
486 while (prev_it.next()) |kv| {
489 if (count > self.per_page) {
496 try self.res.write("<a href=\"{s}?starting_at={x}\">Prev</a> ", .{ self.res.req.target, @intFromEnum(oldest_idx) });
500 try self.res.write("<a href=\"{s}?starting_at={x}\">Next</a>", .{ self.res.req.target, @intFromEnum(kv.key) });
505 fn html_form(res: *http.Response, action: []const u8, inputs: anytype) !void {
506 try res.write("<form action=\"{s}\" method=\"post\">", .{action});
508 inline for (inputs) |input| {
509 switch (@typeInfo(@TypeOf(input))) {
511 if (s.fields.len == 3) {
512 try res.write("<{s} ", .{input[0]});
513 try res.write(input[1], input[2]);
514 try res.write("></{s}>", .{input[0]});
516 try res.write("<input ", .{});
517 try res.write(input[0], input[1]);
518 try res.write(" />", .{});
522 try res.write("<input ", .{});
523 try res.write(input, .{});
524 try res.write(" />", .{});
529 try res.write("</form>", .{});
534 const TimeStr = std.BoundedArray(u8, 256);
536 // http://howardhinnant.github.io/date_algorithms.html
537 fn time_str(_t: i64) TimeStr {
538 const t: u64 = @intCast(_t);
539 var result = TimeStr.init(0) catch unreachable;
541 const nD = @divFloor(t, std.time.s_per_day);
542 const z: u64 = nD + 719468;
543 const era: u64 = (if (z >= 0) z else z - 146096) / 146097;
544 const doe: u64 = z - era * 146097; // [0, 146096]
545 const yoe: u64 = (doe - doe / 1460 + doe / 36524 - doe / 146096) / 365; // [0, 399]
546 const Y: u64 = yoe + era * 400;
547 const doy: u64 = doe - (365 * yoe + yoe / 4 - yoe / 100); // [0, 365]
548 const mp: u64 = (5 * doy + 2) / 153; // [0, 11]
549 const D: u64 = doy - (153 * mp + 2) / 5 + 1; // [1, 31]
550 const M: u64 = if (mp < 10) mp + 3 else mp - 9;
552 const h: u64 = @divFloor(t - nD * std.time.s_per_day, std.time.s_per_hour);
553 const m: u64 = @divFloor(t - nD * std.time.s_per_day - h * std.time.s_per_hour, std.time.s_per_min);
554 const s: u64 = t - nD * std.time.s_per_day - h * std.time.s_per_hour - m * std.time.s_per_min;
556 std.fmt.format(result.writer(), "<time><span>{:0>4}-{:0>2}-{:0>2} {:0>2}:{:0>2}:{:0>2} UTC</span><script>document.currentScript.parentElement.innerHTML=new Date({}).toLocaleString()</script></time>", .{ Y, M, D, h, m, s, t * 1000 }) catch unreachable;
560 fn write_header(res: *http.Response, logged_in: ?Login) !void {
561 if (logged_in) |login| {
563 \\<a href="/">Home</a><br />
566 \\<a href="/user/{s}">Profile</a><br />
567 , .{login.user.name.constSlice()});
569 \\<a href="/post">Post</a><br />
571 try html_form(res, "/logout", .{
572 \\type="submit" value="Logout"
574 try res.write("<br /><br />", .{});
577 \\<a href="/">Home</a><br />
578 \\<form action="/register" method="post">
579 \\<input type="text" name="username" />
580 \\<input type="password" name="password" />
581 \\<input type="submit" value="Register" />
583 \\<form action="/login" method="post">
584 \\<input type="text" name="username" />
585 \\<input type="password" name="password" />
586 \\<input type="submit" value="Login" />
587 \\</form><br /><br />
591 fn write_start(res: *http.Response) !void {
596 \\<link rel="icon" href="data:image/svg+xml,<svg xmlns=%22http://www.w3.org/2000/svg%22 viewBox=%220 0 100 100%22><text y=%22.9em%22 font-size=%2290%22>🐣</text></svg>">
597 \\<meta name="viewport" content="width=device-width, initial-scale=1.0" />
600 \\ display: inline-block;
603 \\ margin:40px auto;max-width:650px;line-height:1.6;font-size:18px;color:#444;padding:0 10px;
605 \\ h1,h2,h3{line-height:1.2}
611 fn write_end(res: *http.Response) !void {
612 try res.write("</body></html>", .{});
614 fn write_post(res: *http.Response, txn: lmdb.Txn, logged_in: ?Login, post_id: PostId, options: struct { recurse: u8 = 0, show_comment_field: bool = false }) !void {
615 const posts = try Db.posts(txn);
616 const post = posts.get(post_id) catch {
617 res.redirect("/") catch {};
620 const users = try Db.users(txn);
621 const user = try users.get(post.user_id);
625 \\<span><a href="/user/{s}">{s}</a>
626 , .{ @intFromEnum(post_id), user.name.constSlice(), user.display_name.constSlice() });
627 if (post.parent_id) |id| {
628 try res.write(" <a href=\"/post/{x}\">..</a>", .{@intFromEnum(id)});
632 \\<span>{s}</span><br />
633 , .{ time_str(post.time).constSlice(), post.text.constSlice() });
635 if (logged_in != null and post.user_id == logged_in.?.user.id) {
639 \\<a href="/upvotes/{0x}">{1} Upvotes</a>
640 \\<a href="/downvotes/{0x}">{2} Downvotes</a>
643 , .{ @intFromEnum(post_id), post.upvotes, post.downvotes });
646 if (post.quote_id) |quote_id| {
647 try res.write("<div style=\"border: 1px solid black;\">", .{});
648 if (options.recurse > 0) {
649 try write_post(res, txn, logged_in, quote_id, .{ .recurse = options.recurse - 1 });
651 try res.write("<a href=\"/post/{x}\">...</a>", .{@intFromEnum(post_id)});
653 try res.write("</div>", .{});
656 const comments_view = try post.comments.open(txn);
657 const quotes_view = try post.quotes.open(txn);
658 const votes_view = try post.votes.open(txn);
661 const vote: ?Vote = if (logged_in != null and try votes_view.has(logged_in.?.user.id)) try votes_view.get(logged_in.?.user.id) else null;
663 if (vote != null and vote.?.kind == .Up) {
664 try html_form(res, "/upvote", .{
665 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
666 .{ "type=\"submit\" value=\"⬆ {}\"", .{post.upvotes} },
669 try html_form(res, "/upvote", .{
670 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
671 .{ "type=\"submit\" value=\"⇧ {}\"", .{post.upvotes} },
674 if (vote != null and vote.?.kind == .Down) {
675 try html_form(res, "/downvote", .{
676 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
677 .{ "type=\"submit\" value=\"⬇ {}\"", .{post.downvotes} },
680 try html_form(res, "/downvote", .{
681 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
682 .{ "type=\"submit\" value=\"⇩ {}\"", .{post.downvotes} },
688 \\<a href="/post/{x}">💭 {}</a>
689 , .{ @intFromEnum(post.id), comments_view.len() });
693 \\<a href="/quoted/{x}">🔁 {}</a>
694 , .{ @intFromEnum(post.id), quotes_view.len() });
697 if (logged_in) |login| {
698 const lists_view = try login.user.post_lists.open(txn);
699 try res.write("<form action=\"/list_add\" method=\"post\">", .{});
700 try res.write("<select name=\"list_id\">", .{});
701 var it = lists_view.iterator();
702 // TODO: mark lists that already contain post
703 while (it.next()) |kv| {
704 const name = kv.val.name;
705 const id = kv.val.list.base.idx.?;
706 const list_view = try kv.val.list.open(txn);
707 try res.write("<option value=\"{x}\">{s}{s}</option>", .{ id, name.constSlice(), if (list_view.has(post_id) catch false) " *" else "" });
709 try res.write("</select>", .{});
710 try res.write("<input type=\"hidden\" name=\"post_id\" value=\"{x}\" />", .{@intFromEnum(post_id)});
711 try res.write("<input type=\"submit\" value=\"Save\" />", .{});
712 try res.write("</form>", .{});
716 // TODO: maybe always show comment field and prompt for login
717 if (options.show_comment_field and logged_in != null) {
718 try res.write("<br /><br />", .{});
719 try html_form(res, "/comment", .{
720 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
721 .{ "textarea", "type=\"text\" name=\"text\" placeholder=\"Text\"", .{} },
722 "type=\"submit\" value=\"Comment\"",
724 try res.write("<br />", .{});
728 if (options.recurse > 0 and comments_view.len() > 0) {
731 \\<summary>Comments</summary>
732 , .{if (options.recurse > 1) " open" else ""});
733 try res.write("<div style=\"margin: 10px;\">", .{});
734 var it = comments_view.iterator();
736 while (it.next()) |comment_id| {
737 try write_post(res, txn, logged_in, comment_id.key, .{ .recurse = options.recurse - 1 });
738 try res.write("<br />", .{});
739 if (options.recurse == 1) {
741 if (count >= 3) break;
754 fn write_profile(res: *http.Response, txn: lmdb.Txn, logged_in: ?Login, user: User) !void {
755 const following = try user.following.open(txn);
756 const followers = try user.followers.open(txn);
759 \\<h2 style="display: inline;"><a href="/user/{s}">{s}</a></h2>
761 user.name.constSlice(), user.display_name.constSlice(),
763 if (logged_in != null and user.id != logged_in.?.user.id) {
764 const login = logged_in.?;
767 if (try followers.has(login.user.id)) {
768 try html_form(res, "/follow", .{
769 .{ "type=\"hidden\" name=\"user_id\" value=\"{x}\"", .{@intFromEnum(user.id)} },
770 \\type="submit" value="Unfollow"
773 try html_form(res, "/follow", .{
774 .{ "type=\"hidden\" name=\"user_id\" value=\"{x}\"", .{@intFromEnum(user.id)} },
775 \\type="submit" value="Follow"
780 const feeds_view = try login.user.feeds.open(txn);
781 try res.write("<form action=\"/feed_add\" method=\"post\">", .{});
782 try res.write("<select name=\"feed_id\">", .{});
783 var it = feeds_view.iterator();
784 while (it.next()) |kv| {
785 const name = kv.val.name;
786 const id = kv.val.list.base.idx.?;
787 const list_view = try kv.val.list.open(txn);
788 try res.write("<option value=\"{x}\">{s}{s}</option>", .{ id, name.constSlice(), if (list_view.has(user.id) catch false) " *" else "" });
790 try res.write("</select>", .{});
791 try res.write("<input type=\"hidden\" name=\"user_id\" value=\"{x}\" />", .{@intFromEnum(user.id)});
792 try res.write("<input type=\"submit\" value=\"Add to feed\" />", .{});
793 try res.write("</form>", .{});
796 \\ <a href="/following/{s}">{} following</a>
797 \\ <a href="/followers/{s}">{} followers</a>
800 user.name.constSlice(), following.len(),
801 user.name.constSlice(), followers.len(),
805 \\<a href="/all/{0s}">All Posts</a>
806 \\ <a href="/comments/{0s}">Comments</a>
807 \\ <a href="/quotes/{0s}">Quotes</a><br />
809 user.name.constSlice(),
812 if (logged_in != null and user.id == logged_in.?.user.id) {
814 \\<a href="/lists">Lists</a>
815 \\<a href="/feeds">Feeds</a>
816 \\<a href="/edit">Edit</a><br />
821 if (user.description.len > 0) {
823 \\<div style="padding-left: 5px; border-left: 1px solid grey;">
824 // \\« {s} »
827 , .{user.description.constSlice()});
830 try res.write("<br />", .{});
832 fn write_posts(res: *http.Response, txn: lmdb.Txn, logged_in: ?Login, post_list: PostList, options: struct {
837 const posts_view = try post_list.open(txn);
839 var paginate = try Paginate(PostList).init(res, posts_view, Chirp.PostsPerPage);
841 while (paginate.next()) |post_id| {
842 const posts = try Db.posts(txn);
843 const post = try posts.get(post_id.key);
844 if ((options.show_posts and (post.parent_id == null and post.quote_id == null)) or
845 (options.show_quotes and (post.quote_id != null)) or
846 (options.show_comments and (post.parent_id != null)))
848 try write_post(res, txn, logged_in, post_id.key, .{ .recurse = 1 });
849 try res.write("<br />", .{});
853 try paginate.write_navigation();
855 fn write_timeline(res: *http.Response, txn: lmdb.Txn, logged_in: ?Login, user_list: UserList) !void {
856 const users = try Db.users(txn);
857 const posts = try Db.posts(txn);
859 var newest_post_ids = try std.BoundedArray(PostId, 10).init(0); // TODO: TimelinePostsCount
860 var prev_newest_post: ?Post = null;
862 const following = try user_list.open(txn);
863 if (following.len() == 0) {
864 try res.write("Empty timeline (no users)", .{});
869 var newest_post: ?Post = null;
871 var following_it = following.iterator();
872 while (following_it.next()) |following_id| {
873 const followed_user = try users.get(following_id.key);
874 const followed_posts = try followed_user.posts.open(txn);
876 if (followed_posts.len() == 0) {
880 var followed_posts_it = followed_posts.reverse_iterator();
881 while (followed_posts_it.next()) |followed_post_id| {
882 const p = try posts.get(followed_post_id.key);
884 if ((prev_newest_post == null or p.time < prev_newest_post.?.time) and (newest_post == null or newest_post.?.time < p.time)) {
890 if (newest_post) |post| {
891 newest_post_ids.append(post.id) catch break;
892 prev_newest_post = post;
898 for (newest_post_ids.constSlice()) |post_id| {
899 try write_post(res, txn, logged_in, post_id, .{ .recurse = 1 });
900 try res.write("<br />", .{});
903 fn write_user(res: *http.Response, txn: lmdb.Txn, user_id: UserId) !void {
904 const users = try Db.users(txn);
905 const user = try users.get(user_id);
907 \\<a href="/user/{s}">{s}</a>
908 , .{ user.name.constSlice(), user.display_name.constSlice() });
910 fn write_votes(res: *http.Response, txn: lmdb.Txn, votes: VoteList, options: struct {
911 show_upvotes: bool = true,
912 show_downvotes: bool = true,
914 const votes_view = try votes.open(txn);
916 var paginate = try Paginate(VoteList).init(res, votes_view, Chirp.UsersPerPage);
918 while (paginate.next()) |kv| {
919 const user_id = kv.key;
922 if ((options.show_upvotes and vote.kind == .Up) or
923 (options.show_downvotes and vote.kind == .Down))
925 try write_user(res, txn, user_id);
926 try res.write(" <small>{s}</small><br />", .{time_str(vote.time).constSlice()});
930 try paginate.write_navigation();
932 fn check_login(env: lmdb.Env, req: http.Request, res: *http.Response) !?Login {
933 var result: ?Login = null;
935 if (req.get_cookie("session_token")) |session_token_str| {
936 var remove_session_token = true;
938 if (std.fmt.parseUnsigned(SessionToken, session_token_str, 16) catch null) |session_token| {
939 // const session_token = try std.fmt.parseUnsigned(SessionToken, session_token_str, 10);
940 // const session_token = std.mem.bytesToValue(SessionToken, session_token_str);
941 if (Chirp.get_session_user_id(env, session_token) catch null) |user_id| {
942 const txn = try env.txn();
944 const users = try Db.users(txn);
947 .user = try users.get(user_id),
948 .session_token = session_token,
951 remove_session_token = false;
955 if (remove_session_token) {
958 .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
969 const Self = @This();
976 fn handle(self: Self) !bool {
977 const ti = @typeInfo(Self);
978 inline for (ti.@"struct".decls) |f_decl| {
979 const has_arg = f_decl.name.len > 1 and f_decl.name[f_decl.name.len - 1] == '/';
980 const match = if (has_arg) std.mem.startsWith(u8, self.req.target, f_decl.name) else std.mem.eql(u8, self.req.target, f_decl.name);
983 const f = @field(Self, f_decl.name);
984 const fi = @typeInfo(@TypeOf(f));
985 if (fi.@"fn".params.len == 1) {
986 try @call(.auto, f, .{self});
988 const arg_type = fi.@"fn".params[1].type.?;
989 const arg_info = @typeInfo(arg_type);
990 var arg: arg_type = undefined;
991 const field = arg_info.@"struct".fields[0];
992 if (self.req.target.len <= f_decl.name.len) {
993 return error.NoArgProvided;
995 const str = self.req.target[f_decl.name.len..self.req.target.len];
996 const field_ti = @typeInfo(field.type);
998 // TODO: maybe handle BoundedArray?
1000 @field(arg, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
1003 @field(arg, field.name) = try parse_enum(field.type, str, 16);
1006 @field(arg, field.name) = str;
1010 try @call(.auto, f, .{ self, arg });
1018 pub fn @"/user/"(self: Self, args: struct { username: []const u8 }) !void {
1019 const user_ids = try Db.user_ids(self.txn);
1020 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
1021 const users = try Db.users(self.txn);
1022 const user = try users.get(user_id);
1024 try write_profile(self.res, self.txn, self.logged_in, user);
1026 try write_posts(self.res, self.txn, self.logged_in, user.posts, .{
1028 .show_quotes = false,
1029 .show_comments = false,
1033 \\<p>User not found [{}]</p>
1037 pub fn @"/comments/"(self: Self, args: struct { username: []const u8 }) !void {
1038 const user_ids = try Db.user_ids(self.txn);
1039 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
1040 const users = try Db.users(self.txn);
1041 const user = try users.get(user_id);
1043 try write_profile(self.res, self.txn, self.logged_in, user);
1045 try write_posts(self.res, self.txn, self.logged_in, user.posts, .{
1046 .show_posts = false,
1047 .show_quotes = false,
1048 .show_comments = true,
1052 \\<p>User not found [{}]</p>
1056 pub fn @"/quotes/"(self: Self, args: struct { username: []const u8 }) !void {
1057 const user_ids = try Db.user_ids(self.txn);
1058 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
1059 const users = try Db.users(self.txn);
1060 const user = try users.get(user_id);
1062 try write_profile(self.res, self.txn, self.logged_in, user);
1064 try write_posts(self.res, self.txn, self.logged_in, user.posts, .{
1065 .show_posts = false,
1066 .show_quotes = true,
1067 .show_comments = false,
1071 \\<p>User not found [{}]</p>
1075 pub fn @"/all/"(self: Self, args: struct { username: []const u8 }) !void {
1076 const user_ids = try Db.user_ids(self.txn);
1077 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
1078 const users = try Db.users(self.txn);
1079 const user = try users.get(user_id);
1081 try write_profile(self.res, self.txn, self.logged_in, user);
1083 try write_posts(self.res, self.txn, self.logged_in, user.posts, .{
1085 .show_quotes = true,
1086 .show_comments = true,
1090 \\<p>User not found [{}]</p>
1094 pub fn @"/following/"(self: Self, args: struct { username: []const u8 }) !void {
1095 const user_ids = try Db.user_ids(self.txn);
1096 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
1097 const users = try Db.users(self.txn);
1098 const user = try users.get(user_id);
1100 const following_view = try user.following.open(self.txn);
1102 var paginate = try Paginate(UserList).init(self.res, following_view, Chirp.UsersPerPage);
1105 \\<h2><a href="/user/{s}">{s}</a> follows:</h2>
1106 , .{ user.name.constSlice(), user.display_name.constSlice() });
1108 while (paginate.next()) |following_id| {
1109 const following_user = try users.get(following_id.key);
1112 \\<a href="/user/{s}">{s}</a><br />
1113 , .{ following_user.name.constSlice(), following_user.display_name.constSlice() });
1116 try paginate.write_navigation();
1119 \\<p>User not found [{}]</p>
1123 pub fn @"/followers/"(self: Self, args: struct { username: []const u8 }) !void {
1124 const user_ids = try Db.user_ids(self.txn);
1125 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
1126 const users = try Db.users(self.txn);
1127 const user = try users.get(user_id);
1129 const followers_view = try user.followers.open(self.txn);
1130 var paginate = try Paginate(UserList).init(self.res, followers_view, Chirp.UsersPerPage);
1133 \\<h2><a href="/user/{s}">{s}</a> followers:</h2>
1134 , .{ user.name.constSlice(), user.display_name.constSlice() });
1136 while (paginate.next()) |follower_id| {
1137 const follower_user = try users.get(follower_id.key);
1140 \\<a href="/user/{s}">{s}</a><br />
1141 , .{ follower_user.name.constSlice(), follower_user.display_name.constSlice() });
1144 try paginate.write_navigation();
1147 \\<p>User not found [{}]</p>
1151 pub fn @"/post/"(self: Self, args: struct { post_id: PostId }) !void {
1152 try write_post(self.res, self.txn, self.logged_in, args.post_id, .{
1153 .recurse = 3, // TODO: factor out
1154 .show_comment_field = true,
1157 pub fn @"/upvotes/"(self: Self, args: struct { post_id: PostId }) !void {
1158 const posts = try Db.posts(self.txn);
1159 const post = try posts.get(args.post_id);
1160 try self.res.write("{} upvotes:<br />", .{post.upvotes});
1161 try write_votes(self.res, self.txn, post.votes, .{});
1163 pub fn @"/quoted/"(self: Self, args: struct { post_id: PostId }) !void {
1164 const posts = try Db.posts(self.txn);
1165 const post = try posts.get(args.post_id);
1167 const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
1169 if (self.logged_in != null) {
1170 try html_form(self.res, "/quote", .{
1171 .{ "type=\"hidden\" name=\"referer\" value=\"{s}\"", .{referer} },
1172 .{ "type=\"hidden\" name=\"post_id\" value=\"{x}\"", .{@intFromEnum(post.id)} },
1173 .{ "textarea", "type=\"text\" name=\"text\" placeholder=\"Text\" autofocus", .{} },
1174 "type=\"submit\" value=\"Quote\"",
1176 try self.res.write("<br />", .{});
1179 // TODO: show all bc this only contains quotes?
1180 try write_posts(self.res, self.txn, self.logged_in, post.quotes, .{
1181 .show_posts = false,
1182 .show_quotes = true,
1183 .show_comments = false,
1186 pub fn @"/list/"(self: Self, args: struct { list_id: PostList.Base.Index }) !void {
1187 try write_posts(self.res, self.txn, self.logged_in, PostList{ .base = .{ .idx = args.list_id } }, .{
1189 .show_quotes = true,
1190 .show_comments = true,
1193 pub fn @"/lists"(self: Self) !void {
1194 if (self.logged_in) |login| {
1195 const post_lists_view = try login.user.post_lists.open(self.txn);
1197 try html_form(self.res, "/create_list", .{
1198 "type=\"text\" name=\"name\"",
1199 "type=\"submit\" value=\"Add\"",
1202 try self.res.write("<br /><br />", .{});
1204 var it = post_lists_view.iterator();
1205 while (it.next()) |kv| {
1206 const name = kv.val.name;
1207 const post_list = kv.val.list;
1209 \\<a href="/list/{x}">{s}</a>
1210 , .{ post_list.base.idx.?, name.constSlice() });
1211 try html_form(self.res, "/delete_list", .{
1212 .{ "type=\"hidden\" name=\"list_id\" value=\"{x}\"", .{kv.key} },
1213 "type=\"submit\" value=\"Delete\"",
1215 try self.res.write("<br />", .{});
1218 try self.res.write("not logged in", .{});
1221 pub fn @"/feed/"(self: Self, args: struct { feed_id: UserList.Base.Index }) !void {
1222 try write_timeline(self.res, self.txn, self.logged_in, UserList{ .base = .{ .idx = args.feed_id } });
1224 pub fn @"/feeds"(self: Self) !void {
1225 if (self.logged_in) |login| {
1226 const feeds_view = try login.user.feeds.open(self.txn);
1228 try html_form(self.res, "/create_feed", .{
1229 "type=\"text\" name=\"name\"",
1230 "type=\"submit\" value=\"Add\"",
1233 try self.res.write("<br /><br />", .{});
1235 var it = feeds_view.iterator();
1236 while (it.next()) |kv| {
1237 const name = kv.val.name;
1238 const user_list = kv.val.list;
1240 \\<a href="/feed/{x}">{s}</a>
1241 , .{ user_list.base.idx.?, name.constSlice() });
1242 try html_form(self.res, "/delete_feed", .{
1243 .{ "type=\"hidden\" name=\"list_id\" value=\"{x}\"", .{kv.key} },
1244 "type=\"submit\" value=\"Delete\"",
1246 try self.res.write("<br />", .{});
1249 try self.res.write("not logged in", .{});
1252 pub fn @"/post"(self: Self) !void {
1253 if (self.logged_in) |login| {
1255 const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
1257 try html_form(self.res, "/post", .{
1258 .{ "type=\"hidden\" name=\"referer\" value=\"{s}\"", .{referer} },
1259 .{ "textarea", "type=\"text\" name=\"text\" placeholder=\"Text\" autofocus", .{} },
1260 "type=\"submit\" value=\"Post\"",
1263 try self.res.write("not logged in", .{});
1266 pub fn @"/edit"(self: Self) !void {
1267 if (self.logged_in) |login| {
1268 try self.res.write("<br />Username: ", .{});
1269 try html_form(self.res, "/set_username", .{
1270 .{ "type=\"text\" name=\"username\" placeholder=\"{s}\"", .{login.user.name.constSlice()} },
1271 "type=\"submit\" value=\"Change\"",
1273 try self.res.write("<br />Display Name: ", .{});
1274 try html_form(self.res, "/set_display_name", .{
1275 .{ "type=\"text\" name=\"display_name\" placeholder=\"{s}\"", .{login.user.display_name.constSlice()} },
1276 "type=\"submit\" value=\"Change\"",
1278 try self.res.write("<br />Description: ", .{});
1279 try html_form(self.res, "/set_description", .{
1280 .{ "textarea", "type=\"text\" name=\"description\" placeholder=\"{s}\"", .{login.user.description.constSlice()} },
1281 "type=\"submit\" value=\"Change\"",
1283 try self.res.write("<br />Password: ", .{});
1284 try html_form(self.res, "/set_password", .{
1285 "type=\"text\" name=\"password\"",
1286 "type=\"submit\" value=\"Change\"",
1289 try self.res.write("not logged in", .{});
1292 pub fn @"/"(self: Self) !void {
1293 if (self.logged_in) |login| {
1294 try write_timeline(self.res, self.txn, self.logged_in, login.user.following);
1296 // TODO: generic home
1297 try self.res.write("Homepage", .{});
1304 const POST = struct {
1305 const Self = @This();
1309 res: *http.Response,
1312 pub fn handle(self: Self) !bool {
1313 const ti = @typeInfo(Self);
1314 inline for (ti.@"struct".decls) |f_decl| {
1315 if (std.mem.eql(u8, f_decl.name, self.req.target)) {
1316 const f = @field(Self, f_decl.name);
1317 const fi = @typeInfo(@TypeOf(f));
1318 if (fi.@"fn".params.len == 1) {
1319 _ = try @call(.auto, f, .{self});
1321 const args_type = fi.@"fn".params[fi.@"fn".params.len - 1].type.?;
1322 const argsi = @typeInfo(args_type);
1323 var args: args_type = undefined;
1324 inline for (argsi.@"struct".fields) |field| {
1325 const str = self.req.get_value(field.name) orelse return error.ArgNotFound;
1326 const field_ti = @typeInfo(field.type);
1329 @field(args, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
1332 @field(args, field.name) = try parse_enum(field.type, str, 16);
1335 @field(args, field.name) = str;
1339 try @call(.auto, f, .{ self, args });
1347 pub fn @"/register"(self: Self, args: struct { username: []const u8, password: []const u8 }) !void {
1348 // TODO: handle args not supplied
1349 std.debug.print("New user: {s} {s}\n", .{ args.username, args.password });
1350 _ = try Chirp.register_user(self.env, args.username, args.password);
1352 pub fn @"/login"(self: Self, args: struct { username: []const u8, password: []const u8 }) !void {
1353 // TODO: handle args not supplied
1354 std.debug.print("New login: {s} {s}\n", .{ args.username, args.password });
1355 if (Chirp.login_user(self.env, args.username, args.password)) |session_token| {
1356 self.res.status = .see_other;
1357 try self.res.add_header(
1359 .{ "session_token={x}; HttpOnly", .{session_token} },
1362 std.debug.print("login_user err: {}\n", .{err});
1365 pub fn @"/logout"(self: Self) !void {
1366 if (self.logged_in) |login| {
1367 try Chirp.logout_user(self.env, login.session_token);
1369 try self.res.add_header(
1371 .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
1375 pub fn @"/set_username"(self: Self, args: struct { username: []const u8 }) !void {
1376 const login = self.logged_in orelse return error.NotLoggedIn;
1377 const username = try Username.fromSlice(args.username);
1379 const txn = try self.env.txn();
1380 defer txn.commit() catch {};
1382 const user_ids = try Db.user_ids(txn);
1384 if (!try user_ids.has(username)) {
1385 try user_ids.del(login.user.name);
1386 try user_ids.put(username, login.user.id);
1388 const users = try Db.users(txn);
1389 var user = login.user;
1390 user.name = username;
1391 try users.put(login.user.id, user);
1394 pub fn @"/set_display_name"(self: Self, args: struct { display_name: []const u8 }) !void {
1395 const login = self.logged_in orelse return error.NotLoggedIn;
1396 const display_name = try DisplayName.fromSlice(args.display_name);
1398 const txn = try self.env.txn();
1399 defer txn.commit() catch {};
1401 const users = try Db.users(txn);
1402 var user = login.user;
1403 user.display_name = display_name;
1404 try users.put(login.user.id, user);
1406 pub fn @"/set_description"(self: Self, args: struct { description: []const u8 }) !void {
1407 const login = self.logged_in orelse return error.NotLoggedIn;
1408 const description = try reencode(UserDescription, args.description);
1410 const txn = try self.env.txn();
1411 defer txn.commit() catch {};
1413 const users = try Db.users(txn);
1414 var user = login.user;
1415 user.description = description;
1416 try users.put(login.user.id, user);
1418 pub fn @"/set_password"(self: Self, args: struct { password: []const u8 }) !void {
1419 const login = self.logged_in orelse return error.NotLoggedIn;
1421 const txn = try self.env.txn();
1422 defer txn.commit() catch {};
1424 const users = try Db.users(txn);
1425 var user = login.user;
1426 user.password_hash = try Chirp.hash_password(args.password);
1427 try users.put(login.user.id, user);
1429 pub fn @"/post"(self: Self) !void {
1430 if (self.logged_in) |login| {
1431 const text = self.req.get_value("text").?;
1432 const has_referer = self.req.get_value("referer");
1434 try Chirp.post(self.env, login.user.id, text);
1436 if (has_referer) |r| {
1437 const decoded = try decode(r);
1438 try self.res.redirect(decoded.constSlice());
1442 pub fn @"/comment"(self: Self) !void {
1443 if (self.logged_in) |login| {
1444 const text = self.req.get_value("text") orelse return error.NoText;
1445 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1446 const post_id = try parse_enum(PostId, post_id_str, 16);
1448 try Chirp.comment(self.env, login.user.id, post_id, text);
1451 pub fn @"/quote"(self: Self) !void {
1452 if (self.logged_in) |login| {
1453 const text = self.req.get_value("text") orelse return error.NoText;
1454 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1455 const has_referer = self.req.get_value("referer");
1457 const post_id = try parse_enum(PostId, post_id_str, 16);
1459 try Chirp.quote(self.env, login.user.id, post_id, text);
1461 if (has_referer) |r| {
1462 const decoded = try decode(r);
1463 try self.res.redirect(decoded.constSlice());
1467 // TODO: add arguments instead of parsing manually
1468 pub fn @"/create_list"(self: Self) !void {
1469 if (self.logged_in) |login| {
1470 const name_str = self.req.get_value("name") orelse return error.NoName;
1471 const name = try Name.fromSlice(name_str);
1472 // TODO: decode name
1474 var txn = try self.env.txn();
1475 const postlist = try PostList.init(txn);
1478 txn = try self.env.txn();
1479 var post_lists_view = try login.user.post_lists.open(txn);
1480 _ = try post_lists_view.append(.{ .name = name, .list = postlist });
1484 pub fn @"/delete_list"(self: Self, args: struct { list_id: PostList.Base.Index }) !void {
1485 if (self.logged_in) |login| {
1486 var post_list: ?PostList = null;
1488 const txn = try self.env.txn();
1489 defer txn.commit() catch {};
1490 var post_lists_view = try login.user.post_lists.open(txn);
1491 post_list = (try post_lists_view.get(args.list_id)).list;
1492 try post_lists_view.del(args.list_id);
1494 if (post_list != null) {
1495 const txn = try self.env.txn();
1496 defer txn.commit() catch {};
1497 var list_view = try post_list.?.open(txn);
1498 try list_view.clear();
1502 pub fn @"/list_add"(self: Self, args: struct { list_id: PostList.Base.Index, post_id: PostId }) !void {
1503 if (self.logged_in) |login| {
1506 const txn = try self.env.txn();
1507 defer txn.commit() catch {};
1509 const post_list = PostList{ .base = .{ .idx = args.list_id } };
1510 var post_list_view = try post_list.open(txn);
1511 if (try post_list_view.has(args.post_id)) {
1512 try post_list_view.del(args.post_id);
1514 try post_list_view.append(args.post_id);
1518 pub fn @"/create_feed"(self: Self) !void {
1519 if (self.logged_in) |login| {
1520 const name_str = self.req.get_value("name") orelse return error.NoName;
1521 const name = try Name.fromSlice(name_str);
1523 var txn = try self.env.txn();
1524 const userlist = try UserList.init(txn);
1527 txn = try self.env.txn();
1528 var feeds_view = try login.user.feeds.open(txn);
1529 _ = try feeds_view.append(.{ .name = name, .list = userlist });
1533 pub fn @"/delete_feed"(self: Self, args: struct { list_id: UserList.Base.Index }) !void {
1534 if (self.logged_in) |login| {
1535 var user_list: ?UserList = null;
1538 const txn = try self.env.txn();
1539 defer txn.commit() catch {};
1540 var feeds_view = try login.user.feeds.open(txn);
1541 user_list = (try feeds_view.get(args.list_id)).list;
1542 try feeds_view.del(args.list_id);
1544 if (user_list != null) {
1545 const txn = try self.env.txn();
1546 defer txn.commit() catch {};
1547 var list_view = try user_list.?.open(txn);
1548 try list_view.clear();
1552 pub fn @"/feed_add"(self: Self, args: struct { feed_id: UserList.Base.Index, user_id: UserId }) !void {
1553 if (self.logged_in) |login| {
1556 const txn = try self.env.txn();
1557 defer txn.commit() catch {};
1559 const user_list = UserList{ .base = .{ .idx = args.feed_id } };
1560 var user_list_view = try user_list.open(txn);
1561 if (try user_list_view.has(args.user_id)) {
1562 try user_list_view.del(args.user_id);
1564 try user_list_view.append(args.user_id);
1568 pub fn @"/upvote"(self: Self) !void {
1569 const login = self.logged_in orelse return error.NotLoggedIn;
1571 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1572 const post_id: PostId = @enumFromInt(try std.fmt.parseUnsigned(u64, post_id_str, 16));
1574 try Chirp.vote(self.env, post_id, login.user.id, .Up);
1576 pub fn @"/downvote"(self: Self) !void {
1577 const login = self.logged_in orelse return error.NotLoggedIn;
1579 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1580 const post_id: PostId = @enumFromInt(try std.fmt.parseUnsigned(u64, post_id_str, 16));
1582 try Chirp.vote(self.env, post_id, login.user.id, .Down);
1584 pub fn @"/follow"(self: Self) !void {
1585 const login = self.logged_in orelse return error.NotLoggedIn;
1587 const user_id_str = self.req.get_value("user_id") orelse return error.NoUserId;
1588 const user_id: UserId = @enumFromInt(try std.fmt.parseUnsigned(u64, user_id_str, 16));
1590 try Chirp.follow(self.env, login.user.id, user_id);
1592 pub fn @"/quit"(self: Self) !void {
1593 if (self.req.get_header("Referer")) |ref| {
1594 try self.res.redirect(ref);
1596 try self.res.redirect("/");
1598 try self.res.send();
1604 fn list_users(env: lmdb.Env) !void {
1605 const txn = try env.txn();
1608 const users = try Db.users(txn);
1609 var it = try users.iterator();
1611 while (it.next()) |kv| {
1613 const user = kv.val;
1614 std.debug.print("[{}] {s}\n", .{ key, user.name.constSlice() });
1617 fn list_user_ids(env: lmdb.Env) !void {
1618 const txn = try env.txn();
1621 const user_ids = try Db.user_ids(txn);
1622 var it = try user_ids.iterator();
1624 while (it.next()) |kv| {
1626 const user_id = kv.val;
1627 std.debug.print("[{s}] {}\n", .{ key.constSlice(), user_id });
1631 fn list_sessions(env: lmdb.Env) !void {
1632 const txn = try env.txn();
1635 const sessions = try Db.sessions(txn);
1636 var it = try sessions.iterator();
1638 while (it.next()) |kv| {
1640 const user_id = kv.val;
1641 std.debug.print("[{x}] {}\n", .{ key, user_id });
1645 fn list_posts(env: lmdb.Env) !void {
1646 const txn = try env.txn();
1649 const posts = try Db.posts(txn);
1650 var it = try posts.iterator();
1652 while (it.next()) |kv| {
1654 const post = kv.val;
1655 std.debug.print("[{}] {s}\n", .{ key, post.text.constSlice() });
1659 const ReqBufferSize = 4096;
1660 const ResHeadBufferSize = 1024 * 64;
1661 const ResBodyBufferSize = 1024 * 64;
1664 var req_buffer: [ReqBufferSize]u8 = undefined;
1665 var res_head_buffer: [ResHeadBufferSize]u8 = undefined;
1666 var res_body_buffer: [ResBodyBufferSize]u8 = undefined;
1668 pub fn main() !void {
1670 var server = try http.Server.init("::", 8080);
1671 defer server.deinit();
1674 var env = try lmdb.Env.open("db", 1024 * 1024 * 10);
1677 std.debug.print("Users:\n", .{});
1678 try list_users(env);
1679 std.debug.print("User IDs:\n", .{});
1680 try list_user_ids(env);
1681 std.debug.print("Sessions:\n", .{});
1682 try list_sessions(env);
1683 std.debug.print("Posts:\n", .{});
1684 try list_posts(env);
1689 const req = (server.next_request(&req_buffer) catch break) orelse break;
1690 // handle_request(env, req) catch {
1691 // try handle_error(env, req);
1693 try handle_request(env, req);
1696 // const ThreadCount = 1;
1697 // var ts: [ThreadCount]std.Thread = undefined;
1699 // for (0..ThreadCount) |i| {
1700 // ts[i] = try std.Thread.spawn(.{}, handle_connection, .{ &server, &env });
1702 // for (0..ThreadCount) |i| {
1706 std.debug.print("done\n", .{});
1709 fn handle_error(env: lmdb.Env, req: http.Request) !void {
1711 var res = http.Response.init(req, &res_head_buffer, &res_body_buffer);
1712 try write_start(&res);
1713 try res.write("Oops, something went terribly wrong there D:", .{});
1714 try write_end(&res);
1717 fn handle_request(env: lmdb.Env, req: http.Request) !void {
1718 // std.debug.print("[{}]: {s}\n", .{ req.method, req.target });
1719 // std.debug.print("[{}]: {s}\n", .{ req.method, req.head.? });
1722 var res = http.Response.init(req, &res_head_buffer, &res_body_buffer);
1724 // check session token
1725 const logged_in: ?Login = try check_login(env, req, &res);
1728 if (req.method == .GET) {
1729 try write_start(&res);
1730 try write_header(&res, logged_in);
1732 const txn = try env.txn();
1739 .logged_in = logged_in,
1741 if (try get.handle()) {} else {
1742 try res.redirect("/");
1745 try write_end(&res);
1754 .logged_in = logged_in,
1756 if (try post.handle()) {} else {
1757 try res.write("<p>[POST] {s}</p>", .{req.target});
1760 if (!res.has_header("Location")) {
1761 if (req.get_header("Referer")) |ref| {
1762 try res.redirect(ref);
1764 try res.redirect("/");