gochan/pkg/posting/formatting.go

package posting

import (
	"errors"
	"fmt"
	"html/template"
	"math/rand"
	"regexp"
	"strconv"
	"strings"

	"github.com/frustra/bbcode"
	"github.com/gochan-org/gochan/pkg/config"
	"github.com/gochan-org/gochan/pkg/gcsql"
	"github.com/rs/zerolog"
)

var (
	msgfmtr          MessageFormatter
	urlRE            = regexp.MustCompile(`https?://(\S+)`)
	unsetBBcodeTags  = []string{"center", "color", "img", "quote", "size"}
	diceRollRE       = regexp.MustCompile(`\[(\d*)d(\d+)(?:([+-])(\d+))?\]`)
	hashTagRE        = regexp.MustCompile(`\[#([^\]]+)\]`)
	brRE             = regexp.MustCompile(`<br\s*/?>`)
	ErrWorksafeBoard = errors.New("this board does not allow NSFW content")
)

// InitPosting prepares the formatter and the temp post pruner
func InitPosting() {
	msgfmtr.Init()
	go tempCleaner()
}

type MessageFormatter struct {
	// Go's garbage collection does weird things with bbcode's internal tag map.
	// Moving the bbcode compiler isntance (and eventually a Markdown compiler) to a struct
	// appears to fix this
	bbCompiler bbcode.Compiler
	linkFixer  *strings.Replacer // used for fixing [url=http://...] being turned into [url=[url]http://...
}

func (mf *MessageFormatter) Init() {
	mf.bbCompiler = bbcode.NewCompiler(true, true)
	for _, tag := range unsetBBcodeTags {
		mf.bbCompiler.SetTag(tag, nil)
	}
	mf.bbCompiler.SetTag("?", func(_ *bbcode.BBCodeNode) (*bbcode.HTMLTag, bool) {
		return &bbcode.HTMLTag{Name: "span", Attrs: map[string]string{"class": "spoiler"}}, true
	})
	mf.bbCompiler.SetTag("hide", func(_ *bbcode.BBCodeNode) (*bbcode.HTMLTag, bool) {
		return &bbcode.HTMLTag{Name: "div", Attrs: map[string]string{"class": "hideblock hidden"}}, true
	})
	mf.linkFixer = strings.NewReplacer(
		"[url=[url]", "[url=",
		"[/url][/url]", "[/url]",
		"[url][url]", "[url]",
	)
}

func (*MessageFormatter) ApplyWordFilters(message string, boardDir string) (string, error) {
	var filters []gcsql.Wordfilter
	var err error
	if boardDir == "" {
		filters, err = gcsql.GetWordfilters(gcsql.OnlyTrue)
	} else {
		filters, err = gcsql.GetBoardWordfilters(boardDir)
	}
	if err != nil {
		return message, err
	}
	for _, wf := range filters {
		if message, err = wf.Apply(message); err != nil {
			return message, err
		}
	}
	return message, nil
}

func (mf *MessageFormatter) Compile(msg string, boardDir string) string {
	if config.GetBoardConfig(boardDir).DisableBBcode {
		return msg
	}
	return mf.bbCompiler.Compile(msg)
}

func ApplyWordFilters(message string, boardDir string) (string, error) {
	return msgfmtr.ApplyWordFilters(message, boardDir)
}

func wrapLinksInURL(urlStr string) string {
	return "[url]" + urlStr + "[/url]"
}

func FormatMessage(message string, boardDir string, warnEv, errEv *zerolog.Event) (template.HTML, error) {
	if config.GetBoardConfig(boardDir).RenderURLsAsLinks {
		message = urlRE.ReplaceAllStringFunc(message, wrapLinksInURL)
		message = msgfmtr.linkFixer.Replace(message)
	}
	message = msgfmtr.Compile(message, boardDir)
	// prepare each line to be formatted
	postLines := brRE.Split(message, -1)
	boardConfig := config.GetBoardConfig(boardDir)
	var err error
	for i, line := range postLines {
		trimmedLine := strings.TrimSpace(line)
		lineWords := strings.Split(trimmedLine, " ")
		isGreentext := false // if true, append </span> to end of line
		WebRoot := config.GetSystemCriticalConfig().WebRoot
		for w, word := range lineWords {
			if strings.LastIndex(word, "&gt;&gt;") == 0 {
				//word is a backlink
				if postID, err := strconv.Atoi(word[8:]); err == nil {
					// the link is in fact, a valid int
					var boardDir string
					var linkParent int
					if linkParent, boardDir, err = gcsql.GetTopPostAndBoardDirFromPostID(postID); err != nil {
						errEv.Caller().Int("childPostID", postID).Msg("Unable to get top post and board")
						return "", fmt.Errorf("unable to get top post and board for post #%d", postID)
					}

					if linkParent == 0 {
						// board or op not found
						lineWords[w] = `<a href="javascript:;"><strike>` + word + `</strike></a>`
					} else {
						lineWords[w] = fmt.Sprintf(`<a href="%s%s/res/%d.html#%s" class="postref">%s</a>`, WebRoot, boardDir, linkParent, word[8:], word)
					}
				}
			} else if strings.Index(word, "&gt;") == 0 && w == 0 {
				// word is at the beginning of a line, and is greentext
				isGreentext = true
				lineWords[w] = `<span class="greentext">` + word
			}
		}
		line = strings.Join(lineWords, " ")
		if isGreentext {
			line += "</span>"
		}

		err = nil
		var classList string
		line = hashTagRE.ReplaceAllStringFunc(line, func(tag string) string {
			if err != nil {
				return tag // don't bother processing if there's already an error
			}
			tagNoBrackets := tag[1 : len(tag)-1]
			classList = "hashtag"
			if strings.ToLower(tagNoBrackets) == "#nsfw" {
				if boardConfig.Worksafe {
					err = ErrWorksafeBoard
					return ""
				}
				classList += " nsfw"
			}
			return fmt.Sprintf(`<span class="%s">%s</span>`, classList, tagNoBrackets)
		})
		if err != nil {
			warnEv.Str("board", boardDir).Msg("NSFW tag found on worksafe board")
			return "", err
		}
		postLines[i] = line
	}
	return template.HTML(strings.Join(postLines, "<br />")), nil // skipcq: GSC-G203
}

func diceRoller(numDice int, diceSides int, modifier int) int {
	rollSum := 0
	for i := 0; i < numDice; i++ {
		rollSum += rand.Intn(diceSides) + 1 // skipcq: GSC-G404
	}
	return rollSum + modifier
}

func ApplyDiceRoll(p *gcsql.Post) error {
	var err error
	result := diceRollRE.ReplaceAllStringFunc(string(p.Message), func(roll string) string {
		rollMatch := diceRollRE.FindStringSubmatch(roll)
		numDice := 1
		if rollMatch[1] != "" {
			numDice, _ = strconv.Atoi(rollMatch[1])
		}
		if numDice < 1 {
			err = errors.New("number of dice must be at least 1")
			return roll
		}
		dieSize, _ := strconv.Atoi(rollMatch[2])
		if dieSize <= 1 {
			err = errors.New("die size must be greater than 1")
			return roll
		}
		modifierIsNegative := rollMatch[3] == "-"
		modifier, _ := strconv.Atoi(rollMatch[4])
		if modifierIsNegative {
			modifier = -modifier
		}
		rollSum := diceRoller(numDice, dieSize, modifier)
		return fmt.Sprintf(`<span class="dice-roll">%dd%d%s%s = %d</span>`, numDice, dieSize, rollMatch[3], rollMatch[4], rollSum)
	})
	if err != nil {
		return err
	}
	p.Message = template.HTML(result) // skipcq: GSC-G203
	return nil
}