// 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() }
// 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() }