X-Git-Url: https://gitweb.ps.run/chirp/blobdiff_plain/0b2f4de9c1db5ec1059f6cd9b8c44e07f3af0b51..e4dc8452f905c132418055d7f2a49f9ea2eb1ddb:/src/main.zig diff --git a/src/main.zig b/src/main.zig index 1871dcc..7aa34bb 100644 --- a/src/main.zig +++ b/src/main.zig @@ -1,416 +1,1735 @@ const std = @import("std"); const lmdb = @import("lmdb"); +const db = @import("db"); +const http = @import("http"); // db {{{ const Db = struct { - const Self = @This(); + fn users(txn: lmdb.Txn) !db.Db(UserId, User) { + return try db.Db(UserId, User).init(txn, "users"); + } + fn user_ids(txn: lmdb.Txn) !db.Db(Username, UserId) { + return try db.Db(Username, UserId).init(txn, "user_ids"); + } + fn sessions(txn: lmdb.Txn) !db.Db(SessionToken, UserId) { + return try db.Db(SessionToken, UserId).init(txn, "sessions"); + } + fn posts(txn: lmdb.Txn) !db.Db(PostId, Post) { + return try db.Db(PostId, Post).init(txn, "posts"); + } +}; - env: ?*lmdb.MDB_env = undefined, - txn: ?*lmdb.MDB_txn = undefined, - dbi: lmdb.MDB_dbi = undefined, - prng: std.Random.DefaultPrng = std.Random.DefaultPrng.init(0), +// }}} - pub fn gen_id(self: *Self) Id { - var id = self.prng.next(); +// content {{{ - while (self.has(id)) { - id = self.prng.next(); - } +const User = struct { + // TODO: choose sizes + id: UserId, + name: Username, + display_name: DisplayName, + password_hash: PasswordHash, + posts: PostList, + following: UserList, + followers: UserList, + + post_lists: PostListList, + feeds: UserListList, +}; + +const Post = struct { + id: PostId, + parent_id: ?PostId, + quote_id: ?PostId, + + user_id: UserId, + time: Timestamp, + + upvotes: u64 = 0, + downvotes: u64 = 0, + votes: VoteList, + comments: PostList, + quotes: PostList, + + text: PostText, +}; + +const SavedPostList = struct { + name: Name, + list: PostList, +}; +const SavedUserList = struct { + name: Name, + list: UserList, +}; - return id; +const Vote = struct { + const Kind = enum { Up, Down }; + + kind: Kind, + time: Timestamp, +}; + +const Id = u64; +const Login = struct { + user: User, + session_token: SessionToken, +}; +const Name = std.BoundedArray(u8, 32); +const UserId = enum(u64) { _ }; +const PostId = enum(u64) { _ }; +const Timestamp = i64; +const Username = std.BoundedArray(u8, 32); +const DisplayName = std.BoundedArray(u8, 64); +const PasswordHash = std.BoundedArray(u8, 128); +const SessionToken = u64; +const CookieValue = std.BoundedArray(u8, 128); +const PostText = std.BoundedArray(u8, 1024); +const PostList = db.Set(PostId); +const UserList = db.Set(UserId); +const VoteList = db.SetList(UserId, Vote); +const PostListList = db.List(SavedPostList); +const UserListList = db.List(SavedUserList); + +fn parse_enum(comptime E: type, buf: []const u8, base: u8) !E { + return @enumFromInt(try std.fmt.parseUnsigned(@typeInfo(E).Enum.tag_type, buf, base)); +} + +// https://developer.mozilla.org/en-US/docs/Glossary/Percent-encoding +fn reencode(comptime T: type, text: []const u8) !T { + var result = try T.init(0); + + const len = @min(text.len, 1024); // TODO: PostText length + + var idx: usize = 0; + while (idx < len) : (idx += 1) { + const c = text[idx]; + if (c == '+') { + try result.append(' '); + } else if (c == '%') { + // special case of ... + // assume only , no + if (idx + 6 < text.len and std.mem.eql(u8, text[idx .. idx + 6], "%26%23")) { + const num_start = idx + 6; + var num_end = num_start; + while (num_end < text.len and std.ascii.isDigit(text[num_end])) { + num_end += 1; + } + + if (num_end + 2 < text.len and + text[num_end] == '%' and + text[num_end + 1] == '3' and + std.ascii.toLower(text[num_end + 2]) == 'b') + { + try std.fmt.format(result.writer(), "{s};", .{text[num_start..num_end]}); + idx = num_end + 2; + continue; + } + } + + try std.fmt.format(result.writer(), "{s};", .{text[idx + 1 .. idx + 3]}); + idx += 2; + } else { + try result.append(c); + } } - pub fn open(self: *Self, name: [*c]const u8) void { - _ = lmdb.mdb_env_create(&self.env); - _ = lmdb.mdb_env_set_maxdbs(self.env, 10); - _ = lmdb.mdb_env_set_mapsize(self.env, 1024 * 1024 * 1); - _ = lmdb.mdb_env_open(self.env, name, lmdb.MDB_WRITEMAP, 0o664); - // _ = lmdb.mdb_env_open(self.env, name, lmdb.MDB_NOSYNC | lmdb.MDB_WRITEMAP, 0o664); + return result; +} + +fn decode(text: []const u8) !std.BoundedArray(u8, 1024) { + var result = try std.BoundedArray(u8, 1024).init(0); + + const max_len = @min(text.len, 1024); // TODO: PostText length + + var idx: usize = 0; + var len: usize = 0; + while (len < max_len and idx < text.len) : ({ + idx += 1; + len += 1; + }) { + const c = text[idx]; + if (c == '+') { + try result.append(' '); + } else if (c == '%') { + if (idx + 2 < text.len) { + try std.fmt.format(result.writer(), "{c}", .{try std.fmt.parseUnsigned(u8, text[idx + 1 .. idx + 3], 16)}); + } + idx += 2; + } else { + try result.append(c); + } } - pub fn close(self: *Self) void { - lmdb.mdb_env_close(self.env); + return result; +} + +const Chirp = struct { + const PostsPerPage = 10; + const UsersPerPage = 10; + + pub fn hash_password(password: []const u8) !PasswordHash { + var hash_buffer = try PasswordHash.init(128); + + // TODO: choose buffer size + // TODO: dont allocate on stack, maybe zero memory? + var buffer: [1024 * 10]u8 = undefined; + var alloc = std.heap.FixedBufferAllocator.init(&buffer); + + // TODO: choose limits + const result = try std.crypto.pwhash.argon2.strHash(password, .{ + .allocator = alloc.allocator(), + .params = std.crypto.pwhash.argon2.Params.fromLimits(1000, 1024), + }, hash_buffer.slice()); + + try hash_buffer.resize(result.len); + + return hash_buffer; } - pub fn begin(self: *Self, name: [*c]const u8) void { - switch (lmdb.mdb_txn_begin(self.env, null, 0, &self.txn)) { - 0 => {}, - else => |err| { - std.debug.print("txn err: {}\n", .{err}); - }, + pub fn verify_password(password: []const u8, hash: PasswordHash) bool { + var buffer: [1024 * 10]u8 = undefined; + var alloc = std.heap.FixedBufferAllocator.init(&buffer); + + if (std.crypto.pwhash.argon2.strVerify(hash.constSlice(), password, .{ + .allocator = alloc.allocator(), + })) { + return true; + } else |err| { + std.debug.print("verify error: {}\n", .{err}); + return false; } + } - // TODO: lmdb.MDB_INTEGERKEY? - _ = lmdb.mdb_dbi_open(self.txn, name, lmdb.MDB_CREATE, &self.dbi); + pub fn register_user(env: lmdb.Env, username: []const u8, password: []const u8) !bool { + const username_array = try Username.fromSlice(username); + const display_name = try DisplayName.fromSlice(username); + + const txn = try env.txn(); + defer txn.commit() catch |err| { + std.debug.print("error registering user: {}\n", .{err}); + }; + + const users = try Db.users(txn); + const user_ids = try Db.user_ids(txn); + + if (try user_ids.has(username_array)) { + return false; + } else { + const user_id = try db.Prng.gen(users.dbi, UserId); + + try users.put(user_id, User{ + .id = user_id, + .name = username_array, + .display_name = display_name, + .password_hash = try hash_password(password), + .posts = try PostList.init(txn), + .following = try UserList.init(txn), + .followers = try UserList.init(txn), + .post_lists = try PostListList.init(txn), + .feeds = try UserListList.init(txn), + }); + + try user_ids.put(username_array, user_id); + + return true; + } } - pub fn commit(self: *Self) void { - switch (lmdb.mdb_txn_commit(self.txn)) { - 0 => {}, - lmdb.MDB_MAP_FULL => { - std.debug.print("resize\n", .{}); - _ = lmdb.mdb_env_set_mapsize(self.env, 0); - }, - else => |err| { - std.debug.print("commit err: {}\n", .{err}); - }, + pub fn login_user( + env: lmdb.Env, + username: []const u8, + password: []const u8, + ) !SessionToken { + const username_array = try Username.fromSlice(username); + + const txn = try env.txn(); + defer txn.commit() catch {}; + + const user_ids = try Db.user_ids(txn); + const user_id = try user_ids.get(username_array); + std.debug.print("user logging in, id: {}\n", .{user_id}); + + const users = try Db.users(txn); + const user = try users.get(user_id); + + if (verify_password(password, user.password_hash)) { + const sessions = try Db.sessions(txn); + const session_token = try db.Prng.gen(sessions.dbi, SessionToken); + try sessions.put(session_token, user_id); + return session_token; + } else { + return error.IncorrectPassword; } + } - // TODO: necessary? - lmdb.mdb_dbi_close(self.env, self.dbi); + fn logout_user(env: lmdb.Env, session_token: SessionToken) !void { + const txn = try env.txn(); + defer txn.commit() catch {}; + + const sessions = try Db.sessions(txn); + try sessions.del(session_token); } - pub fn sync(self: *Self) void { - switch (lmdb.mdb_env_sync(self.env, 1)) { - 0 => {}, - else => |err| { - std.debug.print("sync err: {}\n", .{err}); - }, + fn append_post(env: lmdb.Env, user_id: UserId, post_list: PostList, parent_id: ?PostId, quote_id: ?PostId, text: []const u8) !PostId { + var post_id: PostId = undefined; + + // TODO: do this in one commit + + var txn: lmdb.Txn = undefined; + { + // create post + txn = try env.txn(); + defer txn.commit() catch {}; + + const posts = try Db.posts(txn); + post_id = try db.Prng.gen(posts.dbi, PostId); + + const decoded_text = try reencode(PostText, text); + try posts.put(post_id, Post{ + .id = post_id, + .parent_id = parent_id, + .quote_id = quote_id, + .user_id = user_id, + .time = std.time.timestamp(), + .votes = try VoteList.init(txn), + .comments = try PostList.init(txn), + .quotes = try PostList.init(txn), + .text = decoded_text, + }); + } + + { + // append to user's posts + txn = try env.txn(); + defer txn.commit() catch {}; + + var posts_view = try post_list.open(txn); + try posts_view.append(post_id); } - } - pub fn put(self: *Self, key: anytype, value: anytype) void { - lmdb.put(self.txn, self.dbi, key, value); + if (quote_id != null) { + txn = try env.txn(); + defer txn.commit() catch {}; + + const posts = try Db.posts(txn); + const quote_post = try posts.get(quote_id.?); + var quotes = try quote_post.quotes.open(txn); + try quotes.append(post_id); + } + + return post_id; } - pub fn get(self: *Self, key: anytype, comptime T: type) ?T { - return lmdb.get(self.txn, self.dbi, key, T); + fn post(env: lmdb.Env, user_id: UserId, text: []const u8) !void { + var txn = try env.txn(); + const users = try Db.users(txn); + const user = try users.get(user_id); + txn.abort(); + + const post_id = try append_post(env, user_id, user.posts, null, null, text); + _ = post_id; } - pub fn del(self: *Self, key: anytype) void { - lmdb.del(self.txn, self.dbi, key); + fn comment(env: lmdb.Env, user_id: UserId, parent_post_id: PostId, text: []const u8) !void { + var txn = try env.txn(); + const users = try Db.users(txn); + const user = try users.get(user_id); + + const posts = try Db.posts(txn); + const parent_post = try posts.get(parent_post_id); + txn.abort(); + + const post_id = try append_post(env, user_id, parent_post.comments, parent_post_id, null, text); + + txn = try env.txn(); + var replies_view = try user.posts.open(txn); + try replies_view.append(post_id); + try txn.commit(); } - pub fn has(self: *Self, key: anytype) bool { - return lmdb.has(self.txn, self.dbi, key); + fn quote(env: lmdb.Env, user_id: UserId, quote_post_id: PostId, text: []const u8) !void { + var txn = try env.txn(); + const users = try Db.users(txn); + const user = try users.get(user_id); + txn.abort(); + + const post_id = try append_post(env, user_id, user.posts, null, quote_post_id, text); + _ = post_id; } -}; -// }}} + fn vote(env: lmdb.Env, post_id: PostId, user_id: UserId, kind: Vote.Kind) !void { + const txn = try env.txn(); + defer txn.commit() catch {}; -// http stuff {{{ + const posts = try Db.posts(txn); -pub fn redirect(req: *std.http.Server.Request, location: []const u8) !void { - try req.respond("", .{ .status = .see_other, .extra_headers = &.{.{ .name = "Location", .value = location }} }); -} + var p = try posts.get(post_id); + var votes_view = try p.votes.open(txn); -pub fn get_body(req: *std.http.Server.Request) []const u8 { - return req.server.read_buffer[req.head_end .. req.head_end + (req.head.content_length orelse 0)]; -} + var add_vote = true; + + if (try votes_view.has(user_id)) { + const old_vote = try votes_view.get(user_id); + + add_vote = old_vote.kind != kind; -pub fn get_value(req: *std.http.Server.Request, name: []const u8) ?[]const u8 { - const body = get_body(req); - if (std.mem.indexOf(u8, body, name)) |name_index| { - if (std.mem.indexOfScalarPos(u8, body, name_index, '=')) |eql_index| { - if (std.mem.indexOfScalarPos(u8, body, name_index, '&')) |amp_index| { - return body[eql_index + 1 .. amp_index]; + try votes_view.del(user_id); + + switch (old_vote.kind) { + .Up => p.upvotes -= 1, + .Down => p.downvotes -= 1, } + try posts.put(post_id, p); + } - return body[eql_index + 1 .. body.len]; + if (add_vote) { + try votes_view.append(user_id, Vote{ + .kind = kind, + .time = std.time.timestamp(), + }); + + if (kind == .Up) { + p.upvotes += 1; + } else { + p.downvotes += 1; + } + try posts.put(post_id, p); } } - return null; -} -pub fn get_cookie(req: *std.http.Server.Request, name: []const u8) ?CookieValue { - var header_it = req.iterateHeaders(); - while (header_it.next()) |header| { - if (std.mem.eql(u8, header.name, "Cookie")) { - if (std.mem.indexOf(u8, header.value, name)) |name_index| { - if (std.mem.indexOfScalarPos(u8, header.value, name_index, '=')) |eql_index| { - if (std.mem.indexOfPos(u8, header.value, name_index, "; ")) |semi_index| { - return CookieValue.fromSlice(header.value[eql_index + 1 .. semi_index]) catch null; - } + fn follow(env: lmdb.Env, user_id: UserId, user_id_to_follow: UserId) !void { + const txn = try env.txn(); + defer txn.commit() catch {}; - return CookieValue.fromSlice(header.value[eql_index + 1 .. header.value.len]) catch null; - } - } + const users = try Db.users(txn); + + const user = try users.get(user_id); + const user_to_follow = try users.get(user_id_to_follow); + + var user_following = try user.following.open(txn); + var user_to_follow_followers = try user_to_follow.followers.open(txn); + + if ((user_following.has(user_id_to_follow) catch false) and (user_to_follow_followers.has(user_id) catch false)) { + try user_following.del(user_id_to_follow); + try user_to_follow_followers.del(user_id); + } else if (!(user_following.has(user_id_to_follow) catch true) and !(user_to_follow_followers.has(user_id) catch true)) { + try user_following.append(user_id_to_follow); + try user_to_follow_followers.append(user_id); + } else { + std.debug.print("Something went wrong when trying to unfollow\n", .{}); } } - return null; -} -// }}} + fn get_session_user_id(env: lmdb.Env, session_token: SessionToken) !UserId { + const txn = try env.txn(); + defer txn.abort(); -// content {{{ + const sessions = try Db.sessions(txn); -const User = struct { - // TODO: choose sizes - username: Username, - password_hash: PasswordHash, + return try sessions.get(session_token); + } + + fn get_user(env: lmdb.Env, user_id: UserId) !User { + const txn = try env.txn(); + defer txn.abort(); + + const users = try Db.users(txn); + return try users.get(user_id); + } }; -const Id = u64; -const Username = std.BoundedArray(u8, 16); -const PasswordHash = std.BoundedArray(u8, 128); -const SessionToken = u64; -const CookieValue = std.BoundedArray(u8, 128); +// }}} -pub fn hash_password(password: []const u8) !PasswordHash { - var hash_buffer = try PasswordHash.init(128); +// html {{{ +pub fn Paginate(comptime T: type) type { + return struct { + const Self = @This(); - // TODO: choose buffer size - // TODO: dont allocate on stack, maybe zero memory? - var buffer: [1024 * 10]u8 = undefined; - var alloc = std.heap.FixedBufferAllocator.init(&buffer); + const IterateResult = T.Base.View.Iterator.Result; - // TODO: choose limits - const result = try std.crypto.pwhash.argon2.strHash(password, .{ - .allocator = alloc.allocator(), - .params = std.crypto.pwhash.argon2.Params.fromLimits(1000, 1024), - }, hash_buffer.slice()); + res: *http.Response, + view: T.View, + per_page: u64, - try hash_buffer.resize(result.len); + it: T.Base.View.Iterator, + starting_idx: ?T.Base.Key, + count: u64 = 0, - return hash_buffer; -} + pub fn init(res: *http.Response, view: T.View, per_page: u64) !Self { + var it = view.reverse_iterator(); + if (res.req.get_param("starting_at")) |starting_at_str| { + it.idx = try parse_enum(T.Base.Key, starting_at_str, 16); + } -pub fn verify_password(password: []const u8, hash: PasswordHash) bool { - var buffer: [1024 * 10]u8 = undefined; - var alloc = std.heap.FixedBufferAllocator.init(&buffer); + return .{ + .res = res, + .view = view, + .per_page = per_page, + .it = it, + .starting_idx = it.idx, + }; + } + pub fn next(self: *Self) IterateResult { + if (self.it.next()) |kv| { + if (self.count < self.per_page) { + self.count += 1; + return kv; + } + } + return null; + } + pub fn write_navigation(self: *Self) !void { + const next_idx = self.it.next(); - if (std.crypto.pwhash.argon2.strVerify(hash.constSlice(), password, .{ - .allocator = alloc.allocator(), - })) { - return true; - } else |err| { - std.debug.print("verify error: {}\n", .{err}); - return false; - } -} + if (self.view.base.head.last != self.starting_idx) { + var prev_it = self.view.iterator(); + prev_it.idx = self.starting_idx.?; + var oldest_idx = self.starting_idx.?; -pub fn register_user(db: *Db, username: []const u8, password: []const u8) !void { - const username_array = try Username.fromSlice(username); + var count: u64 = 0; + while (prev_it.next()) |kv| { + oldest_idx = kv.key; - db.begin("users"); - const user_id = db.gen_id(); - db.put(user_id, User{ - .username = username_array, - .password_hash = try hash_password(password), - }); - db.commit(); + if (count > self.per_page) { + break; + } else { + count += 1; + } + } - db.begin("ids"); - db.put(username_array.buffer, user_id); - db.commit(); + try self.res.write("Prev ", .{ self.res.req.target, @intFromEnum(oldest_idx) }); + } - std.debug.print("id: {}\n", .{user_id}); + if (next_idx) |kv| { + try self.res.write("Next", .{ self.res.req.target, @intFromEnum(kv.key) }); + } + } + }; +} +fn html_form(res: *http.Response, action: []const u8, inputs: anytype) !void { + try res.write("
", .{}); } +// }}} -pub fn login_user(db: *Db, username: []const u8, password: []const u8) ?SessionToken { - const username_array = Username.fromSlice(username) catch return null; +// write {{{ +const TimeStr = std.BoundedArray(u8, 256); - db.begin("ids"); - const user_id = db.get(username_array.buffer, Id) orelse return null; - std.debug.print("id: {}\n", .{user_id}); - // TODO: maybe no commit? - db.commit(); +// http://howardhinnant.github.io/date_algorithms.html +fn time_str(_t: i64) TimeStr { + const t: u64 = @intCast(_t); + var result = TimeStr.init(0) catch unreachable; - db.begin("users"); - const user = db.get(user_id, User) orelse return null; - db.commit(); + const nD = @divFloor(t, std.time.s_per_day); + const z: u64 = nD + 719468; + const era: u64 = (if (z >= 0) z else z - 146096) / 146097; + const doe: u64 = z - era * 146097; // [0, 146096] + const yoe: u64 = (doe - doe / 1460 + doe / 36524 - doe / 146096) / 365; // [0, 399] + const Y: u64 = yoe + era * 400; + const doy: u64 = doe - (365 * yoe + yoe / 4 - yoe / 100); // [0, 365] + const mp: u64 = (5 * doy + 2) / 153; // [0, 11] + const D: u64 = doy - (153 * mp + 2) / 5 + 1; // [1, 31] + const M: u64 = if (mp < 10) mp + 3 else mp - 9; - if (verify_password(password, user.password_hash)) { - db.begin("sessions"); - const session_token = db.gen_id(); - db.put(session_token, user_id); - db.commit(); + const h: u64 = @divFloor(t - nD * std.time.s_per_day, std.time.s_per_hour); + const m: u64 = @divFloor(t - nD * std.time.s_per_day - h * std.time.s_per_hour, std.time.s_per_min); + const s: u64 = t - nD * std.time.s_per_day - h * std.time.s_per_hour - m * std.time.s_per_min; - db.sync(); + std.fmt.format(result.writer(), "", .{ Y, M, D, h, m, s, t * 1000 }) catch unreachable; - return session_token; + return result; +} +fn write_header(res: *http.Response, logged_in: ?Login) !void { + if (logged_in) |login| { + try res.write( + \\HomeUser not found [{}]
+ , .{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( + \\User not found [{}]
+ , .{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( + \\User not found [{}]
+ , .{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( + \\User not found [{}]
+ , .{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); - accept: while (true) { - const conn = try server.accept(); + const following_view = try user.following.open(self.txn); - std.debug.print("new connection: {}\n", .{conn}); + var paginate = try Paginate(UserList).init(self.res, following_view, Chirp.UsersPerPage); - var read_buffer: [1024]u8 = undefined; - var http_server = std.http.Server.init(conn, &read_buffer); + try self.res.write( + \\User not found [{}]
+ , .{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); - if (get_cookie(&req, "session_token")) |session_token_str| { - const session_token = try std.fmt.parseUnsigned(SessionToken, session_token_str.constSlice(), 10); - if (get_session_user(&db, session_token)) |user| { - logged_in = .{ - .user = user, - .session_token = session_token, - }; - } - // TODO: delete session token - // TODO: add changeable headers (set, delete cookies) + 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( + \\User not found [{}]
+ , .{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:POST
- , .{}); + try @call(.auto, f, .{ self, args }); } + return true; + } + } + return false; + } + + pub fn @"/register"(self: Self, args: struct { username: []const u8, password: []const u8 }) !void { + // TODO: handle args not supplied + std.debug.print("New user: {s} {s}\n", .{ args.username, args.password }); + _ = try Chirp.register_user(self.env, args.username, args.password); + } + pub fn @"/login"(self: Self, args: struct { username: []const u8, password: []const u8 }) !void { + // TODO: handle args not supplied + std.debug.print("New login: {s} {s}\n", .{ args.username, args.password }); + if (Chirp.login_user(self.env, args.username, args.password)) |session_token| { + self.res.status = .see_other; + try self.res.add_header( + "Set-Cookie", + .{ "session_token={x}; HttpOnly", .{session_token} }, + ); + } else |err| { + std.debug.print("login_user err: {}\n", .{err}); + } + } + pub fn @"/logout"(self: Self) !void { + if (self.logged_in) |login| { + try Chirp.logout_user(self.env, login.session_token); + + try self.res.add_header( + "Set-Cookie", + .{"session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT"}, + ); + } + } + pub fn @"/set_username"(self: Self, args: struct { username: []const u8 }) !void { + const login = self.logged_in orelse return error.NotLoggedIn; + const username = try Username.fromSlice(args.username); + + const txn = try self.env.txn(); + defer txn.commit() catch {}; + + const user_ids = try Db.user_ids(txn); + + if (!try user_ids.has(username)) { + try user_ids.del(login.user.name); + try user_ids.put(username, login.user.id); + + const users = try Db.users(txn); + var user = login.user; + user.name = username; + try users.put(login.user.id, user); + } + } + pub fn @"/set_display_name"(self: Self, args: struct { display_name: []const u8 }) !void { + const login = self.logged_in orelse return error.NotLoggedIn; + const display_name = try DisplayName.fromSlice(args.display_name); + + const txn = try self.env.txn(); + defer txn.commit() catch {}; + + const users = try Db.users(txn); + var user = login.user; + user.display_name = display_name; + try users.put(login.user.id, user); + } + pub fn @"/set_password"(self: Self, args: struct { password: []const u8 }) !void { + const login = self.logged_in orelse return error.NotLoggedIn; + + const txn = try self.env.txn(); + defer txn.commit() catch {}; + + const users = try Db.users(txn); + var user = login.user; + user.password_hash = try Chirp.hash_password(args.password); + try users.put(login.user.id, user); + } + pub fn @"/post"(self: Self) !void { + if (self.logged_in) |login| { + const text = self.req.get_value("text").?; + const has_referer = self.req.get_value("referer"); + + try Chirp.post(self.env, login.user.id, text); + + if (has_referer) |r| { + const decoded = try decode(r); + try self.res.redirect(decoded.constSlice()); + } + } + } + pub fn @"/comment"(self: Self) !void { + if (self.logged_in) |login| { + const text = self.req.get_value("text") orelse return error.NoText; + const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId; + const post_id = try parse_enum(PostId, post_id_str, 16); + + try Chirp.comment(self.env, login.user.id, post_id, text); + } + } + pub fn @"/quote"(self: Self) !void { + if (self.logged_in) |login| { + const text = self.req.get_value("text") orelse return error.NoText; + const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId; + const has_referer = self.req.get_value("referer"); + + const post_id = try parse_enum(PostId, post_id_str, 16); + + try Chirp.quote(self.env, login.user.id, post_id, text); + + if (has_referer) |r| { + const decoded = try decode(r); + try self.res.redirect(decoded.constSlice()); + } + } + } + // TODO: add arguments instead of parsing manually + pub fn @"/create_list"(self: Self) !void { + if (self.logged_in) |login| { + const name_str = self.req.get_value("name") orelse return error.NoName; + const name = try Name.fromSlice(name_str); + // TODO: decode name + + var txn = try self.env.txn(); + const postlist = try PostList.init(txn); + try txn.commit(); + + txn = try self.env.txn(); + var post_lists_view = try login.user.post_lists.open(txn); + _ = try post_lists_view.append(.{ .name = name, .list = postlist }); + try txn.commit(); + } + } + pub fn @"/delete_list"(self: Self, args: struct { list_id: PostList.Base.Index }) !void { + if (self.logged_in) |login| { + var post_list: ?PostList = null; + { + const txn = try self.env.txn(); + defer txn.commit() catch {}; + var post_lists_view = try login.user.post_lists.open(txn); + post_list = (try post_lists_view.get(args.list_id)).list; + try post_lists_view.del(args.list_id); + } + if (post_list != null) { + const txn = try self.env.txn(); + defer txn.commit() catch {}; + var list_view = try post_list.?.open(txn); + try list_view.clear(); + } + } + } + pub fn @"/list_add"(self: Self, args: struct { list_id: PostList.Base.Index, post_id: PostId }) !void { + if (self.logged_in) |login| { + _ = login; + + const txn = try self.env.txn(); + defer txn.commit() catch {}; + + const post_list = PostList{ .base = .{ .idx = args.list_id } }; + var post_list_view = try post_list.open(txn); + if (try post_list_view.has(args.post_id)) { + try post_list_view.del(args.post_id); + } else { + try post_list_view.append(args.post_id); + } + } + } + pub fn @"/create_feed"(self: Self) !void { + if (self.logged_in) |login| { + const name_str = self.req.get_value("name") orelse return error.NoName; + const name = try Name.fromSlice(name_str); + + var txn = try self.env.txn(); + const userlist = try UserList.init(txn); + try txn.commit(); + + txn = try self.env.txn(); + var feeds_view = try login.user.feeds.open(txn); + _ = try feeds_view.append(.{ .name = name, .list = userlist }); + try txn.commit(); + } + } + pub fn @"/delete_feed"(self: Self, args: struct { list_id: UserList.Base.Index }) !void { + if (self.logged_in) |login| { + var user_list: ?UserList = null; + + { + const txn = try self.env.txn(); + defer txn.commit() catch {}; + var feeds_view = try login.user.feeds.open(txn); + user_list = (try feeds_view.get(args.list_id)).list; + try feeds_view.del(args.list_id); + } + if (user_list != null) { + const txn = try self.env.txn(); + defer txn.commit() catch {}; + var list_view = try user_list.?.open(txn); + try list_view.clear(); + } + } + } + pub fn @"/feed_add"(self: Self, args: struct { feed_id: UserList.Base.Index, user_id: UserId }) !void { + if (self.logged_in) |login| { + _ = login; + + const txn = try self.env.txn(); + defer txn.commit() catch {}; + + const user_list = UserList{ .base = .{ .idx = args.feed_id } }; + var user_list_view = try user_list.open(txn); + if (try user_list_view.has(args.user_id)) { + try user_list_view.del(args.user_id); + } else { + try user_list_view.append(args.user_id); + } + } + } + pub fn @"/upvote"(self: Self) !void { + const login = self.logged_in orelse return error.NotLoggedIn; + + const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId; + const post_id: PostId = @enumFromInt(try std.fmt.parseUnsigned(u64, post_id_str, 16)); + + try Chirp.vote(self.env, post_id, login.user.id, .Up); + } + pub fn @"/downvote"(self: Self) !void { + const login = self.logged_in orelse return error.NotLoggedIn; + + const post_id_str = self.req.get_value("post_id") orelse return error.NoPostId; + const post_id: PostId = @enumFromInt(try std.fmt.parseUnsigned(u64, post_id_str, 16)); + + try Chirp.vote(self.env, post_id, login.user.id, .Down); + } + pub fn @"/follow"(self: Self) !void { + const login = self.logged_in orelse return error.NotLoggedIn; + + const user_id_str = self.req.get_value("user_id") orelse return error.NoUserId; + const user_id: UserId = @enumFromInt(try std.fmt.parseUnsigned(u64, user_id_str, 16)); + + try Chirp.follow(self.env, login.user.id, user_id); + } + pub fn @"/quit"(self: Self) !void { + if (self.req.get_header("Referer")) |ref| { + try self.res.redirect(ref); + } else { + try self.res.redirect("/"); + } + try self.res.send(); + // break :accept; + } +}; +// }}} + +fn list_users(env: lmdb.Env) !void { + const txn = try env.txn(); + defer txn.abort(); + + const users = try Db.users(txn); + var it = try users.iterator(); + + while (it.next()) |kv| { + const key = kv.key; + const user = kv.val; + std.debug.print("[{}] {s}\n", .{ key, user.name.constSlice() }); + } +} +fn list_user_ids(env: lmdb.Env) !void { + const txn = try env.txn(); + defer txn.abort(); + + const user_ids = try Db.user_ids(txn); + var it = try user_ids.iterator(); + + while (it.next()) |kv| { + const key = kv.key; + const user_id = kv.val; + std.debug.print("[{s}] {}\n", .{ key.constSlice(), user_id }); + } +} + +fn list_sessions(env: lmdb.Env) !void { + const txn = try env.txn(); + defer txn.abort(); + + const sessions = try Db.sessions(txn); + var it = try sessions.iterator(); + + while (it.next()) |kv| { + const key = kv.key; + const user_id = kv.val; + std.debug.print("[{x}] {}\n", .{ key, user_id }); + } +} + +fn list_posts(env: lmdb.Env) !void { + const txn = try env.txn(); + defer txn.abort(); + + const posts = try Db.posts(txn); + var it = try posts.iterator(); + + while (it.next()) |kv| { + const key = kv.key; + const post = kv.val; + std.debug.print("[{}] {s}\n", .{ key, post.text.constSlice() }); + } +} + +const ReqBufferSize = 4096; +const ResHeadBufferSize = 1024 * 64; +const ResBodyBufferSize = 1024 * 64; + +// TODO: static? +var req_buffer: [ReqBufferSize]u8 = undefined; +var res_head_buffer: [ResHeadBufferSize]u8 = undefined; +var res_body_buffer: [ResBodyBufferSize]u8 = undefined; + +pub fn main() !void { + // server + var server = try http.Server.init("::", 8080); + defer server.deinit(); + + // lmdb + var env = try lmdb.Env.open("db", 1024 * 1024 * 10); + defer env.close(); + + std.debug.print("Users:\n", .{}); + try list_users(env); + std.debug.print("User IDs:\n", .{}); + try list_user_ids(env); + std.debug.print("Sessions:\n", .{}); + try list_sessions(env); + std.debug.print("Posts:\n", .{}); + try list_posts(env); + + while (true) { + server.wait(); + while (true) { + const req = (server.next_request(&req_buffer) catch break) orelse break; + // handle_request(env, req) catch { + // try handle_error(env, req); + // }; + try handle_request(env, req); + } + } + // const ThreadCount = 1; + // var ts: [ThreadCount]std.Thread = undefined; + + // for (0..ThreadCount) |i| { + // ts[i] = try std.Thread.spawn(.{}, handle_connection, .{ &server, &env }); + // } + // for (0..ThreadCount) |i| { + // ts[i].join(); + // } + + std.debug.print("done\n", .{}); +} + +fn handle_error(env: lmdb.Env, req: http.Request) !void { + _ = env; + var res = http.Response.init(req, &res_head_buffer, &res_body_buffer); + try write_start(&res); + try res.write("Oops, something went terribly wrong there D:", .{}); + try write_end(&res); + try res.send(); +} +fn handle_request(env: lmdb.Env, req: http.Request) !void { + // std.debug.print("[{}]: {s}\n", .{ req.method, req.target }); + // std.debug.print("[{}]: {s}\n", .{ req.method, req.head.? }); + + // reponse + var res = http.Response.init(req, &res_head_buffer, &res_body_buffer); + + // check session token + const logged_in: ?Login = try check_login(env, req, &res); + + // html + if (req.method == .GET) { + try write_start(&res); + try write_header(&res, logged_in); + + const txn = try env.txn(); + defer txn.abort(); + + const get = GET{ + .txn = txn, + .req = req, + .res = &res, + .logged_in = logged_in, + }; + if (try get.handle()) {} else { + try res.redirect("/"); + } + + try write_end(&res); + try res.send(); + } + // api + else { + const post = POST{ + .env = env, + .req = req, + .res = &res, + .logged_in = logged_in, + }; + if (try post.handle()) {} else { + try res.write("[POST] {s}
", .{req.target}); + } + + if (!res.has_header("Location")) { + if (req.get_header("Referer")) |ref| { + try res.redirect(ref); + } else { + try res.redirect("/"); } } + try res.send(); } }