1
0
Fork 0
mirror of https://github.com/Eggbertx/gochan.git synced 2025-08-02 10:56:25 -07:00
gochan/pkg/posting/formatting.go

101 lines
3.1 KiB
Go

package posting
import (
"html/template"
"strconv"
"strings"
"time"
"github.com/frustra/bbcode"
"github.com/gochan-org/gochan/pkg/config"
"github.com/gochan-org/gochan/pkg/gclog"
"github.com/gochan-org/gochan/pkg/gcsql"
)
var (
msgfmtr *MessageFormatter
)
// InitPosting prepares the formatter and the temp post pruner
func InitPosting() {
msgfmtr = new(MessageFormatter)
msgfmtr.InitBBcode()
tempCleanerTicker = time.NewTicker(time.Minute * 5)
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
}
func (mf *MessageFormatter) InitBBcode() {
if config.GetBoardConfig("").DisableBBcode {
return
}
mf.bbCompiler = bbcode.NewCompiler(true, true)
mf.bbCompiler.SetTag("center", nil)
mf.bbCompiler.SetTag("code", nil)
mf.bbCompiler.SetTag("color", nil)
mf.bbCompiler.SetTag("img", nil)
mf.bbCompiler.SetTag("quote", nil)
mf.bbCompiler.SetTag("size", nil)
}
func (mf *MessageFormatter) Compile(msg string) string {
if config.GetBoardConfig("").DisableBBcode {
return msg
}
return mf.bbCompiler.Compile(msg)
}
func FormatMessage(message string) template.HTML {
message = msgfmtr.Compile(message)
// prepare each line to be formatted
postLines := strings.Split(message, "<br>")
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
var boardIDFound bool
if boardDir, boardIDFound, err = gcsql.GetBoardFromPostID(postID); err != nil {
gclog.Print(gclog.LErrorLog, "Error getting board dir for backlink: ", err.Error())
}
if linkParent, err = gcsql.GetThreadIDZeroIfTopPost(postID); err != nil {
gclog.Print(gclog.LErrorLog, "Error getting post parent for backlink: ", err.Error())
}
// get post board dir
if !boardIDFound {
lineWords[w] = `<a href="javascript:;"><strike>` + word + `</strike></a>`
} else if linkParent == 0 {
lineWords[w] = `<a href="` + WebRoot + boardDir + `/res/` + word[8:] + `.html" class="postref">` + word + `</a>`
} else {
lineWords[w] = `<a href="` + WebRoot + boardDir + `/res/` + strconv.Itoa(linkParent) + `.html#` + word[8:] + `" class="postref">` + word + `</a>`
}
}
} 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>"
}
postLines[i] = line
}
return template.HTML(strings.Join(postLines, "<br />"))
}