]> gitweb.ps.run Git - chirp/blob - src/main.zig
change non-logged-in formatting
[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><br />
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 />
530         , .{});
531     }
532 }
533 fn write_start(res: *http.Response) !void {
534     try res.write(
535         \\<!doctype html>
536         \\<html>
537         \\<head>
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>&#128035;</text></svg>">
539         \\<style>
540         \\  form {
541         \\    display: inline-block;
542         \\  }
543         \\</style>
544         \\</head>
545         \\<body>
546     , .{});
547 }
548 fn write_end(res: *http.Response) !void {
549     try res.write("</body></html>", .{});
550 }
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 {};
555         return;
556     };
557     const users = try Db.users(txn);
558     const user = try users.get(post.user_id);
559
560     try res.write(
561         \\<div>
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)});
566     }
567     try res.write(
568         \\ {s}</span><br />
569         \\<span>{s}</span><br />
570     , .{ time_str(post.time).constSlice(), post.text.constSlice() });
571
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 });
576         } else {
577             try res.write("<a href=\"/post/{x}\">...</a>", .{@intFromEnum(post_id)});
578         }
579         try res.write("</div>", .{});
580     }
581
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);
585
586     // Votes
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;
588
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=\"&#x2B06; {}\"", .{post.upvotes} },
593         });
594     } else {
595         try html_form(res, "/upvote", .{
596             .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
597             .{ "type=\"submit\" value=\"&#x21E7; {}\"", .{post.upvotes} },
598         });
599     }
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=\"&#x2B07; {}\"", .{post.downvotes} },
604         });
605     } else {
606         try html_form(res, "/downvote", .{
607             .{ "type=\"hidden\" value=\"{x}\" name=\"post_id\"", .{@intFromEnum(post.id)} },
608             .{ "type=\"submit\" value=\"&#x21E9; {}\"", .{post.downvotes} },
609         });
610     }
611
612     // Comment Count
613     try res.write(
614         \\<a href="/post/{x}">&#x1F4AD; {}</a> 
615     , .{ @intFromEnum(post.id), comments_view.len() });
616
617     // Quote
618     try res.write(
619         \\<a href="/quotes/{x}">&#x1F501; {}</a> 
620     , .{ @intFromEnum(post.id), quotes_view.len() });
621
622     // Save to List
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() });
633         }
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>", .{});
638     }
639
640     // Comment field
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\"",
648         });
649         try res.write("<br />", .{});
650     }
651
652     // Comments
653     if (options.recurse > 0 and comments_view.len() > 0) {
654         try res.write(
655             \\<details{s}>
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();
660         var count: u8 = 0;
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) {
665                 count += 1;
666                 if (count >= 3) break;
667             }
668         }
669         try res.write(
670             \\</div>
671             \\</details>
672         , .{});
673     }
674
675     try res.write(
676         \\</div>
677     , .{});
678 }
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);
682
683     try res.write(
684         \\<h2 style="display: inline;"><a href="/user/{s}">{s}</a></h2>
685     , .{
686         user.name.constSlice(), user.display_name.constSlice(),
687     });
688     if (logged_in != null and user.id != logged_in.?.user.id) {
689         const login = logged_in.?;
690
691         // follow/unfollow
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"
696             });
697         } else {
698             try html_form(res, "/follow", .{
699                 .{ "type=\"hidden\" name=\"user.id\" value=\"{x}\"", .{@intFromEnum(user.id)} },
700                 \\type="submit" value="Follow"
701             });
702         }
703
704         // add to feed
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() });
713         }
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>", .{});
718     }
719     try res.write(
720         \\ <a href="/following/{s}">{} following</a>
721         \\ <a href="/followers/{s}">{} followers</a>
722         \\<br />
723     , .{
724         user.name.constSlice(), following.len(),
725         user.name.constSlice(), followers.len(),
726     });
727
728     try res.write("<a href=\"/replies/{s}\">Replies</a><br />", .{
729         user.name.constSlice(),
730     });
731
732     if (logged_in != null and user.id == logged_in.?.user.id) {
733         try res.write(
734             \\<a href="/lists">Lists</a>
735             \\<a href="/feeds">Feeds</a>
736             \\<a href="/edit">Edit</a><br />
737             \\<br />
738         , .{});
739     }
740
741     try res.write("<br />", .{});
742 }
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);
745
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 />", .{});
750     }
751 }
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);
755
756     var newest_post_ids = try std.BoundedArray(PostId, 10).init(0); // TODO: TimelinePostsCount
757     var prev_newest_post: ?Post = null;
758
759     const following = try user_list.open(txn);
760
761     while (true) {
762         var newest_post: ?Post = null;
763
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);
768
769             if (followed_posts.len() == 0) {
770                 continue;
771             }
772
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);
776
777                 if ((prev_newest_post == null or p.time < prev_newest_post.?.time) and (newest_post == null or newest_post.?.time < p.time)) {
778                     newest_post = p;
779                     break;
780                 }
781             }
782         }
783         if (newest_post) |post| {
784             newest_post_ids.append(post.id) catch break;
785             prev_newest_post = post;
786         } else {
787             break;
788         }
789     }
790
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 />", .{});
794     }
795 }
796 fn check_login(env: lmdb.Env, req: http.Request, res: *http.Response) !?Login {
797     var result: ?Login = null;
798
799     if (req.get_cookie("session_token")) |session_token_str| {
800         var remove_session_token = true;
801
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();
807                 defer txn.abort();
808                 const users = try Db.users(txn);
809
810                 result = .{
811                     .user = try users.get(user_id),
812                     .session_token = session_token,
813                 };
814
815                 remove_session_token = false;
816             }
817         }
818
819         if (remove_session_token) {
820             try res.add_header(
821                 "Set-Cookie",
822                 .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
823             );
824         }
825     }
826
827     return result;
828 }
829 // }}}
830
831 // GET {{{
832 const GET = struct {
833     const Self = @This();
834
835     txn: lmdb.Txn,
836     req: http.Request,
837     res: *http.Response,
838     logged_in: ?Login,
839
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);
845
846             if (match) {
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});
851                 } else {
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;
858                     }
859                     const str = self.req.target[f_decl.name.len..self.req.target.len];
860                     const field_ti = @typeInfo(field.type);
861                     switch (field_ti) {
862                         // TODO: maybe handle BoundedArray?
863                         .Int => {
864                             @field(arg, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
865                         },
866                         .Enum => {
867                             @field(arg, field.name) = try parse_enum(field.type, str, 16);
868                         },
869                         else => {
870                             @field(arg, field.name) = str;
871                         },
872                     }
873
874                     try @call(.auto, f, .{ self, arg });
875                 }
876                 return true;
877             }
878         }
879         return false;
880     }
881
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);
887
888             try write_profile(self.res, self.txn, self.logged_in, user);
889
890             try write_posts(self.res, self.txn, self.logged_in, user.posts);
891         } else |err| {
892             try self.res.write(
893                 \\<p>User not found [{}]</p>
894             , .{err});
895         }
896     }
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);
902
903             try write_profile(self.res, self.txn, self.logged_in, user);
904
905             try write_posts(self.res, self.txn, self.logged_in, user.replies);
906         } else |err| {
907             try self.res.write(
908                 \\<p>User not found [{}]</p>
909             , .{err});
910         }
911     }
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);
917
918             const following = try user.following.open(self.txn);
919             var it = following.iterator();
920
921             try self.res.write(
922                 \\<h2><a href="/user/{s}">{s}</a> follows:</h2>
923             , .{ user.name.constSlice(), user.display_name.constSlice() });
924
925             while (it.next()) |following_id| {
926                 const following_user = try users.get(following_id);
927
928                 try self.res.write(
929                     \\<a href="/user/{s}">{s}</a><br />
930                 , .{ following_user.name.constSlice(), following_user.display_name.constSlice() });
931             }
932         } else |err| {
933             try self.res.write(
934                 \\<p>User not found [{}]</p>
935             , .{err});
936         }
937     }
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);
943
944             const followers = try user.followers.open(self.txn);
945             var it = followers.iterator();
946
947             try self.res.write(
948                 \\<h2><a href="/user/{s}">{s}</a> followers:</h2>
949             , .{ user.name.constSlice(), user.display_name.constSlice() });
950
951             while (it.next()) |follower_id| {
952                 const follower_user = try users.get(follower_id);
953
954                 try self.res.write(
955                     \\<a href="/user/{s}">{s}</a><br />
956                 , .{ follower_user.name.constSlice(), follower_user.display_name.constSlice() });
957             }
958         } else |err| {
959             try self.res.write(
960                 \\<p>User not found [{}]</p>
961             , .{err});
962         }
963     }
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,
968         });
969     }
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);
973
974         const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
975
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\"",
982             });
983             try self.res.write("<br />", .{});
984         }
985
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 />", .{});
991         }
992     }
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 });
995     }
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);
999
1000             try html_form(self.res, "/create_list", .{
1001                 "type=\"text\" name=\"name\"",
1002                 "type=\"submit\" value=\"Add\"",
1003             });
1004
1005             try self.res.write("<br /><br />", .{});
1006
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;
1011                 try self.res.write(
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\"",
1017                 });
1018                 try self.res.write("<br />", .{});
1019             }
1020         } else {
1021             try self.res.write("not logged in", .{});
1022         }
1023     }
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 });
1026     }
1027     pub fn @"/feeds"(self: Self) !void {
1028         if (self.logged_in) |login| {
1029             const feeds_view = try login.user.feeds.open(self.txn);
1030
1031             try html_form(self.res, "/create_feed", .{
1032                 "type=\"text\" name=\"name\"",
1033                 "type=\"submit\" value=\"Add\"",
1034             });
1035
1036             try self.res.write("<br /><br />", .{});
1037
1038             var it = feeds_view.iterator();
1039             while (it.next()) |kv| {
1040                 const name = kv.val.name;
1041                 const user_list = kv.val.list;
1042                 try self.res.write(
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\"",
1048                 });
1049                 try self.res.write("<br />", .{});
1050             }
1051         } else {
1052             try self.res.write("not logged in", .{});
1053         }
1054     }
1055     pub fn @"/post"(self: Self) !void {
1056         if (self.logged_in) |login| {
1057             _ = login;
1058             const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
1059
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\"",
1064             });
1065         } else {
1066             try self.res.write("not logged in", .{});
1067         }
1068     }
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\"",
1075             });
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\"",
1080             });
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\"",
1085             });
1086         } else {
1087             try self.res.write("not logged in", .{});
1088         }
1089     }
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);
1093         } else {
1094             // TODO: generic home
1095             try self.res.write("Homepage", .{});
1096         }
1097     }
1098 };
1099 // }}}
1100
1101 // POST {{{
1102 const POST = struct {
1103     const Self = @This();
1104
1105     env: lmdb.Env,
1106     req: http.Request,
1107     res: *http.Response,
1108     logged_in: ?Login,
1109
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});
1118                 } else {
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);
1125                         switch (field_ti) {
1126                             .Int => {
1127                                 @field(args, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
1128                             },
1129                             .Enum => {
1130                                 @field(args, field.name) = try parse_enum(field.type, str, 16);
1131                             },
1132                             else => {
1133                                 @field(args, field.name) = str;
1134                             },
1135                         }
1136                     }
1137                     try @call(.auto, f, .{ self, args });
1138                 }
1139                 return true;
1140             }
1141         }
1142         return false;
1143     }
1144
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);
1149     }
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(
1156                 "Set-Cookie",
1157                 .{ "session_token={x}; HttpOnly", .{session_token} },
1158             );
1159         } else |err| {
1160             std.debug.print("login_user err: {}\n", .{err});
1161         }
1162     }
1163     pub fn @"/logout"(self: Self) !void {
1164         if (self.logged_in) |login| {
1165             try Chirp.logout_user(self.env, login.session_token);
1166
1167             try self.res.add_header(
1168                 "Set-Cookie",
1169                 .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
1170             );
1171         }
1172     }
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);
1176
1177         const txn = try self.env.txn();
1178         defer txn.commit() catch {};
1179
1180         const user_ids = try Db.user_ids(txn);
1181
1182         if (!try user_ids.has(username)) {
1183             try user_ids.del(login.user.name);
1184             try user_ids.put(username, login.user.id);
1185
1186             const users = try Db.users(txn);
1187             var user = login.user;
1188             user.name = username;
1189             try users.put(login.user.id, user);
1190         }
1191     }
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);
1195
1196         const txn = try self.env.txn();
1197         defer txn.commit() catch {};
1198
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);
1203     }
1204     pub fn @"/set_password"(self: Self, args: struct { password: []const u8 }) !void {
1205         const login = self.logged_in orelse return error.NotLoggedIn;
1206
1207         const txn = try self.env.txn();
1208         defer txn.commit() catch {};
1209
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);
1214     }
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");
1219
1220             try Chirp.post(self.env, login.user.id, text);
1221
1222             if (has_referer) |r| {
1223                 const decoded = try decode(r);
1224                 try self.res.redirect(decoded.constSlice());
1225             }
1226         }
1227     }
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);
1233
1234             try Chirp.comment(self.env, login.user.id, post_id, text);
1235         }
1236     }
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");
1242
1243             const post_id = try parse_enum(PostId, post_id_str, 16);
1244
1245             try Chirp.quote(self.env, login.user.id, post_id, text);
1246
1247             if (has_referer) |r| {
1248                 const decoded = try decode(r);
1249                 try self.res.redirect(decoded.constSlice());
1250             }
1251         }
1252     }
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
1259
1260             var txn = try self.env.txn();
1261             const postlist = try PostList.init(txn);
1262             try txn.commit();
1263
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 });
1267             try txn.commit();
1268         }
1269     }
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;
1273             {
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);
1279             }
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();
1285             }
1286         }
1287     }
1288     pub fn @"/list_add"(self: Self, args: struct { list_id: PostList.Index, post_id: PostId }) !void {
1289         if (self.logged_in) |login| {
1290             _ = login;
1291
1292             const txn = try self.env.txn();
1293             defer txn.commit() catch {};
1294
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);
1299             } else {
1300                 try post_list_view.append(args.post_id);
1301             }
1302         }
1303     }
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);
1308
1309             var txn = try self.env.txn();
1310             const userlist = try UserList.init(txn);
1311             try txn.commit();
1312
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 });
1316             try txn.commit();
1317         }
1318     }
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;
1322
1323             {
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);
1329             }
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();
1335             }
1336         }
1337     }
1338     pub fn @"/feed_add"(self: Self, args: struct { feed_id: UserList.Index, user_id: UserId }) !void {
1339         if (self.logged_in) |login| {
1340             _ = login;
1341
1342             const txn = try self.env.txn();
1343             defer txn.commit() catch {};
1344
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);
1349             } else {
1350                 try user_list_view.append(args.user_id);
1351             }
1352         }
1353     }
1354     pub fn @"/upvote"(self: Self) !void {
1355         const login = self.logged_in orelse return error.NotLoggedIn;
1356
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));
1359
1360         try Chirp.vote(self.env, post_id, login.user.id, .Up);
1361     }
1362     pub fn @"/downvote"(self: Self) !void {
1363         const login = self.logged_in orelse return error.NotLoggedIn;
1364
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));
1367
1368         try Chirp.vote(self.env, post_id, login.user.id, .Down);
1369     }
1370     pub fn @"/follow"(self: Self) !void {
1371         const login = self.logged_in orelse return error.NotLoggedIn;
1372
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));
1375
1376         try Chirp.follow(self.env, login.user.id, user_id);
1377     }
1378     pub fn @"/quit"(self: Self) !void {
1379         if (self.req.get_header("Referer")) |ref| {
1380             try self.res.redirect(ref);
1381         } else {
1382             try self.res.redirect("/");
1383         }
1384         try self.res.send();
1385         // break :accept;
1386     }
1387 };
1388 // }}}
1389
1390 fn list_users(env: lmdb.Env) !void {
1391     const txn = try env.txn();
1392     defer txn.abort();
1393
1394     const users = try Db.users(txn);
1395     var it = try users.iterator();
1396
1397     while (it.next()) |kv| {
1398         const key = kv.key;
1399         const user = kv.val;
1400         std.debug.print("[{}] {s}\n", .{ key, user.name.constSlice() });
1401     }
1402 }
1403 fn list_user_ids(env: lmdb.Env) !void {
1404     const txn = try env.txn();
1405     defer txn.abort();
1406
1407     const user_ids = try Db.user_ids(txn);
1408     var it = try user_ids.iterator();
1409
1410     while (it.next()) |kv| {
1411         const key = kv.key;
1412         const user_id = kv.val;
1413         std.debug.print("[{s}] {}\n", .{ key.constSlice(), user_id });
1414     }
1415 }
1416
1417 fn list_sessions(env: lmdb.Env) !void {
1418     const txn = try env.txn();
1419     defer txn.abort();
1420
1421     const sessions = try Db.sessions(txn);
1422     var it = try sessions.iterator();
1423
1424     while (it.next()) |kv| {
1425         const key = kv.key;
1426         const user_id = kv.val;
1427         std.debug.print("[{x}] {}\n", .{ key, user_id });
1428     }
1429 }
1430
1431 fn list_posts(env: lmdb.Env) !void {
1432     const txn = try env.txn();
1433     defer txn.abort();
1434
1435     const posts = try Db.posts(txn);
1436     var it = try posts.iterator();
1437
1438     while (it.next()) |kv| {
1439         const key = kv.key;
1440         const post = kv.val;
1441         std.debug.print("[{}] {s}\n", .{ key, post.text.constSlice() });
1442     }
1443 }
1444
1445 const ReqBufferSize = 4096;
1446 const ResHeadBufferSize = 1024 * 64;
1447 const ResBodyBufferSize = 1024 * 64;
1448
1449 // TODO: static?
1450 var req_buffer: [ReqBufferSize]u8 = undefined;
1451 var res_head_buffer: [ResHeadBufferSize]u8 = undefined;
1452 var res_body_buffer: [ResBodyBufferSize]u8 = undefined;
1453
1454 pub fn main() !void {
1455     // server
1456     var server = try http.Server.init("::", 8080);
1457     defer server.deinit();
1458
1459     // lmdb
1460     var env = try lmdb.Env.open("db", 1024 * 1024 * 10);
1461     defer env.close();
1462
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);
1471
1472     while (true) {
1473         server.wait();
1474         while (true) {
1475             const req = (server.next_request(&req_buffer) catch break) orelse break;
1476             handle_request(env, req) catch {
1477                 try handle_error(env, req);
1478             };
1479         }
1480     }
1481     // const ThreadCount = 1;
1482     // var ts: [ThreadCount]std.Thread = undefined;
1483
1484     // for (0..ThreadCount) |i| {
1485     //     ts[i] = try std.Thread.spawn(.{}, handle_connection, .{ &server, &env });
1486     // }
1487     // for (0..ThreadCount) |i| {
1488     //     ts[i].join();
1489     // }
1490
1491     std.debug.print("done\n", .{});
1492 }
1493
1494 fn handle_error(env: lmdb.Env, req: http.Request) !void {
1495     _ = env;
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);
1500     try res.send();
1501 }
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.? });
1505
1506     // reponse
1507     var res = http.Response.init(req.fd, &res_head_buffer, &res_body_buffer);
1508
1509     // check session token
1510     const logged_in: ?Login = try check_login(env, req, &res);
1511
1512     // html
1513     if (req.method == .GET) {
1514         try write_start(&res);
1515         try write_header(&res, logged_in);
1516
1517         const txn = try env.txn();
1518         defer txn.abort();
1519
1520         const get = GET{
1521             .txn = txn,
1522             .req = req,
1523             .res = &res,
1524             .logged_in = logged_in,
1525         };
1526         if (try get.handle()) {} else {
1527             try res.redirect("/");
1528         }
1529
1530         try write_end(&res);
1531         try res.send();
1532     }
1533     // api
1534     else {
1535         const post = POST{
1536             .env = env,
1537             .req = req,
1538             .res = &res,
1539             .logged_in = logged_in,
1540         };
1541         if (try post.handle()) {} else {
1542             try res.write("<p>[POST] {s}</p>", .{req.target});
1543         }
1544
1545         if (!res.has_header("Location")) {
1546             if (req.get_header("Referer")) |ref| {
1547                 try res.redirect(ref);
1548             } else {
1549                 try res.redirect("/");
1550             }
1551         }
1552         try res.send();
1553     }
1554 }