- // api
- else {
- if (std.mem.eql(u8, req.head.target, "/register")) {
- // TODO: handle args not supplied
- const username = get_value(&req, "username").?;
- const password = get_value(&req, "password").?;
-
- std.debug.print("New user: {s} {s}\n", .{ username, password });
- try register_user(&db, username, password);
-
- try redirect(&req, "/login");
- } else if (std.mem.eql(u8, req.head.target, "/login")) {
- // TODO: handle args not supplied
- const username = get_value(&req, "username").?;
- const password = get_value(&req, "password").?;
-
- std.debug.print("New login: {s} {s}\n", .{ username, password });
- if (login_user(&db, username, password)) |session_token| {
- var redirect_buffer = try std.BoundedArray(u8, 128).init(0);
- try std.fmt.format(redirect_buffer.writer(), "/user/{s}", .{username});
-
- var cookie_buffer = try std.BoundedArray(u8, 128).init(0);
- try std.fmt.format(cookie_buffer.writer(), "session_token={}; Secure; HttpOnly", .{session_token});
-
- try req.respond("", .{
- .status = .see_other,
- .extra_headers = &.{
- .{ .name = "Location", .value = redirect_buffer.constSlice() },
- .{ .name = "Set-Cookie", .value = cookie_buffer.constSlice() },
- },
- });
- } else {
- try redirect(&req, "/login");
- }
- } else if (std.mem.eql(u8, req.head.target, "/logout")) {
- if (logged_in) |login| {
- logout_user(&db, login.session_token);
- try req.respond("", .{
- .status = .see_other,
- .extra_headers = &.{
- .{ .name = "Location", .value = "/" },
- .{ .name = "Set-Cookie", .value = "session_token=deleted; Expires=Thu, 01 Jan 1970 00:00:00 GMT" },
- },
- });
- }
- } else if (std.mem.eql(u8, req.head.target, "/quit")) {
- try redirect(&req, "/");
- break :accept;
- } else {
- try req.respond(
- \\<p>POST</p>
- , .{});
- }
+ }
+ }
+ // 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 PostSet.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: PostSet.Base.Index }) !void {
+ if (self.logged_in) |login| {
+ var post_list: ?PostSet = 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: PostSet.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 = PostSet{ .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 UserSet.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: UserSet.Base.Index }) !void {
+ if (self.logged_in) |login| {
+ var user_list: ?UserSet = 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: UserSet.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 = UserSet{ .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("/");