+ 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("<p>[POST] {s}</p>", .{req.target});
+ }
+
+ if (!res.has_header("Location")) {
+ if (req.get_header("Referer")) |ref| {
+ try res.redirect(ref);
+ } else {
+ try res.redirect("/");
projects / chirp / blobdiff