James Tombleson
0e0058506a
* added feature flags around background workers * background workers are moved to a new package as outputs are starting to get added * package name was updated * updated refs to the new input package * query and sql updates on routes * moved the services and starting to add discord web hook * query update
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
package output
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/jtom38/newsbot/collector/database"
|
|
)
|
|
|
|
type discordField struct {
|
|
Name string `json:"name,omitempty"`
|
|
Value string `json:"value,omitempty"`
|
|
Inline bool `json:"inline,omitempty"`
|
|
}
|
|
|
|
type discordAuthor struct {
|
|
Name string `json:"name,omitempty"`
|
|
Url string `json:"url,omitempty"`
|
|
IconUrl string `json:"icon_url,omitempty"`
|
|
}
|
|
|
|
type discordImage struct {
|
|
Url string `json:"url,omitempty"`
|
|
}
|
|
|
|
type discordEmbed struct {
|
|
Title string `json:"title,omitempty"`
|
|
Description string `json:"description,omitempty"`
|
|
Url string `json:"url,omitempty"`
|
|
Color int32 `json:"color,omitempty"`
|
|
Timestamp time.Time `json:"timestamp,omitempty"`
|
|
Fields []discordField `json:"fields,omitempty"`
|
|
Author discordAuthor `json:"author,omitempty"`
|
|
Image discordImage `json:"image,omitempty"`
|
|
Thumbnail discordImage `json:"thumbnail,omitempty"`
|
|
}
|
|
|
|
// Root object for Discord Webhook messages
|
|
type discordMessage struct {
|
|
Content string `json:"content,omitempty"`
|
|
Embeds []discordEmbed `json:"embeds,omitempty"`
|
|
}
|
|
|
|
type Discord struct {
|
|
Subscriptions []string
|
|
article database.Article
|
|
Message discordMessage
|
|
}
|
|
|
|
func NewDiscordWebHookMessage(Subscriptions []string, Article database.Article) Discord {
|
|
return Discord{
|
|
Subscriptions: Subscriptions,
|
|
article: Article,
|
|
Message: discordMessage{
|
|
Embeds: []discordEmbed{},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (dwh Discord) GeneratePayload() error {
|
|
// Convert the message
|
|
embed := discordEmbed {
|
|
Title: dwh.article.Title,
|
|
Description: dwh.convertFromHtml(dwh.article.Description),
|
|
Url: dwh.article.Url,
|
|
Thumbnail: discordImage{
|
|
Url: dwh.article.Thumbnail,
|
|
},
|
|
}
|
|
var arr []discordEmbed
|
|
|
|
arr = append(arr, embed)
|
|
dwh.Message.Embeds = arr
|
|
return nil
|
|
}
|
|
|
|
func (dwh Discord) SendPayload() error {
|
|
return nil
|
|
}
|
|
|
|
func (dwh Discord) convertFromHtml(body string) string {
|
|
clean := body
|
|
clean = strings.ReplaceAll(clean, "<h2>", "**")
|
|
clean = strings.ReplaceAll(clean, "</h2>", "**")
|
|
clean = strings.ReplaceAll(clean, "<h3>", "**")
|
|
clean = strings.ReplaceAll(clean, "</h3>", "**\r\n")
|
|
clean = strings.ReplaceAll(clean, "<strong>", "**")
|
|
clean = strings.ReplaceAll(clean, "</strong>", "**\r\n")
|
|
clean = strings.ReplaceAll(clean, "<ul>", "\r\n")
|
|
clean = strings.ReplaceAll(clean, "</ul>", "")
|
|
clean = strings.ReplaceAll(clean, "</li>", "\r\n")
|
|
clean = strings.ReplaceAll(clean, "<li>", "> ")
|
|
clean = strings.ReplaceAll(clean, "“", "\"")
|
|
clean = strings.ReplaceAll(clean, "”", "\"")
|
|
clean = strings.ReplaceAll(clean, "…", "...")
|
|
clean = strings.ReplaceAll(clean, "<b>", "**")
|
|
clean = strings.ReplaceAll(clean, "</b>", "**")
|
|
clean = strings.ReplaceAll(clean, "<br>", "\r\n")
|
|
clean = strings.ReplaceAll(clean, "<br/>", "\r\n")
|
|
clean = strings.ReplaceAll(clean, "\xe2\x96\xa0", "*")
|
|
clean = strings.ReplaceAll(clean, "\xa0", "\r\n")
|
|
clean = strings.ReplaceAll(clean, "<p>", "")
|
|
clean = strings.ReplaceAll(clean, "</p>", "\r\n")
|
|
return clean
|
|
}
|
|
|
|
func (dwh Discord) convertLinks(body string) string {
|
|
//items := regexp.MustCompile("<a(.*?)a>")
|
|
return ""
|
|
}
|