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 password_hash: PasswordHash,
38 post_lists: PostListList,
59 const SavedPostList = struct {
63 const SavedUserList = struct {
69 const Kind = enum { Up, Down };
76 const Login = struct {
78 session_token: SessionToken,
80 const Name = std.BoundedArray(u8, 32);
81 const UserId = enum(u64) { _ };
82 const PostId = enum(u64) { _ };
83 const Timestamp = i64;
84 const Username = std.BoundedArray(u8, 32);
85 const DisplayName = std.BoundedArray(u8, 64);
86 const PasswordHash = std.BoundedArray(u8, 128);
87 const SessionToken = u64;
88 const CookieValue = std.BoundedArray(u8, 128);
89 const PostText = std.BoundedArray(u8, 1024);
90 const PostList = db.Set(PostId);
91 const UserList = db.Set(UserId);
92 const VoteList = db.SetList(UserId, Vote);
93 const PostListList = db.List(SavedPostList);
94 const UserListList = db.List(SavedUserList);
96 fn parse_enum(comptime E: type, buf: []const u8, base: u8) !E {
97 return @enumFromInt(try std.fmt.parseUnsigned(@typeInfo(E).Enum.tag_type, buf, base));
100 // https://developer.mozilla.org/en-US/docs/Glossary/Percent-encoding
101 fn reencode(text: []const u8) !PostText {
102 var result = try PostText.init(0);
104 const len = @min(text.len, 1024); // TODO: PostText length
107 while (idx < len) : (idx += 1) {
110 try result.append(' ');
111 } else if (c == '%') {
112 // special case of &#...
113 // assume only &#, no &#x
114 if (idx + 6 < text.len and std.mem.eql(u8, text[idx .. idx + 6], "%26%23")) {
115 const num_start = idx + 6;
116 var num_end = num_start;
117 while (num_end < text.len and std.ascii.isDigit(text[num_end])) {
121 if (num_end + 2 < text.len and
122 text[num_end] == '%' and
123 text[num_end + 1] == '3' and
124 std.ascii.toLower(text[num_end + 2]) == 'b')
126 try std.fmt.format(result.writer(), "&#{s};", .{text[num_start..num_end]});
132 try std.fmt.format(result.writer(), "&#x{s};", .{text[idx + 1 .. idx + 3]});
135 try result.append(c);
142 fn decode(text: []const u8) !std.BoundedArray(u8, 1024) {
143 var result = try std.BoundedArray(u8, 1024).init(0);
145 const max_len = @min(text.len, 1024); // TODO: PostText length
149 while (len < max_len and idx < text.len) : ({
155 try result.append(' ');
156 } else if (c == '%') {
157 if (idx + 2 < text.len) {
158 try std.fmt.format(result.writer(), "{c}", .{try std.fmt.parseUnsigned(u8, text[idx + 1 .. idx + 3], 16)});
162 try result.append(c);
169 const Chirp = struct {
170 pub fn hash_password(password: []const u8) !PasswordHash {
171 var hash_buffer = try PasswordHash.init(128);
173 // TODO: choose buffer size
174 // TODO: dont allocate on stack, maybe zero memory?
175 var buffer: [1024 * 10]u8 = undefined;
176 var alloc = std.heap.FixedBufferAllocator.init(&buffer);
178 // TODO: choose limits
179 const result = try std.crypto.pwhash.argon2.strHash(password, .{
180 .allocator = alloc.allocator(),
181 .params = std.crypto.pwhash.argon2.Params.fromLimits(1000, 1024),
182 }, hash_buffer.slice());
184 try hash_buffer.resize(result.len);
189 pub fn verify_password(password: []const u8, hash: PasswordHash) bool {
190 var buffer: [1024 * 10]u8 = undefined;
191 var alloc = std.heap.FixedBufferAllocator.init(&buffer);
193 if (std.crypto.pwhash.argon2.strVerify(hash.constSlice(), password, .{
194 .allocator = alloc.allocator(),
198 std.debug.print("verify error: {}\n", .{err});
203 pub fn register_user(env: lmdb.Env, username: []const u8, password: []const u8) !bool {
204 const username_array = try Username.fromSlice(username);
205 const display_name = try DisplayName.fromSlice(username);
207 const txn = try env.txn();
208 defer txn.commit() catch |err| {
209 std.debug.print("error registering user: {}\n", .{err});
212 const users = try Db.users(txn);
213 const user_ids = try Db.user_ids(txn);
215 if (try user_ids.has(username_array)) {
218 const user_id = try db.Prng.gen(users.dbi, UserId);
220 try users.put(user_id, User{
222 .name = username_array,
223 .display_name = display_name,
224 .password_hash = try hash_password(password),
225 .posts = try PostList.init(txn),
226 .replies = try PostList.init(txn),
227 .following = try UserList.init(txn),
228 .followers = try UserList.init(txn),
229 .post_lists = try PostListList.init(txn),
230 .feeds = try UserListList.init(txn),
233 try user_ids.put(username_array, user_id);
241 username: []const u8,
242 password: []const u8,
244 const username_array = try Username.fromSlice(username);
246 const txn = try env.txn();
247 defer txn.commit() catch {};
249 const user_ids = try Db.user_ids(txn);
250 const user_id = try user_ids.get(username_array);
251 std.debug.print("user logging in, id: {}\n", .{user_id});
253 const users = try Db.users(txn);
254 const user = try users.get(user_id);
256 if (verify_password(password, user.password_hash)) {
257 const sessions = try Db.sessions(txn);
258 const session_token = try db.Prng.gen(sessions.dbi, SessionToken);
259 try sessions.put(session_token, user_id);
260 return session_token;
262 return error.IncorrectPassword;
266 fn logout_user(env: lmdb.Env, session_token: SessionToken) !void {
267 const txn = try env.txn();
268 defer txn.commit() catch {};
270 const sessions = try Db.sessions(txn);
271 try sessions.del(session_token);
274 fn append_post(env: lmdb.Env, user_id: UserId, post_list: PostList, parent_id: ?PostId, quote_id: ?PostId, text: []const u8) !PostId {
275 var post_id: PostId = undefined;
277 // TODO: do this in one commit
279 var txn: lmdb.Txn = undefined;
283 defer txn.commit() catch {};
285 const posts = try Db.posts(txn);
286 post_id = try db.Prng.gen(posts.dbi, PostId);
288 const decoded_text = try reencode(text);
289 try posts.put(post_id, Post{
291 .parent_id = parent_id,
292 .quote_id = quote_id,
294 .time = std.time.timestamp(),
295 .votes = try VoteList.init(txn),
296 .comments = try PostList.init(txn),
297 .quotes = try PostList.init(txn),
298 .text = decoded_text,
303 // append to user's posts
305 defer txn.commit() catch {};
307 var posts_view = try post_list.open(txn);
308 try posts_view.append(post_id);
311 if (quote_id != null) {
313 defer txn.commit() catch {};
315 const posts = try Db.posts(txn);
316 const quote_post = try posts.get(quote_id.?);
317 var quotes = try quote_post.quotes.open(txn);
318 try quotes.append(post_id);
324 fn post(env: lmdb.Env, user_id: UserId, text: []const u8) !void {
325 var txn = try env.txn();
326 const users = try Db.users(txn);
327 const user = try users.get(user_id);
330 const post_id = try append_post(env, user_id, user.posts, null, null, text);
333 var replies_view = try user.replies.open(txn);
334 try replies_view.append(post_id);
338 fn comment(env: lmdb.Env, user_id: UserId, parent_post_id: PostId, text: []const u8) !void {
339 var txn = try env.txn();
340 const users = try Db.users(txn);
341 const user = try users.get(user_id);
343 const posts = try Db.posts(txn);
344 const parent_post = try posts.get(parent_post_id);
347 const post_id = try append_post(env, user_id, parent_post.comments, parent_post_id, null, text);
350 var replies_view = try user.replies.open(txn);
351 try replies_view.append(post_id);
355 fn quote(env: lmdb.Env, user_id: UserId, quote_post_id: PostId, text: []const u8) !void {
356 var txn = try env.txn();
357 const users = try Db.users(txn);
358 const user = try users.get(user_id);
361 const post_id = try append_post(env, user_id, user.posts, null, quote_post_id, text);
364 var replies_view = try user.replies.open(txn);
365 try replies_view.append(post_id);
369 fn vote(env: lmdb.Env, post_id: PostId, user_id: UserId, kind: Vote.Kind) !void {
370 const txn = try env.txn();
371 defer txn.commit() catch {};
373 const posts = try Db.posts(txn);
375 var p = try posts.get(post_id);
376 var votes_view = try p.votes.open(txn);
380 if (try votes_view.has(user_id)) {
381 const old_vote = try votes_view.get(user_id);
383 add_vote = old_vote.kind != kind;
385 try votes_view.del(user_id);
387 switch (old_vote.kind) {
388 .Up => p.upvotes -= 1,
389 .Down => p.downvotes -= 1,
391 try posts.put(post_id, p);
395 try votes_view.append(user_id, Vote{
397 .time = std.time.timestamp(),
405 try posts.put(post_id, p);
409 fn follow(env: lmdb.Env, user_id: UserId, user_id_to_follow: UserId) !void {
410 const txn = try env.txn();
411 defer txn.commit() catch {};
413 const users = try Db.users(txn);
415 const user = try users.get(user_id);
416 const user_to_follow = try users.get(user_id_to_follow);
418 var user_following = try user.following.open(txn);
419 var user_to_follow_followers = try user_to_follow.followers.open(txn);
421 if ((user_following.has(user_id_to_follow) catch false) and (user_to_follow_followers.has(user_id) catch false)) {
422 try user_following.del(user_id_to_follow);
423 try user_to_follow_followers.del(user_id);
424 } else if (!(user_following.has(user_id_to_follow) catch true) and !(user_to_follow_followers.has(user_id) catch true)) {
425 try user_following.append(user_id_to_follow);
426 try user_to_follow_followers.append(user_id);
428 std.debug.print("Something went wrong when trying to unfollow\n", .{});
432 fn get_session_user_id(env: lmdb.Env, session_token: SessionToken) !UserId {
433 const txn = try env.txn();
436 const sessions = try Db.sessions(txn);
438 return try sessions.get(session_token);
441 fn get_user(env: lmdb.Env, user_id: UserId) !User {
442 const txn = try env.txn();
445 const users = try Db.users(txn);
446 return try users.get(user_id);
453 fn html_form(res: *http.Response, action: []const u8, inputs: anytype) !void {
454 try res.write("<form action=\"{s}\" method=\"post\">", .{action});
456 inline for (inputs) |input| {
457 switch (@typeInfo(@TypeOf(input))) {
459 try res.write("<input ", .{});
460 try res.write(input[0], input[1]);
461 try res.write(" />", .{});
464 try res.write("<input ", .{});
465 try res.write(input, .{});
466 try res.write(" />", .{});
471 try res.write("</form>", .{});
476 const TimeStr = std.BoundedArray(u8, 256);
478 // http://howardhinnant.github.io/date_algorithms.html
479 fn time_str(_t: i64) TimeStr {
480 const t: u64 = @intCast(_t);
481 var result = TimeStr.init(0) catch unreachable;
483 const nD = @divFloor(t, std.time.s_per_day);
484 const z: u64 = nD + 719468;
485 const era: u64 = (if (z >= 0) z else z - 146096) / 146097;
486 const doe: u64 = z - era * 146097; // [0, 146096]
487 const yoe: u64 = (doe - doe / 1460 + doe / 36524 - doe / 146096) / 365; // [0, 399]
488 const Y: u64 = yoe + era * 400;
489 const doy: u64 = doe - (365 * yoe + yoe / 4 - yoe / 100); // [0, 365]
490 const mp: u64 = (5 * doy + 2) / 153; // [0, 11]
491 const D: u64 = doy - (153 * mp + 2) / 5 + 1; // [1, 31]
492 const M: u64 = if (mp < 10) mp + 3 else mp - 9;
494 const h: u64 = @divFloor(t - nD * std.time.s_per_day, std.time.s_per_hour);
495 const m: u64 = @divFloor(t - nD * std.time.s_per_day - h * std.time.s_per_hour, std.time.s_per_min);
496 const s: u64 = t - nD * std.time.s_per_day - h * std.time.s_per_hour - m * std.time.s_per_min;
498 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;
502 fn write_header(res: *http.Response, logged_in: ?Login) !void {
503 if (logged_in) |login| {
505 \\<a href="/">Home</a><br />
508 \\<a href="/user/{s}">Profile</a><br />
509 , .{login.user.name.constSlice()});
511 \\<a href="/post">Post</a><br />
513 try html_form(res, "/logout", .{
514 \\type="submit" value="Logout"
516 try res.write("<br /><br />", .{});
519 \\<a href="/">Home</a><br />
520 \\<form action="/register" method="post">
521 \\<input type="text" name="username" />
522 \\<input type="password" name="password" />
523 \\<input type="submit" value="Register" />
525 \\<form action="/login" method="post">
526 \\<input type="text" name="username" />
527 \\<input type="password" name="password" />
528 \\<input type="submit" value="Login" />
531 try html_form(res, "/quit", .{
532 \\type="submit" value="Quit"
534 try res.write("<br /><br />", .{});
537 fn write_start(res: *http.Response) !void {
542 \\<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>">
545 \\ display: inline-block;
552 fn write_end(res: *http.Response) !void {
553 try res.write("</body></html>", .{});
555 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 {
556 const posts = try Db.posts(txn);
557 const post = posts.get(post_id) catch {
558 res.redirect("/") catch {};
561 const users = try Db.users(txn);
562 const user = try users.get(post.user_id);
566 \\<span><a href="/user/{s}">{s}</a>
567 , .{ user.name.constSlice(), user.display_name.constSlice() });
568 if (post.parent_id) |id| {
569 try res.write(" <a href=\"/post/{x}\">..</a>", .{@intFromEnum(id)});
573 \\<span>{s}</span><br />
574 , .{ time_str(post.time).constSlice(), post.text.constSlice() });
576 if (post.quote_id) |quote_id| {
577 try res.write("<div style=\"border: 1px solid black;\">", .{});
578 if (options.recurse > 0) {
579 try write_post(res, txn, logged_in, quote_id, .{ .recurse = options.recurse - 1 });
581 try res.write("<a href=\"/post/{x}\">...</a>", .{@intFromEnum(post_id)});
583 try res.write("</div>", .{});
586 const comments_view = try post.comments.open(txn);
587 const quotes_view = try post.quotes.open(txn);
588 const votes_view = try post.votes.open(txn);
591 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;
593 if (vote != null and vote.?.kind == .Up) {
594 try html_form(res, "/upvote", .{
595 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
596 .{ "type=\"submit\" value=\"⬆ {}\"", .{post.upvotes} },
599 try html_form(res, "/upvote", .{
600 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
601 .{ "type=\"submit\" value=\"⇧ {}\"", .{post.upvotes} },
604 if (vote != null and vote.?.kind == .Down) {
605 try html_form(res, "/downvote", .{
606 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
607 .{ "type=\"submit\" value=\"⬇ {}\"", .{post.downvotes} },
610 try html_form(res, "/downvote", .{
611 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
612 .{ "type=\"submit\" value=\"⇩ {}\"", .{post.downvotes} },
618 \\<a href="/post/{x}">💭 {}</a>
619 , .{ @intFromEnum(post.id), comments_view.len() });
623 \\<a href="/quotes/{x}">🔁 {}</a>
624 , .{ @intFromEnum(post.id), quotes_view.len() });
627 if (logged_in) |login| {
628 const lists_view = try login.user.post_lists.open(txn);
629 try res.write("<form action=\"/list_add\" method=\"post\">", .{});
630 try res.write("<select name=\"list_id\">", .{});
631 var it = lists_view.iterator();
632 // TODO: mark lists that already contain post
633 while (it.next()) |kv| {
634 const name = kv.val.name;
635 const id = kv.val.list.idx.?;
636 try res.write("<option value=\"{x}\">{s}</option>", .{ id, name.constSlice() });
638 try res.write("</select>", .{});
639 try res.write("<input type=\"hidden\" name=\"post_id\" value=\"{x}\"></input>", .{@intFromEnum(post_id)});
640 try res.write("<input type=\"submit\" value=\"Save\"></input>", .{});
641 try res.write("</form>", .{});
645 // TODO: maybe always show comment field and prompt for login
646 if (options.show_comment_field and logged_in != null) {
647 try res.write("<br /><br />", .{});
648 try html_form(res, "/comment", .{
649 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
650 "type=\"text\" name=\"text\" placeholder=\"Text\"",
651 "type=\"submit\" value=\"Comment\"",
653 try res.write("<br />", .{});
657 if (options.recurse > 0 and comments_view.len() > 0) {
660 \\<summary>Comments</summary>
661 , .{if (options.recurse > 1) " open" else ""});
662 try res.write("<div style=\"margin: 10px;\">", .{});
663 var it = comments_view.iterator();
665 while (it.next()) |comment_id| {
666 try write_post(res, txn, logged_in, comment_id, .{ .recurse = options.recurse - 1 });
667 try res.write("<br />", .{});
668 if (options.recurse == 1) {
670 if (count >= 3) break;
683 fn write_profile(res: *http.Response, txn: lmdb.Txn, logged_in: ?Login, user: User) !void {
684 const following = try user.following.open(txn);
685 const followers = try user.followers.open(txn);
688 \\<h2 style="display: inline;"><a href="/user/{s}">{s}</a></h2>
690 user.name.constSlice(), user.display_name.constSlice(),
692 if (logged_in != null and user.id != logged_in.?.user.id) {
693 const login = logged_in.?;
696 if (try followers.has(login.user.id)) {
697 try html_form(res, "/follow", .{
698 .{ "type=\"hidden\" name=\"user.id\" value=\"{x}\"", .{@intFromEnum(user.id)} },
699 \\type="submit" value="Unfollow"
702 try html_form(res, "/follow", .{
703 .{ "type=\"hidden\" name=\"user.id\" value=\"{x}\"", .{@intFromEnum(user.id)} },
704 \\type="submit" value="Follow"
709 const feeds_view = try login.user.feeds.open(txn);
710 try res.write("<form action=\"/feed_add\" method=\"post\">", .{});
711 try res.write("<select name=\"feed_id\">", .{});
712 var it = feeds_view.iterator();
713 while (it.next()) |kv| {
714 const name = kv.val.name;
715 const id = kv.val.list.idx.?;
716 try res.write("<option value=\"{x}\">{s}</option>", .{ id, name.constSlice() });
718 try res.write("</select>", .{});
719 try res.write("<input type=\"hidden\" name=\"user.id\" value=\"{x}\"></input>", .{@intFromEnum(user.id)});
720 try res.write("<input type=\"submit\" value=\"Add to feed\"></input>", .{});
721 try res.write("</form>", .{});
724 \\ <a href="/following/{s}">{} following</a>
725 \\ <a href="/followers/{s}">{} followers</a>
728 user.name.constSlice(), following.len(),
729 user.name.constSlice(), followers.len(),
732 try res.write("<a href=\"/replies/{s}\">Replies</a><br />", .{
733 user.name.constSlice(),
736 if (logged_in != null and user.id == logged_in.?.user.id) {
738 \\<a href="/lists">Lists</a>
739 \\<a href="/feeds">Feeds</a>
740 \\<a href="/edit">Edit</a><br />
745 try res.write("<br />", .{});
747 fn write_posts(res: *http.Response, txn: lmdb.Txn, logged_in: ?Login, post_list: PostList) !void {
748 const posts_view = try post_list.open(txn);
750 var it = posts_view.reverse_iterator();
751 while (it.next()) |post_id| {
752 try write_post(res, txn, logged_in, post_id, .{ .recurse = 1 });
753 try res.write("<br />", .{});
756 fn write_timeline(res: *http.Response, txn: lmdb.Txn, logged_in: ?Login, user_list: UserList) !void {
757 const users = try Db.users(txn);
758 const posts = try Db.posts(txn);
760 var newest_post_ids = try std.BoundedArray(PostId, 10).init(0); // TODO: TimelinePostsCount
761 var prev_newest_post: ?Post = null;
763 const following = try user_list.open(txn);
766 var newest_post: ?Post = null;
768 var following_it = following.iterator();
769 while (following_it.next()) |following_id| {
770 const followed_user = try users.get(following_id);
771 const followed_posts = try followed_user.posts.open(txn);
773 if (followed_posts.len() == 0) {
777 var followed_posts_it = followed_posts.reverse_iterator();
778 while (followed_posts_it.next()) |followed_post_id| {
779 const p = try posts.get(followed_post_id);
781 if ((prev_newest_post == null or p.time < prev_newest_post.?.time) and (newest_post == null or newest_post.?.time < p.time)) {
787 if (newest_post) |post| {
788 newest_post_ids.append(post.id) catch break;
789 prev_newest_post = post;
795 for (newest_post_ids.constSlice()) |post_id| {
796 try write_post(res, txn, logged_in, post_id, .{ .recurse = 1 });
797 try res.write("<br />", .{});
800 fn check_login(env: lmdb.Env, req: http.Request, res: *http.Response) !?Login {
801 var result: ?Login = null;
803 if (req.get_cookie("session_token")) |session_token_str| {
804 var remove_session_token = true;
806 if (std.fmt.parseUnsigned(SessionToken, session_token_str, 16) catch null) |session_token| {
807 // const session_token = try std.fmt.parseUnsigned(SessionToken, session_token_str, 10);
808 // const session_token = std.mem.bytesToValue(SessionToken, session_token_str);
809 if (Chirp.get_session_user_id(env, session_token) catch null) |user_id| {
810 const txn = try env.txn();
812 const users = try Db.users(txn);
815 .user = try users.get(user_id),
816 .session_token = session_token,
819 remove_session_token = false;
823 if (remove_session_token) {
826 .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
837 const Self = @This();
844 fn handle(self: Self) !bool {
845 const ti = @typeInfo(Self);
846 inline for (ti.Struct.decls) |f_decl| {
847 const has_arg = f_decl.name.len > 1 and f_decl.name[f_decl.name.len - 1] == '/';
848 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);
851 const f = @field(Self, f_decl.name);
852 const fi = @typeInfo(@TypeOf(f));
853 if (fi.Fn.params.len == 1) {
854 try @call(.auto, f, .{self});
856 const arg_type = fi.Fn.params[1].type.?;
857 const arg_info = @typeInfo(arg_type);
858 var arg: arg_type = undefined;
859 const field = arg_info.Struct.fields[0];
860 if (self.req.target.len <= f_decl.name.len) {
861 return error.NoArgProvided;
863 const str = self.req.target[f_decl.name.len..self.req.target.len];
864 const field_ti = @typeInfo(field.type);
866 // TODO: maybe handle BoundedArray?
868 @field(arg, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
871 @field(arg, field.name) = try parse_enum(field.type, str, 16);
874 @field(arg, field.name) = str;
878 try @call(.auto, f, .{ self, arg });
886 pub fn @"/user/"(self: Self, args: struct { username: []const u8 }) !void {
887 const user_ids = try Db.user_ids(self.txn);
888 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
889 const users = try Db.users(self.txn);
890 const user = try users.get(user_id);
892 try write_profile(self.res, self.txn, self.logged_in, user);
894 try write_posts(self.res, self.txn, self.logged_in, user.posts);
897 \\<p>User not found [{}]</p>
901 pub fn @"/replies/"(self: Self, args: struct { username: []const u8 }) !void {
902 const user_ids = try Db.user_ids(self.txn);
903 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
904 const users = try Db.users(self.txn);
905 const user = try users.get(user_id);
907 try write_profile(self.res, self.txn, self.logged_in, user);
909 try write_posts(self.res, self.txn, self.logged_in, user.replies);
912 \\<p>User not found [{}]</p>
916 pub fn @"/following/"(self: Self, args: struct { username: []const u8 }) !void {
917 const user_ids = try Db.user_ids(self.txn);
918 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
919 const users = try Db.users(self.txn);
920 const user = try users.get(user_id);
922 const following = try user.following.open(self.txn);
923 var it = following.iterator();
926 \\<h2><a href="/user/{s}">{s}</a> follows:</h2>
927 , .{ user.name.constSlice(), user.display_name.constSlice() });
929 while (it.next()) |following_id| {
930 const following_user = try users.get(following_id);
933 \\<a href="/user/{s}">{s}</a><br />
934 , .{ following_user.name.constSlice(), following_user.display_name.constSlice() });
938 \\<p>User not found [{}]</p>
942 pub fn @"/followers/"(self: Self, args: struct { username: []const u8 }) !void {
943 const user_ids = try Db.user_ids(self.txn);
944 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
945 const users = try Db.users(self.txn);
946 const user = try users.get(user_id);
948 const followers = try user.followers.open(self.txn);
949 var it = followers.iterator();
952 \\<h2><a href="/user/{s}">{s}</a> followers:</h2>
953 , .{ user.name.constSlice(), user.display_name.constSlice() });
955 while (it.next()) |follower_id| {
956 const follower_user = try users.get(follower_id);
959 \\<a href="/user/{s}">{s}</a><br />
960 , .{ follower_user.name.constSlice(), follower_user.display_name.constSlice() });
964 \\<p>User not found [{}]</p>
968 pub fn @"/post/"(self: Self, args: struct { post_id: PostId }) !void {
969 try write_post(self.res, self.txn, self.logged_in, args.post_id, .{
970 .recurse = 3, // TODO: factor out
971 .show_comment_field = true,
974 pub fn @"/quotes/"(self: Self, args: struct { post_id: PostId }) !void {
975 const posts = try Db.posts(self.txn);
976 const post = try posts.get(args.post_id);
978 const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
980 if (self.logged_in != null) {
981 try html_form(self.res, "/quote", .{
982 .{ "type=\"hidden\" name=\"referer\" value=\"{s}\"", .{referer} },
983 .{ "type=\"hidden\" name=\"post_id\" value=\"{x}\"", .{@intFromEnum(post.id)} },
984 "type=\"text\" name=\"text\" placeholder=\"Text\"",
985 "type=\"submit\" value=\"Quote\"",
987 try self.res.write("<br />", .{});
990 const quotes_view = try post.quotes.open(self.txn);
991 var it = quotes_view.iterator();
992 while (it.next()) |quote_id| {
993 try write_post(self.res, self.txn, self.logged_in, quote_id, .{ .recurse = 1 });
994 try self.res.write("<br />", .{});
997 pub fn @"/list/"(self: Self, args: struct { list_id: PostList.Index }) !void {
998 try write_posts(self.res, self.txn, self.logged_in, PostList{ .idx = args.list_id });
1000 pub fn @"/lists"(self: Self) !void {
1001 if (self.logged_in) |login| {
1002 const post_lists_view = try login.user.post_lists.open(self.txn);
1004 try html_form(self.res, "/create_list", .{
1005 "type=\"text\" name=\"name\"",
1006 "type=\"submit\" value=\"Add\"",
1009 try self.res.write("<br /><br />", .{});
1011 var it = post_lists_view.iterator();
1012 while (it.next()) |kv| {
1013 const name = kv.val.name;
1014 const post_list = kv.val.list;
1016 \\<a href="/list/{x}">{s}</a>
1017 , .{ post_list.idx.?, name.constSlice() });
1018 try html_form(self.res, "/delete_list", .{
1019 .{ "type=\"hidden\" name=\"list_id\" value=\"{x}\"", .{kv.key} },
1020 "type=\"submit\" value=\"Delete\"",
1022 try self.res.write("<br />", .{});
1025 try self.res.write("not logged in", .{});
1028 pub fn @"/feed/"(self: Self, args: struct { feed_id: UserList.Index }) !void {
1029 try write_timeline(self.res, self.txn, self.logged_in, UserList{ .idx = args.feed_id });
1031 pub fn @"/feeds"(self: Self) !void {
1032 if (self.logged_in) |login| {
1033 const feeds_view = try login.user.feeds.open(self.txn);
1035 try html_form(self.res, "/create_feed", .{
1036 "type=\"text\" name=\"name\"",
1037 "type=\"submit\" value=\"Add\"",
1040 try self.res.write("<br /><br />", .{});
1042 var it = feeds_view.iterator();
1043 while (it.next()) |kv| {
1044 const name = kv.val.name;
1045 const user_list = kv.val.list;
1047 \\<a href="/feed/{x}">{s}</a>
1048 , .{ user_list.idx.?, name.constSlice() });
1049 try html_form(self.res, "/delete_feed", .{
1050 .{ "type=\"hidden\" name=\"list_id\" value=\"{x}\"", .{kv.key} },
1051 "type=\"submit\" value=\"Delete\"",
1053 try self.res.write("<br />", .{});
1056 try self.res.write("not logged in", .{});
1059 pub fn @"/post"(self: Self) !void {
1060 if (self.logged_in) |login| {
1062 const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
1064 try html_form(self.res, "/post", .{
1065 .{ "type=\"hidden\" name=\"referer\" value=\"{s}\"", .{referer} },
1066 "type=\"text\" name=\"text\"",
1067 "type=\"submit\" value=\"Post\"",
1070 try self.res.write("not logged in", .{});
1073 pub fn @"/edit"(self: Self) !void {
1074 if (self.logged_in) |login| {
1075 try self.res.write("<br />Username: ", .{});
1076 try html_form(self.res, "/set_username", .{
1077 .{ "type=\"text\" name=\"username\" placeholder=\"{s}\"", .{login.user.name.constSlice()} },
1078 "type=\"submit\" value=\"Change\"",
1080 try self.res.write("<br />Display Name: ", .{});
1081 try html_form(self.res, "/set_display_name", .{
1082 .{ "type=\"text\" name=\"display_name\" placeholder=\"{s}\"", .{login.user.display_name.constSlice()} },
1083 "type=\"submit\" value=\"Change\"",
1085 try self.res.write("<br />Password: ", .{});
1086 try html_form(self.res, "/set_password", .{
1087 "type=\"text\" name=\"password\"",
1088 "type=\"submit\" value=\"Change\"",
1091 try self.res.write("not logged in", .{});
1094 pub fn @"/"(self: Self) !void {
1095 if (self.logged_in) |login| {
1096 try write_timeline(self.res, self.txn, self.logged_in, login.user.following);
1098 // TODO: generic home
1099 try self.res.write("Homepage", .{});
1106 const POST = struct {
1107 const Self = @This();
1111 res: *http.Response,
1114 pub fn handle(self: Self) !bool {
1115 const ti = @typeInfo(Self);
1116 inline for (ti.Struct.decls) |f_decl| {
1117 if (std.mem.eql(u8, f_decl.name, self.req.target)) {
1118 const f = @field(Self, f_decl.name);
1119 const fi = @typeInfo(@TypeOf(f));
1120 if (fi.Fn.params.len == 1) {
1121 _ = try @call(.auto, f, .{self});
1123 const args_type = fi.Fn.params[fi.Fn.params.len - 1].type.?;
1124 const argsi = @typeInfo(args_type);
1125 var args: args_type = undefined;
1126 inline for (argsi.Struct.fields) |field| {
1127 const str = self.req.get_value(field.name) orelse return error.ArgNotFound;
1128 const field_ti = @typeInfo(field.type);
1131 @field(args, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
1134 @field(args, field.name) = try parse_enum(field.type, str, 16);
1137 @field(args, field.name) = str;
1141 try @call(.auto, f, .{ self, args });
1149 pub fn @"/register"(self: Self, args: struct { username: []const u8, password: []const u8 }) !void {
1150 // TODO: handle args not supplied
1151 std.debug.print("New user: {s} {s}\n", .{ args.username, args.password });
1152 _ = try Chirp.register_user(self.env, args.username, args.password);
1154 pub fn @"/login"(self: Self, args: struct { username: []const u8, password: []const u8 }) !void {
1155 // TODO: handle args not supplied
1156 std.debug.print("New login: {s} {s}\n", .{ args.username, args.password });
1157 if (Chirp.login_user(self.env, args.username, args.password)) |session_token| {
1158 self.res.status = .see_other;
1159 try self.res.add_header(
1161 .{ "session_token={x}; HttpOnly", .{session_token} },
1164 std.debug.print("login_user err: {}\n", .{err});
1167 pub fn @"/logout"(self: Self) !void {
1168 if (self.logged_in) |login| {
1169 try Chirp.logout_user(self.env, login.session_token);
1171 try self.res.add_header(
1173 .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
1177 pub fn @"/set_username"(self: Self, args: struct { username: []const u8 }) !void {
1178 const login = self.logged_in orelse return error.NotLoggedIn;
1179 const username = try Username.fromSlice(args.username);
1181 const txn = try self.env.txn();
1182 defer txn.commit() catch {};
1184 const user_ids = try Db.user_ids(txn);
1186 if (!try user_ids.has(username)) {
1187 try user_ids.del(login.user.name);
1188 try user_ids.put(username, login.user.id);
1190 const users = try Db.users(txn);
1191 var user = login.user;
1192 user.name = username;
1193 try users.put(login.user.id, user);
1196 pub fn @"/set_display_name"(self: Self, args: struct { display_name: []const u8 }) !void {
1197 const login = self.logged_in orelse return error.NotLoggedIn;
1198 const display_name = try DisplayName.fromSlice(args.display_name);
1200 const txn = try self.env.txn();
1201 defer txn.commit() catch {};
1203 const users = try Db.users(txn);
1204 var user = login.user;
1205 user.display_name = display_name;
1206 try users.put(login.user.id, user);
1208 pub fn @"/set_password"(self: Self, args: struct { password: []const u8 }) !void {
1209 const login = self.logged_in orelse return error.NotLoggedIn;
1211 const txn = try self.env.txn();
1212 defer txn.commit() catch {};
1214 const users = try Db.users(txn);
1215 var user = login.user;
1216 user.password_hash = try Chirp.hash_password(args.password);
1217 try users.put(login.user.id, user);
1219 pub fn @"/post"(self: Self) !void {
1220 if (self.logged_in) |login| {
1221 const text = self.req.get_value("text").?;
1222 const has_referer = self.req.get_value("referer");
1224 try Chirp.post(self.env, login.user.id, text);
1226 if (has_referer) |r| {
1227 const decoded = try decode(r);
1228 try self.res.redirect(decoded.constSlice());
1232 pub fn @"/comment"(self: Self) !void {
1233 if (self.logged_in) |login| {
1234 const text = self.req.get_value("text") orelse return error.NoText;
1235 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1236 const post_id = try parse_enum(PostId, post_id_str, 16);
1238 try Chirp.comment(self.env, login.user.id, post_id, text);
1241 pub fn @"/quote"(self: Self) !void {
1242 if (self.logged_in) |login| {
1243 const text = self.req.get_value("text") orelse return error.NoText;
1244 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1245 const has_referer = self.req.get_value("referer");
1247 const post_id = try parse_enum(PostId, post_id_str, 16);
1249 try Chirp.quote(self.env, login.user.id, post_id, text);
1251 if (has_referer) |r| {
1252 const decoded = try decode(r);
1253 try self.res.redirect(decoded.constSlice());
1257 // TODO: add arguments instead of parsing manually
1258 pub fn @"/create_list"(self: Self) !void {
1259 if (self.logged_in) |login| {
1260 const name_str = self.req.get_value("name") orelse return error.NoName;
1261 const name = try Name.fromSlice(name_str);
1262 // TODO: decode name
1264 var txn = try self.env.txn();
1265 const postlist = try PostList.init(txn);
1268 txn = try self.env.txn();
1269 var post_lists_view = try login.user.post_lists.open(txn);
1270 _ = try post_lists_view.append(.{ .name = name, .list = postlist });
1274 pub fn @"/delete_list"(self: Self, args: struct { list_id: PostList.Index }) !void {
1275 if (self.logged_in) |login| {
1276 var post_list: ?PostList = null;
1278 const txn = try self.env.txn();
1279 defer txn.commit() catch {};
1280 var post_lists_view = try login.user.post_lists.open(txn);
1281 post_list = (try post_lists_view.get(args.list_id)).list;
1282 try post_lists_view.del(args.list_id);
1284 if (post_list != null) {
1285 const txn = try self.env.txn();
1286 defer txn.commit() catch {};
1287 var list_view = try post_list.?.open(txn);
1288 try list_view.clear();
1292 pub fn @"/list_add"(self: Self, args: struct { list_id: PostList.Index, post_id: PostId }) !void {
1293 if (self.logged_in) |login| {
1296 const txn = try self.env.txn();
1297 defer txn.commit() catch {};
1299 const post_list = PostList{ .idx = args.list_id };
1300 var post_list_view = try post_list.open(txn);
1301 if (try post_list_view.has(args.post_id)) {
1302 try post_list_view.del(args.post_id);
1304 try post_list_view.append(args.post_id);
1308 pub fn @"/create_feed"(self: Self) !void {
1309 if (self.logged_in) |login| {
1310 const name_str = self.req.get_value("name") orelse return error.NoName;
1311 const name = try Name.fromSlice(name_str);
1313 var txn = try self.env.txn();
1314 const userlist = try UserList.init(txn);
1317 txn = try self.env.txn();
1318 var feeds_view = try login.user.feeds.open(txn);
1319 _ = try feeds_view.append(.{ .name = name, .list = userlist });
1323 pub fn @"/delete_feed"(self: Self, args: struct { list_id: UserList.Index }) !void {
1324 if (self.logged_in) |login| {
1325 var user_list: ?UserList = null;
1328 const txn = try self.env.txn();
1329 defer txn.commit() catch {};
1330 var feeds_view = try login.user.feeds.open(txn);
1331 user_list = (try feeds_view.get(args.list_id)).list;
1332 try feeds_view.del(args.list_id);
1334 if (user_list != null) {
1335 const txn = try self.env.txn();
1336 defer txn.commit() catch {};
1337 var list_view = try user_list.?.open(txn);
1338 try list_view.clear();
1342 pub fn @"/feed_add"(self: Self, args: struct { feed_id: UserList.Index, user_id: UserId }) !void {
1343 if (self.logged_in) |login| {
1346 const txn = try self.env.txn();
1347 defer txn.commit() catch {};
1349 const user_list = UserList{ .idx = args.feed_id };
1350 var user_list_view = try user_list.open(txn);
1351 if (try user_list_view.has(args.user_id)) {
1352 try user_list_view.del(args.user_id);
1354 try user_list_view.append(args.user_id);
1358 pub fn @"/upvote"(self: Self) !void {
1359 const login = self.logged_in orelse return error.NotLoggedIn;
1361 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1362 const post_id: PostId = @enumFromInt(try std.fmt.parseUnsigned(u64, post_id_str, 16));
1364 try Chirp.vote(self.env, post_id, login.user.id, .Up);
1366 pub fn @"/downvote"(self: Self) !void {
1367 const login = self.logged_in orelse return error.NotLoggedIn;
1369 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1370 const post_id: PostId = @enumFromInt(try std.fmt.parseUnsigned(u64, post_id_str, 16));
1372 try Chirp.vote(self.env, post_id, login.user.id, .Down);
1374 pub fn @"/follow"(self: Self) !void {
1375 const login = self.logged_in orelse return error.NotLoggedIn;
1377 const user_id_str = self.req.get_value("user_id") orelse return error.NoUserId;
1378 const user_id: UserId = @enumFromInt(try std.fmt.parseUnsigned(u64, user_id_str, 16));
1380 try Chirp.follow(self.env, login.user.id, user_id);
1382 pub fn @"/quit"(self: Self) !void {
1383 if (self.req.get_header("Referer")) |ref| {
1384 try self.res.redirect(ref);
1386 try self.res.redirect("/");
1388 try self.res.send();
1394 fn list_users(env: lmdb.Env) !void {
1395 const txn = try env.txn();
1398 const users = try Db.users(txn);
1399 var it = try users.iterator();
1401 while (it.next()) |kv| {
1403 const user = kv.val;
1404 std.debug.print("[{}] {s}\n", .{ key, user.name.constSlice() });
1407 fn list_user_ids(env: lmdb.Env) !void {
1408 const txn = try env.txn();
1411 const user_ids = try Db.user_ids(txn);
1412 var it = try user_ids.iterator();
1414 while (it.next()) |kv| {
1416 const user_id = kv.val;
1417 std.debug.print("[{s}] {}\n", .{ key.constSlice(), user_id });
1421 fn list_sessions(env: lmdb.Env) !void {
1422 const txn = try env.txn();
1425 const sessions = try Db.sessions(txn);
1426 var it = try sessions.iterator();
1428 while (it.next()) |kv| {
1430 const user_id = kv.val;
1431 std.debug.print("[{x}] {}\n", .{ key, user_id });
1435 fn list_posts(env: lmdb.Env) !void {
1436 const txn = try env.txn();
1439 const posts = try Db.posts(txn);
1440 var it = try posts.iterator();
1442 while (it.next()) |kv| {
1444 const post = kv.val;
1445 std.debug.print("[{}] {s}\n", .{ key, post.text.constSlice() });
1449 const ReqBufferSize = 4096;
1450 const ResHeadBufferSize = 1024 * 64;
1451 const ResBodyBufferSize = 1024 * 64;
1454 var req_buffer: [ReqBufferSize]u8 = undefined;
1455 var res_head_buffer: [ResHeadBufferSize]u8 = undefined;
1456 var res_body_buffer: [ResBodyBufferSize]u8 = undefined;
1458 pub fn main() !void {
1460 var server = try http.Server.init("::", 8080);
1461 defer server.deinit();
1464 var env = try lmdb.Env.open("db", 1024 * 1024 * 10);
1467 std.debug.print("Users:\n", .{});
1468 try list_users(env);
1469 std.debug.print("User IDs:\n", .{});
1470 try list_user_ids(env);
1471 std.debug.print("Sessions:\n", .{});
1472 try list_sessions(env);
1473 std.debug.print("Posts:\n", .{});
1474 try list_posts(env);
1479 const req = (server.next_request(&req_buffer) catch break) orelse break;
1480 handle_request(env, req) catch {
1481 try handle_error(env, req);
1485 // const ThreadCount = 1;
1486 // var ts: [ThreadCount]std.Thread = undefined;
1488 // for (0..ThreadCount) |i| {
1489 // ts[i] = try std.Thread.spawn(.{}, handle_connection, .{ &server, &env });
1491 // for (0..ThreadCount) |i| {
1495 std.debug.print("done\n", .{});
1498 fn handle_error(env: lmdb.Env, req: http.Request) !void {
1500 var res = http.Response.init(req.fd, &res_head_buffer, &res_body_buffer);
1501 try write_start(&res);
1502 try res.write("Oops, something went terribly wrong there D:", .{});
1503 try write_end(&res);
1506 fn handle_request(env: lmdb.Env, req: http.Request) !void {
1507 // std.debug.print("[{}]: {s}\n", .{ req.method, req.target });
1508 // std.debug.print("[{}]: {s}\n", .{ req.method, req.head.? });
1511 var res = http.Response.init(req.fd, &res_head_buffer, &res_body_buffer);
1513 // check session token
1514 const logged_in: ?Login = try check_login(env, req, &res);
1517 if (req.method == .GET) {
1518 try write_start(&res);
1519 try write_header(&res, logged_in);
1521 const txn = try env.txn();
1528 .logged_in = logged_in,
1530 if (try get.handle()) {} else {
1531 try res.redirect("/");
1534 try write_end(&res);
1543 .logged_in = logged_in,
1545 if (try post.handle()) {} else {
1546 try res.write("<p>[POST] {s}</p>", .{req.target});
1549 if (!res.has_header("Location")) {
1550 if (req.get_header("Referer")) |ref| {
1551 try res.redirect(ref);
1553 try res.redirect("/");