mirror of https://github.com/velour/catbase.git
97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
package slack
|
|
|
|
import (
|
|
"unicode/utf8"
|
|
)
|
|
|
|
// fixText strips all of the Slack-specific annotations from message text,
|
|
// replacing it with the equivalent display form.
|
|
// Currently it:
|
|
// • Replaces user mentions like <@U124356> with @ followed by the user's nick.
|
|
// This uses the lookupUser function, which must map U1243456 to the nick.
|
|
// • Replaces user mentions like <U123456|nick> with the user's nick.
|
|
// • Strips < and > surrounding links.
|
|
//
|
|
// This was directly bogarted from velour/chat with emoji conversion removed.
|
|
func fixText(findUser func(id string) (string, bool), text string) string {
|
|
var output []rune
|
|
for len(text) > 0 {
|
|
r, i := utf8.DecodeRuneInString(text)
|
|
text = text[i:]
|
|
switch {
|
|
case r == '<':
|
|
var tag []rune
|
|
for {
|
|
r, i := utf8.DecodeRuneInString(text)
|
|
text = text[i:]
|
|
switch {
|
|
case r == '>':
|
|
if t, ok := fixTag(findUser, tag); ok {
|
|
output = append(output, t...)
|
|
break
|
|
}
|
|
fallthrough
|
|
case len(text) == 0:
|
|
output = append(output, '<')
|
|
output = append(output, tag...)
|
|
output = append(output, r)
|
|
default:
|
|
tag = append(tag, r)
|
|
continue
|
|
}
|
|
break
|
|
}
|
|
default:
|
|
output = append(output, r)
|
|
}
|
|
}
|
|
return string(output)
|
|
}
|
|
|
|
func fixTag(findUser func(string) (string, bool), tag []rune) ([]rune, bool) {
|
|
switch {
|
|
case hasPrefix(tag, "@U"):
|
|
if i := indexRune(tag, '|'); i >= 0 {
|
|
return tag[i+1:], true
|
|
}
|
|
if findUser != nil {
|
|
if u, ok := findUser(string(tag[1:])); ok {
|
|
return []rune("@" + u), true
|
|
}
|
|
}
|
|
return tag, true
|
|
|
|
case hasPrefix(tag, "#C"):
|
|
if i := indexRune(tag, '|'); i >= 0 {
|
|
return append([]rune{'#'}, tag[i+1:]...), true
|
|
}
|
|
|
|
case hasPrefix(tag, "http"):
|
|
if i := indexRune(tag, '|'); i >= 0 {
|
|
tag = tag[:i]
|
|
}
|
|
return tag, true
|
|
}
|
|
|
|
return nil, false
|
|
}
|
|
|
|
func hasPrefix(text []rune, prefix string) bool {
|
|
for _, r := range prefix {
|
|
if len(text) == 0 || text[0] != r {
|
|
return false
|
|
}
|
|
text = text[1:]
|
|
}
|
|
return true
|
|
}
|
|
|
|
func indexRune(text []rune, find rune) int {
|
|
for i, r := range text {
|
|
if r == find {
|
|
return i
|
|
}
|
|
}
|
|
return -1
|
|
}
|