]> gitweb.ps.run Git - chirp/blob - src/main.zig
remove html form args, add replies, change handle_connection
[chirp] / src / main.zig
1 const std = @import("std");
2 const lmdb = @import("lmdb");
3 const db = @import("db");
4 const http = @import("http");
5
6 // db {{{
7
8 const Db = struct {
9     fn users(txn: lmdb.Txn) !db.Db(UserId, User) {
10         return try db.Db(UserId, User).init(txn, "users");
11     }
12     fn user_ids(txn: lmdb.Txn) !db.Db(Username, UserId) {
13         return try db.Db(Username, UserId).init(txn, "user_ids");
14     }
15     fn sessions(txn: lmdb.Txn) !db.Db(SessionToken, UserId) {
16         return try db.Db(SessionToken, UserId).init(txn, "sessions");
17     }
18     fn posts(txn: lmdb.Txn) !db.Db(PostId, Post) {
19         return try db.Db(PostId, Post).init(txn, "posts");
20     }
21 };
22
23 // }}}
24
25 // content {{{
26
27 const User = struct {
28     // TODO: choose sizes
29     id: UserId,
30     name: Username,
31     display_name: DisplayName,
32     password_hash: PasswordHash,
33     posts: PostList,
34     replies: PostList,
35     following: UserList,
36     followers: UserList,
37
38     post_lists: PostListList,
39     feeds: UserListList,
40 };
41
42 const Post = struct {
43     id: PostId,
44     parent_id: ?PostId,
45     quote_id: ?PostId,
46
47     user_id: UserId,
48     time: Timestamp,
49
50     upvotes: u64 = 0,
51     downvotes: u64 = 0,
52     votes: VoteList,
53     comments: PostList,
54     quotes: PostList,
55
56     text: PostText,
57 };
58
59 const SavedPostList = struct {
60     name: Name,
61     list: PostList,
62 };
63 const SavedUserList = struct {
64     name: Name,
65     list: UserList,
66 };
67
68 const Vote = struct {
69     const Kind = enum { Up, Down };
70
71     kind: Kind,
72     time: Timestamp,
73 };
74
75 const Id = u64;
76 const Login = struct {
77     user: User,
78     session_token: SessionToken,
79 };
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);
95
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));
98 }
99
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);
103
104     const len = @min(text.len, 1024); // TODO: PostText length
105
106     var idx: usize = 0;
107     while (idx < len) : (idx += 1) {
108         const c = text[idx];
109         if (c == '+') {
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])) {
118                     num_end += 1;
119                 }
120
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')
125                 {
126                     try std.fmt.format(result.writer(), "&#{s};", .{text[num_start..num_end]});
127                     idx = num_end + 2;
128                     continue;
129                 }
130             }
131
132             try std.fmt.format(result.writer(), "&#x{s};", .{text[idx + 1 .. idx + 3]});
133             idx += 2;
134         } else {
135             try result.append(c);
136         }
137     }
138
139     return result;
140 }
141
142 fn decode(text: []const u8) !std.BoundedArray(u8, 1024) {
143     var result = try std.BoundedArray(u8, 1024).init(0);
144
145     const max_len = @min(text.len, 1024); // TODO: PostText length
146
147     var idx: usize = 0;
148     var len: usize = 0;
149     while (len < max_len and idx < text.len) : ({
150         idx += 1;
151         len += 1;
152     }) {
153         const c = text[idx];
154         if (c == '+') {
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)});
159             }
160             idx += 2;
161         } else {
162             try result.append(c);
163         }
164     }
165
166     return result;
167 }
168
169 const Chirp = struct {
170     pub fn hash_password(password: []const u8) !PasswordHash {
171         var hash_buffer = try PasswordHash.init(128);
172
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);
177
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());
183
184         try hash_buffer.resize(result.len);
185
186         return hash_buffer;
187     }
188
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);
192
193         if (std.crypto.pwhash.argon2.strVerify(hash.constSlice(), password, .{
194             .allocator = alloc.allocator(),
195         })) {
196             return true;
197         } else |err| {
198             std.debug.print("verify error: {}\n", .{err});
199             return false;
200         }
201     }
202
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);
206
207         const txn = try env.txn();
208         defer txn.commit() catch |err| {
209             std.debug.print("error registering user: {}\n", .{err});
210         };
211
212         const users = try Db.users(txn);
213         const user_ids = try Db.user_ids(txn);
214
215         if (try user_ids.has(username_array)) {
216             return false;
217         } else {
218             const user_id = try db.Prng.gen(users.dbi, UserId);
219
220             try users.put(user_id, User{
221                 .id = user_id,
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),
231             });
232
233             try user_ids.put(username_array, user_id);
234
235             return true;
236         }
237     }
238
239     pub fn login_user(
240         env: lmdb.Env,
241         username: []const u8,
242         password: []const u8,
243     ) !SessionToken {
244         const username_array = try Username.fromSlice(username);
245
246         const txn = try env.txn();
247         defer txn.commit() catch {};
248
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});
252
253         const users = try Db.users(txn);
254         const user = try users.get(user_id);
255
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;
261         } else {
262             return error.IncorrectPassword;
263         }
264     }
265
266     fn logout_user(env: lmdb.Env, session_token: SessionToken) !void {
267         const txn = try env.txn();
268         defer txn.commit() catch {};
269
270         const sessions = try Db.sessions(txn);
271         try sessions.del(session_token);
272     }
273
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;
276
277         // TODO: do this in one commit
278
279         var txn: lmdb.Txn = undefined;
280         {
281             // create post
282             txn = try env.txn();
283             defer txn.commit() catch {};
284
285             const posts = try Db.posts(txn);
286             post_id = try db.Prng.gen(posts.dbi, PostId);
287
288             const decoded_text = try reencode(text);
289             try posts.put(post_id, Post{
290                 .id = post_id,
291                 .parent_id = parent_id,
292                 .quote_id = quote_id,
293                 .user_id = user_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,
299             });
300         }
301
302         {
303             // append to user's posts
304             txn = try env.txn();
305             defer txn.commit() catch {};
306
307             var posts_view = try post_list.open(txn);
308             try posts_view.append(post_id);
309         }
310
311         if (quote_id != null) {
312             txn = try env.txn();
313             defer txn.commit() catch {};
314
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);
319         }
320
321         return post_id;
322     }
323
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);
328         txn.abort();
329
330         const post_id = try append_post(env, user_id, user.posts, null, null, text);
331
332         txn = try env.txn();
333         var replies_view = try user.replies.open(txn);
334         try replies_view.append(post_id);
335         try txn.commit();
336     }
337
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);
342
343         const posts = try Db.posts(txn);
344         const parent_post = try posts.get(parent_post_id);
345         txn.abort();
346
347         const post_id = try append_post(env, user_id, parent_post.comments, parent_post_id, null, text);
348
349         txn = try env.txn();
350         var replies_view = try user.replies.open(txn);
351         try replies_view.append(post_id);
352         try txn.commit();
353     }
354
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);
359         txn.abort();
360
361         const post_id = try append_post(env, user_id, user.posts, null, quote_post_id, text);
362
363         txn = try env.txn();
364         var replies_view = try user.replies.open(txn);
365         try replies_view.append(post_id);
366         try txn.commit();
367     }
368
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 {};
372
373         const posts = try Db.posts(txn);
374
375         var p = try posts.get(post_id);
376         var votes_view = try p.votes.open(txn);
377
378         var add_vote = true;
379
380         if (try votes_view.has(user_id)) {
381             const old_vote = try votes_view.get(user_id);
382
383             add_vote = old_vote.kind != kind;
384
385             try votes_view.del(user_id);
386
387             switch (old_vote.kind) {
388                 .Up => p.upvotes -= 1,
389                 .Down => p.downvotes -= 1,
390             }
391             try posts.put(post_id, p);
392         }
393
394         if (add_vote) {
395             try votes_view.append(user_id, Vote{
396                 .kind = kind,
397                 .time = std.time.timestamp(),
398             });
399
400             if (kind == .Up) {
401                 p.upvotes += 1;
402             } else {
403                 p.downvotes += 1;
404             }
405             try posts.put(post_id, p);
406         }
407     }
408
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 {};
412
413         const users = try Db.users(txn);
414
415         const user = try users.get(user_id);
416         const user_to_follow = try users.get(user_id_to_follow);
417
418         var user_following = try user.following.open(txn);
419         var user_to_follow_followers = try user_to_follow.followers.open(txn);
420
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);
427         } else {
428             std.debug.print("Something went wrong when trying to unfollow\n", .{});
429         }
430     }
431
432     fn get_session_user_id(env: lmdb.Env, session_token: SessionToken) !UserId {
433         const txn = try env.txn();
434         defer txn.abort();
435
436         const sessions = try Db.sessions(txn);
437
438         return try sessions.get(session_token);
439     }
440
441     fn get_user(env: lmdb.Env, user_id: UserId) !User {
442         const txn = try env.txn();
443         defer txn.abort();
444
445         const users = try Db.users(txn);
446         return try users.get(user_id);
447     }
448 };
449
450 // }}}
451
452 // html {{{
453 fn html_form(res: *http.Response, action: []const u8, inputs: anytype) !void {
454     try res.write("<form action=\"{s}\" method=\"post\">", .{action});
455
456     inline for (inputs) |input| {
457         switch (@typeInfo(@TypeOf(input))) {
458             .Struct => {
459                 try res.write("<input ", .{});
460                 try res.write(input[0], input[1]);
461                 try res.write(" />", .{});
462             },
463             else => {
464                 try res.write("<input ", .{});
465                 try res.write(input, .{});
466                 try res.write(" />", .{});
467             },
468         }
469     }
470
471     try res.write("</form>", .{});
472 }
473 // }}}
474
475 // write {{{
476 const TimeStr = std.BoundedArray(u8, 256);
477
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;
482
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;
493
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;
497
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;
499
500     return result;
501 }
502 fn write_header(res: *http.Response, logged_in: ?Login) !void {
503     if (logged_in) |login| {
504         try res.write(
505             \\<a href="/">Home</a><br />
506         , .{});
507         try res.write(
508             \\<a href="/user/{s}">Profile</a><br />
509         , .{login.user.name.constSlice()});
510         try res.write(
511             \\<a href="/post">Post</a><br />
512         , .{});
513         try html_form(res, "/logout", .{
514             \\type="submit" value="Logout"
515         });
516         try res.write("<br /><br />", .{});
517     } else {
518         try res.write(
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" />
524             \\</form>
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>
530         , .{});
531         try html_form(res, "/quit", .{
532             \\type="submit" value="Quit"
533         });
534         try res.write("<br /><br />", .{});
535     }
536 }
537 fn write_start(res: *http.Response) !void {
538     try res.write(
539         \\<!doctype html>
540         \\<html>
541         \\<head>
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>&#128035;</text></svg>">
543         \\<style>
544         \\  form {
545         \\    display: inline-block;
546         \\  }
547         \\</style>
548         \\</head>
549         \\<body>
550     , .{});
551 }
552 fn write_end(res: *http.Response) !void {
553     try res.write("</body></html>", .{});
554 }
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 {};
559         return;
560     };
561     const users = try Db.users(txn);
562     const user = try users.get(post.user_id);
563
564     try res.write(
565         \\<div>
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)});
570     }
571     try res.write(
572         \\ {s}</span><br />
573         \\<span>{s}</span><br />
574     , .{ time_str(post.time).constSlice(), post.text.constSlice() });
575
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 });
580         } else {
581             try res.write("<a href=\"/post/{x}\">...</a>", .{@intFromEnum(post_id)});
582         }
583         try res.write("</div>", .{});
584     }
585
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);
589
590     // Votes
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;
592
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=\"&#x2B06; {}\"", .{post.upvotes} },
597         });
598     } else {
599         try html_form(res, "/upvote", .{
600             .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
601             .{ "type=\"submit\" value=\"&#x21E7; {}\"", .{post.upvotes} },
602         });
603     }
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=\"&#x2B07; {}\"", .{post.downvotes} },
608         });
609     } else {
610         try html_form(res, "/downvote", .{
611             .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
612             .{ "type=\"submit\" value=\"&#x21E9; {}\"", .{post.downvotes} },
613         });
614     }
615
616     // Comment Count
617     try res.write(
618         \\<a href="/post/{x}">&#x1F4AD; {}</a> 
619     , .{ @intFromEnum(post.id), comments_view.len() });
620
621     // Quote
622     try res.write(
623         \\<a href="/quotes/{x}">&#x1F501; {}</a> 
624     , .{ @intFromEnum(post.id), quotes_view.len() });
625
626     // Save to List
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() });
637         }
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>", .{});
642     }
643
644     // Comment field
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\"",
652         });
653         try res.write("<br />", .{});
654     }
655
656     // Comments
657     if (options.recurse > 0 and comments_view.len() > 0) {
658         try res.write(
659             \\<details{s}>
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();
664         var count: u8 = 0;
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) {
669                 count += 1;
670                 if (count >= 3) break;
671             }
672         }
673         try res.write(
674             \\</div>
675             \\</details>
676         , .{});
677     }
678
679     try res.write(
680         \\</div>
681     , .{});
682 }
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);
686
687     try res.write(
688         \\<h2 style="display: inline;"><a href="/user/{s}">{s}</a></h2>
689     , .{
690         user.name.constSlice(), user.display_name.constSlice(),
691     });
692     if (logged_in != null and user.id != logged_in.?.user.id) {
693         const login = logged_in.?;
694
695         // follow/unfollow
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"
700             });
701         } else {
702             try html_form(res, "/follow", .{
703                 .{ "type=\"hidden\" name=\"user.id\" value=\"{x}\"", .{@intFromEnum(user.id)} },
704                 \\type="submit" value="Follow"
705             });
706         }
707
708         // add to feed
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() });
717         }
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>", .{});
722     }
723     try res.write(
724         \\ <a href="/following/{s}">{} following</a>
725         \\ <a href="/followers/{s}">{} followers</a>
726         \\<br />
727     , .{
728         user.name.constSlice(), following.len(),
729         user.name.constSlice(), followers.len(),
730     });
731
732     try res.write("<a href=\"/replies/{s}\">Replies</a><br />", .{
733         user.name.constSlice(),
734     });
735
736     if (logged_in != null and user.id == logged_in.?.user.id) {
737         try res.write(
738             \\<a href="/lists">Lists</a>
739             \\<a href="/feeds">Feeds</a>
740             \\<a href="/edit">Edit</a><br />
741             \\<br />
742         , .{});
743     }
744
745     try res.write("<br />", .{});
746 }
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);
749
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 />", .{});
754     }
755 }
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);
759
760     var newest_post_ids = try std.BoundedArray(PostId, 10).init(0); // TODO: TimelinePostsCount
761     var prev_newest_post: ?Post = null;
762
763     const following = try user_list.open(txn);
764
765     while (true) {
766         var newest_post: ?Post = null;
767
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);
772
773             if (followed_posts.len() == 0) {
774                 continue;
775             }
776
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);
780
781                 if ((prev_newest_post == null or p.time < prev_newest_post.?.time) and (newest_post == null or newest_post.?.time < p.time)) {
782                     newest_post = p;
783                     break;
784                 }
785             }
786         }
787         if (newest_post) |post| {
788             newest_post_ids.append(post.id) catch break;
789             prev_newest_post = post;
790         } else {
791             break;
792         }
793     }
794
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 />", .{});
798     }
799 }
800 fn check_login(env: lmdb.Env, req: http.Request, res: *http.Response) !?Login {
801     var result: ?Login = null;
802
803     if (req.get_cookie("session_token")) |session_token_str| {
804         var remove_session_token = true;
805
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();
811                 defer txn.abort();
812                 const users = try Db.users(txn);
813
814                 result = .{
815                     .user = try users.get(user_id),
816                     .session_token = session_token,
817                 };
818
819                 remove_session_token = false;
820             }
821         }
822
823         if (remove_session_token) {
824             try res.add_header(
825                 "Set-Cookie",
826                 .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
827             );
828         }
829     }
830
831     return result;
832 }
833 // }}}
834
835 // GET {{{
836 const GET = struct {
837     const Self = @This();
838
839     txn: lmdb.Txn,
840     req: http.Request,
841     res: *http.Response,
842     logged_in: ?Login,
843
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);
849
850             if (match) {
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});
855                 } else {
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;
862                     }
863                     const str = self.req.target[f_decl.name.len..self.req.target.len];
864                     const field_ti = @typeInfo(field.type);
865                     switch (field_ti) {
866                         // TODO: maybe handle BoundedArray?
867                         .Int => {
868                             @field(arg, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
869                         },
870                         .Enum => {
871                             @field(arg, field.name) = try parse_enum(field.type, str, 16);
872                         },
873                         else => {
874                             @field(arg, field.name) = str;
875                         },
876                     }
877
878                     try @call(.auto, f, .{ self, arg });
879                 }
880                 return true;
881             }
882         }
883         return false;
884     }
885
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);
891
892             try write_profile(self.res, self.txn, self.logged_in, user);
893
894             try write_posts(self.res, self.txn, self.logged_in, user.posts);
895         } else |err| {
896             try self.res.write(
897                 \\<p>User not found [{}]</p>
898             , .{err});
899         }
900     }
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);
906
907             try write_profile(self.res, self.txn, self.logged_in, user);
908
909             try write_posts(self.res, self.txn, self.logged_in, user.replies);
910         } else |err| {
911             try self.res.write(
912                 \\<p>User not found [{}]</p>
913             , .{err});
914         }
915     }
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);
921
922             const following = try user.following.open(self.txn);
923             var it = following.iterator();
924
925             try self.res.write(
926                 \\<h2><a href="/user/{s}">{s}</a> follows:</h2>
927             , .{ user.name.constSlice(), user.display_name.constSlice() });
928
929             while (it.next()) |following_id| {
930                 const following_user = try users.get(following_id);
931
932                 try self.res.write(
933                     \\<a href="/user/{s}">{s}</a><br />
934                 , .{ following_user.name.constSlice(), following_user.display_name.constSlice() });
935             }
936         } else |err| {
937             try self.res.write(
938                 \\<p>User not found [{}]</p>
939             , .{err});
940         }
941     }
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);
947
948             const followers = try user.followers.open(self.txn);
949             var it = followers.iterator();
950
951             try self.res.write(
952                 \\<h2><a href="/user/{s}">{s}</a> followers:</h2>
953             , .{ user.name.constSlice(), user.display_name.constSlice() });
954
955             while (it.next()) |follower_id| {
956                 const follower_user = try users.get(follower_id);
957
958                 try self.res.write(
959                     \\<a href="/user/{s}">{s}</a><br />
960                 , .{ follower_user.name.constSlice(), follower_user.display_name.constSlice() });
961             }
962         } else |err| {
963             try self.res.write(
964                 \\<p>User not found [{}]</p>
965             , .{err});
966         }
967     }
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,
972         });
973     }
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);
977
978         const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
979
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\"",
986             });
987             try self.res.write("<br />", .{});
988         }
989
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 />", .{});
995         }
996     }
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 });
999     }
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);
1003
1004             try html_form(self.res, "/create_list", .{
1005                 "type=\"text\" name=\"name\"",
1006                 "type=\"submit\" value=\"Add\"",
1007             });
1008
1009             try self.res.write("<br /><br />", .{});
1010
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;
1015                 try self.res.write(
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\"",
1021                 });
1022                 try self.res.write("<br />", .{});
1023             }
1024         } else {
1025             try self.res.write("not logged in", .{});
1026         }
1027     }
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 });
1030     }
1031     pub fn @"/feeds"(self: Self) !void {
1032         if (self.logged_in) |login| {
1033             const feeds_view = try login.user.feeds.open(self.txn);
1034
1035             try html_form(self.res, "/create_feed", .{
1036                 "type=\"text\" name=\"name\"",
1037                 "type=\"submit\" value=\"Add\"",
1038             });
1039
1040             try self.res.write("<br /><br />", .{});
1041
1042             var it = feeds_view.iterator();
1043             while (it.next()) |kv| {
1044                 const name = kv.val.name;
1045                 const user_list = kv.val.list;
1046                 try self.res.write(
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\"",
1052                 });
1053                 try self.res.write("<br />", .{});
1054             }
1055         } else {
1056             try self.res.write("not logged in", .{});
1057         }
1058     }
1059     pub fn @"/post"(self: Self) !void {
1060         if (self.logged_in) |login| {
1061             _ = login;
1062             const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
1063
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\"",
1068             });
1069         } else {
1070             try self.res.write("not logged in", .{});
1071         }
1072     }
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\"",
1079             });
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\"",
1084             });
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\"",
1089             });
1090         } else {
1091             try self.res.write("not logged in", .{});
1092         }
1093     }
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);
1097         } else {
1098             // TODO: generic home
1099             try self.res.write("Homepage", .{});
1100         }
1101     }
1102 };
1103 // }}}
1104
1105 // POST {{{
1106 const POST = struct {
1107     const Self = @This();
1108
1109     env: lmdb.Env,
1110     req: http.Request,
1111     res: *http.Response,
1112     logged_in: ?Login,
1113
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});
1122                 } else {
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);
1129                         switch (field_ti) {
1130                             .Int => {
1131                                 @field(args, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
1132                             },
1133                             .Enum => {
1134                                 @field(args, field.name) = try parse_enum(field.type, str, 16);
1135                             },
1136                             else => {
1137                                 @field(args, field.name) = str;
1138                             },
1139                         }
1140                     }
1141                     try @call(.auto, f, .{ self, args });
1142                 }
1143                 return true;
1144             }
1145         }
1146         return false;
1147     }
1148
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);
1153     }
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(
1160                 "Set-Cookie",
1161                 .{ "session_token={x}; HttpOnly", .{session_token} },
1162             );
1163         } else |err| {
1164             std.debug.print("login_user err: {}\n", .{err});
1165         }
1166     }
1167     pub fn @"/logout"(self: Self) !void {
1168         if (self.logged_in) |login| {
1169             try Chirp.logout_user(self.env, login.session_token);
1170
1171             try self.res.add_header(
1172                 "Set-Cookie",
1173                 .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
1174             );
1175         }
1176     }
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);
1180
1181         const txn = try self.env.txn();
1182         defer txn.commit() catch {};
1183
1184         const user_ids = try Db.user_ids(txn);
1185
1186         if (!try user_ids.has(username)) {
1187             try user_ids.del(login.user.name);
1188             try user_ids.put(username, login.user.id);
1189
1190             const users = try Db.users(txn);
1191             var user = login.user;
1192             user.name = username;
1193             try users.put(login.user.id, user);
1194         }
1195     }
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);
1199
1200         const txn = try self.env.txn();
1201         defer txn.commit() catch {};
1202
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);
1207     }
1208     pub fn @"/set_password"(self: Self, args: struct { password: []const u8 }) !void {
1209         const login = self.logged_in orelse return error.NotLoggedIn;
1210
1211         const txn = try self.env.txn();
1212         defer txn.commit() catch {};
1213
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);
1218     }
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");
1223
1224             try Chirp.post(self.env, login.user.id, text);
1225
1226             if (has_referer) |r| {
1227                 const decoded = try decode(r);
1228                 try self.res.redirect(decoded.constSlice());
1229             }
1230         }
1231     }
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);
1237
1238             try Chirp.comment(self.env, login.user.id, post_id, text);
1239         }
1240     }
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");
1246
1247             const post_id = try parse_enum(PostId, post_id_str, 16);
1248
1249             try Chirp.quote(self.env, login.user.id, post_id, text);
1250
1251             if (has_referer) |r| {
1252                 const decoded = try decode(r);
1253                 try self.res.redirect(decoded.constSlice());
1254             }
1255         }
1256     }
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
1263
1264             var txn = try self.env.txn();
1265             const postlist = try PostList.init(txn);
1266             try txn.commit();
1267
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 });
1271             try txn.commit();
1272         }
1273     }
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;
1277             {
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);
1283             }
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();
1289             }
1290         }
1291     }
1292     pub fn @"/list_add"(self: Self, args: struct { list_id: PostList.Index, post_id: PostId }) !void {
1293         if (self.logged_in) |login| {
1294             _ = login;
1295
1296             const txn = try self.env.txn();
1297             defer txn.commit() catch {};
1298
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);
1303             } else {
1304                 try post_list_view.append(args.post_id);
1305             }
1306         }
1307     }
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);
1312
1313             var txn = try self.env.txn();
1314             const userlist = try UserList.init(txn);
1315             try txn.commit();
1316
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 });
1320             try txn.commit();
1321         }
1322     }
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;
1326
1327             {
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);
1333             }
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();
1339             }
1340         }
1341     }
1342     pub fn @"/feed_add"(self: Self, args: struct { feed_id: UserList.Index, user_id: UserId }) !void {
1343         if (self.logged_in) |login| {
1344             _ = login;
1345
1346             const txn = try self.env.txn();
1347             defer txn.commit() catch {};
1348
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);
1353             } else {
1354                 try user_list_view.append(args.user_id);
1355             }
1356         }
1357     }
1358     pub fn @"/upvote"(self: Self) !void {
1359         const login = self.logged_in orelse return error.NotLoggedIn;
1360
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));
1363
1364         try Chirp.vote(self.env, post_id, login.user.id, .Up);
1365     }
1366     pub fn @"/downvote"(self: Self) !void {
1367         const login = self.logged_in orelse return error.NotLoggedIn;
1368
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));
1371
1372         try Chirp.vote(self.env, post_id, login.user.id, .Down);
1373     }
1374     pub fn @"/follow"(self: Self) !void {
1375         const login = self.logged_in orelse return error.NotLoggedIn;
1376
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));
1379
1380         try Chirp.follow(self.env, login.user.id, user_id);
1381     }
1382     pub fn @"/quit"(self: Self) !void {
1383         if (self.req.get_header("Referer")) |ref| {
1384             try self.res.redirect(ref);
1385         } else {
1386             try self.res.redirect("/");
1387         }
1388         try self.res.send();
1389         // break :accept;
1390     }
1391 };
1392 // }}}
1393
1394 fn list_users(env: lmdb.Env) !void {
1395     const txn = try env.txn();
1396     defer txn.abort();
1397
1398     const users = try Db.users(txn);
1399     var it = try users.iterator();
1400
1401     while (it.next()) |kv| {
1402         const key = kv.key;
1403         const user = kv.val;
1404         std.debug.print("[{}] {s}\n", .{ key, user.name.constSlice() });
1405     }
1406 }
1407 fn list_user_ids(env: lmdb.Env) !void {
1408     const txn = try env.txn();
1409     defer txn.abort();
1410
1411     const user_ids = try Db.user_ids(txn);
1412     var it = try user_ids.iterator();
1413
1414     while (it.next()) |kv| {
1415         const key = kv.key;
1416         const user_id = kv.val;
1417         std.debug.print("[{s}] {}\n", .{ key.constSlice(), user_id });
1418     }
1419 }
1420
1421 fn list_sessions(env: lmdb.Env) !void {
1422     const txn = try env.txn();
1423     defer txn.abort();
1424
1425     const sessions = try Db.sessions(txn);
1426     var it = try sessions.iterator();
1427
1428     while (it.next()) |kv| {
1429         const key = kv.key;
1430         const user_id = kv.val;
1431         std.debug.print("[{x}] {}\n", .{ key, user_id });
1432     }
1433 }
1434
1435 fn list_posts(env: lmdb.Env) !void {
1436     const txn = try env.txn();
1437     defer txn.abort();
1438
1439     const posts = try Db.posts(txn);
1440     var it = try posts.iterator();
1441
1442     while (it.next()) |kv| {
1443         const key = kv.key;
1444         const post = kv.val;
1445         std.debug.print("[{}] {s}\n", .{ key, post.text.constSlice() });
1446     }
1447 }
1448
1449 const ReqBufferSize = 4096;
1450 const ResHeadBufferSize = 1024 * 64;
1451 const ResBodyBufferSize = 1024 * 64;
1452
1453 // TODO: static?
1454 var req_buffer: [ReqBufferSize]u8 = undefined;
1455 var res_head_buffer: [ResHeadBufferSize]u8 = undefined;
1456 var res_body_buffer: [ResBodyBufferSize]u8 = undefined;
1457
1458 pub fn main() !void {
1459     // server
1460     var server = try http.Server.init("::", 8080);
1461     defer server.deinit();
1462
1463     // lmdb
1464     var env = try lmdb.Env.open("db", 1024 * 1024 * 10);
1465     defer env.close();
1466
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);
1475
1476     while (true) {
1477         server.wait();
1478         while (true) {
1479             const req = (server.next_request(&req_buffer) catch break) orelse break;
1480             handle_request(env, req) catch {
1481                 try handle_error(env, req);
1482             };
1483         }
1484     }
1485     // const ThreadCount = 1;
1486     // var ts: [ThreadCount]std.Thread = undefined;
1487
1488     // for (0..ThreadCount) |i| {
1489     //     ts[i] = try std.Thread.spawn(.{}, handle_connection, .{ &server, &env });
1490     // }
1491     // for (0..ThreadCount) |i| {
1492     //     ts[i].join();
1493     // }
1494
1495     std.debug.print("done\n", .{});
1496 }
1497
1498 fn handle_error(env: lmdb.Env, req: http.Request) !void {
1499     _ = env;
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);
1504     try res.send();
1505 }
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.? });
1509
1510     // reponse
1511     var res = http.Response.init(req.fd, &res_head_buffer, &res_body_buffer);
1512
1513     // check session token
1514     const logged_in: ?Login = try check_login(env, req, &res);
1515
1516     // html
1517     if (req.method == .GET) {
1518         try write_start(&res);
1519         try write_header(&res, logged_in);
1520
1521         const txn = try env.txn();
1522         defer txn.abort();
1523
1524         const get = GET{
1525             .txn = txn,
1526             .req = req,
1527             .res = &res,
1528             .logged_in = logged_in,
1529         };
1530         if (try get.handle()) {} else {
1531             try res.redirect("/");
1532         }
1533
1534         try write_end(&res);
1535         try res.send();
1536     }
1537     // api
1538     else {
1539         const post = POST{
1540             .env = env,
1541             .req = req,
1542             .res = &res,
1543             .logged_in = logged_in,
1544         };
1545         if (try post.handle()) {} else {
1546             try res.write("<p>[POST] {s}</p>", .{req.target});
1547         }
1548
1549         if (!res.has_header("Location")) {
1550             if (req.get_header("Referer")) |ref| {
1551                 try res.redirect(ref);
1552             } else {
1553                 try res.redirect("/");
1554             }
1555         }
1556         try res.send();
1557     }
1558 }