+ var newest_post_ids = try std.BoundedArray(PostId, 10).init(0); // TODO: TimelinePostsCount
+ var prev_newest_post: ?Post = null;
+
+ const following = try user_list.open(txn);
+
+ while (true) {
+ var newest_post: ?Post = null;
+
+ var following_it = following.iterator();
+ while (following_it.next()) |following_id| {
+ const followed_user = try users.get(following_id.key);
+ const followed_posts = try followed_user.posts.open(txn);
+
+ if (followed_posts.len() == 0) {
+ continue;
+ }
+
+ var followed_posts_it = followed_posts.reverse_iterator();
+ while (followed_posts_it.next()) |followed_post_id| {
+ const p = try posts.get(followed_post_id.key);
+
+ if ((prev_newest_post == null or p.time < prev_newest_post.?.time) and (newest_post == null or newest_post.?.time < p.time)) {
+ newest_post = p;
+ break;
+ }
+ }
+ }
+ if (newest_post) |post| {
+ newest_post_ids.append(post.id) catch break;
+ prev_newest_post = post;
+ } else {
+ break;
+ }
+ }
+
+ for (newest_post_ids.constSlice()) |post_id| {
+ try write_post(res, txn, logged_in, post_id, .{ .recurse = 1 });
+ try res.write("<br />", .{});
+ }
+}
+fn write_user(res: *http.Response, txn: lmdb.Txn, user_id: UserId) !void {
+ const users = try Db.users(txn);
+ const user = try users.get(user_id);
+ try res.write(
+ \\<a href="/user/{s}">{s}</a>
+ , .{ user.name.constSlice(), user.display_name.constSlice() });
+}
+fn write_votes(res: *http.Response, txn: lmdb.Txn, votes: VoteList, options: struct {
+ show_upvotes: bool = true,
+ show_downvotes: bool = true,
+}) !void {
+ const votes_view = try votes.open(txn);
+
+ var paginate = try Paginate(VoteList).init(res, votes_view, Chirp.UsersPerPage);
+
+ while (paginate.next()) |kv| {
+ const user_id = kv.key;
+ const vote = kv.val;
+
+ if ((options.show_upvotes and vote.kind == .Up) or
+ (options.show_downvotes and vote.kind == .Down))
+ {
+ try write_user(res, txn, user_id);
+ try res.write(" <small>{s}</small><br />", .{time_str(vote.time).constSlice()});
+ }
+ }
+
+ try paginate.write_navigation();
+}
+fn check_login(env: lmdb.Env, req: http.Request, res: *http.Response) !?Login {
+ var result: ?Login = null;
+
+ if (req.get_cookie("session_token")) |session_token_str| {
+ var remove_session_token = true;
+
+ if (std.fmt.parseUnsigned(SessionToken, session_token_str, 16) catch null) |session_token| {
+ // const session_token = try std.fmt.parseUnsigned(SessionToken, session_token_str, 10);
+ // const session_token = std.mem.bytesToValue(SessionToken, session_token_str);
+ if (Chirp.get_session_user_id(env, session_token) catch null) |user_id| {
+ const txn = try env.txn();
+ defer txn.abort();
+ const users = try Db.users(txn);
+
+ result = .{
+ .user = try users.get(user_id),
+ .session_token = session_token,
+ };
+
+ remove_session_token = false;
+ }
+ }
+
+ if (remove_session_token) {
+ try res.add_header(
+ "Set-Cookie",
+ .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"},
+ );
+ }
+ }
+
+ return result;
+}
+// }}}
+
+// GET {{{
+const GET = struct {
+ const Self = @This();
+
+ txn: lmdb.Txn,
+ req: http.Request,
+ res: *http.Response,
+ logged_in: ?Login,
+
+ fn handle(self: Self) !bool {
+ const ti = @typeInfo(Self);
+ inline for (ti.Struct.decls) |f_decl| {
+ const has_arg = f_decl.name.len > 1 and f_decl.name[f_decl.name.len - 1] == '/';
+ 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);
+
+ if (match) {
+ const f = @field(Self, f_decl.name);
+ const fi = @typeInfo(@TypeOf(f));
+ if (fi.Fn.params.len == 1) {
+ try @call(.auto, f, .{self});
+ } else {
+ const arg_type = fi.Fn.params[1].type.?;
+ const arg_info = @typeInfo(arg_type);
+ var arg: arg_type = undefined;
+ const field = arg_info.Struct.fields[0];
+ if (self.req.target.len <= f_decl.name.len) {
+ return error.NoArgProvided;
+ }
+ const str = self.req.target[f_decl.name.len..self.req.target.len];
+ const field_ti = @typeInfo(field.type);
+ switch (field_ti) {
+ // TODO: maybe handle BoundedArray?
+ .Int => {
+ @field(arg, field.name) = try std.fmt.parseUnsigned(field.type, str, 16);
+ },
+ .Enum => {
+ @field(arg, field.name) = try parse_enum(field.type, str, 16);
+ },
+ else => {
+ @field(arg, field.name) = str;
+ },
+ }
+
+ try @call(.auto, f, .{ self, arg });
+ }
+ return true;
+ }
+ }
+ return false;
+ }
+
+ pub fn @"/user/"(self: Self, args: struct { username: []const u8 }) !void {
+ const user_ids = try Db.user_ids(self.txn);
+ if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
+ const users = try Db.users(self.txn);
+ const user = try users.get(user_id);
+
+ try write_profile(self.res, self.txn, self.logged_in, user);
+
+ try write_posts(self.res, self.txn, self.logged_in, user.posts, .{
+ .show_posts = true,
+ .show_quotes = false,
+ .show_comments = false,
+ });
+ } else |err| {
+ try self.res.write(
+ \\<p>User not found [{}]</p>
+ , .{err});
+ }
+ }
+ pub fn @"/comments/"(self: Self, args: struct { username: []const u8 }) !void {
+ const user_ids = try Db.user_ids(self.txn);
+ if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
+ const users = try Db.users(self.txn);
+ const user = try users.get(user_id);
+
+ try write_profile(self.res, self.txn, self.logged_in, user);
+
+ try write_posts(self.res, self.txn, self.logged_in, user.posts, .{
+ .show_posts = false,
+ .show_quotes = false,
+ .show_comments = true,
+ });
+ } else |err| {
+ try self.res.write(
+ \\<p>User not found [{}]</p>
+ , .{err});
+ }
+ }
+ pub fn @"/quotes/"(self: Self, args: struct { username: []const u8 }) !void {
+ const user_ids = try Db.user_ids(self.txn);
+ if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
+ const users = try Db.users(self.txn);
+ const user = try users.get(user_id);
+
+ try write_profile(self.res, self.txn, self.logged_in, user);
+
+ try write_posts(self.res, self.txn, self.logged_in, user.posts, .{
+ .show_posts = false,
+ .show_quotes = true,
+ .show_comments = false,
+ });
+ } else |err| {
+ try self.res.write(
+ \\<p>User not found [{}]</p>
+ , .{err});
+ }
+ }
+ pub fn @"/all/"(self: Self, args: struct { username: []const u8 }) !void {
+ const user_ids = try Db.user_ids(self.txn);
+ if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
+ const users = try Db.users(self.txn);
+ const user = try users.get(user_id);
+
+ try write_profile(self.res, self.txn, self.logged_in, user);
+
+ try write_posts(self.res, self.txn, self.logged_in, user.posts, .{
+ .show_posts = true,
+ .show_quotes = true,
+ .show_comments = true,
+ });
+ } else |err| {
+ try self.res.write(
+ \\<p>User not found [{}]</p>
+ , .{err});
+ }
+ }
+ pub fn @"/following/"(self: Self, args: struct { username: []const u8 }) !void {
+ const user_ids = try Db.user_ids(self.txn);
+ if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
+ const users = try Db.users(self.txn);
+ const user = try users.get(user_id);
+
+ const following_view = try user.following.open(self.txn);
+
+ var paginate = try Paginate(UserList).init(self.res, following_view, Chirp.UsersPerPage);
+
+ try self.res.write(
+ \\<h2><a href="/user/{s}">{s}</a> follows:</h2>
+ , .{ user.name.constSlice(), user.display_name.constSlice() });
+
+ while (paginate.next()) |following_id| {
+ const following_user = try users.get(following_id.key);
+
+ try self.res.write(
+ \\<a href="/user/{s}">{s}</a><br />
+ , .{ following_user.name.constSlice(), following_user.display_name.constSlice() });
+ }
+
+ try paginate.write_navigation();
+ } else |err| {
+ try self.res.write(
+ \\<p>User not found [{}]</p>
+ , .{err});
+ }
+ }
+ pub fn @"/followers/"(self: Self, args: struct { username: []const u8 }) !void {
+ const user_ids = try Db.user_ids(self.txn);
+ if (user_ids.get(try Username.fromSlice(args.username))) |user_id| {
+ const users = try Db.users(self.txn);
+ const user = try users.get(user_id);
+
+ const followers_view = try user.followers.open(self.txn);
+ var paginate = try Paginate(UserList).init(self.res, followers_view, Chirp.UsersPerPage);
+
+ try self.res.write(
+ \\<h2><a href="/user/{s}">{s}</a> followers:</h2>
+ , .{ user.name.constSlice(), user.display_name.constSlice() });
+
+ while (paginate.next()) |follower_id| {
+ const follower_user = try users.get(follower_id.key);
+
+ try self.res.write(
+ \\<a href="/user/{s}">{s}</a><br />
+ , .{ follower_user.name.constSlice(), follower_user.display_name.constSlice() });
+ }
+
+ try paginate.write_navigation();
+ } else |err| {
+ try self.res.write(
+ \\<p>User not found [{}]</p>
+ , .{err});
+ }
+ }
+ pub fn @"/post/"(self: Self, args: struct { post_id: PostId }) !void {
+ try write_post(self.res, self.txn, self.logged_in, args.post_id, .{
+ .recurse = 3, // TODO: factor out
+ .show_comment_field = true,
+ });
+ }
+ pub fn @"/upvotes/"(self: Self, args: struct { post_id: PostId }) !void {
+ const posts = try Db.posts(self.txn);
+ const post = try posts.get(args.post_id);
+ try self.res.write("{} upvotes:<br />", .{post.upvotes});
+ try write_votes(self.res, self.txn, post.votes, .{});
+ }
+ pub fn @"/quoted/"(self: Self, args: struct { post_id: PostId }) !void {
+ const posts = try Db.posts(self.txn);
+ const post = try posts.get(args.post_id);
+
+ const referer = if (self.req.get_header("Referer")) |ref| ref else self.req.target;
+
+ if (self.logged_in != null) {
+ try html_form(self.res, "/quote", .{
+ .{ "type=\"hidden\" name=\"referer\" value=\"{s}\"", .{referer} },
+ .{ "type=\"hidden\" name=\"post_id\" value=\"{x}\"", .{@intFromEnum(post.id)} },
+ "type=\"text\" name=\"text\" placeholder=\"Text\" autofocus",
+ "type=\"submit\" value=\"Quote\"",
+ });
+ try self.res.write("<br />", .{});
+ }
+
+ // TODO: show all bc this only contains quotes?
+ try write_posts(self.res, self.txn, self.logged_in, post.quotes, .{
+ .show_posts = false,
+ .show_quotes = true,
+ .show_comments = false,
+ });
+ }
+ pub fn @"/list/"(self: Self, args: struct { list_id: PostList.Base.Index }) !void {
+ try write_posts(self.res, self.txn, self.logged_in, PostList{ .base = .{ .idx = args.list_id } }, .{
+ .show_posts = true,
+ .show_quotes = true,
+ .show_comments = true,
+ });
+ }
+ pub fn @"/lists"(self: Self) !void {
+ if (self.logged_in) |login| {
+ const post_lists_view = try login.user.post_lists.open(self.txn);
+
+ try html_form(self.res, "/create_list", .{
+ "type=\"text\" name=\"name\"",
+ "type=\"submit\" value=\"Add\"",
projects / chirp / blobdiff