132 lines
3.0 KiB
Go
132 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
"github.com/dustin/go-humanize"
|
|
"github.com/joho/godotenv"
|
|
"github.com/namsral/flag"
|
|
)
|
|
|
|
var (
|
|
DiscordToken string
|
|
dg *discordgo.Session
|
|
DiscordChannelId string
|
|
)
|
|
|
|
func init() {
|
|
err := godotenv.Load()
|
|
if err != nil {
|
|
log.Fatal("Error loading .env file")
|
|
}
|
|
flag.StringVar(&DiscordToken, "DISCORD_TOKEN", "", "Discord bot token")
|
|
flag.StringVar(&DiscordChannelId, "DISCORD_CHANNEL_ID", "", "Discord channel id")
|
|
flag.Parse()
|
|
if DiscordToken == "" {
|
|
log.Fatal("DISCORD_TOKEN not set")
|
|
}
|
|
if DiscordChannelId == "" {
|
|
log.Fatal("DISCORD_CHANNEL_ID not set")
|
|
}
|
|
initBot()
|
|
}
|
|
|
|
func initBot() {
|
|
var err error
|
|
dg, err = discordgo.New("Bot " + DiscordToken)
|
|
if err != nil {
|
|
log.Fatal("error creating discord session:", err)
|
|
}
|
|
dg.AddHandler(func(s *discordgo.Session, event *discordgo.Ready) {
|
|
log.Println("Logged in as", event.User.Username)
|
|
})
|
|
dg.AddHandler(onMessage)
|
|
dg.Identify.Intents = discordgo.IntentsGuildMessages | discordgo.IntentsMessageContent
|
|
err = dg.Open()
|
|
if err != nil {
|
|
log.Fatal("error opening connection to discord: ", err, " -- Is your token correct?")
|
|
}
|
|
}
|
|
|
|
func onMessage(s *discordgo.Session, m *discordgo.MessageCreate) {
|
|
// Ignore all messages created by the bot itself
|
|
if m.Author.ID == s.State.User.ID {
|
|
return
|
|
}
|
|
id, err := ParseHackerNewsLink(m.Content)
|
|
if err != nil {
|
|
return
|
|
}
|
|
story := FetchStoryById(id)
|
|
id, err = PostStoryToStackerNews(&story)
|
|
if err != nil {
|
|
var dupesErr *DupesError
|
|
if errors.As(err, &dupesErr) {
|
|
SendDupesErrorToDiscord(dupesErr)
|
|
} else {
|
|
log.Fatal("unexpected error returned")
|
|
}
|
|
}
|
|
}
|
|
|
|
func SendDupesErrorToDiscord(dupesErr *DupesError) {
|
|
title := fmt.Sprintf("%d dupe(s) found for %s:", len(dupesErr.Dupes), dupesErr.Url)
|
|
color := 0xffc107
|
|
var fields []*discordgo.MessageEmbedField
|
|
for _, dupe := range dupesErr.Dupes {
|
|
fields = append(fields,
|
|
&discordgo.MessageEmbedField{
|
|
Name: "Title",
|
|
Value: dupe.Title,
|
|
Inline: false,
|
|
},
|
|
&discordgo.MessageEmbedField{
|
|
Name: "Id",
|
|
Value: StackerNewsItemLink(dupe.Id),
|
|
Inline: true,
|
|
},
|
|
&discordgo.MessageEmbedField{
|
|
Name: "Url",
|
|
Value: dupe.Url,
|
|
Inline: true,
|
|
},
|
|
&discordgo.MessageEmbedField{
|
|
Name: "User",
|
|
Value: dupe.User.Name,
|
|
Inline: true,
|
|
},
|
|
&discordgo.MessageEmbedField{
|
|
Name: "Created",
|
|
Value: humanize.Time(dupe.CreatedAt),
|
|
Inline: true,
|
|
},
|
|
&discordgo.MessageEmbedField{
|
|
Name: "Sats",
|
|
Value: fmt.Sprint(dupe.Sats),
|
|
Inline: true,
|
|
},
|
|
&discordgo.MessageEmbedField{
|
|
Name: "Comments",
|
|
Value: fmt.Sprint(dupe.NComments),
|
|
Inline: true,
|
|
},
|
|
)
|
|
}
|
|
embed := discordgo.MessageEmbed{
|
|
Title: title,
|
|
Color: color,
|
|
Fields: fields,
|
|
}
|
|
SendEmbedToDiscord(&embed)
|
|
}
|
|
|
|
func SendEmbedToDiscord(embed *discordgo.MessageEmbed) {
|
|
_, err := dg.ChannelMessageSendEmbed(DiscordChannelId, embed)
|
|
if err != nil {
|
|
log.Fatal("Error during json.Marshal:", err)
|
|
}
|
|
}
|