MovieNight/chatclient.go

package main

import (
	"fmt"
	"html"
	"net"
	"strings"
	"unicode"

	"github.com/gorilla/websocket"
)

type Client struct {
	name          string // Display name
	conn          *websocket.Conn
	belongsTo     *ChatRoom
	color         string
	IsMod         bool
	IsAdmin       bool
	IsColorForced bool
}

var emotes map[string]string

func ParseEmotes(msg string) string {
	words := strings.Split(msg, " ")
	newWords := []string{}
	for _, word := range words {
		word = strings.Trim(word, "[]")

		found := false
		for key, val := range emotes {
			if key == word {
				newWords = append(newWords, fmt.Sprintf("<img src=\"/emotes/%s\" title=\"%s\" />", val, key))
				//fmt.Printf("[emote] %s\n", val)
				found = true
			}
		}
		if !found {
			newWords = append(newWords, word)
		}
	}
	return strings.Join(newWords, " ")
}

//Client has a new message to broadcast
func (cl *Client) NewMsg(msg string) {
	msg = html.EscapeString(msg)
	msg = removeDumbSpaces(msg)
	msg = strings.Trim(msg, " ")

	// Don't send zero-length messages
	if len(msg) == 0 {
		return
	}

	if strings.HasPrefix(msg, "/") {
		// is a command
		msg = msg[1:len(msg)]
		fullcmd := strings.Split(msg, " ")
		cmd := strings.ToLower(fullcmd[0])
		args := fullcmd[1:len(fullcmd)]

		response := commands.RunCommand(cmd, args, cl)
		if response != "" {
			cl.ServerMessage(response)
			return
		}

	} else {
		// Trim long messages
		if len(msg) > 400 {
			msg = msg[0:400]
		}

		fmt.Printf("[chat] <%s> %q\n", cl.name, msg)

		// Enable links for mods and admins
		if cl.IsMod || cl.IsAdmin {
			msg = formatLinks(msg)
		}

		cl.Message(msg)
	}
}

// Make links clickable
func formatLinks(input string) string {
	newMsg := []string{}
	for _, word := range strings.Split(input, " ") {
		if strings.HasPrefix(word, "http:&#x2F;&#x2F;") || strings.HasPrefix(word, "https:&#x2F;&#x2F;") {
			//word = unescape(word)
			word = html.UnescapeString(word)
			word = fmt.Sprintf(`<a href="%s" target="_blank">%s</a>`, word, word)
		}
		newMsg = append(newMsg, word)
	}
	return strings.Join(newMsg, " ")
}

//Exiting out
func (cl *Client) Exit() {
	cl.belongsTo.Leave(cl.name, cl.color)
}

//Sending message block to the client
func (cl *Client) Send(msgs string) {
	cl.conn.WriteMessage(websocket.TextMessage, []byte(msgs))
}

// Send server message to this client
func (cl *Client) ServerMessage(msg string) {
	msg = ParseEmotes(msg)
	cl.Send(`<span class="svmsg">` + msg + `</span><br />`)
}

// Outgoing messages
func (cl *Client) Message(msg string) {
	msg = ParseEmotes(msg)
	cl.belongsTo.AddMsg(
		`<span class="name" style="color:` + cl.color + `">` + cl.name +
			`</span><b>:</b> <span class="msg">` + msg + `</span><br />`)
}

// Outgoing /me command
func (cl *Client) Me(msg string) {
	msg = ParseEmotes(msg)
	cl.belongsTo.AddMsg(fmt.Sprintf(`<span style="color:%s"><span class="name">%s</span> <span class="cmdme">%s</span><br />`, cl.color, cl.name, msg))
}

func (cl *Client) Mod() {
	cl.IsMod = true
}

func (cl *Client) Unmod() {
	cl.IsMod = false
}

func (cl *Client) Host() string {
	host, _, err := net.SplitHostPort(cl.conn.RemoteAddr().String())
	if err != nil {
		host = "err"
	}
	return host
}

var dumbSpaces []string = []string{
	"\n",
	"\t",
	"\r",
	"\u200b",
}

func removeDumbSpaces(msg string) string {
	for _, ds := range dumbSpaces {
		msg = strings.ReplaceAll(msg, ds, " ")
	}

	newMsg := ""
	for _, r := range msg {
		if unicode.IsSpace(r) {
			newMsg += " "
		} else {
			newMsg += string(r)
		}
	}
	return newMsg
}