Mirror
/
go-neb
mirror of https://github.com/matrix-org/go-neb.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

slackapi enhancements, use html/template properly

pull/68/head
Aviral Dasgupta 8 years ago
parent
768e097b7c
commit
67c68dbd6a
2 changed files with 141 additions and 84 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 136
      src/github.com/matrix-org/go-neb/services/slackapi/message.go
  2. 21
      src/github.com/matrix-org/go-neb/services/slackapi/slackapi.go

136
src/github.com/matrix-org/go-neb/services/slackapi/message.go
View File

@ -8,33 +8,40 @@ import (
log "github.com/Sirupsen/logrus"
"github.com/matrix-org/go-neb/matrix"
"github.com/russross/blackfriday"
"html/template"
"io/ioutil"
"mime"
"net/http"
"regexp"
"text/template"
"time"
)
type slackAttachment struct {
Fallback string `json:"fallback"`
FallbackRendered template.HTML
Color *string `json:"color"`
ColorRendered template.HTMLAttr
Pretext string `json:"pretext"`
PretextRendered template.HTML
AuthorName *string `json:"author_name"`
AuthorLink *string `json:"author_link"`
AuthorLink template.URL `json:"author_link"`
AuthorIcon *string `json:"author_icon"`
AuthorIconURL template.URL
Title *string `json:"title"`
TitleLink *string `json:"title_link"`
Text string `json:"text"`
TextRendered template.HTML
MrkdwnIn []string `json:"mrkdwn_in"`
Ts *int64 `json:"ts"`
}
type slackMessage struct {
Text string `json:"text"`
TextRendered template.HTML
Username string `json:"username"`
Channel string `json:"channel"`
Mrkdwn *bool `json:"mrkdwn"`
@ -46,19 +53,31 @@ type slackMessage struct {
// We do not do this yet, since it's assumed that clients also escape the content we send them.
var htmlTemplate, _ = template.New("htmlTemplate").Parse(`
<strong>@{{ .Username }}</strong> via <strong>#{{ .Channel }}</strong><br />
{{ if .Text }}{{ .Text }}<br />{{ end }}
{{ range .Attachments }}
{{ if .AuthorName }}
{{if .AuthorLink }}<a href="{{ .AuthorLink }}">{{ end }}
{{ if .AuthorIcon }}{{ .AuthorIcon }}{{ end }}
{{ .AuthorName }}
{{if .AuthorLink }}</a>{{ end }}
{{- with (or .TextRendered .Text nil) }}
{{- if . }}
{{- . }}<br />
{{- end }}
{{- end }}
{{- range .Attachments }}
{{- if .AuthorName }}
{{- if .AuthorLink }}<a href="{{ .AuthorLink }}">{{ end }}
{{- if .AuthorIconUrl }}<img src="{{ .AuthorIconUrl }}" />{{ end }}
{{- .AuthorName }}
{{- if .AuthorLink }}</a>{{ end }}
<br />
{{- end }}
<strong>
<font color="{{- .ColorRendered }}">▌</font>
{{- if .TitleLink }}
<a href="{{ .TitleLink}}">{{ .Title }}</a>
{{- else }}
{{- .Title }}
{{- end }}
<br />
{{ end }}
<strong><font color="{{ .Color }}">▌</font>{{ if .TitleLink }}<a href="{{ .TitleLink}}">{{ .Title }}</a>{{ else }}{{ .Title }}{{ end }}<br /></strong>
{{ if .Pretext }}{{ .Pretext }}<br />{{ end }}
{{ if .Text }}{{ .Text }}<br />{{ end }}
{{ end }}
</strong>
{{- if .Pretext }}{{ or .PretextRendered .Pretext }}<br />{{ end }}
{{- if .Text }}{{ or .TextRendered .Text }}<br />{{ end }}
{{- end }}
`)
var netClient = &http.Client{
@ -80,7 +99,7 @@ func getSlackMessage(req http.Request) (message slackMessage, err error) {
decoder := json.NewDecoder(req.Body)
err = decoder.Decode(&message)
} else {
message.Text = fmt.Sprint("**Error:** unknown Content-Type `%s`", ct)
message.Text = fmt.Sprintf("**Error:** unknown Content-Type `%s`", ct)
log.Error(message.Text)
}
@ -92,7 +111,10 @@ func linkifyString(text string) string {
}
func getColor(color *string) string {
if color != nil {
if color == nil {
return "black"
}
// https://api.slack.com/docs/message-attachments defines these aliases
mappedColor, ok := map[string]string{
"good": "green",
@ -103,65 +125,85 @@ func getColor(color *string) string {
return mappedColor
}
return *color
}
return "black"
}
func slackMessageToHTMLMessage(message slackMessage) (html matrix.HTMLMessage, err error) {
processedMessage := message
if message.Mrkdwn == nil || *message.Mrkdwn == true {
text := linkifyString(message.Text)
processedMessage.Text = string(blackfriday.MarkdownBasic([]byte(text)))
// fetches an image and encodes it as a data URL
// returns nil if fetch fails
func fetchAndEncodeImage(url *string) (data template.URL) {
if url == nil {
return
}
for attachmentID, attachment := range message.Attachments {
target := &processedMessage.Attachments[attachmentID]
color := getColor(attachment.Color)
target.Color = &color
if attachment.AuthorIcon != nil {
var resp *http.Response
resp, err = netClient.Get(*attachment.AuthorIcon)
resp, err := netClient.Get(*url)
if err == nil {
body, _ := ioutil.ReadAll(resp.Body)
ct, _, _ := mime.ParseMediaType(resp.Header.Get("Content-Type"))
b64body := base64.StdEncoding.EncodeToString(body)
*target.AuthorIcon = fmt.Sprintf("<img src=\"data:%s;base64,%s\" />", ct, b64body)
} else {
*target.AuthorIcon = ""
var (
body []byte
contentType string
)
if body, err = ioutil.ReadAll(resp.Body); err != nil {
return
}
if contentType, _, err = mime.ParseMediaType(resp.Header.Get("Content-Type")); err != nil {
return
}
base64Body := base64.StdEncoding.EncodeToString(body)
data = template.URL(fmt.Sprintf("data:%s;base64,%s", contentType, base64Body))
}
return
}
func renderSlackAttachment(attachment *slackAttachment) {
if attachment == nil {
return
}
attachment.ColorRendered = template.HTMLAttr(getColor(attachment.Color))
attachment.AuthorIconURL = fetchAndEncodeImage(attachment.AuthorIcon)
for _, fieldName := range attachment.MrkdwnIn {
var targetField, srcField *string
var (
srcField *string
targetField *template.HTML
)
switch fieldName {
case "text":
srcField = &attachment.Text
targetField = &target.Text
break
targetField = &attachment.TextRendered
case "pretext":
srcField = &attachment.Pretext
targetField = &target.Pretext
break
targetField = &attachment.PretextRendered
case "fallback":
srcField = &attachment.Fallback
targetField = &attachment.FallbackRendered
}
if targetField != nil && srcField != nil {
value := string(
log.Info(targetField)
*targetField = template.HTML(
blackfriday.MarkdownBasic([]byte(linkifyString(*srcField))))
targetField = &value
}
}
}
func slackMessageToHTMLMessage(message slackMessage) (html matrix.HTMLMessage, err error) {
text := linkifyString(message.Text)
if message.Mrkdwn == nil || *message.Mrkdwn == true {
message.TextRendered = template.HTML(blackfriday.MarkdownBasic([]byte(text)))
}
for attachmentID := range message.Attachments {
renderSlackAttachment(&message.Attachments[attachmentID])
}
var buffer bytes.Buffer
html.MsgType = "m.text"
html.Format = "org.matrix.custom.html"
html.Body, _ = slackMessageToMarkdown(message)
err = htmlTemplate.ExecuteTemplate(&buffer, "htmlTemplate", processedMessage)
err = htmlTemplate.ExecuteTemplate(&buffer, "htmlTemplate", message)
html.FormattedBody = buffer.String()
return
}

21
src/github.com/matrix-org/go-neb/services/slackapi/slackapi.go
View File

@ -13,7 +13,11 @@ type slackAPIService struct {
serviceUserID string
webhookEndpointURL string
ClientUserID string
Hooks map[string]string
// maps from hookID -> roomID
Hooks map[string]struct {
RoomID string
MessageType string
}
}
func (s *slackAPIService) ServiceUserID() string { return s.serviceUserID }
@ -30,15 +34,26 @@ func (s *slackAPIService) Plugin(cli *matrix.Client, roomID string) plugin.Plugi
func (s *slackAPIService) OnReceiveWebhook(w http.ResponseWriter, req *http.Request, cli *matrix.Client) {
segments := strings.Split(req.URL.Path, "/")
if len(segments) < 2 {
w.WriteHeader(400)
}
hookID := segments[len(segments)-2]
messageType := segments[len(segments)-3]
roomID := s.Hooks[hookID]
messageType := s.Hooks[hookID].MessageType
if messageType == "" {
messageType = "m.text"
}
roomID := s.Hooks[hookID].RoomID
slackMessage, err := getSlackMessage(*req)
if err != nil {
return
}
htmlMessage, err := slackMessageToHTMLMessage(slackMessage)
if err != nil {
return
}
htmlMessage.MsgType = messageType
cli.SendMessageEvent(
roomID,

Write Preview
Loading…
Cancel
Save