mirror of
https://github.com/idanoo/go-flat-finder.git
synced 2024-11-22 00:11:57 +00:00
55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package flatfinder
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/disgoorg/disgo/discord"
|
|
"github.com/disgoorg/disgo/webhook"
|
|
"github.com/disgoorg/snowflake/v2"
|
|
)
|
|
|
|
// Load discord client
|
|
func (c *LocalConfig) initDiscord() {
|
|
// Webhook URL splitting
|
|
webhookString := strings.ReplaceAll(c.DiscordWebhook, "https://discord.com/api/webhooks/", "")
|
|
webhookParts := strings.Split(webhookString, "/")
|
|
if len(webhookParts) != 2 {
|
|
log.Fatal("Invalid DISCORD_WEBHOOK")
|
|
}
|
|
|
|
// Convert snowflakeID to uint64
|
|
i, err := strconv.ParseInt(webhookParts[0], 10, 64)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
// Start client!
|
|
client := webhook.New(snowflake.ID(i), webhookParts[1])
|
|
c.DiscordClient = client
|
|
|
|
log.Print("Discord client loaded succesfully")
|
|
}
|
|
|
|
// sendEmbeddedMessage - Build an embedded message from listing data
|
|
func (c *LocalConfig) sendEmbeddedMessage(listing TradeMeListing) {
|
|
log.Printf("New listing: %s", listing.Title)
|
|
|
|
embed := discord.NewEmbedBuilder().
|
|
SetTitle(listing.Title).
|
|
SetURL(fmt.Sprintf("https://trademe.co.nz/%d", listing.ListingID)).
|
|
SetColor(1127128).
|
|
SetImage(listing.PictureHref).
|
|
AddField("Location", listing.Address, true).
|
|
AddField("Bedrooms", fmt.Sprintf("%d", listing.Bedrooms), true)
|
|
|
|
embeds := []discord.Embed{}
|
|
embeds = append(embeds, embed.Build())
|
|
_, err := c.DiscordClient.CreateEmbeds(embeds)
|
|
if err != nil {
|
|
log.Print(err)
|
|
}
|
|
}
|