// PubHandle implements the handle for the room's public directory (GET).
func PubHandle(newConn net.Conn, roomName, httpPayload string, rooms *config.CherryRooms, preprocessor *html.Preprocessor) {
pubdir := rooms.GetPublicDirectory(roomName)
var httpReq string
var spaceNr int
for _, h := range httpPayload {
if h == ' ' {
spaceNr++
}
if h == '\n' || h == '\r' || spaceNr > 1 {
break
}
httpReq += string(h)
}
var replyBuffer []byte
if len(pubdir) == 0 || !strings.HasPrefix(httpReq, "GET /pub/"+pubdir) {
replyBuffer = rawhttp.MakeReplyBuffer(html.GetBadAssErrorData(), 404, true)
} else {
relativeLocalPath := httpReq[9:]
_, err := os.Stat(relativeLocalPath)
if os.IsNotExist(err) {
replyBuffer = rawhttp.MakeReplyBuffer(html.GetBadAssErrorData(), 404, true)
} else {
replyBuffer = rawhttp.MakeReplyBufferByFilePath(relativeLocalPath, 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()
}
// 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()
}
// 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()
}
// BadAssErrorHandle implements the handle for the any unexpected HTTP request (GET/POST/Whatever).
func BadAssErrorHandle(newConn net.Conn, roomName, httpPayload string, rooms *config.CherryRooms, preprocessor *html.Preprocessor) {
newConn.Write(rawhttp.MakeReplyBuffer(html.GetBadAssErrorData(), 404, true))
newConn.Close()
}
