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" />
529 \\</form><br /><br />
533 fn write_start(res: *http.Response) !void {
538 \\<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>">
541 \\ display: inline-block;
548 fn write_end(res: *http.Response) !void {
549 try res.write("</body></html>", .{});
551 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 {
552 const posts = try Db.posts(txn);
553 const post = posts.get(post_id) catch {
554 res.redirect("/") catch {};
557 const users = try Db.users(txn);
558 const user = try users.get(post.user_id);
562 \\<span><a href="/user/{s}">{s}</a>
563 , .{ user.name.constSlice(), user.display_name.constSlice() });
564 if (post.parent_id) |id| {
565 try res.write(" <a href=\"/post/{x}\">..</a>", .{@intFromEnum(id)});
569 \\<span>{s}</span><br />
570 , .{ time_str(post.time).constSlice(), post.text.constSlice() });
572 if (post.quote_id) |quote_id| {
573 try res.write("<div style=\"border: 1px solid black;\">", .{});
574 if (options.recurse > 0) {
575 try write_post(res, txn, logged_in, quote_id, .{ .recurse = options.recurse - 1 });
577 try res.write("<a href=\"/post/{x}\">...</a>", .{@intFromEnum(post_id)});
579 try res.write("</div>", .{});
582 const comments_view = try post.comments.open(txn);
583 const quotes_view = try post.quotes.open(txn);
584 const votes_view = try post.votes.open(txn);
587 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;
589 if (vote != null and vote.?.kind == .Up) {
590 try html_form(res, "/upvote", .{
591 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
592 .{ "type=\"submit\" value=\"⬆ {}\"", .{post.upvotes} },
595 try html_form(res, "/upvote", .{
596 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
597 .{ "type=\"submit\" value=\"⇧ {}\"", .{post.upvotes} },
600 if (vote != null and vote.?.kind == .Down) {
601 try html_form(res, "/downvote", .{
602 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
603 .{ "type=\"submit\" value=\"⬇ {}\"", .{post.downvotes} },
606 try html_form(res, "/downvote", .{
607 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
608 .{ "type=\"submit\" value=\"⇩ {}\"", .{post.downvotes} },
614 \\<a href="/post/{x}">💭 {}</a>
615 , .{ @intFromEnum(post.id), comments_view.len() });
619 \\<a href="/quotes/{x}">🔁 {}</a>
620 , .{ @intFromEnum(post.id), quotes_view.len() });
623 if (logged_in) |login| {
624 const lists_view = try login.user.post_lists.open(txn);
625 try res.write("<form action=\"/list_add\" method=\"post\">", .{});
626 try res.write("<select name=\"list_id\">", .{});
627 var it = lists_view.iterator();
628 // TODO: mark lists that already contain post
629 while (it.next()) |kv| {
630 const name = kv.val.name;
631 const id = kv.val.list.idx.?;
632 try res.write("<option value=\"{x}\">{s}</option>", .{ id, name.constSlice() });
634 try res.write("</select>", .{});
635 try res.write("<input type=\"hidden\" name=\"post_id\" value=\"{x}\"></input>", .{@intFromEnum(post_id)});
636 try res.write("<input type=\"submit\" value=\"Save\"></input>", .{});
637 try res.write("</form>", .{});
641 // TODO: maybe always show comment field and prompt for login
642 if (options.show_comment_field and logged_in != null) {
643 try res.write("<br /><br />", .{});
644 try html_form(res, "/comment", .{
645 .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
646 "type=\"text\" name=\"text\" placeholder=\"Text\"",
647 "type=\"submit\" value=\"Comment\"",
649 try res.write("<br />", .{});
653 if (options.recurse > 0 and comments_view.len() > 0) {
656 \\<summary>Comments</summary>
657 , .{if (options.recurse > 1) " open" else ""});
658 try res.write("<div style=\"margin: 10px;\">", .{});
659 var it = comments_view.iterator();
661 while (it.next()) |comment_id| {
662 try write_post(res, txn, logged_in, comment_id, .{ .recurse = options.recurse - 1 });
663 try res.write("<br />", .{});
664 if (options.recurse == 1) {
666 if (count >= 3) break;
679 fn write_profile(res: *http.Response, txn: lmdb.Txn, logged_in: ?Login, user: User) !void {
680 const following = try user.following.open(txn);
681 const followers = try user.followers.open(txn);
684 \\<h2 style="display: inline;"><a href="/user/{s}">{s}</a></h2>
686 user.name.constSlice(), user.display_name.constSlice(),
688 if (logged_in != null and user.id != logged_in.?.user.id) {
689 const login = logged_in.?;
692 if (try followers.has(login.user.id)) {
693 try html_form(res, "/follow", .{
694 .{ "type=\"hidden\" name=\"user.id\" value=\"{x}\"", .{@intFromEnum(user.id)} },
695 \\type="submit" value="Unfollow"
698 try html_form(res, "/follow", .{
699 .{ "type=\"hidden\" name=\"user.id\" value=\"{x}\"", .{@intFromEnum(user.id)} },
700 \\type="submit" value="Follow"
705 const feeds_view = try login.user.feeds.open(txn);
706 try res.write("<form action=\"/feed_add\" method=\"post\">", .{});
707 try res.write("<select name=\"feed_id\">", .{});
708 var it = feeds_view.iterator();
709 while (it.next()) |kv| {
710 const name = kv.val.name;
711 const id = kv.val.list.idx.?;
712 try res.write("<option value=\"{x}\">{s}</option>", .{ id, name.constSlice() });
714 try res.write("</select>", .{});
715 try res.write("<input type=\"hidden\" name=\"user.id\" value=\"{x}\"></input>", .{@intFromEnum(user.id)});
716 try res.write("<input type=\"submit\" value=\"Add to feed\"></input>", .{});
717 try res.write("</form>", .{});
720 \\ <a href="/following/{s}">{} following</a>
721 \\ <a href="/followers/{s}">{} followers</a>
724 user.name.constSlice(), following.len(),
725 user.name.constSlice(), followers.len(),
728 try res.write("<a href=\"/replies/{s}\">Replies</a><br />", .{
729 user.name.constSlice(),
732 if (logged_in != null and user.id == logged_in.?.user.id) {
734 \\<a href="/lists">Lists</a>
735 \\<a href="/feeds">Feeds</a>
736 \\<a href="/edit">Edit</a><br />
741 try res.write("<br />", .{});
743 fn write_posts(res: *http.Response, txn: lmdb.Txn, logged_in: ?Login, post_list: PostList) !void {
744 const posts_view = try post_list.open(txn);
746 var it = posts_view.reverse_iterator();
747 while (it.next()) |post_id| {
748 try write_post(res, txn, logged_in, post_id, .{ .recurse = 1 });
749 try res.write("<br />", .{});
752 fn write_timeline(res: *http.Response, txn: lmdb.Txn, logged_in: ?Login, user_list: UserList) !void {
753 const users = try Db.users(txn);
754 const posts = try Db.posts(txn);
756 var newest_post_ids = try std.BoundedArray(PostId, 10).init(0); // TODO: TimelinePostsCount
757 var prev_newest_post: ?Post = null;
759 const following = try user_list.open(txn);
762 var newest_post: ?Post = null;
764 var following_it = following.iterator();
765 while (following_it.next()) |following_id| {
766 const followed_user = try users.get(following_id);
767 const followed_posts = try followed_user.posts.open(txn);
769 if (followed_posts.len() == 0) {
773 var followed_posts_it = followed_posts.reverse_iterator();
774 while (followed_posts_it.next()) |followed_post_id| {
775 const p = try posts.get(followed_post_id);
777 if ((prev_newest_post == null or p.time < prev_newest_post.?.time) and (newest_post == null or newest_post.?.time < p.time)) {
783 if (newest_post) |post| {
784 newest_post_ids.append(post.id) catch break;
785 prev_newest_post = post;
791 for (newest_post_ids.constSlice()) |post_id| {
792 try write_post(res, txn, logged_in, post_id, .{ .recurse = 1 });
793 try res.write("<br />", .{});
796 fn check_login(env: lmdb.Env, req: http.Request, res: *http.Response) !?Login {
797 var result: ?Login = null;
799 if (req.get_cookie("session_token")) |session_token_str| {
800 var remove_session_token = true;
802 if (std.fmt.parseUnsigned(SessionToken, session_token_str, 16) catch null) |session_token| {
803 // const session_token = try std.fmt.parseUnsigned(SessionToken, session_token_str, 10);
804 // const session_token = std.mem.bytesToValue(SessionToken, session_token_str);
805 if (Chirp.get_session_user_id(env, session_token) catch null) |user_id| {
806 const txn = try env.txn();
808 const users = try Db.users(txn);
811 .user = try users.get(user_id),
812 .session_token = session_token,
815 remove_session_token = false;
819 if (remove_session_token) {
822 .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
833 const Self = @This();
840 fn handle(self: Self) !bool {
841 const ti = @typeInfo(Self);
842 inline for (ti.Struct.decls) |f_decl| {
843 const has_arg = f_decl.name.len > 1 and f_decl.name[f_decl.name.len - 1] == '/';
844 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);
847 const f = @field(Self, f_decl.name);
848 const fi = @typeInfo(@TypeOf(f));
849 if (fi.Fn.params.len == 1) {
850 try @call(.auto, f, .{self});
852 const arg_type = fi.Fn.params[1].type.?;
853 const arg_info = @typeInfo(arg_type);
854 var arg: arg_type = undefined;
855 const field = arg_info.Struct.fields[0];
856 if (self.req.target.len <= f_decl.name.len) {
857 return error.NoArgProvided;
859 const str = self.req.target[f_decl.name.len..self.req.target.len];
860 const field_ti = @typeInfo(field.type);
862 // TODO: maybe handle BoundedArray?
864 @field(arg, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
867 @field(arg, field.name) = try parse_enum(field.type, str, 16);
870 @field(arg, field.name) = str;
874 try @call(.auto, f, .{ self, arg });
882 pub fn @"/user/"(self: Self, args: struct { username: []const u8 }) !void {
883 const user_ids = try Db.user_ids(self.txn);
884 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
885 const users = try Db.users(self.txn);
886 const user = try users.get(user_id);
888 try write_profile(self.res, self.txn, self.logged_in, user);
890 try write_posts(self.res, self.txn, self.logged_in, user.posts);
893 \\<p>User not found [{}]</p>
897 pub fn @"/replies/"(self: Self, args: struct { username: []const u8 }) !void {
898 const user_ids = try Db.user_ids(self.txn);
899 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
900 const users = try Db.users(self.txn);
901 const user = try users.get(user_id);
903 try write_profile(self.res, self.txn, self.logged_in, user);
905 try write_posts(self.res, self.txn, self.logged_in, user.replies);
908 \\<p>User not found [{}]</p>
912 pub fn @"/following/"(self: Self, args: struct { username: []const u8 }) !void {
913 const user_ids = try Db.user_ids(self.txn);
914 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
915 const users = try Db.users(self.txn);
916 const user = try users.get(user_id);
918 const following = try user.following.open(self.txn);
919 var it = following.iterator();
922 \\<h2><a href="/user/{s}">{s}</a> follows:</h2>
923 , .{ user.name.constSlice(), user.display_name.constSlice() });
925 while (it.next()) |following_id| {
926 const following_user = try users.get(following_id);
929 \\<a href="/user/{s}">{s}</a><br />
930 , .{ following_user.name.constSlice(), following_user.display_name.constSlice() });
934 \\<p>User not found [{}]</p>
938 pub fn @"/followers/"(self: Self, args: struct { username: []const u8 }) !void {
939 const user_ids = try Db.user_ids(self.txn);
940 if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
941 const users = try Db.users(self.txn);
942 const user = try users.get(user_id);
944 const followers = try user.followers.open(self.txn);
945 var it = followers.iterator();
948 \\<h2><a href="/user/{s}">{s}</a> followers:</h2>
949 , .{ user.name.constSlice(), user.display_name.constSlice() });
951 while (it.next()) |follower_id| {
952 const follower_user = try users.get(follower_id);
955 \\<a href="/user/{s}">{s}</a><br />
956 , .{ follower_user.name.constSlice(), follower_user.display_name.constSlice() });
960 \\<p>User not found [{}]</p>
964 pub fn @"/post/"(self: Self, args: struct { post_id: PostId }) !void {
965 try write_post(self.res, self.txn, self.logged_in, args.post_id, .{
966 .recurse = 3, // TODO: factor out
967 .show_comment_field = true,
970 pub fn @"/quotes/"(self: Self, args: struct { post_id: PostId }) !void {
971 const posts = try Db.posts(self.txn);
972 const post = try posts.get(args.post_id);
974 const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
976 if (self.logged_in != null) {
977 try html_form(self.res, "/quote", .{
978 .{ "type=\"hidden\" name=\"referer\" value=\"{s}\"", .{referer} },
979 .{ "type=\"hidden\" name=\"post_id\" value=\"{x}\"", .{@intFromEnum(post.id)} },
980 "type=\"text\" name=\"text\" placeholder=\"Text\"",
981 "type=\"submit\" value=\"Quote\"",
983 try self.res.write("<br />", .{});
986 const quotes_view = try post.quotes.open(self.txn);
987 var it = quotes_view.iterator();
988 while (it.next()) |quote_id| {
989 try write_post(self.res, self.txn, self.logged_in, quote_id, .{ .recurse = 1 });
990 try self.res.write("<br />", .{});
993 pub fn @"/list/"(self: Self, args: struct { list_id: PostList.Index }) !void {
994 try write_posts(self.res, self.txn, self.logged_in, PostList{ .idx = args.list_id });
996 pub fn @"/lists"(self: Self) !void {
997 if (self.logged_in) |login| {
998 const post_lists_view = try login.user.post_lists.open(self.txn);
1000 try html_form(self.res, "/create_list", .{
1001 "type=\"text\" name=\"name\"",
1002 "type=\"submit\" value=\"Add\"",
1005 try self.res.write("<br /><br />", .{});
1007 var it = post_lists_view.iterator();
1008 while (it.next()) |kv| {
1009 const name = kv.val.name;
1010 const post_list = kv.val.list;
1012 \\<a href="/list/{x}">{s}</a>
1013 , .{ post_list.idx.?, name.constSlice() });
1014 try html_form(self.res, "/delete_list", .{
1015 .{ "type=\"hidden\" name=\"list_id\" value=\"{x}\"", .{kv.key} },
1016 "type=\"submit\" value=\"Delete\"",
1018 try self.res.write("<br />", .{});
1021 try self.res.write("not logged in", .{});
1024 pub fn @"/feed/"(self: Self, args: struct { feed_id: UserList.Index }) !void {
1025 try write_timeline(self.res, self.txn, self.logged_in, UserList{ .idx = args.feed_id });
1027 pub fn @"/feeds"(self: Self) !void {
1028 if (self.logged_in) |login| {
1029 const feeds_view = try login.user.feeds.open(self.txn);
1031 try html_form(self.res, "/create_feed", .{
1032 "type=\"text\" name=\"name\"",
1033 "type=\"submit\" value=\"Add\"",
1036 try self.res.write("<br /><br />", .{});
1038 var it = feeds_view.iterator();
1039 while (it.next()) |kv| {
1040 const name = kv.val.name;
1041 const user_list = kv.val.list;
1043 \\<a href="/feed/{x}">{s}</a>
1044 , .{ user_list.idx.?, name.constSlice() });
1045 try html_form(self.res, "/delete_feed", .{
1046 .{ "type=\"hidden\" name=\"list_id\" value=\"{x}\"", .{kv.key} },
1047 "type=\"submit\" value=\"Delete\"",
1049 try self.res.write("<br />", .{});
1052 try self.res.write("not logged in", .{});
1055 pub fn @"/post"(self: Self) !void {
1056 if (self.logged_in) |login| {
1058 const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
1060 try html_form(self.res, "/post", .{
1061 .{ "type=\"hidden\" name=\"referer\" value=\"{s}\"", .{referer} },
1062 "type=\"text\" name=\"text\"",
1063 "type=\"submit\" value=\"Post\"",
1066 try self.res.write("not logged in", .{});
1069 pub fn @"/edit"(self: Self) !void {
1070 if (self.logged_in) |login| {
1071 try self.res.write("<br />Username: ", .{});
1072 try html_form(self.res, "/set_username", .{
1073 .{ "type=\"text\" name=\"username\" placeholder=\"{s}\"", .{login.user.name.constSlice()} },
1074 "type=\"submit\" value=\"Change\"",
1076 try self.res.write("<br />Display Name: ", .{});
1077 try html_form(self.res, "/set_display_name", .{
1078 .{ "type=\"text\" name=\"display_name\" placeholder=\"{s}\"", .{login.user.display_name.constSlice()} },
1079 "type=\"submit\" value=\"Change\"",
1081 try self.res.write("<br />Password: ", .{});
1082 try html_form(self.res, "/set_password", .{
1083 "type=\"text\" name=\"password\"",
1084 "type=\"submit\" value=\"Change\"",
1087 try self.res.write("not logged in", .{});
1090 pub fn @"/"(self: Self) !void {
1091 if (self.logged_in) |login| {
1092 try write_timeline(self.res, self.txn, self.logged_in, login.user.following);
1094 // TODO: generic home
1095 try self.res.write("Homepage", .{});
1102 const POST = struct {
1103 const Self = @This();
1107 res: *http.Response,
1110 pub fn handle(self: Self) !bool {
1111 const ti = @typeInfo(Self);
1112 inline for (ti.Struct.decls) |f_decl| {
1113 if (std.mem.eql(u8, f_decl.name, self.req.target)) {
1114 const f = @field(Self, f_decl.name);
1115 const fi = @typeInfo(@TypeOf(f));
1116 if (fi.Fn.params.len == 1) {
1117 _ = try @call(.auto, f, .{self});
1119 const args_type = fi.Fn.params[fi.Fn.params.len - 1].type.?;
1120 const argsi = @typeInfo(args_type);
1121 var args: args_type = undefined;
1122 inline for (argsi.Struct.fields) |field| {
1123 const str = self.req.get_value(field.name) orelse return error.ArgNotFound;
1124 const field_ti = @typeInfo(field.type);
1127 @field(args, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
1130 @field(args, field.name) = try parse_enum(field.type, str, 16);
1133 @field(args, field.name) = str;
1137 try @call(.auto, f, .{ self, args });
1145 pub fn @"/register"(self: Self, args: struct { username: []const u8, password: []const u8 }) !void {
1146 // TODO: handle args not supplied
1147 std.debug.print("New user: {s} {s}\n", .{ args.username, args.password });
1148 _ = try Chirp.register_user(self.env, args.username, args.password);
1150 pub fn @"/login"(self: Self, args: struct { username: []const u8, password: []const u8 }) !void {
1151 // TODO: handle args not supplied
1152 std.debug.print("New login: {s} {s}\n", .{ args.username, args.password });
1153 if (Chirp.login_user(self.env, args.username, args.password)) |session_token| {
1154 self.res.status = .see_other;
1155 try self.res.add_header(
1157 .{ "session_token={x}; HttpOnly", .{session_token} },
1160 std.debug.print("login_user err: {}\n", .{err});
1163 pub fn @"/logout"(self: Self) !void {
1164 if (self.logged_in) |login| {
1165 try Chirp.logout_user(self.env, login.session_token);
1167 try self.res.add_header(
1169 .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
1173 pub fn @"/set_username"(self: Self, args: struct { username: []const u8 }) !void {
1174 const login = self.logged_in orelse return error.NotLoggedIn;
1175 const username = try Username.fromSlice(args.username);
1177 const txn = try self.env.txn();
1178 defer txn.commit() catch {};
1180 const user_ids = try Db.user_ids(txn);
1182 if (!try user_ids.has(username)) {
1183 try user_ids.del(login.user.name);
1184 try user_ids.put(username, login.user.id);
1186 const users = try Db.users(txn);
1187 var user = login.user;
1188 user.name = username;
1189 try users.put(login.user.id, user);
1192 pub fn @"/set_display_name"(self: Self, args: struct { display_name: []const u8 }) !void {
1193 const login = self.logged_in orelse return error.NotLoggedIn;
1194 const display_name = try DisplayName.fromSlice(args.display_name);
1196 const txn = try self.env.txn();
1197 defer txn.commit() catch {};
1199 const users = try Db.users(txn);
1200 var user = login.user;
1201 user.display_name = display_name;
1202 try users.put(login.user.id, user);
1204 pub fn @"/set_password"(self: Self, args: struct { password: []const u8 }) !void {
1205 const login = self.logged_in orelse return error.NotLoggedIn;
1207 const txn = try self.env.txn();
1208 defer txn.commit() catch {};
1210 const users = try Db.users(txn);
1211 var user = login.user;
1212 user.password_hash = try Chirp.hash_password(args.password);
1213 try users.put(login.user.id, user);
1215 pub fn @"/post"(self: Self) !void {
1216 if (self.logged_in) |login| {
1217 const text = self.req.get_value("text").?;
1218 const has_referer = self.req.get_value("referer");
1220 try Chirp.post(self.env, login.user.id, text);
1222 if (has_referer) |r| {
1223 const decoded = try decode(r);
1224 try self.res.redirect(decoded.constSlice());
1228 pub fn @"/comment"(self: Self) !void {
1229 if (self.logged_in) |login| {
1230 const text = self.req.get_value("text") orelse return error.NoText;
1231 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1232 const post_id = try parse_enum(PostId, post_id_str, 16);
1234 try Chirp.comment(self.env, login.user.id, post_id, text);
1237 pub fn @"/quote"(self: Self) !void {
1238 if (self.logged_in) |login| {
1239 const text = self.req.get_value("text") orelse return error.NoText;
1240 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1241 const has_referer = self.req.get_value("referer");
1243 const post_id = try parse_enum(PostId, post_id_str, 16);
1245 try Chirp.quote(self.env, login.user.id, post_id, text);
1247 if (has_referer) |r| {
1248 const decoded = try decode(r);
1249 try self.res.redirect(decoded.constSlice());
1253 // TODO: add arguments instead of parsing manually
1254 pub fn @"/create_list"(self: Self) !void {
1255 if (self.logged_in) |login| {
1256 const name_str = self.req.get_value("name") orelse return error.NoName;
1257 const name = try Name.fromSlice(name_str);
1258 // TODO: decode name
1260 var txn = try self.env.txn();
1261 const postlist = try PostList.init(txn);
1264 txn = try self.env.txn();
1265 var post_lists_view = try login.user.post_lists.open(txn);
1266 _ = try post_lists_view.append(.{ .name = name, .list = postlist });
1270 pub fn @"/delete_list"(self: Self, args: struct { list_id: PostList.Index }) !void {
1271 if (self.logged_in) |login| {
1272 var post_list: ?PostList = null;
1274 const txn = try self.env.txn();
1275 defer txn.commit() catch {};
1276 var post_lists_view = try login.user.post_lists.open(txn);
1277 post_list = (try post_lists_view.get(args.list_id)).list;
1278 try post_lists_view.del(args.list_id);
1280 if (post_list != null) {
1281 const txn = try self.env.txn();
1282 defer txn.commit() catch {};
1283 var list_view = try post_list.?.open(txn);
1284 try list_view.clear();
1288 pub fn @"/list_add"(self: Self, args: struct { list_id: PostList.Index, post_id: PostId }) !void {
1289 if (self.logged_in) |login| {
1292 const txn = try self.env.txn();
1293 defer txn.commit() catch {};
1295 const post_list = PostList{ .idx = args.list_id };
1296 var post_list_view = try post_list.open(txn);
1297 if (try post_list_view.has(args.post_id)) {
1298 try post_list_view.del(args.post_id);
1300 try post_list_view.append(args.post_id);
1304 pub fn @"/create_feed"(self: Self) !void {
1305 if (self.logged_in) |login| {
1306 const name_str = self.req.get_value("name") orelse return error.NoName;
1307 const name = try Name.fromSlice(name_str);
1309 var txn = try self.env.txn();
1310 const userlist = try UserList.init(txn);
1313 txn = try self.env.txn();
1314 var feeds_view = try login.user.feeds.open(txn);
1315 _ = try feeds_view.append(.{ .name = name, .list = userlist });
1319 pub fn @"/delete_feed"(self: Self, args: struct { list_id: UserList.Index }) !void {
1320 if (self.logged_in) |login| {
1321 var user_list: ?UserList = null;
1324 const txn = try self.env.txn();
1325 defer txn.commit() catch {};
1326 var feeds_view = try login.user.feeds.open(txn);
1327 user_list = (try feeds_view.get(args.list_id)).list;
1328 try feeds_view.del(args.list_id);
1330 if (user_list != null) {
1331 const txn = try self.env.txn();
1332 defer txn.commit() catch {};
1333 var list_view = try user_list.?.open(txn);
1334 try list_view.clear();
1338 pub fn @"/feed_add"(self: Self, args: struct { feed_id: UserList.Index, user_id: UserId }) !void {
1339 if (self.logged_in) |login| {
1342 const txn = try self.env.txn();
1343 defer txn.commit() catch {};
1345 const user_list = UserList{ .idx = args.feed_id };
1346 var user_list_view = try user_list.open(txn);
1347 if (try user_list_view.has(args.user_id)) {
1348 try user_list_view.del(args.user_id);
1350 try user_list_view.append(args.user_id);
1354 pub fn @"/upvote"(self: Self) !void {
1355 const login = self.logged_in orelse return error.NotLoggedIn;
1357 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1358 const post_id: PostId = @enumFromInt(try std.fmt.parseUnsigned(u64, post_id_str, 16));
1360 try Chirp.vote(self.env, post_id, login.user.id, .Up);
1362 pub fn @"/downvote"(self: Self) !void {
1363 const login = self.logged_in orelse return error.NotLoggedIn;
1365 const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId;
1366 const post_id: PostId = @enumFromInt(try std.fmt.parseUnsigned(u64, post_id_str, 16));
1368 try Chirp.vote(self.env, post_id, login.user.id, .Down);
1370 pub fn @"/follow"(self: Self) !void {
1371 const login = self.logged_in orelse return error.NotLoggedIn;
1373 const user_id_str = self.req.get_value("user_id") orelse return error.NoUserId;
1374 const user_id: UserId = @enumFromInt(try std.fmt.parseUnsigned(u64, user_id_str, 16));
1376 try Chirp.follow(self.env, login.user.id, user_id);
1378 pub fn @"/quit"(self: Self) !void {
1379 if (self.req.get_header("Referer")) |ref| {
1380 try self.res.redirect(ref);
1382 try self.res.redirect("/");
1384 try self.res.send();
1390 fn list_users(env: lmdb.Env) !void {
1391 const txn = try env.txn();
1394 const users = try Db.users(txn);
1395 var it = try users.iterator();
1397 while (it.next()) |kv| {
1399 const user = kv.val;
1400 std.debug.print("[{}] {s}\n", .{ key, user.name.constSlice() });
1403 fn list_user_ids(env: lmdb.Env) !void {
1404 const txn = try env.txn();
1407 const user_ids = try Db.user_ids(txn);
1408 var it = try user_ids.iterator();
1410 while (it.next()) |kv| {
1412 const user_id = kv.val;
1413 std.debug.print("[{s}] {}\n", .{ key.constSlice(), user_id });
1417 fn list_sessions(env: lmdb.Env) !void {
1418 const txn = try env.txn();
1421 const sessions = try Db.sessions(txn);
1422 var it = try sessions.iterator();
1424 while (it.next()) |kv| {
1426 const user_id = kv.val;
1427 std.debug.print("[{x}] {}\n", .{ key, user_id });
1431 fn list_posts(env: lmdb.Env) !void {
1432 const txn = try env.txn();
1435 const posts = try Db.posts(txn);
1436 var it = try posts.iterator();
1438 while (it.next()) |kv| {
1440 const post = kv.val;
1441 std.debug.print("[{}] {s}\n", .{ key, post.text.constSlice() });
1445 const ReqBufferSize = 4096;
1446 const ResHeadBufferSize = 1024 * 64;
1447 const ResBodyBufferSize = 1024 * 64;
1450 var req_buffer: [ReqBufferSize]u8 = undefined;
1451 var res_head_buffer: [ResHeadBufferSize]u8 = undefined;
1452 var res_body_buffer: [ResBodyBufferSize]u8 = undefined;
1454 pub fn main() !void {
1456 var server = try http.Server.init("::", 8080);
1457 defer server.deinit();
1460 var env = try lmdb.Env.open("db", 1024 * 1024 * 10);
1463 std.debug.print("Users:\n", .{});
1464 try list_users(env);
1465 std.debug.print("User IDs:\n", .{});
1466 try list_user_ids(env);
1467 std.debug.print("Sessions:\n", .{});
1468 try list_sessions(env);
1469 std.debug.print("Posts:\n", .{});
1470 try list_posts(env);
1475 const req = (server.next_request(&req_buffer) catch break) orelse break;
1476 handle_request(env, req) catch {
1477 try handle_error(env, req);
1481 // const ThreadCount = 1;
1482 // var ts: [ThreadCount]std.Thread = undefined;
1484 // for (0..ThreadCount) |i| {
1485 // ts[i] = try std.Thread.spawn(.{}, handle_connection, .{ &server, &env });
1487 // for (0..ThreadCount) |i| {
1491 std.debug.print("done\n", .{});
1494 fn handle_error(env: lmdb.Env, req: http.Request) !void {
1496 var res = http.Response.init(req.fd, &res_head_buffer, &res_body_buffer);
1497 try write_start(&res);
1498 try res.write("Oops, something went terribly wrong there D:", .{});
1499 try write_end(&res);
1502 fn handle_request(env: lmdb.Env, req: http.Request) !void {
1503 // std.debug.print("[{}]: {s}\n", .{ req.method, req.target });
1504 // std.debug.print("[{}]: {s}\n", .{ req.method, req.head.? });
1507 var res = http.Response.init(req.fd, &res_head_buffer, &res_body_buffer);
1509 // check session token
1510 const logged_in: ?Login = try check_login(env, req, &res);
1513 if (req.method == .GET) {
1514 try write_start(&res);
1515 try write_header(&res, logged_in);
1517 const txn = try env.txn();
1524 .logged_in = logged_in,
1526 if (try get.handle()) {} else {
1527 try res.redirect("/");
1530 try write_end(&res);
1539 .logged_in = logged_in,
1541 if (try post.handle()) {} else {
1542 try res.write("<p>[POST] {s}</p>", .{req.target});
1545 if (!res.has_header("Location")) {
1546 if (req.get_header("Referer")) |ref| {
1547 try res.redirect(ref);
1549 try res.redirect("/");