// GetJoinHandle implements the handle for the join document (GET). func GetJoinHandle(newConn net.Conn, roomName, httpPayload string, rooms *config.CherryRooms, preprocessor *html.Preprocessor) { // INFO(Santiago): The form for room joining was requested, so we will flush it to client. var replyBuffer []byte replyBuffer = rawhttp.MakeReplyBuffer(preprocessor.ExpandData(roomName, rooms.GetEntranceTemplate(roomName)), 200, true) newConn.Write(replyBuffer) newConn.Close() }
// PostJoinHandle implements the handle for the join document (POST). func PostJoinHandle(newConn net.Conn, roomName, httpPayload string, rooms *config.CherryRooms, preprocessor *html.Preprocessor) { // INFO(Santiago): Here, we need firstly parse the posted fields, check for "nickclash", if this is the case // flush the page informing it. Otherwise we add the user basic info and flush the room skeleton // [TOP/BODY/BANNER]. Then we finally close the connection. var userData map[string]string var replyBuffer []byte userData = rawhttp.GetFieldsFromPost(httpPayload) if _, posted := userData["user"]; !posted { newConn.Close() return } if _, posted := userData["color"]; !posted { newConn.Close() return } preprocessor.SetDataValue("{{.nickname}}", userData["user"]) preprocessor.SetDataValue("{{.session-id}}", "0") if rooms.HasUser(roomName, userData["user"]) || userData["user"] == rooms.GetAllUsersAlias(roomName) || strings.Contains(userData["user"], "<") || strings.Contains(userData["user"], ">") || strings.Contains(userData["user"], "<") || strings.Contains(userData["user"], ">") { replyBuffer = rawhttp.MakeReplyBuffer(preprocessor.ExpandData(roomName, rooms.GetNickclashTemplate(roomName)), 200, true) } else { rooms.AddUser(roomName, userData["user"], userData["color"], true) preprocessor.SetDataValue("{{.session-id}}", rooms.GetSessionID(userData["user"], roomName)) replyBuffer = rawhttp.MakeReplyBuffer(preprocessor.ExpandData(roomName, rooms.GetSkeletonTemplate(roomName)), 200, true) rooms.EnqueueMessage(roomName, userData["user"], "", "", "", rooms.GetJoinMessage(roomName), "") } newConn.Write(replyBuffer) newConn.Close() }
// GetBriefHandle implements the handle for the brief document (GET). func GetBriefHandle(newConn net.Conn, roomName, httpPayload string, rooms *config.CherryRooms, preprocessor *html.Preprocessor) { var replyBuffer []byte if rooms.IsAllowingBriefs(roomName) { replyBuffer = rawhttp.MakeReplyBuffer(preprocessor.ExpandData(roomName, rooms.GetBriefTemplate(roomName)), 200, true) } else { replyBuffer = rawhttp.MakeReplyBuffer(html.GetBadAssErrorData(), 404, true) } newConn.Write(replyBuffer) newConn.Close() }
// GetTopHandle implements the handle for the top document (GET). func GetTopHandle(newConn net.Conn, roomName, httpPayload string, rooms *config.CherryRooms, preprocessor *html.Preprocessor) { var userData map[string]string userData = rawhttp.GetFieldsFromGet(httpPayload) var replyBuffer []byte if !rooms.IsValidUserRequest(roomName, userData["user"], userData["id"], newConn) { replyBuffer = rawhttp.MakeReplyBuffer(html.GetBadAssErrorData(), 404, true) } else { replyBuffer = rawhttp.MakeReplyBuffer(preprocessor.ExpandData(roomName, rooms.GetTopTemplate(roomName)), 200, true) } newConn.Write(replyBuffer) newConn.Close() }
// GetExitHandle implements the handle for the exit document (GET). func GetExitHandle(newConn net.Conn, roomName, httpPayload string, rooms *config.CherryRooms, preprocessor *html.Preprocessor) { var userData map[string]string var replyBuffer []byte userData = rawhttp.GetFieldsFromGet(httpPayload) if !rooms.IsValidUserRequest(roomName, userData["user"], userData["id"], newConn) { replyBuffer = rawhttp.MakeReplyBuffer(html.GetBadAssErrorData(), 404, true) } else { preprocessor.SetDataValue("{{.nickname}}", userData["user"]) preprocessor.SetDataValue("{{.session-id}}", userData["id"]) replyBuffer = rawhttp.MakeReplyBuffer(preprocessor.ExpandData(roomName, rooms.GetExitTemplate(roomName)), 200, true) } rooms.EnqueueMessage(roomName, userData["user"], "", "", "", rooms.GetExitMessage(roomName), "") newConn.Write(replyBuffer) rooms.RemoveUser(roomName, userData["user"]) newConn.Close() }
// PostBannerHandle implements the handle for the banner document (POST). func PostBannerHandle(newConn net.Conn, roomName, httpPayload string, rooms *config.CherryRooms, preprocessor *html.Preprocessor) { var userData map[string]string var replyBuffer []byte var invalidRequest = false userData = rawhttp.GetFieldsFromPost(httpPayload) if _, has := userData["user"]; !has { invalidRequest = true } else if _, has := userData["id"]; !has { invalidRequest = true } else if _, has := userData["action"]; !has { invalidRequest = true } else if _, has := userData["whoto"]; !has { invalidRequest = true } else if _, has := userData["image"]; !has { invalidRequest = true } else if _, has := userData["says"]; !has { invalidRequest = true } var restoreBanner = true if invalidRequest || !rooms.IsValidUserRequest(roomName, userData["user"], userData["id"], newConn) { replyBuffer = rawhttp.MakeReplyBuffer(html.GetBadAssErrorData(), 404, true) } else if userData["action"] == rooms.GetIgnoreAction(roomName) { if userData["user"] != userData["whoto"] && !rooms.IsIgnored(userData["user"], userData["whoto"], roomName) { rooms.AddToIgnoreList(userData["user"], userData["whoto"], roomName) rooms.EnqueueMessage(roomName, userData["user"], "", "", "", rooms.GetOnIgnoreMessage(roomName)+userData["whoto"], "1") restoreBanner = false } } else if userData["action"] == rooms.GetDeIgnoreAction(roomName) { if rooms.IsIgnored(userData["user"], userData["whoto"], roomName) { rooms.DelFromIgnoreList(userData["user"], userData["whoto"], roomName) rooms.EnqueueMessage(roomName, userData["user"], "", "", "", rooms.GetOnDeIgnoreMessage(roomName)+userData["whoto"], "1") restoreBanner = false } } else { var somethingToSay = (len(userData["says"]) > 0 || len(userData["image"]) > 0 || len(userData["sound"]) > 0) if somethingToSay { // INFO(Santiago): Any further antiflood control would go from here. rooms.EnqueueMessage(roomName, userData["user"], userData["whoto"], userData["action"], userData["image"], userData["says"], userData["priv"]) } } preprocessor.SetDataValue("{{.nickname}}", userData["user"]) preprocessor.SetDataValue("{{.session-id}}", userData["id"]) if userData["priv"] == "1" { preprocessor.SetDataValue("{{.priv}}", "checked") } tempBanner := preprocessor.ExpandData(roomName, rooms.GetBannerTemplate(roomName)) if restoreBanner { tempBanner = strings.Replace(tempBanner, "<option value = \""+userData["whoto"]+"\">", "<option value = \""+userData["whoto"]+"\" selected>", -1) tempBanner = strings.Replace(tempBanner, "<option value = \""+userData["action"]+"\">", "<option value = \""+userData["action"]+"\" selected>", -1) } replyBuffer = rawhttp.MakeReplyBuffer(tempBanner, 200, true) newConn.Write(replyBuffer) newConn.Close() }
// GetFindHandle implements the handle for the find document (GET). func GetFindHandle(newConn net.Conn, roomName, httpPayload string, rooms *config.CherryRooms, preprocessor *html.Preprocessor) { var replyBuffer []byte replyBuffer = rawhttp.MakeReplyBuffer(preprocessor.ExpandData(roomName, rooms.GetFindBotTemplate(roomName)), 200, true) newConn.Write(replyBuffer) newConn.Close() }
// PostFindHandle implements the handle for the find document (POST). func PostFindHandle(newConn net.Conn, roomName, httpPayload string, rooms *config.CherryRooms, preprocessor *html.Preprocessor) { var userData map[string]string userData = rawhttp.GetFieldsFromPost(httpPayload) var replyBuffer []byte if _, posted := userData["user"]; !posted { replyBuffer = rawhttp.MakeReplyBuffer(html.GetBadAssErrorData(), 404, true) } else { var result string result = preprocessor.ExpandData(roomName, rooms.GetFindResultsHeadTemplate(roomName)) listing := rooms.GetFindResultsBodyTemplate(roomName) availRooms := rooms.GetRooms() user := strings.ToUpper(userData["user"]) if len(user) > 0 { for _, r := range availRooms { users := rooms.GetRoomUsers(r) preprocessor.SetDataValue("{{.find-result-users-total}}", rooms.GetUsersTotal(r)) preprocessor.SetDataValue("{{.find-result-room-name}}", r) for _, u := range users { if strings.HasPrefix(strings.ToUpper(u), user) { preprocessor.SetDataValue("{{.find-result-user}}", u) result += preprocessor.ExpandData(roomName, listing) } } } } result += preprocessor.ExpandData(roomName, rooms.GetFindResultsTailTemplate(roomName)) replyBuffer = rawhttp.MakeReplyBuffer(result, 200, true) } newConn.Write(replyBuffer) newConn.Close() }